File tree Expand file tree Collapse file tree 1 file changed +13
-1
lines changed
Expand file tree Collapse file tree 1 file changed +13
-1
lines changed Original file line number Diff line number Diff line change @@ -417,9 +417,21 @@ server_params = StdioServerParameters(
417417 env = None # Optional environment variables
418418)
419419
420+ # Optional: create a sampling callback
421+ async def handle_sampling_message (message : types.CreateMessageRequestParams) -> types.CreateMessageResult:
422+ return types.CreateMessageResult(
423+ role = " assistant" ,
424+ content = types.TextContent(
425+ type = " text" ,
426+ text = " Hello, world! from model" ,
427+ ),
428+ model = " gpt-3.5-turbo" ,
429+ stopReason = " endTurn" ,
430+ )
431+
420432async def run ():
421433 async with stdio_client(server_params) as (read, write):
422- async with ClientSession(read, write) as session:
434+ async with ClientSession(read, write, sampling_callback = handle_sampling_message ) as session:
423435 # Initialize the connection
424436 await session.initialize()
425437
You can’t perform that action at this time.
0 commit comments