stack-service-base 0.0.63 → 0.0.65

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 13692ed5603f5f7f604c33ea62544337e03a91b763edb32608cbcecdc42b492b
-  data.tar.gz: 977236928246a312407ef70bfe316a605d57672eed48249c92eb36f1d3d32a88
+  metadata.gz: 93e5d152f7456e78c982d7d872cb726e864c8c7d8b590e138aadec698b34e144
+  data.tar.gz: 658cb94d84c307962c345bb5c7a6a342a90c2f7f8fa31f424e7b3dcf56041aad
 SHA512:
-  metadata.gz: d0966902c26ccda5d2573f37449ebc6c3e1a284d3585c0627a62009da90961828bc9ca8f90fe675e76a2af098d42b45efdca9b07cb713979870427d2bd8ab789
-  data.tar.gz: 519b03a4803d7af6c3360d29cfc6b39bc33fb219984be86c60244263f57eeaddca4822fc6b921962a5b4faa18f659258e020cfe02ee75a14139e10d0f82cf274
+  metadata.gz: 5d9c4d21e2628fe4122e52ff8f8f1525f558139c2398d0f2b1279aa51175357323e3aea8ec1e1319ef07496cd1d2d3555c6182286ba4f44ec8f2ee360d191377
+  data.tar.gz: eb1cec3620d180ba68569ba9a3a5faa68c6aa42ca9b0dfe971793f419532adaf38acece7cb3f6000c187c55af3535b3659a269477ae29313e66fa0b7b7364697

data/lib/stack-service-base/examples/mcp_config.ru

@@ -11,10 +11,26 @@ SERVICES = {
   }
 }
 
-require 'stack-service-base/mcp_processor'
-require 'stack-service-base/mcp_tool_registry'
+require 'stack-service-base/mcp_helper'
+helpers McpHelper
 
-MCP_PROCESSOR = McpProcessor.new
+Tool :search do
+  description 'Search for a term in the database'
+  input query: { type: "string", description: "Term to search for", required: true }
+  execute do |inputs|
+    query = inputs[:query]
+    { results: [{id:"doc-1",title:"...",url:"..."}] }
+  end
+end
+
+Tool :fetch do
+  description 'Fetch a resource from the database'
+  input resource_id: { type: "string", description: "Resource ID to fetch", required: true }
+  execute do |inputs|
+    id = inputs[:id]
+    { id: "doc-1", title: "...", text: "full text...", url: "https://example.com/doc",  metadata: { source: "vector_store" } }
+  end
+end
 
 Tool :service_status do
   description 'Check current status of a service'
@@ -24,12 +40,10 @@ Tool :service_status do
     service = SERVICES[service_name]
     rpc_error!(404, "Unknown service #{service_name}") unless service
     {
-      toolResult: {
-        service_name: service_name,
-        status: service[:status],
-        uptime_sec: service[:uptime],
-        last_restart: service[:last_restart].utc.iso8601,
-      }
+      service_name: service_name,
+      status: service[:status],
+      uptime_sec: service[:uptime],
+      last_restart: service[:last_restart].utc.iso8601,
     }
   end
 end
@@ -46,28 +60,13 @@ Tool :restart_service do
     service[:status]       = "running"
     service[:last_restart] = Time.now
     service[:uptime]       = 0
-
     {
-      toolResult: {
-        service_name: service_name,
-        status: service[:status],
-        restarted_at: service[:last_restart].utc.iso8601,
-        force: inputs.fetch(:force, false)
-      }
+      service_name: service_name,
+      status: service[:status],
+      restarted_at: service[:last_restart].utc.iso8601,
+      force: inputs.fetch(:force, false)
     }
   end
 end
 
-before { content_type :json }
-error McpProcessor::ParseError do |err|
-  status err.status
-  err.body
-end
-
-get  '/', &MCP_PROCESSOR.method(:root_endpoint)
-post '/' do
-  request.body.rewind
-  MCP_PROCESSOR.rpc_endpoint(request.body.read)
-end
-
 run Sinatra::Application

data/lib/stack-service-base/mcp_helper.rb

@@ -0,0 +1,47 @@
+require_relative 'mcp_processor'
+require_relative 'mcp_tool_registry'
+
+MCP_PROCESSOR = McpProcessor.new
+
+module McpHelper
+  def self.included(base)
+    base.class_eval do
+
+      error McpProcessor::ParseError do |err|
+        status err.status
+        err.body
+      end
+
+      get  '/mcp' do
+        content_type :json
+        MCP_PROCESSOR.root_endpoint
+      end
+
+      post '/mcp' do
+        content_type 'text/event-stream'
+        headers['Cache-Control'] = 'no-cache'
+        headers['X-Accel-Buffering'] = 'no'
+        headers['mcp-session-id'] = SecureRandom.uuid
+        request.body&.rewind
+        body = request.body.read.to_s
+
+        response_body =
+          begin
+            MCP_PROCESSOR.rpc_endpoint(body)
+          rescue McpProcessor::ParseError => e
+            status e.status
+            e.body
+          end
+
+        LOGGER.debug "request body: #{body}"
+        LOGGER.debug "response body: #{response_body}"
+
+        stream true do |s|
+          s.callback { LOGGER.debug "stream closed: #{s}" }
+          s << "event: message\ndata: #{response_body}\n\n"
+          s.close
+        end
+      end
+    end
+  end
+end

data/lib/stack-service-base/version.rb

@@ -1,3 +1,3 @@
 module StackServiceBase
-  VERSION = '0.0.63'
+  VERSION = '0.0.65'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: stack-service-base
 version: !ruby/object:Gem::Version
-  version: 0.0.63
+  version: 0.0.65
 platform: ruby
 authors:
 - Artyom B
@@ -401,6 +401,7 @@ files:
 - lib/stack-service-base/examples/mcp_config.ru
 - lib/stack-service-base/fiber_pool.rb
 - lib/stack-service-base/logging.rb
+- lib/stack-service-base/mcp_helper.rb
 - lib/stack-service-base/mcp_processor.rb
 - lib/stack-service-base/mcp_tool_registry.rb
 - lib/stack-service-base/nats_patch_1.rb