From cfee865f165627a9f55d0d4762bbf6d4e2b3f2f2 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Fri, 20 Jun 2025 12:09:45 -0700 Subject: [PATCH] fix test MCP --- tests/store_model_in_db_tests/test_mcp_servers.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/store_model_in_db_tests/test_mcp_servers.py b/tests/store_model_in_db_tests/test_mcp_servers.py index 2402b012b4..21b12fe182 100644 --- a/tests/store_model_in_db_tests/test_mcp_servers.py +++ b/tests/store_model_in_db_tests/test_mcp_servers.py @@ -9,7 +9,8 @@ import os from starlette import status from litellm.constants import LITELLM_PROXY_ADMIN_NAME -from litellm.proxy._types import MCPAuth, MCPSpecVersion, MCPSpecVersionType, MCPTransportType, MCPTransport, NewMCPServerRequest, LiteLLM_MCPServerTable +from litellm.proxy._types import MCPSpecVersion, MCPSpecVersionType, MCPTransportType, MCPTransport, NewMCPServerRequest, LiteLLM_MCPServerTable +from litellm.types.mcp import MCPAuth from litellm.proxy.management_endpoints.mcp_management_endpoints import does_mcp_server_exist TEST_MASTER_KEY = os.getenv("LITELLM_MASTER_KEY", "sk-1234")