Narrow message_handler's parameter to notifications and exceptions (#3168)
This commit is contained in:
parent
47bfa85e83
commit
814072c94d
1 changed files with 74 additions and 147 deletions
|
|
@ -3,7 +3,7 @@
|
|||
import anyio
|
||||
import mcp_types as types
|
||||
|
||||
from mcp.client import Client
|
||||
from mcp.client import Client, IncomingMessage
|
||||
from stories._harness import Target, run_client
|
||||
|
||||
|
||||
|
|
@ -13,7 +13,7 @@ async def main(target: Target, *, mode: str = "auto") -> None:
|
|||
received: list[types.ResourceListChangedNotification] = []
|
||||
seen = anyio.Event()
|
||||
|
||||
async def on_message(message: object) -> None:
|
||||
async def on_message(message: IncomingMessage) -> None:
|
||||
if isinstance(message, types.ResourceListChangedNotification):
|
||||
received.append(message)
|
||||
seen.set()
|
||||
|
|
|
|||
Loading…
Reference in a new issue