Merge pull request #393 from yuki3738/add_size_field_to_resource
Add `size` field to `MCP::Resource` per MCP specification
This commit is contained in:
commit
ac07f8f3a7
2 changed files with 22 additions and 2 deletions
|
|
@ -5,15 +5,16 @@ require_relative "resource/embedded"
|
|||
|
||||
module MCP
|
||||
class Resource
|
||||
attr_reader :uri, :name, :title, :description, :icons, :mime_type, :meta
|
||||
attr_reader :uri, :name, :title, :description, :icons, :mime_type, :size, :meta
|
||||
|
||||
def initialize(uri:, name:, title: nil, description: nil, icons: [], mime_type: nil, meta: nil)
|
||||
def initialize(uri:, name:, title: nil, description: nil, icons: [], mime_type: nil, size: nil, meta: nil)
|
||||
@uri = uri
|
||||
@name = name
|
||||
@title = title
|
||||
@description = description
|
||||
@icons = icons
|
||||
@mime_type = mime_type
|
||||
@size = size
|
||||
@meta = meta
|
||||
end
|
||||
|
||||
|
|
@ -25,6 +26,7 @@ module MCP
|
|||
description: description,
|
||||
icons: icons&.then { |icons| icons.empty? ? nil : icons.map(&:to_h) },
|
||||
mimeType: mime_type,
|
||||
size: size,
|
||||
_meta: meta,
|
||||
}.compact
|
||||
end
|
||||
|
|
|
|||
|
|
@ -49,5 +49,23 @@ module MCP
|
|||
|
||||
assert_equal meta, resource.to_h[:_meta]
|
||||
end
|
||||
|
||||
test "#to_h omits size when nil" do
|
||||
resource = Resource.new(uri: "file:///test.txt", name: "resource_without_size")
|
||||
|
||||
refute resource.to_h.key?(:size)
|
||||
end
|
||||
|
||||
test "#to_h includes size when present" do
|
||||
resource = Resource.new(uri: "file:///test.txt", name: "resource_with_size", size: 12_345)
|
||||
|
||||
assert_equal 12_345, resource.to_h[:size]
|
||||
end
|
||||
|
||||
test "#to_h includes size when zero" do
|
||||
resource = Resource.new(uri: "file:///empty.txt", name: "empty_resource", size: 0)
|
||||
|
||||
assert_equal 0, resource.to_h[:size]
|
||||
end
|
||||
end
|
||||
end
|
||||
|
|
|
|||
Loading…
Reference in a new issue