Fix typing

This commit is contained in:
Jeremiah Lowin 2025-05-20 09:36:00 -04:00
commit 5a89e0cc2b
2 changed files with 4 additions and 4 deletions

View file

@ -22,7 +22,6 @@ from mcp.client.stdio import stdio_client
from mcp.client.streamable_http import streamablehttp_client
from mcp.client.websocket import websocket_client
from mcp.shared.memory import create_connected_server_and_client_session
from mcp.shared.session import ProgressFnT
from pydantic import AnyUrl
from typing_extensions import Unpack
@ -36,7 +35,6 @@ class SessionKwargs(TypedDict, total=False):
list_roots_callback: ListRootsFnT | None
logging_callback: LoggingFnT | None
message_handler: MessageHandlerFnT | None
progress_callback: ProgressFnT | None
read_timeout_seconds: datetime.timedelta | None

View file

@ -59,8 +59,10 @@ async def test_progress_handler_can_be_supplied_on_tool_call(fastmcp_server: Fas
async def test_progress_handler_supplied_on_tool_call_overrides_default(
fastmcp_server: FastMCP,
):
async def bad_progress_handler(*args, **kwargs):
1 / 0
async def bad_progress_handler(
progress: float, total: float | None, message: str | None
) -> None:
raise Exception("This should not be called")
async with Client(fastmcp_server, progress_handler=bad_progress_handler) as client:
await client.call_tool("progress_tool", {}, progress_handler=progress_handler)