Add Content::EmbeddedResource class for embedded resource content type
The MCP spec defines an embedded resource content type, but the SDK had no corresponding class. Add `Content::EmbeddedResource` that wraps a resource object and serializes it via `to_h`. https://modelcontextprotocol.io/specification/2025-11-25/server/tools#embedded-resources
This commit is contained in:
parent
60f69351ee
commit
ba7a08aa04
2 changed files with 52 additions and 0 deletions
|
|
@ -42,5 +42,18 @@ module MCP
|
||||||
{ data: data, mimeType: mime_type, annotations: annotations, type: "audio" }.compact
|
{ data: data, mimeType: mime_type, annotations: annotations, type: "audio" }.compact
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
|
class EmbeddedResource
|
||||||
|
attr_reader :resource, :annotations
|
||||||
|
|
||||||
|
def initialize(resource, annotations: nil)
|
||||||
|
@resource = resource
|
||||||
|
@annotations = annotations
|
||||||
|
end
|
||||||
|
|
||||||
|
def to_h
|
||||||
|
{ resource: resource.to_h, annotations: annotations, type: "resource" }.compact
|
||||||
|
end
|
||||||
|
end
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
|
||||||
|
|
@ -54,5 +54,44 @@ module MCP
|
||||||
refute result.key?(:annotations)
|
refute result.key?(:annotations)
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
|
class EmbeddedResourceTest < ActiveSupport::TestCase
|
||||||
|
test "#to_h returns correct format per MCP spec" do
|
||||||
|
resource = Object.new
|
||||||
|
def resource.to_h
|
||||||
|
{ uri: "test://example", mimeType: "text/plain", text: "content" }
|
||||||
|
end
|
||||||
|
|
||||||
|
embedded = EmbeddedResource.new(resource)
|
||||||
|
result = embedded.to_h
|
||||||
|
|
||||||
|
assert_equal "resource", result[:type]
|
||||||
|
assert_equal({ uri: "test://example", mimeType: "text/plain", text: "content" }, result[:resource])
|
||||||
|
end
|
||||||
|
|
||||||
|
test "#to_h with annotations" do
|
||||||
|
resource = Object.new
|
||||||
|
def resource.to_h
|
||||||
|
{ uri: "test://x" }
|
||||||
|
end
|
||||||
|
|
||||||
|
embedded = EmbeddedResource.new(resource, annotations: { role: "data" })
|
||||||
|
result = embedded.to_h
|
||||||
|
|
||||||
|
assert_equal({ role: "data" }, result[:annotations])
|
||||||
|
end
|
||||||
|
|
||||||
|
test "#to_h without annotations omits the key" do
|
||||||
|
resource = Object.new
|
||||||
|
def resource.to_h
|
||||||
|
{ uri: "test://x" }
|
||||||
|
end
|
||||||
|
|
||||||
|
embedded = EmbeddedResource.new(resource)
|
||||||
|
result = embedded.to_h
|
||||||
|
|
||||||
|
refute result.key?(:annotations)
|
||||||
|
end
|
||||||
|
end
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue