From 15f5dc38c47aaf8c5dc61d3bfeb665495df6769c Mon Sep 17 00:00:00 2001 From: Sameer Kankute Date: Thu, 26 Mar 2026 19:41:43 +0530 Subject: [PATCH] Fix tests --- .../model_management_endpoints.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/litellm/proxy/management_endpoints/model_management_endpoints.py b/litellm/proxy/management_endpoints/model_management_endpoints.py index 355e2011237..2ca8e3daba3 100644 --- a/litellm/proxy/management_endpoints/model_management_endpoints.py +++ b/litellm/proxy/management_endpoints/model_management_endpoints.py @@ -42,6 +42,9 @@ from litellm.proxy.management_endpoints.team_endpoints import ( team_model_add, team_model_delete, ) +from litellm.proxy.management_endpoints.team_endpoints import ( + update_team as _legacy_update_team, +) from litellm.proxy.management_helpers.audit_logs import create_object_audit_log from litellm.proxy.utils import PrismaClient from litellm.types.proxy.management_endpoints.model_management_endpoints import ( @@ -58,6 +61,14 @@ from litellm.utils import get_utc_datetime router = APIRouter() +async def update_team(*args, **kwargs): + """ + Backward-compatible shim for tests/legacy call sites that patch this symbol. + Team model management now uses team_model_add/team_model_delete directly. + """ + return await _legacy_update_team(*args, **kwargs) + + class UpdatePublicModelGroupsRequest(BaseModel): """Request model for updating public model groups"""