diff --git a/litellm/proxy/management_endpoints/mcp_management_endpoints.py b/litellm/proxy/management_endpoints/mcp_management_endpoints.py index 949c9bed1cf..8c4d4e7937e 100644 --- a/litellm/proxy/management_endpoints/mcp_management_endpoints.py +++ b/litellm/proxy/management_endpoints/mcp_management_endpoints.py @@ -15,6 +15,8 @@ Endpoints here: """ import importlib +import json +import os from dataclasses import dataclass from datetime import datetime, timedelta from typing import Any, Dict, Iterable, List, Literal, Optional @@ -1181,7 +1183,7 @@ if MCP_AVAILABLE: # --- MCP Discovery --- _MCP_REGISTRY_PATH = os.path.join( - os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))), + os.path.dirname(os.path.dirname(os.path.abspath(__file__))), "mcp_registry.json", ) diff --git a/mcp_registry.json b/litellm/proxy/mcp_registry.json similarity index 100% rename from mcp_registry.json rename to litellm/proxy/mcp_registry.json diff --git a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_discovery.py b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_discovery.py index 9dacec1904b..dde73016271 100644 --- a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_discovery.py +++ b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_discovery.py @@ -21,6 +21,8 @@ class TestMCPRegistryFile: "..", "..", "..", + "litellm", + "proxy", "mcp_registry.json", ) diff --git a/ui/litellm-dashboard/src/components/mcp_tools/mcp_discovery.tsx b/ui/litellm-dashboard/src/components/mcp_tools/mcp_discovery.tsx index ed8ffdf5d3e..558814f4dc8 100644 --- a/ui/litellm-dashboard/src/components/mcp_tools/mcp_discovery.tsx +++ b/ui/litellm-dashboard/src/components/mcp_tools/mcp_discovery.tsx @@ -1,11 +1,11 @@ import React, { useState, useMemo, useEffect } from "react"; -import { Modal, Input } from "antd"; -import { Typography } from "antd"; +import { Modal, Input, Typography } from "antd"; import { fetchDiscoverableMCPServers } from "../networking"; import { DiscoverableMCPServer, DiscoverMCPServersResponse } from "./types"; import { mcpLogoImg } from "./create_mcp_server"; const { Search } = Input; +const { Text } = Typography; interface MCPDiscoveryProps { isVisible: boolean;