Skip to content

Commit cc137e2

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 993a298 commit cc137e2

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
@@ -34,7 +34,10 @@
3434
)
3535
from mcp.client.session import ClientSession
3636
from mcp.client.sse import sse_client
37-
from mcp.client.streamable_http import streamable_http_client
37+
from mcp.client.streamable_http import GetSessionIdCallback, streamable_http_client
38+
from mcp.shared.context import RequestContext
39+
from mcp.shared.message import SessionMessage
40+
from mcp.shared.session import RequestResponder
3841
from mcp.types import (
3942
ClientResult,
4043
CreateMessageRequestParams,

0 commit comments

Comments
 (0)