@@ -379,7 +379,7 @@ def call(message:, server_context: nil)
379379 test "#tools_call_handler sets the tools/call handler" do
380380 @server . tools_call_handler do |request |
381381 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
383383 end
384384
385385 request = {
@@ -391,7 +391,7 @@ def call(message:, server_context: nil)
391391
392392 response = @server . handle ( request )
393393 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" } )
395395 end
396396
397397 test "#handle prompts/list returns list of prompts" do
@@ -484,10 +484,10 @@ def call(message:, server_context: nil)
484484 test "#prompts_get_handler sets the prompts/get handler" do
485485 @server . prompts_get_handler do |request |
486486 prompt_name = request [ :name ]
487- return Prompt ::Result . new (
487+ Prompt ::Result . new (
488488 description : prompt_name ,
489489 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" ] ) ) ,
491491 ] ,
492492 ) . to_h
493493 end
@@ -501,10 +501,10 @@ def call(message:, server_context: nil)
501501
502502 response = @server . handle ( request )
503503 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" } } ] } ,
505505 response [ :result ] ,
506506 )
507- assert_instrumentation_data ( { method : "prompts/get" , prompt_name : "foo_bar_prompt" } )
507+ assert_instrumentation_data ( { method : "prompts/get" } )
508508 end
509509
510510 test "#handle resources/list returns a list of resources" do
@@ -555,9 +555,9 @@ def call(message:, server_context: nil)
555555
556556 test "#resources_read_handler sets the resources/read handler" do
557557 @server . resources_read_handler do |request |
558- return {
558+ {
559559 uri : request [ :uri ] ,
560- mime_type : "text/plain" ,
560+ mimeType : "text/plain" ,
561561 text : "Lorem ipsum dolor sit amet" ,
562562 }
563563 end
@@ -573,7 +573,7 @@ def call(message:, server_context: nil)
573573
574574 response = @server . handle ( request )
575575 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" } } ,
577577 response [ :result ] ,
578578 )
579579 end
0 commit comments