scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-11 22:51:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 154

fix: ownership transfer DB writes and registry reload ordering

Browse source 
Move team_id clear-to-null and ObjectPermissionTable sync before
registry reload so the cache reflects the new ownership state.
Also update the response object when clearing team_id to null.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
yuneng-jiang 2026-03-21 12:36:50 -07:00
parent bdab813f21
commit bb019c8920
1 changed files with 8 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
litellm/proxy/management_endpoints/mcp_management_endpoints.py
View file

@ -1994,12 +1994,8 @@ if MCP_AVAILABLE:
"error": f"MCP Server not found, passed server_id={payload.server_id}"
},
)
await global_mcp_server_manager.update_server(mcp_server_record_updated)
# Ensure registry is up to date by reloading from database
await global_mcp_server_manager.reload_servers_from_database()
# Handle explicit team_id changes (including clearing to null/global)
# Must happen before registry reload so the cache reflects the new state.
if is_admin and team_id_explicitly_set:
new_team_id = payload.team_id # could be a team ID or None (global)
@ -2010,6 +2006,8 @@ if MCP_AVAILABLE:
where={"server_id": payload.server_id},
data={"team_id": None},
)
if mcp_server_record_updated is not None:
mcp_server_record_updated.team_id = None
# Sync ObjectPermissionTable
if old_team_id != new_team_id:
@ -2032,6 +2030,11 @@ if MCP_AVAILABLE:
f"Failed to add server {payload.server_id} to new team {new_team_id}: {e}"
)
await global_mcp_server_manager.update_server(mcp_server_record_updated)
# Ensure registry is up to date by reloading from database
await global_mcp_server_manager.reload_servers_from_database()
# TODO: Enterprise: Finish audit log trail
if litellm.store_audit_logs:
pass