File tree Expand file tree Collapse file tree 2 files changed +7
-3
lines changed Expand file tree Collapse file tree 2 files changed +7
-3
lines changed Original file line number Diff line number Diff line change @@ -259,13 +259,12 @@ async def send_roots_list_changed(self) -> None:
259
259
async def _received_request (
260
260
self , responder : RequestResponder ["types.ServerRequest" , "types.ClientResult" ]
261
261
) -> None :
262
-
263
262
if isinstance (responder .request .root , types .CreateMessageRequest ):
264
263
# handle create message request (sampling)
265
264
266
265
if self .sampling_callback is None :
267
266
raise RuntimeError ("Sampling callback is not set" )
268
-
267
+
269
268
response = await self .sampling_callback (responder .request .root .params )
270
269
client_response = types .ClientResult (** response .model_dump ())
271
270
await responder .respond (client_response )
Original file line number Diff line number Diff line change
1
+ import shutil
2
+
1
3
import pytest
2
4
3
5
from mcp .client .stdio import StdioServerParameters , stdio_client
4
6
from mcp .types import JSONRPCMessage , JSONRPCRequest , JSONRPCResponse
5
7
8
+ tee : str = shutil .which ("tee" ) # type: ignore
9
+ assert tee is not None , "could not find tee command"
10
+
6
11
7
12
@pytest .mark .anyio
8
13
async def test_stdio_client ():
9
- server_parameters = StdioServerParameters (command = "/usr/bin/ tee" )
14
+ server_parameters = StdioServerParameters (command = tee )
10
15
11
16
async with stdio_client (server_parameters ) as (read_stream , write_stream ):
12
17
# Test sending and receiving messages
You can’t perform that action at this time.
0 commit comments