Spaces:
Running
Running
Jeremiah Lowin commited on
Commit ·
f0e5fd6
1
Parent(s): c27f039
Fix typing
Browse files
src/fastmcp/client/transports.py
CHANGED
|
@@ -22,7 +22,6 @@ from mcp.client.stdio import stdio_client
|
|
| 22 |
from mcp.client.streamable_http import streamablehttp_client
|
| 23 |
from mcp.client.websocket import websocket_client
|
| 24 |
from mcp.shared.memory import create_connected_server_and_client_session
|
| 25 |
-
from mcp.shared.session import ProgressFnT
|
| 26 |
from pydantic import AnyUrl
|
| 27 |
from typing_extensions import Unpack
|
| 28 |
|
|
@@ -36,7 +35,6 @@ class SessionKwargs(TypedDict, total=False):
|
|
| 36 |
list_roots_callback: ListRootsFnT | None
|
| 37 |
logging_callback: LoggingFnT | None
|
| 38 |
message_handler: MessageHandlerFnT | None
|
| 39 |
-
progress_callback: ProgressFnT | None
|
| 40 |
read_timeout_seconds: datetime.timedelta | None
|
| 41 |
|
| 42 |
|
|
|
|
| 22 |
from mcp.client.streamable_http import streamablehttp_client
|
| 23 |
from mcp.client.websocket import websocket_client
|
| 24 |
from mcp.shared.memory import create_connected_server_and_client_session
|
|
|
|
| 25 |
from pydantic import AnyUrl
|
| 26 |
from typing_extensions import Unpack
|
| 27 |
|
|
|
|
| 35 |
list_roots_callback: ListRootsFnT | None
|
| 36 |
logging_callback: LoggingFnT | None
|
| 37 |
message_handler: MessageHandlerFnT | None
|
|
|
|
| 38 |
read_timeout_seconds: datetime.timedelta | None
|
| 39 |
|
| 40 |
|
tests/client/test_progress.py
CHANGED
|
@@ -59,8 +59,10 @@ async def test_progress_handler_can_be_supplied_on_tool_call(fastmcp_server: Fas
|
|
| 59 |
async def test_progress_handler_supplied_on_tool_call_overrides_default(
|
| 60 |
fastmcp_server: FastMCP,
|
| 61 |
):
|
| 62 |
-
async def bad_progress_handler(
|
| 63 |
-
|
|
|
|
|
|
|
| 64 |
|
| 65 |
async with Client(fastmcp_server, progress_handler=bad_progress_handler) as client:
|
| 66 |
await client.call_tool("progress_tool", {}, progress_handler=progress_handler)
|
|
|
|
| 59 |
async def test_progress_handler_supplied_on_tool_call_overrides_default(
|
| 60 |
fastmcp_server: FastMCP,
|
| 61 |
):
|
| 62 |
+
async def bad_progress_handler(
|
| 63 |
+
progress: float, total: float | None, message: str | None
|
| 64 |
+
) -> None:
|
| 65 |
+
raise Exception("This should not be called")
|
| 66 |
|
| 67 |
async with Client(fastmcp_server, progress_handler=bad_progress_handler) as client:
|
| 68 |
await client.call_tool("progress_tool", {}, progress_handler=progress_handler)
|