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

Use Capabilities object instead of private method

Browse source
This commit is contained in:
Kevin Fischer 2025-06-04 22:55:08 +09:00
commit 4a666ed0d9
No known key found for this signature in database
GPG key ID: 8E4CB46A9AC8D1C1
1 changed files with 9 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

22
lib/mcp/server.rb
View file

@ -46,6 +46,10 @@ module MCP
@resource_index = index_resources_by_uri(resources) @resource_index = index_resources_by_uri(resources)
@server_context = server_context @server_context = server_context
@configuration = MCP.configuration.merge(configuration) @configuration = MCP.configuration.merge(configuration)
@capabilities = Capabilities.new(capabilities)
@capabilities.support_tools if tools.any?
@capabilities.support_prompts if prompts.any?
@capabilities.support_resources if resources.any? || resource_templates.any?
@handlers = { @handlers = {
Methods::RESOURCES_LIST => method(:list_resources), Methods::RESOURCES_LIST => method(:list_resources),
@ -67,7 +71,7 @@ module MCP
end end
def capabilities def capabilities
@capabilities ||= determine_capabilities @capabilities.to_h
end end
def handle(request) def handle(request)
@ -93,6 +97,7 @@ module MCP
end end
def resources_list_handler(&block) def resources_list_handler(&block)
@capabilities.support_resources
@handlers[Methods::RESOURCES_LIST] = block @handlers[Methods::RESOURCES_LIST] = block
end end
@ -101,10 +106,12 @@ module MCP
end end
def resources_templates_list_handler(&block) def resources_templates_list_handler(&block)
@capabilities.support_resources
@handlers[Methods::RESOURCES_TEMPLATES_LIST] = block @handlers[Methods::RESOURCES_TEMPLATES_LIST] = block
end end
def tools_list_handler(&block) def tools_list_handler(&block)
@capabilities.support_tools
@handlers[Methods::TOOLS_LIST] = block @handlers[Methods::TOOLS_LIST] = block
end end
@ -113,6 +120,7 @@ module MCP
end end
def prompts_list_handler(&block) def prompts_list_handler(&block)
@capabilities.support_prompts
@handlers[Methods::PROMPTS_LIST] = block @handlers[Methods::PROMPTS_LIST] = block
end end
@ -160,18 +168,6 @@ module MCP
} }
end end
def determine_capabilities
defines_prompts = @prompts.any? || @handlers[Methods::PROMPTS_LIST] != method(:list_prompts)
defines_tools = @tools.any? || @handlers[Methods::TOOLS_LIST] != method(:list_tools)
defines_resources = @resources.any? || @handlers[Methods::RESOURCES_LIST] != method(:list_resources)
defines_resource_templates = @resource_templates.any? || @handlers[Methods::RESOURCES_TEMPLATES_LIST] != method(:list_resource_templates)
{
prompts: defines_prompts ? {} : nil,
resources: defines_resources || defines_resource_templates ? {} : nil,
tools: defines_tools ? {} : nil,
}.compact
end
def server_info def server_info
@server_info ||= { @server_info ||= {
name:, name:,