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

Allow specifying a server version

Browse source
This commit is contained in:
Kevin Fischer 2025-05-13 23:51:02 +09:00
commit ea4828987c
No known key found for this signature in database
GPG key ID: 8E4CB46A9AC8D1C1
2 changed files with 20 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

8
lib/model_context_protocol/server.rb
View file

@ -6,6 +6,8 @@ require_relative "methods"
module ModelContextProtocol module ModelContextProtocol
class Server class Server
DEFAULT_VERSION = "0.1.0"
class RequestHandlerError < StandardError class RequestHandlerError < StandardError
attr_reader :error_type attr_reader :error_type
attr_reader :original_error attr_reader :original_error
@ -20,10 +22,11 @@ module ModelContextProtocol
include Instrumentation include Instrumentation
attr_accessor :name, :tools, :prompts, :resources, :server_context, :configuration, :capabilities attr_accessor :name, :version, :tools, :prompts, :resources, :server_context, :configuration, :capabilities
def initialize( def initialize(
name: "model_context_protocol", name: "model_context_protocol",
version: DEFAULT_VERSION,
tools: [], tools: [],
prompts: [], prompts: [],
resources: [], resources: [],
@ -33,6 +36,7 @@ module ModelContextProtocol
capabilities: { prompts: {}, resources: {}, tools: {} } capabilities: { prompts: {}, resources: {}, tools: {} }
) )
@name = name @name = name
@version = version
@tools = tools.to_h { |t| [t.name_value, t] } @tools = tools.to_h { |t| [t.name_value, t] }
@prompts = prompts.to_h { |p| [p.name_value, p] } @prompts = prompts.to_h { |p| [p.name_value, p] }
@resources = resources @resources = resources
@ -153,7 +157,7 @@ module ModelContextProtocol
def server_info def server_info
@server_info ||= { @server_info ||= {
name:, name:,
version: ModelContextProtocol::VERSION, version:,
} }
end end

15
test/model_context_protocol/server_test.rb
View file

@ -51,6 +51,7 @@ module ModelContextProtocol
@server = Server.new( @server = Server.new(
name: @server_name, name: @server_name,
version: "1.2.3",
tools: [@tool, @tool_that_raises], tools: [@tool, @tool_that_raises],
prompts: [@prompt], prompts: [@prompt],
resources: [@resource], resources: [@resource],
@ -120,7 +121,7 @@ module ModelContextProtocol
}, },
"serverInfo": { "serverInfo": {
"name": @server_name, "name": @server_name,
"version": ModelContextProtocol::VERSION, "version": "1.2.3",
}, },
}, },
} }
@ -648,6 +649,18 @@ module ModelContextProtocol
assert_equal Configuration::DEFAULT_PROTOCOL_VERSION, response[:result][:protocolVersion] assert_equal Configuration::DEFAULT_PROTOCOL_VERSION, response[:result][:protocolVersion]
end end
test "server uses default version when not configured" do
server = Server.new(name: "test_server")
request = {
jsonrpc: "2.0",
method: "initialize",
id: 1,
}
response = server.handle(request)
assert_equal Server::DEFAULT_VERSION, response[:result][:serverInfo][:version]
end
test "#define_tool adds a tool to the server" do test "#define_tool adds a tool to the server" do
@server.define_tool( @server.define_tool(
name: "defined_tool", name: "defined_tool",