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

fix linting errors

Browse source
This commit is contained in:
Ishaan Jaff 2025-03-24 18:48:43 -07:00
parent 3ce0821314
commit f600899ff7
2 changed files with 4 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

@ -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 #######################

2
litellm/proxy/_experimental/mcp_server/sse_transport.py
View file

@ -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]
)