ask-mcp 0.3.0 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 90e04370e4d5b1136abb513220b68beb46a5ca2c74d8f7ca694fa59297f8f32e
-  data.tar.gz: 17f2accea77cefc0ea51fedcc3879e219fff47760999b0ba39e0e168d30dbfe1
+  metadata.gz: ae155e99332cb2a3fcec6b65c6d0ac20aa9b2b81b27f1e3aa3c667dc0c7a42fb
+  data.tar.gz: af84c5f5aae82c640e2c12316cde40444fe7733d6ad5aaddd6780242b98e1a1c
 SHA512:
-  metadata.gz: d8a481b9467e8f4d6d4d305da61346855f4d27ac94de77f109ac528e704599c18aee5a9c5deb64f7edb71a71b4e446cc1801149df21891d15eefdcb441eb1d09
-  data.tar.gz: b78ccc9a910fbfbeaf2e273ce6169279e4195624871f89356d6564ab6cd46380dd876951544e90b2f3b3b57328ae00180c6d218e1d66afa5416090c5703e1560
+  metadata.gz: 62f473ff0fdc35b25adcdadf3aeb2a65c330767f7b73cc85bed204a32bff7fcc28f0348c2ee3f8e8ae1e3d7eaf3977714af67d67038c1be990d9feca2199d81f
+  data.tar.gz: e511364526282fcb429740dbd65bbce3e51d0bde06a06cb672e33a0c30b1b09e1971c663145f1b272d7e01015060349447bb7d53081b8dd9813dea02f19591e9

data/CHANGELOG.md

@@ -1,37 +1,45 @@
+## [0.4.0] - 2026-06-26
+
+### Added
+- Request ID deduplication — retried `tools/call` with the same ID returns
+  cached result instead of re-executing the tool.
+- SIGTERM/SIGHUP graceful shutdown — closes stdin to unblock the read loop
+  so the current tool call can finish before the process exits.
+- Configurable `tool_timeout` option — prevents hung tool calls from
+  blocking the server indefinitely.
+- `ping` handler for MCP keepalive support.
+- 10 new tests: request dedup (3), tool timeout (2), SIGTERM (1),
+  multiline JSON safety (3), ping (1) — 180 total.
+
+### Changed
+- `Server::Stdio` now echoes back the client's requested protocol version
+  instead of hardcoding a value.
+
 ## [0.3.0] - 2026-06-26
 
 ### Changed
-- Renamed `Adapters::AskToolServer` → `Adapters::ToolServer` — the adapter is
-  now duck-typed and works with any Ruby object, not just Ask::Tool instances.
-- README fully rewritten: leads with a general-purpose duck-typed server example
-  instead of ask-tools-specific code. ask-rb integration is now a sub-section.
-- Gemspec description updated to reflect general-purpose positioning.
+- Renamed `Adapters::AskToolServer` → `Adapters::ToolServer` — duck-typed,
+  works with any Ruby object, not just Ask::Tool instances.
+- README rewritten to lead with general-purpose duck-typed server example.
+- ask-rb integration is now a documented sub-section.
 
 ## [0.2.0] - 2026-06-26
 
 ### Added
-- Server Runtime: `Ask::MCP::Server::Stdio` — run as an MCP server over stdio
-  with full initialize handshake, tool discovery, and tool call dispatch.
-- `Adapters::AskToolServer` — converts tools to MCP definitions and dispatches calls.
-- `Ask::MCP::Server.start_stdio` entry point for easy one-line server setup.
-
-### Changed
-- 56 new tests across adapter, server stdio, integration, and start_stdio (170 total).
+- Server Runtime: `Ask::MCP::Server::Stdio` with full initialize handshake.
+- `Server.start_stdio` entry point for one-line server setup.
+- 56 new tests (170 total at time of release).
 
 ## [0.1.1] - 2026-06-25
 
 ### Changed
-- Major test expansion: Transport tests, MessagesParser, OAuth, Client, Server, Tool, Messages, integration. Rubocop, overcommit, CI matrix, SimpleCov.
+- Major test expansion across all modules.
 
 ## [0.1.0] - 2026-06-10
 
 ### Added
-- Core MCP client with full JSON-RPC 2.0 message layer
-- stdio, SSE, and Streamable HTTP transports
-- Tool, Resource, and Prompt data models
-- Client lifecycle, tool calling, resource reading, prompt retrieval
-- Token-based and OAuth 2.1 authentication
-- Ask::Tool adapter for ask-agent integration
-- Thread-safe request/response matching with configurable timeouts
-- Server-side notifications handling
-- Factory methods: `from_stdio`, `from_sse`, `from_http`, `connect`
+- Core MCP client with JSON-RPC 2.0 message layer.
+- stdio, SSE, and Streamable HTTP transports.
+- Tool, Resource, Prompt data models.
+- OAuth 2.1 and token-based authentication.
+- ask-agent integration via AskTool adapter.

data/lib/ask/mcp/server/stdio.rb

@@ -1,70 +1,80 @@
 # frozen_string_literal: true
 
 require "json"
+require "timeout"
 
 module Ask
   module MCP
     class Server
       # MCP server over stdio transport.
-      #
-      # Reads JSON-RPC 2.0 messages from stdin and writes responses to stdout.
-      # Designed to be spawned as a child process by MCP clients (Codex Desktop,
-      # Claude Code, etc.) with command/args config.
-      #
-      # Usage:
-      #   Server::Stdio.new(name: "my-server", tools: my_tools).start
-      #
       class Stdio
-        PROTOCOL_VERSION = "0.1.0"
+        MAX_RESULT_CACHE = 100
 
         attr_reader :name, :tools, :capabilities, :resources, :prompts
 
-        def initialize(name:, tools: [], capabilities: {}, resources: {}, prompts: {}, debug: false)
+        def initialize(name:, tools: [], capabilities: {}, resources: {}, prompts: {},
+                       debug: false, tool_timeout: nil)
           @name = name
           @capabilities = capabilities
           @resources = resources
           @prompts = prompts
           @debug = debug
+          @tool_timeout = tool_timeout
 
           @adapter = Adapters::ToolServer.new(tools || [])
           @initialized = false
           @running = false
+          @shutdown_requested = false
+          @result_cache = {}
         end
 
-        # Start the server (blocking). Processes stdin until EOF.
         def start
           @running = true
           $stdout.sync = true
 
+          # Graceful shutdown: on TERM/HUP, set flag and close stdin to
+          # unblock the read loop so the current tool call can finish.
+          trap("TERM") { graceful_shutdown }
+          trap("HUP")  { graceful_shutdown }
+
           debug_log "Server starting: #{@name} (PID #{Process.pid})"
           debug_log "Tools: #{@adapter.definitions.map { |d| d[:name] }.join(', ')}"
 
-          while @running && (line = $stdin.gets)
+          while @running && !@shutdown_requested && (line = $stdin.gets)
             line = line.strip
             next if line.empty?
-
             process_line(line)
           end
 
           debug_log "stdin closed — exiting"
         rescue Errno::EBADF, IOError
-          # stdin closed externally
+          # stdin closed externally (e.g. from trap handler)
+        rescue SignalException
+          # SIGTERM or SIGHUP during blocked read
         ensure
           @running = false
+          @shutdown_requested = false
+          trap("TERM", "DEFAULT")
+          trap("HUP", "DEFAULT")
         end
 
-        # Stop the server (non-blocking, closes stdin to unblock the read loop).
         def stop
           @running = false
         end
 
-        # Is the server still running?
         def running?
           @running
         end
 
         private
 
+        def graceful_shutdown
+          @shutdown_requested = true
+          # Close stdin to unblock $stdin.gets so the signal handler
+          # returns promptly and the process exits cleanly.
+          $stdin.close rescue nil
+        end
+
         def process_line(line)
           msg = JSON.parse(line, symbolize_names: true)
           handle_message(msg)
@@ -90,17 +100,20 @@ module Ask
           when "tools/call"
             return send_error(id, -32000, "Server not initialized") unless @initialized
             handle_tool_call(id, params)
+          when "ping"
+            send_result(id, {}) if has_id
           else
             debug_log "Unknown method: #{method}"
             send_error(id, -32601, "Method not found: #{method}") if has_id
           end
         end
 
-        def handle_initialize(id, _params)
+        def handle_initialize(id, params)
           @initialized = true
-          debug_log "Handling initialize (id=#{id.inspect})"
+          client_version = params[:protocolVersion] || PROTOCOL_VERSION
+          debug_log "Handling initialize (id=#{id.inspect}, version=#{client_version})"
           send_result(id, {
-            protocolVersion: PROTOCOL_VERSION,
+            protocolVersion: client_version,
             capabilities: @capabilities,
             serverInfo: {
               name: @name,
@@ -111,20 +124,41 @@ module Ask
         end
 
         def handle_tools_list(id)
-          debug_log "Handling tools/list (id=#{id.inspect})"
           defs = @adapter.definitions
           debug_log "tools/list returning #{defs.length} tool definitions"
           send_result(id, { tools: defs })
         end
 
         def handle_tool_call(id, params)
+          cache_key = id.to_s
+
+          # Return cached result for retried requests (same ID, already processed)
+          if @result_cache.key?(cache_key)
+            debug_log "Returning cached result for id=#{id}"
+            return send_result(id, @result_cache[cache_key])
+          end
+
           tool_name = params[:name].to_s
           arguments = params[:arguments] || {}
 
           debug_log "Handling tools/call: #{tool_name} (id=#{id.inspect})"
 
-          result = @adapter.call(tool_name, arguments)
+          result = if @tool_timeout
+                     Timeout.timeout(@tool_timeout) { @adapter.call(tool_name, arguments) }
+                   else
+                     @adapter.call(tool_name, arguments)
+                   end
+
+          @result_cache[cache_key] = result
+          trim_cache
+
           send_result(id, result)
+        rescue Timeout::Error
+          debug_log "Tool call timed out: #{tool_name}"
+          send_result(id, {
+            content: [{ type: "text", text: "Tool call timed out: #{tool_name}" }],
+            isError: true
+          })
         end
 
         def send_result(id, result)
@@ -140,6 +174,11 @@ module Ask
           ts = Time.now.strftime("%H:%M:%S.%L")
           $stderr.puts "[#{ts}] [ask-mcp] #{msg}"
         end
+
+        def trim_cache
+          return if @result_cache.size <= MAX_RESULT_CACHE
+          @result_cache.shift(@result_cache.size - MAX_RESULT_CACHE)
+        end
       end
     end
   end

data/lib/ask/mcp/version.rb

@@ -1,5 +1,5 @@
 module Ask
   module MCP
-    VERSION = "0.3.0"
+    VERSION = "0.4.0"
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ask-mcp
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.4.0
 platform: ruby
 authors:
 - Kaka Ruto