From 002837f962dcdfafd085968be46d2b9f497dce2a Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 24 Mar 2025 18:48:43 -0700 Subject: [PATCH] fix linting errors --- litellm/proxy/_experimental/mcp_server/server.py | 4 ++-- litellm/proxy/_experimental/mcp_server/sse_transport.py | 2 ++ 2 files changed, 4 insertions(+), 2 deletions(-) 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] )