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

Merge pull request #8 from kfischer-okarin/auto-determine-capabilities

Browse source 
Auto determine capabilities
This commit is contained in:
Topher Bullock 2025-05-21 14:23:11 -04:00 • committed by GitHub
commit 18f3e8761e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 81 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

22
lib/model_context_protocol/server.rb
View file

@ -22,7 +22,8 @@ module ModelContextProtocol
include Instrumentation
attr_accessor :name, :version, :tools, :prompts, :resources, :server_context, :configuration, :capabilities
attr_writer :capabilities
attr_accessor :name, :version, :tools, :prompts, :resources, :server_context, :configuration
def initialize(
name: "model_context_protocol",
@ -33,7 +34,7 @@ module ModelContextProtocol
resource_templates: [],
server_context: nil,
configuration: nil,
capabilities: { prompts: {}, resources: {}, tools: {} }
capabilities: nil
)
@name = name
@version = version
@ -44,7 +45,6 @@ module ModelContextProtocol
@resource_index = index_resources_by_uri(resources)
@server_context = server_context
@configuration = ModelContextProtocol.configuration.merge(configuration)
@capabilities = capabilities
@handlers = {
Methods::RESOURCES_LIST => method(:list_resources),
@ -64,6 +64,10 @@ module ModelContextProtocol
}
end
def capabilities
@capabilities ||= determine_capabilities
end
def handle(request)
JsonRpcHandler.handle(request) do |method|
handle_request(request, method)
@ -154,6 +158,18 @@ module ModelContextProtocol
}
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
@server_info ||= {
name:,