modelcontextprotocol/ruby-sdk
Watch
1
0
Fork
You've already forked ruby-sdk
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: add tool_arguments to instrumentation data

Browse source 
Include the tool arguments in instrumentation callback data when
tools/call is invoked. This allows consumers to track which arguments
were passed to each tool call for analytics and debugging purposes.
This commit is contained in:
Mikhail Dubov 2026-01-21 17:18:28 +00:00
commit 7412197e52
2 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

2
lib/mcp/server.rb
View file

@ -322,7 +322,7 @@ module MCP
end end
arguments = request[:arguments] || {} arguments = request[:arguments] || {}
add_instrumentation_data(tool_name: tool_name) add_instrumentation_data(tool_name: tool_name, tool_arguments: arguments)
if tool.input_schema&.missing_required_arguments?(arguments) if tool.input_schema&.missing_required_arguments?(arguments)
add_instrumentation_data(error: :missing_required_arguments) add_instrumentation_data(error: :missing_required_arguments)

8
test/mcp/server_test.rb
View file

@ -264,7 +264,7 @@ module MCP
response = @server.handle(request) response = @server.handle(request)
assert_equal tool_response.to_h, response[:result] assert_equal tool_response.to_h, response[:result]
assert_instrumentation_data({ method: "tools/call", tool_name: tool_name }) assert_instrumentation_data({ method: "tools/call", tool_name: tool_name, tool_arguments: tool_args })
end end
test "#handle tools/call returns error response with isError true if required tool arguments are missing" do test "#handle tools/call returns error response with isError true if required tool arguments are missing" do
@ -320,7 +320,7 @@ module MCP
raw_response = @server.handle_json(request) raw_response = @server.handle_json(request)
response = JSON.parse(raw_response, symbolize_names: true) if raw_response response = JSON.parse(raw_response, symbolize_names: true) if raw_response
assert_equal tool_response.to_h, response[:result] if response assert_equal tool_response.to_h, response[:result] if response
assert_instrumentation_data({ method: "tools/call", tool_name: tool_name }) assert_instrumentation_data({ method: "tools/call", tool_name: tool_name, tool_arguments: { arg: "value" } })
end end
test "#handle_json tools/call executes tool and returns result, when the tool is typed with Sorbet" do test "#handle_json tools/call executes tool and returns result, when the tool is typed with Sorbet" do
@ -389,7 +389,7 @@ module MCP
assert response[:result][:isError] assert response[:result][:isError]
assert_equal "text", response[:result][:content][0][:type] assert_equal "text", response[:result][:content][0][:type]
assert_match(/Internal error calling tool tool_that_raises: /, response[:result][:content][0][:text]) assert_match(/Internal error calling tool tool_that_raises: /, response[:result][:content][0][:text])
assert_instrumentation_data({ method: "tools/call", tool_name: "tool_that_raises" }) assert_instrumentation_data({ method: "tools/call", tool_name: "tool_that_raises", tool_arguments: { message: "test" } })
end end
test "registers tools with the same class name in different namespaces" do test "registers tools with the same class name in different namespaces" do
@ -446,7 +446,7 @@ module MCP
assert response[:result][:isError] assert response[:result][:isError]
assert_equal "text", response[:result][:content][0][:type] assert_equal "text", response[:result][:content][0][:type]
assert_match(/Internal error calling tool tool_that_raises: /, response[:result][:content][0][:text]) assert_match(/Internal error calling tool tool_that_raises: /, response[:result][:content][0][:text])
assert_instrumentation_data({ method: "tools/call", tool_name: "tool_that_raises" }) assert_instrumentation_data({ method: "tools/call", tool_name: "tool_that_raises", tool_arguments: { message: "test" } })
end end
test "#handle tools/call returns error response with isError true if input_schema raises an error during validation" do test "#handle tools/call returns error response with isError true if input_schema raises an error during validation" do