From c600371e6e9a9f7ec4f8866a5d565ae6da037045 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 6 May 2024 15:45:07 -0700 Subject: [PATCH] feat - send alert on adding new model --- litellm/proxy/proxy_server.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index b0469fdd3..83ba3f60f 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -7303,6 +7303,7 @@ async def add_new_model( """ # encrypt litellm params # _litellm_params_dict = model_params.litellm_params.dict(exclude_none=True) + _orignal_litellm_model_name = model_params.litellm_params.model for k, v in _litellm_params_dict.items(): if isinstance(v, str): encrypted_value = encrypt_value(value=v, master_key=master_key) # type: ignore @@ -7329,6 +7330,11 @@ async def add_new_model( prisma_client=prisma_client, proxy_logging_obj=proxy_logging_obj ) + await proxy_logging_obj.slack_alerting_instance.model_added_alert( + model_name=model_params.model_name, + litellm_model_name=_orignal_litellm_model_name, + ) + else: raise HTTPException( status_code=500,