diff --git a/litellm/proxy/_experimental/mcp_server/server.py b/litellm/proxy/_experimental/mcp_server/server.py index 2dcfe60a06..02a9303be5 100644 --- a/litellm/proxy/_experimental/mcp_server/server.py +++ b/litellm/proxy/_experimental/mcp_server/server.py @@ -30,8 +30,8 @@ router = APIRouter( prefix="/mcp", tags=["mcp"], ) -server = Server("litellm-mcp-server") -sse = SseServerTransport("/mcp/sse/messages") +server: Server = Server("litellm-mcp-server") +sse: SseServerTransport = SseServerTransport("/mcp/sse/messages") ######################################################## ############### MCP Server Routes ####################### diff --git a/litellm/proxy/_experimental/mcp_server/sse_transport.py b/litellm/proxy/_experimental/mcp_server/sse_transport.py index 05a75b3b8a..63ffd403c6 100644 --- a/litellm/proxy/_experimental/mcp_server/sse_transport.py +++ b/litellm/proxy/_experimental/mcp_server/sse_transport.py @@ -73,6 +73,8 @@ class SseServerTransport: self._read_stream_writers[session_id] = read_stream_writer verbose_logger.debug(f"Created new session with ID: {session_id}") + sse_stream_writer: MemoryObjectSendStream[dict[str, Any]] + sse_stream_reader: MemoryObjectReceiveStream[dict[str, Any]] sse_stream_writer, sse_stream_reader = anyio.create_memory_object_stream( 0, dict[str, Any] )