From 6e782098eccccdf0920801dcf74378126dc462b9 Mon Sep 17 00:00:00 2001 From: Christopher Butcher Date: Fri, 16 May 2025 14:26:17 -0400 Subject: [PATCH] Update ToolTest to use kwargs in example tool call methods --- test/model_context_protocol/tool_test.rb | 15 ++++----------- 1 file changed, 4 insertions(+), 11 deletions(-) diff --git a/test/model_context_protocol/tool_test.rb b/test/model_context_protocol/tool_test.rb index f8eda3c..1b96022 100644 --- a/test/model_context_protocol/tool_test.rb +++ b/test/model_context_protocol/tool_test.rb @@ -18,7 +18,7 @@ module ModelContextProtocol ) class << self - def call(message, server_context: nil) + def call(message:, server_context: nil) Tool::Response.new([{ type: "text", content: "OK" }]) end end @@ -43,7 +43,7 @@ module ModelContextProtocol test "#call invokes the tool block and returns the response" do tool = TestTool - response = tool.call("test") + response = tool.call(message: "test") assert_equal response.content, [{ type: "text", content: "OK" }] assert_equal response.is_error, false end @@ -210,26 +210,19 @@ module ModelContextProtocol tool_name "test_tool" description "a test tool for testing" input_schema({ properties: { message: { type: "string" } }, required: ["message"] }) - annotations( - title: "Test Tool", - read_only_hint: true, - destructive_hint: false, - idempotent_hint: true, - open_world_hint: false, - ) class << self extend T::Sig sig { params(message: String, server_context: T.nilable(T.untyped)).returns(Tool::Response) } - def call(message, server_context: nil) + def call(message:, server_context: nil) Tool::Response.new([{ type: "text", content: "OK" }]) end end end tool = TypedTestTool - response = tool.call("test") + response = tool.call(message: "test") assert_equal response.content, [{ type: "text", content: "OK" }] assert_equal response.is_error, false end