Skip to content

Commit b2ef979

Browse files
Add missing imports to test_integration.py
Added missing type annotation imports that were causing NameError and preventing test collection: - RequestResponder from mcp.shared.session - SessionMessage from mcp.shared.message - GetSessionIdCallback from mcp.client.streamable_http - RequestContext from mcp.shared.context This fixes 4 NameError collection failures and 10 F821 ruff errors, allowing all 20 tests in the file to be properly collected and executed.
1 parent 261acbb commit b2ef979

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tests/server/fastmcp/test_integration.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,10 @@
3535
)
3636
from mcp.client.session import ClientSession
3737
from mcp.client.sse import sse_client
38-
from mcp.client.streamable_http import streamable_http_client
38+
from mcp.client.streamable_http import GetSessionIdCallback, streamable_http_client
39+
from mcp.shared.context import RequestContext
40+
from mcp.shared.message import SessionMessage
41+
from mcp.shared.session import RequestResponder
3942
from mcp.types import (
4043
ClientResult,
4144
CreateMessageRequestParams,

0 commit comments

Comments
 (0)