From dc33cfa0740a2e84858ad5702ca1b262fb8690ea Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Thu, 20 Mar 2025 17:53:37 -0700 Subject: [PATCH] init global_mcp_tool_registry --- litellm/proxy/proxy_server.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index ae1c8d18af..390c215ce6 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -126,6 +126,9 @@ from litellm.litellm_core_utils.core_helpers import ( from litellm.litellm_core_utils.credential_accessor import CredentialAccessor from litellm.litellm_core_utils.litellm_logging import Logging as LiteLLMLoggingObj from litellm.llms.custom_httpx.http_handler import AsyncHTTPHandler, HTTPHandler +from litellm.proxy._experimental.mcp_server.tool_registry import ( + global_mcp_tool_registry, +) from litellm.proxy._types import * from litellm.proxy.analytics_endpoints.analytics_endpoints import ( router as analytics_router, @@ -2153,6 +2156,11 @@ class ProxyConfig: all_guardrails=guardrails_v2, config_file_path=config_file_path ) + ## MCP TOOLS + mcp_tools_config = config.get("mcp_tools", None) + if mcp_tools_config: + global_mcp_tool_registry.load_tools_from_config(mcp_tools_config) + ## CREDENTIALS credential_list_dict = self.load_credential_list(config=config) litellm.credential_list = credential_list_dict