Move default capabilities into a private method
This commit is contained in:
parent
784b8b8548
commit
81fa3a9efa
1 changed files with 11 additions and 3 deletions
|
|
@ -20,7 +20,8 @@ module ModelContextProtocol
|
||||||
|
|
||||||
include Instrumentation
|
include Instrumentation
|
||||||
|
|
||||||
attr_accessor :name, :tools, :prompts, :resources, :server_context, :configuration, :capabilities
|
attr_writer :capabilities
|
||||||
|
attr_accessor :name, :tools, :prompts, :resources, :server_context, :configuration
|
||||||
|
|
||||||
def initialize(
|
def initialize(
|
||||||
name: "model_context_protocol",
|
name: "model_context_protocol",
|
||||||
|
|
@ -30,7 +31,7 @@ module ModelContextProtocol
|
||||||
resource_templates: [],
|
resource_templates: [],
|
||||||
server_context: nil,
|
server_context: nil,
|
||||||
configuration: nil,
|
configuration: nil,
|
||||||
capabilities: { prompts: {}, resources: {}, tools: {} }
|
capabilities: nil
|
||||||
)
|
)
|
||||||
@name = name
|
@name = name
|
||||||
@tools = tools.to_h { |t| [t.name_value, t] }
|
@tools = tools.to_h { |t| [t.name_value, t] }
|
||||||
|
|
@ -40,7 +41,6 @@ module ModelContextProtocol
|
||||||
@resource_index = index_resources_by_uri(resources)
|
@resource_index = index_resources_by_uri(resources)
|
||||||
@server_context = server_context
|
@server_context = server_context
|
||||||
@configuration = ModelContextProtocol.configuration.merge(configuration)
|
@configuration = ModelContextProtocol.configuration.merge(configuration)
|
||||||
@capabilities = capabilities
|
|
||||||
|
|
||||||
@handlers = {
|
@handlers = {
|
||||||
Methods::RESOURCES_LIST => method(:list_resources),
|
Methods::RESOURCES_LIST => method(:list_resources),
|
||||||
|
|
@ -60,6 +60,10 @@ module ModelContextProtocol
|
||||||
}
|
}
|
||||||
end
|
end
|
||||||
|
|
||||||
|
def capabilities
|
||||||
|
@capabilities ||= determine_capabilities
|
||||||
|
end
|
||||||
|
|
||||||
def handle(request)
|
def handle(request)
|
||||||
JsonRpcHandler.handle(request) do |method|
|
JsonRpcHandler.handle(request) do |method|
|
||||||
handle_request(request, method)
|
handle_request(request, method)
|
||||||
|
|
@ -150,6 +154,10 @@ module ModelContextProtocol
|
||||||
}
|
}
|
||||||
end
|
end
|
||||||
|
|
||||||
|
def determine_capabilities
|
||||||
|
{ prompts: {}, resources: {}, tools: {} }
|
||||||
|
end
|
||||||
|
|
||||||
def server_info
|
def server_info
|
||||||
@server_info ||= {
|
@server_info ||= {
|
||||||
name:,
|
name:,
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue