From 79a6feb7acc1085e491e0de674b330fd695576c8 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Tue, 23 Sep 2025 17:33:03 -0700 Subject: [PATCH] generate_mock_mcp_server_config_record --- .../test_mcp_management_endpoints.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/tests/test_litellm/proxy/management_endpoints/test_mcp_management_endpoints.py b/tests/test_litellm/proxy/management_endpoints/test_mcp_management_endpoints.py index fc97b38913..f1594c36e7 100644 --- a/tests/test_litellm/proxy/management_endpoints/test_mcp_management_endpoints.py +++ b/tests/test_litellm/proxy/management_endpoints/test_mcp_management_endpoints.py @@ -22,7 +22,7 @@ from litellm.proxy._types import ( UserAPIKeyAuth, ) from litellm.types.mcp import MCPAuth -from litellm.types.mcp_server.mcp_server_manager import MCPInfo, MCPServer +from litellm.types.mcp_server.mcp_server_manager import MCPServer def generate_mock_mcp_server_db_record( @@ -63,10 +63,10 @@ def generate_mock_mcp_server_config_record( url=url, transport=MCPTransport.http if transport == "http" else MCPTransport.sse, auth_type=MCPAuth.api_key if auth_type == "api_key" else None, - mcp_info=MCPInfo( - server_name=name, - description="Config server description", - ), + mcp_info={ + "server_name": name, + "description": "Config server description", + }, )