scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-23 00:41:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 99

register OpenAPI tools on server add/update/reload

Browse source
This commit is contained in:
Ishaan Jaffer 2026-02-19 12:14:26 -08:00
parent da50242746
commit b9a864c24c
1 changed files with 18 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
litellm/proxy/_experimental/mcp_server/mcp_server_manager.py
View file

@ -641,11 +641,25 @@ class MCPServerManager:
)
return new_server
async def _maybe_register_openapi_tools(self, server: MCPServer):
"""Register OpenAPI tools if the server has a spec_path configured."""
if server.spec_path:
verbose_logger.info(
f"Loading OpenAPI spec from {server.spec_path} for server {server.name}"
)
await self._register_openapi_tools(
spec_path=server.spec_path,
server=server,
base_url=server.url or "",
)
self.initialize_tool_name_to_mcp_server_name_mapping()
async def add_server(self, mcp_server: LiteLLM_MCPServerTable):
try:
if mcp_server.server_id not in self.registry:
new_server = await self.build_mcp_server_from_table(mcp_server)
self.registry[mcp_server.server_id] = new_server
await self._maybe_register_openapi_tools(new_server)
verbose_logger.debug(f"Added MCP Server: {new_server.name}")
except Exception as e:
@ -657,6 +671,7 @@ class MCPServerManager:
if mcp_server.server_id in self.registry:
new_server = await self.build_mcp_server_from_table(mcp_server)
self.registry[mcp_server.server_id] = new_server
await self._maybe_register_openapi_tools(new_server)
verbose_logger.debug(f"Updated MCP Server: {new_server.name}")
except Exception as e:
@ -2245,9 +2260,9 @@ class MCPServerManager:
verbose_logger.debug(
f"Building server from DB: {server.server_id} ({server.server_name})"
)
new_registry[server.server_id] = await self.build_mcp_server_from_table(
server
)
new_server = await self.build_mcp_server_from_table(server)
new_registry[server.server_id] = new_server
await self._maybe_register_openapi_tools(new_server)
self.registry = new_registry