From e0edadc1f8007d407d1642813ffe889d76497193 Mon Sep 17 00:00:00 2001 From: Jerome Date: Wed, 19 Feb 2025 13:33:45 +0000 Subject: [PATCH] Updated typing on request context for the server to use server session --- src/mcp/server/fastmcp/server.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/mcp/server/fastmcp/server.py b/src/mcp/server/fastmcp/server.py index 5ae30a5ca..e08a161c1 100644 --- a/src/mcp/server/fastmcp/server.py +++ b/src/mcp/server/fastmcp/server.py @@ -34,6 +34,7 @@ from mcp.server.lowlevel.server import ( lifespan as default_lifespan, ) +from mcp.server.session import ServerSession from mcp.server.sse import SseServerTransport from mcp.server.stdio import stdio_server from mcp.shared.context import RequestContext @@ -597,7 +598,7 @@ def my_tool(x: int, ctx: Context) -> str: The context is optional - tools that don't need it can omit the parameter. """ - _request_context: RequestContext | None + _request_context: RequestContext[ServerSession, Any] | None _fastmcp: FastMCP | None def __init__(