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