@@ -379,7 +379,7 @@ def call(message:, server_context: nil)
379
379
test "#tools_call_handler sets the tools/call handler" do
380
380
@server . tools_call_handler do |request |
381
381
tool_name = request [ :name ]
382
- return Tool ::Response . new ( "#{ tool_name } called successfully" ) . to_h
382
+ Tool ::Response . new ( "#{ tool_name } called successfully" ) . to_h
383
383
end
384
384
385
385
request = {
@@ -391,7 +391,7 @@ def call(message:, server_context: nil)
391
391
392
392
response = @server . handle ( request )
393
393
assert_equal ( { content : "my_tool called successfully" , isError : false } , response [ :result ] )
394
- assert_instrumentation_data ( { method : "tools/call" , tool_name : "my_tool" } )
394
+ assert_instrumentation_data ( { method : "tools/call" } )
395
395
end
396
396
397
397
test "#handle prompts/list returns list of prompts" do
@@ -484,10 +484,10 @@ def call(message:, server_context: nil)
484
484
test "#prompts_get_handler sets the prompts/get handler" do
485
485
@server . prompts_get_handler do |request |
486
486
prompt_name = request [ :name ]
487
- return Prompt ::Result . new (
487
+ Prompt ::Result . new (
488
488
description : prompt_name ,
489
489
messages : [
490
- Prompt ::Message . new ( role : "user" , content : Content ::Text . new ( request [ :arguments ] [ : foo] ) ) ,
490
+ Prompt ::Message . new ( role : "user" , content : Content ::Text . new ( request [ :arguments ] [ " foo" ] ) ) ,
491
491
] ,
492
492
) . to_h
493
493
end
@@ -501,10 +501,10 @@ def call(message:, server_context: nil)
501
501
502
502
response = @server . handle ( request )
503
503
assert_equal (
504
- { description : "foo_bar_prompt" , messages : [ { role : "user" , content : { text : "bar" } } ] } ,
504
+ { description : "foo_bar_prompt" , messages : [ { role : "user" , content : { type : "text" , text : "bar" } } ] } ,
505
505
response [ :result ] ,
506
506
)
507
- assert_instrumentation_data ( { method : "prompts/get" , prompt_name : "foo_bar_prompt" } )
507
+ assert_instrumentation_data ( { method : "prompts/get" } )
508
508
end
509
509
510
510
test "#handle resources/list returns a list of resources" do
@@ -555,9 +555,9 @@ def call(message:, server_context: nil)
555
555
556
556
test "#resources_read_handler sets the resources/read handler" do
557
557
@server . resources_read_handler do |request |
558
- return {
558
+ {
559
559
uri : request [ :uri ] ,
560
- mime_type : "text/plain" ,
560
+ mimeType : "text/plain" ,
561
561
text : "Lorem ipsum dolor sit amet" ,
562
562
}
563
563
end
@@ -573,7 +573,7 @@ def call(message:, server_context: nil)
573
573
574
574
response = @server . handle ( request )
575
575
assert_equal (
576
- { contents : [ { uri : "example.com/my_resource" , mimeType : "text/plain" , text : "Lorem ipsum dolor sit amet" } ] } ,
576
+ { contents : { uri : "example.com/my_resource" , mimeType : "text/plain" , text : "Lorem ipsum dolor sit amet" } } ,
577
577
response [ :result ] ,
578
578
)
579
579
end
0 commit comments