Remove transport-specific code from server
This commit is contained in:
parent
08be669b58
commit
8fbb2e86aa
3 changed files with 6 additions and 12 deletions
|
|
@ -94,7 +94,8 @@ server.resources_read_handler do |params|
|
||||||
end
|
end
|
||||||
|
|
||||||
# Create the Streamable HTTP transport
|
# Create the Streamable HTTP transport
|
||||||
server.transport = MCP::Transports::HTTP.new(server)
|
transport = MCP::Transports::HTTP.new(server)
|
||||||
|
server.transport = transport
|
||||||
|
|
||||||
# Create a logger for MCP-specific logging
|
# Create a logger for MCP-specific logging
|
||||||
mcp_logger = Logger.new($stdout)
|
mcp_logger = Logger.new($stdout)
|
||||||
|
|
@ -121,7 +122,7 @@ app = proc do |env|
|
||||||
|
|
||||||
# Handle the request
|
# Handle the request
|
||||||
puts "request: #{request.inspect}"
|
puts "request: #{request.inspect}"
|
||||||
response = server.handle_rack_request(request)
|
response = transport.handle_request(request)
|
||||||
|
|
||||||
# Log the MCP response details
|
# Log the MCP response details
|
||||||
_, _, body = response
|
_, _, body = response
|
||||||
|
|
|
||||||
|
|
@ -64,7 +64,8 @@ server.define_tool(
|
||||||
end
|
end
|
||||||
|
|
||||||
# Create the Streamable HTTP transport
|
# Create the Streamable HTTP transport
|
||||||
server.transport = MCP::Transports::HTTP.new(server)
|
transport = MCP::Transports::HTTP.new(server)
|
||||||
|
server.transport = transport
|
||||||
|
|
||||||
# Create a logger for MCP request/response logging
|
# Create a logger for MCP request/response logging
|
||||||
mcp_logger = Logger.new($stdout)
|
mcp_logger = Logger.new($stdout)
|
||||||
|
|
@ -98,7 +99,7 @@ app = proc do |env|
|
||||||
end
|
end
|
||||||
|
|
||||||
# Handle the request
|
# Handle the request
|
||||||
response = server.handle_rack_request(request)
|
response = transport.handle_request(request)
|
||||||
|
|
||||||
# Log response details
|
# Log response details
|
||||||
status, headers, body = response
|
status, headers, body = response
|
||||||
|
|
|
||||||
|
|
@ -98,14 +98,6 @@ module MCP
|
||||||
@prompts[prompt.name_value] = prompt
|
@prompts[prompt.name_value] = prompt
|
||||||
end
|
end
|
||||||
|
|
||||||
def handle_rack_request(request)
|
|
||||||
@transport.handle_request(request)
|
|
||||||
end
|
|
||||||
|
|
||||||
def close
|
|
||||||
@transport.close
|
|
||||||
end
|
|
||||||
|
|
||||||
def resources_list_handler(&block)
|
def resources_list_handler(&block)
|
||||||
@capabilities.support_resources
|
@capabilities.support_resources
|
||||||
@handlers[Methods::RESOURCES_LIST] = block
|
@handlers[Methods::RESOURCES_LIST] = block
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue