diff --git a/litellm/proxy/_super_secret_config.yaml b/litellm/proxy/_super_secret_config.yaml index 7bbf9d3cd..f349bd09e 100644 --- a/litellm/proxy/_super_secret_config.yaml +++ b/litellm/proxy/_super_secret_config.yaml @@ -18,11 +18,3 @@ model_list: router_settings: enable_pre_call_checks: true - -litellm_settings: - success_callback: ["langfuse"] - failure_callback: ["langfuse"] - -general_settings: - alerting: ["slack"] - \ No newline at end of file diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index d5d2fb329..550405af8 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -2603,6 +2603,11 @@ class ProxyConfig: Return model info w/ id """ + _id: Optional[str] = getattr(model, "model_id", None) + if _id is not None: + model.model_info["id"] = _id + model.model_info["db_model"] = True + if model.model_info is not None and isinstance(model.model_info, dict): if "id" not in model.model_info: model.model_info["id"] = model.model_id diff --git a/litellm/tests/test_config.py b/litellm/tests/test_config.py index e38187e0e..47f632b96 100644 --- a/litellm/tests/test_config.py +++ b/litellm/tests/test_config.py @@ -102,7 +102,7 @@ async def test_delete_deployment(): pc = ProxyConfig() db_model = DBModel( - model_id="12340523", + model_id=deployment.model_info.id, model_name="gpt-3.5-turbo", litellm_params=encrypted_litellm_params, model_info={"id": deployment.model_info.id},