ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

generate_mock_mcp_server_config_record

Browse Source
This commit is contained in:
Ishaan Jaffer 2025-09-23 17:33:03 -07:00
parent 8b9bd9bdb6
commit 79a6feb7ac
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/test_litellm/proxy/management_endpoints/test_mcp_management_endpoints.py
View File

@ -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",
},
)