mirror of
https://github.com/PrefectHQ/fastmcp.git
synced 2026-08-19 20:14:17 +02:00
Fix typing
This commit is contained in:
parent
c71138d6b8
commit
5a89e0cc2b
2 changed files with 4 additions and 4 deletions
|
|
@ -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
|
||||
|
||||
|
||||
|
|
|
|||
|
|
@ -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)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue