Add automatic _meta parameter extraction support
This commit is contained in:
Eric Kreutzer
parent
4bf168840e
commit
50cd86c01c
1 changed files with 282 additions and 4 deletions
|
|
@ -414,5 +414,226 @@ module MCP
|
|||
assert_equal "FlexiblePrompt: Hello (context: present)",
|
||||
response[:result][:messages][0][:content][:text]
|
||||
end
|
||||
|
||||
test "tool receives _meta when provided in request params" do
|
||||
class ToolWithMeta < Tool
|
||||
tool_name "tool_with_meta"
|
||||
description "A tool that uses _meta"
|
||||
input_schema({ properties: { message: { type: "string" } }, required: ["message"] })
|
||||
|
||||
class << self
|
||||
def call(message:, server_context:)
|
||||
meta_info = server_context.dig(:_meta, :provider, :metadata)
|
||||
Tool::Response.new([
|
||||
{ type: "text", content: "Message: #{message}, Metadata: #{meta_info}" },
|
||||
])
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
server = Server.new(
|
||||
name: "test_server",
|
||||
tools: [ToolWithMeta],
|
||||
)
|
||||
|
||||
request = {
|
||||
jsonrpc: "2.0",
|
||||
id: 1,
|
||||
method: "tools/call",
|
||||
params: {
|
||||
name: "tool_with_meta",
|
||||
arguments: { message: "Hello" },
|
||||
_meta: {
|
||||
provider: {
|
||||
metadata: "test_value",
|
||||
},
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
response = server.handle(request)
|
||||
|
||||
assert response[:result]
|
||||
assert_equal "Message: Hello, Metadata: test_value",
|
||||
response[:result][:content][0][:content]
|
||||
end
|
||||
|
||||
test "_meta is nested within server_context" do
|
||||
class ToolWithNestedMeta < Tool
|
||||
tool_name "tool_with_nested_meta"
|
||||
description "A tool that uses nested _meta"
|
||||
input_schema({ properties: { message: { type: "string" } }, required: ["message"] })
|
||||
|
||||
class << self
|
||||
def call(message:, server_context:)
|
||||
user = server_context[:user]
|
||||
session_id = server_context.dig(:_meta, :session_id)
|
||||
Tool::Response.new([
|
||||
{ type: "text", content: "User: #{user}, Session: #{session_id}, Message: #{message}" },
|
||||
])
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
server = Server.new(
|
||||
name: "test_server",
|
||||
tools: [ToolWithNestedMeta],
|
||||
server_context: { user: "test_user", original_field: "value" },
|
||||
)
|
||||
|
||||
request = {
|
||||
jsonrpc: "2.0",
|
||||
id: 1,
|
||||
method: "tools/call",
|
||||
params: {
|
||||
name: "tool_with_nested_meta",
|
||||
arguments: { message: "Hello" },
|
||||
_meta: {
|
||||
session_id: "abc123",
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
response = server.handle(request)
|
||||
|
||||
assert response[:result]
|
||||
assert_equal "User: test_user, Session: abc123, Message: Hello",
|
||||
response[:result][:content][0][:content]
|
||||
end
|
||||
|
||||
test "_meta preserves original server_context" do
|
||||
class ToolPreservesContext < Tool
|
||||
tool_name "tool_preserves_context"
|
||||
description "A tool that checks context preservation"
|
||||
|
||||
class << self
|
||||
def call(server_context:)
|
||||
priority = server_context[:priority]
|
||||
meta_priority = server_context.dig(:_meta, :priority)
|
||||
Tool::Response.new([
|
||||
{ type: "text", content: "Context priority: #{priority}, Meta priority: #{meta_priority}" },
|
||||
])
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
server = Server.new(
|
||||
name: "test_server",
|
||||
tools: [ToolPreservesContext],
|
||||
server_context: { priority: "low" },
|
||||
)
|
||||
|
||||
request = {
|
||||
jsonrpc: "2.0",
|
||||
id: 1,
|
||||
method: "tools/call",
|
||||
params: {
|
||||
name: "tool_preserves_context",
|
||||
arguments: {},
|
||||
_meta: {
|
||||
priority: "high",
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
response = server.handle(request)
|
||||
|
||||
assert response[:result]
|
||||
assert_equal "Context priority: low, Meta priority: high", response[:result][:content][0][:content]
|
||||
end
|
||||
|
||||
test "prompt receives _meta when provided in request params" do
|
||||
class PromptWithMeta < Prompt
|
||||
prompt_name "prompt_with_meta"
|
||||
description "A prompt that uses _meta"
|
||||
arguments [Prompt::Argument.new(name: "message", required: true)]
|
||||
|
||||
class << self
|
||||
def template(args, server_context:)
|
||||
meta_info = server_context.dig(:_meta, :request_id)
|
||||
Prompt::Result.new(
|
||||
messages: [
|
||||
Prompt::Message.new(
|
||||
role: "user",
|
||||
content: Content::Text.new("Message: #{args[:message]}, Request ID: #{meta_info}"),
|
||||
),
|
||||
],
|
||||
)
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
server = Server.new(
|
||||
name: "test_server",
|
||||
prompts: [PromptWithMeta],
|
||||
)
|
||||
|
||||
request = {
|
||||
jsonrpc: "2.0",
|
||||
id: 1,
|
||||
method: "prompts/get",
|
||||
params: {
|
||||
name: "prompt_with_meta",
|
||||
arguments: { message: "Hello" },
|
||||
_meta: {
|
||||
request_id: "req_12345",
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
response = server.handle(request)
|
||||
|
||||
assert response[:result]
|
||||
assert_equal "Message: Hello, Request ID: req_12345",
|
||||
response[:result][:messages][0][:content][:text]
|
||||
end
|
||||
|
||||
test "_meta is nested within server_context for prompts" do
|
||||
class PromptWithNestedContext < Prompt
|
||||
prompt_name "prompt_with_nested_context"
|
||||
description "A prompt that uses nested context"
|
||||
arguments [Prompt::Argument.new(name: "message", required: true)]
|
||||
|
||||
class << self
|
||||
def template(args, server_context:)
|
||||
user = server_context[:user]
|
||||
trace_id = server_context.dig(:_meta, :trace_id)
|
||||
Prompt::Result.new(
|
||||
messages: [
|
||||
Prompt::Message.new(
|
||||
role: "user",
|
||||
content: Content::Text.new("User: #{user}, Trace: #{trace_id}, Message: #{args[:message]}"),
|
||||
),
|
||||
],
|
||||
)
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
server = Server.new(
|
||||
name: "test_server",
|
||||
prompts: [PromptWithNestedContext],
|
||||
server_context: { user: "prompt_user" },
|
||||
)
|
||||
|
||||
request = {
|
||||
jsonrpc: "2.0",
|
||||
id: 1,
|
||||
method: "prompts/get",
|
||||
params: {
|
||||
name: "prompt_with_nested_context",
|
||||
arguments: { message: "World" },
|
||||
_meta: {
|
||||
trace_id: "trace_xyz789",
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
response = server.handle(request)
|
||||
|
||||
assert response[:result]
|
||||
assert_equal "User: prompt_user, Trace: trace_xyz789, Message: World",
|
||||
response[:result][:messages][0][:content][:text]
|
||||
end
|
||||
end
|
||||
end
|
||||
|
|
|
|||
Loading…
Reference in a new issue