From 696974bacb49af2804692ed7a6427f08cc79a67b Mon Sep 17 00:00:00 2001 From: Mubashir Osmani Date: Fri, 21 Nov 2025 23:00:33 -0500 Subject: [PATCH] fix: add mcp server ids (#16904) * fix: add mcp server ids * revert to prev version --- .../proxy/_experimental/mcp_server/mcp_server_manager.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/litellm/proxy/_experimental/mcp_server/mcp_server_manager.py b/litellm/proxy/_experimental/mcp_server/mcp_server_manager.py index f28c9b5acb..320565f7f6 100644 --- a/litellm/proxy/_experimental/mcp_server/mcp_server_manager.py +++ b/litellm/proxy/_experimental/mcp_server/mcp_server_manager.py @@ -1928,6 +1928,14 @@ class MCPServerManager: f"Registry now contains {len(self.get_registry())} servers" ) + def get_mcp_servers_from_ids(self, server_ids: List[str]) -> List[MCPServer]: + servers = [] + registry = self.get_registry() + for server in registry.values(): + if server.server_id in server_ids: + servers.append(server) + return servers + def get_mcp_server_by_id(self, server_id: str) -> Optional[MCPServer]: """ Get the MCP Server from the server id