phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix pydantic import error

Browse source
This commit is contained in:
Ishaan Jaff 2025-03-24 07:11:48 -07:00
parent 3395004f0a
commit 5017332690
2 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/proxy/_experimental/mcp_server/server.py
View file

@ -3,6 +3,7 @@ LiteLLM MCP Server Routes
"""
import asyncio
from typing import Any, Dict
import mcp.types as types
from anyio import BrokenResourceError
@ -16,7 +17,6 @@ from litellm._logging import verbose_logger
from litellm.proxy._experimental.mcp_server.tool_registry import (
global_mcp_tool_registry,
)
from litellm.types.mcp_server.tool_registry import *
from .sse_transport import SseServerTransport
@ -55,7 +55,7 @@ async def list_tools() -> list[types.Tool]:
@server.call_tool()
async def handle_call_tool(
name: str, arguments: dict | None
name: str, arguments: Dict[str, Any] | None
) -> list[types.TextContent | types.ImageContent | types.EmbeddedResource]:
"""
Call a specific tool with the provided arguments