From 23ce5c2fad5f513bc10241333355d1e1aa5ee22c Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 9 Apr 2024 07:32:24 -0700 Subject: [PATCH] instantly add deployment on /model/new --- litellm/proxy/proxy_server.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 5291801e4..bfba52969 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -6900,7 +6900,7 @@ async def add_new_model( model_params: Deployment, user_api_key_dict: UserAPIKeyAuth = Depends(user_api_key_auth), ): - global llm_router, llm_model_list, general_settings, user_config_file_path, proxy_config, prisma_client, master_key, store_model_in_db + global llm_router, llm_model_list, general_settings, user_config_file_path, proxy_config, prisma_client, master_key, store_model_in_db, proxy_logging_obj try: import base64 @@ -6939,6 +6939,11 @@ async def add_new_model( "updated_by": user_api_key_dict.user_id or litellm_proxy_admin_name, } ) + + await proxy_config.add_deployment( + prisma_client=prisma_client, proxy_logging_obj=proxy_logging_obj + ) + else: raise HTTPException( status_code=500,