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(
417
417
env = None # Optional environment variables
418
418
)
419
419
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
+
420
432
async def run ():
421
433
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:
423
435
# Initialize the connection
424
436
await session.initialize()
425
437
You can’t perform that action at this time.
0 commit comments