claude-agent-sdk 0.1.4 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4d8b669f4c3f02dea0763113551838bb0a3a5ec514125876f28dbc5d23ee44aa
-  data.tar.gz: 6ce126558a2d99e3b6aba03527d5b4ce8954ff3a99a13ee25f54dd6fa0141980
+  metadata.gz: db20a5dbe6ebc51b229f4de595489038a26a47d8109600131c45aa2bf27c6d2e
+  data.tar.gz: 563e16e88a48ce331cd48c6ebce3302f0c57292ee6563ebaec7af02bed29433d
 SHA512:
-  metadata.gz: 54e48e83a98845d56d6dc4c1843ebf8728fd74ffd3e7670e826c0fdbf3ef297359c18f5b27c2fac0757da0fab5e28d6c19ab084ac1842c5615d06cf52a0a94f8
-  data.tar.gz: 224aca0535d196fa3b348da69af9e4cbcc46276af8d36b7b58f6e35811704f93506ff31df7449e9aae9ebd252c8b5c035fa581dc4d500a5bade82d0d15f37ddc
+  metadata.gz: 288868f49db2968d92dc2250fe2ca91b7692431c92dc4fce41b04a35610858c146490da45178290877988a9210d67ebdf8257363f651e058a0bab534d24222fc
+  data.tar.gz: 183224c382c68b916da22c20c64d2f4fd625861baee265611bae8f752c25effcc7de2159a61cc65822cd62f24344f85b485cd1e6005ceff0e88ea2c34be4c30c

data/CHANGELOG.md

@@ -5,6 +5,28 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [0.2.0] - 2025-10-17
+
+### Changed
+- **BREAKING:** Updated minimum Ruby version from 3.0+ to 3.2+ (required by official MCP SDK)
+- **Major refactoring:** SDK MCP server now uses official Ruby MCP SDK (`mcp` gem v0.4) internally
+- Internal implementation migrated from custom MCP to wrapping official `MCP::Server`
+
+### Added
+- Official Ruby MCP SDK (`mcp` gem) as runtime dependency
+- Full MCP protocol compliance via official SDK
+- `handle_json` method for protocol-compliant JSON-RPC handling
+
+### Improved
+- Better long-term maintenance by leveraging official SDK updates
+- Aligned with Python SDK implementation pattern (using official MCP library)
+- All 86 tests passing with full backward compatibility maintained
+
+### Technical Details
+- Creates dynamic `MCP::Tool`, `MCP::Resource`, and `MCP::Prompt` classes from block-based definitions
+- User-facing API remains unchanged - no breaking changes for Ruby 3.2+ users
+- Maintains backward-compatible methods (`list_tools`, `call_tool`, etc.)
+
 ## [0.1.3] - 2025-10-15
 
 ### Added

data/README.md

@@ -13,7 +13,7 @@ Ruby SDK for Claude Agent. See the [Claude Agent SDK documentation](https://docs
 Add this line to your application's Gemfile:
 
 ```ruby
-gem 'claude-agent-sdk'
+gem 'claude-agent-sdk', '~> 0.2.0'
 ```
 
 And then execute:
@@ -29,7 +29,7 @@ gem install claude-agent-sdk
 ```
 
 **Prerequisites:**
-- Ruby 3.0+
+- Ruby 3.2+
 - Node.js
 - Claude Code 2.0.0+: `npm install -g @anthropic-ai/claude-code`
 
@@ -143,6 +143,8 @@ A **custom tool** is a Ruby proc/lambda that you can offer to Claude, for Claude
 
 Custom tools are implemented as in-process MCP servers that run directly within your Ruby application, eliminating the need for separate processes that regular MCP servers require.
 
+**Implementation**: This SDK uses the [official Ruby MCP SDK](https://github.com/modelcontextprotocol/ruby-sdk) (`mcp` gem) internally, providing full protocol compliance while offering a simpler block-based API for tool definition.
+
 For a complete example, see [examples/mcp_calculator.rb](examples/mcp_calculator.rb).
 
 #### Creating a Simple Tool

data/lib/claude_agent_sdk/query.rb

@@ -292,7 +292,10 @@ module ClaudeAgentSDK
     end
 
     def handle_sdk_mcp_request(server_name, message)
-      unless @sdk_mcp_servers.key?(server_name)
+      # Convert server_name to symbol if needed for hash lookup
+      server_key = @sdk_mcp_servers.key?(server_name) ? server_name : server_name.to_sym
+
+      unless @sdk_mcp_servers.key?(server_key)
         return {
           jsonrpc: '2.0',
           id: message[:id],
@@ -303,7 +306,7 @@ module ClaudeAgentSDK
         }
       end
 
-      server = @sdk_mcp_servers[server_name]
+      server = @sdk_mcp_servers[server_key]
       method = message[:method]
       params = message[:params] || {}
 

data/lib/claude_agent_sdk/sdk_mcp_server.rb

@@ -1,18 +1,18 @@
 # frozen_string_literal: true
 
+require 'mcp'
+
 module ClaudeAgentSDK
-  # SDK MCP Server - runs in-process within your Ruby application
+  # SDK MCP Server - wraps official MCP::Server with block-based API
   #
   # Unlike external MCP servers that run as separate processes, SDK MCP servers
   # run directly in your application's process, providing better performance
   # and simpler deployment.
   #
-  # Supports:
-  # - Tools: Executable functions that Claude can call
-  # - Resources: Data sources that can be read (files, databases, APIs, etc.)
-  # - Prompts: Reusable prompt templates with arguments
+  # This class wraps the official MCP Ruby SDK and provides a simpler block-based
+  # API for defining tools, resources, and prompts.
   class SdkMcpServer
-    attr_reader :name, :version, :tools, :resources, :prompts
+    attr_reader :name, :version, :tools, :resources, :prompts, :mcp_server
 
     def initialize(name:, version: '1.0.0', tools: [], resources: [], prompts: [])
       @name = name
@@ -20,12 +20,34 @@ module ClaudeAgentSDK
       @tools = tools
       @resources = resources
       @prompts = prompts
-      @tool_map = tools.each_with_object({}) { |tool, hash| hash[tool.name] = tool }
-      @resource_map = resources.each_with_object({}) { |res, hash| hash[res.uri] = res }
-      @prompt_map = prompts.each_with_object({}) { |prompt, hash| hash[prompt.name] = prompt }
+
+      # Create dynamic Tool classes from tool definitions
+      tool_classes = create_tool_classes(tools)
+
+      # Create dynamic Resource classes from resource definitions
+      resource_classes = create_resource_classes(resources)
+
+      # Create dynamic Prompt classes from prompt definitions
+      prompt_classes = create_prompt_classes(prompts)
+
+      # Create the official MCP::Server instance
+      @mcp_server = MCP::Server.new(
+        name: name,
+        version: version,
+        tools: tool_classes,
+        resources: resource_classes,
+        prompts: prompt_classes
+      )
+    end
+
+    # Handle a JSON-RPC request
+    # @param json_string [String] JSON-RPC request
+    # @return [String] JSON-RPC response
+    def handle_json(json_string)
+      @mcp_server.handle_json(json_string)
     end
 
-    # List all available tools
+    # List all available tools (for backward compatibility)
     # @return [Array<Hash>] Array of tool definitions
     def list_tools
       @tools.map do |tool|
@@ -37,12 +59,12 @@ module ClaudeAgentSDK
       end
     end
 
-    # Execute a tool by name
+    # Execute a tool by name (for backward compatibility)
     # @param name [String] Tool name
     # @param arguments [Hash] Tool arguments
     # @return [Hash] Tool result
     def call_tool(name, arguments)
-      tool = @tool_map[name]
+      tool = @tools.find { |t| t.name == name }
       raise "Tool '#{name}' not found" unless tool
 
       # Call the tool's handler
@@ -56,7 +78,7 @@ module ClaudeAgentSDK
       result
     end
 
-    # List all available resources
+    # List all available resources (for backward compatibility)
     # @return [Array<Hash>] Array of resource definitions
     def list_resources
       @resources.map do |resource|
@@ -69,11 +91,11 @@ module ClaudeAgentSDK
       end
     end
 
-    # Read a resource by URI
+    # Read a resource by URI (for backward compatibility)
     # @param uri [String] Resource URI
     # @return [Hash] Resource content
     def read_resource(uri)
-      resource = @resource_map[uri]
+      resource = @resources.find { |r| r.uri == uri }
       raise "Resource '#{uri}' not found" unless resource
 
       # Call the resource's reader
@@ -87,7 +109,7 @@ module ClaudeAgentSDK
       content
     end
 
-    # List all available prompts
+    # List all available prompts (for backward compatibility)
     # @return [Array<Hash>] Array of prompt definitions
     def list_prompts
       @prompts.map do |prompt|
@@ -99,12 +121,12 @@ module ClaudeAgentSDK
       end
     end
 
-    # Get a prompt by name
+    # Get a prompt by name (for backward compatibility)
     # @param name [String] Prompt name
     # @param arguments [Hash] Arguments to fill in the prompt template
     # @return [Hash] Prompt with filled-in arguments
     def get_prompt(name, arguments = {})
-      prompt = @prompt_map[name]
+      prompt = @prompts.find { |p| p.name == name }
       raise "Prompt '#{name}' not found" unless prompt
 
       # Call the prompt's generator
@@ -120,6 +142,177 @@ module ClaudeAgentSDK
 
     private
 
+    # Create dynamic Tool classes from tool definitions
+    def create_tool_classes(tools)
+      tools.map do |tool_def|
+        # Create a new class that extends MCP::Tool
+        Class.new(MCP::Tool) do
+          @tool_def = tool_def
+
+          class << self
+            attr_reader :tool_def
+
+            def name_value
+              @tool_def.name
+            end
+
+            def description_value
+              @tool_def.description
+            end
+
+            def input_schema_value
+              schema = convert_schema(@tool_def.input_schema)
+              MCP::Tool::InputSchema.new(
+                properties: schema[:properties] || {},
+                required: schema[:required] || []
+              )
+            end
+
+            def call(server_context: nil, **args)
+              # Filter out server_context and pass remaining args to handler
+              result = @tool_def.handler.call(args)
+
+              # Convert result to MCP::Tool::Response format
+              content = result[:content].map do |item|
+                {
+                  type: item[:type],
+                  text: item[:text]
+                }
+              end
+
+              MCP::Tool::Response.new(content)
+            end
+
+            private
+
+            def convert_schema(schema)
+              # If it's already a proper JSON schema, return it
+              if schema.is_a?(Hash) && schema[:type] && schema[:properties]
+                return schema
+              end
+
+              # Simple schema: hash mapping parameter names to types
+              if schema.is_a?(Hash)
+                properties = {}
+                schema.each do |param_name, param_type|
+                  properties[param_name] = type_to_json_schema(param_type)
+                end
+
+                return {
+                  type: 'object',
+                  properties: properties,
+                  required: properties.keys.map(&:to_s)
+                }
+              end
+
+              # Default fallback
+              { type: 'object', properties: {} }
+            end
+
+            def type_to_json_schema(type)
+              case type
+              when :string, String
+                { type: 'string' }
+              when :integer, Integer
+                { type: 'integer' }
+              when :float, Float
+                { type: 'number' }
+              when :boolean, TrueClass, FalseClass
+                { type: 'boolean' }
+              when :number
+                { type: 'number' }
+              else
+                { type: 'string' } # Default fallback
+              end
+            end
+          end
+        end
+      end
+    end
+
+    # Create dynamic Resource classes from resource definitions
+    def create_resource_classes(resources)
+      resources.map do |resource_def|
+        # Create a new class that extends MCP::Resource
+        Class.new(MCP::Resource) do
+          @resource_def = resource_def
+
+          class << self
+            attr_reader :resource_def
+
+            def uri
+              @resource_def.uri
+            end
+
+            def name
+              @resource_def.name
+            end
+
+            def description
+              @resource_def.description
+            end
+
+            def mime_type
+              @resource_def.mime_type
+            end
+
+            def read
+              result = @resource_def.reader.call
+
+              # Convert to MCP format
+              result[:contents].map do |content|
+                MCP::ResourceContents.new(
+                  uri: content[:uri],
+                  mime_type: content[:mimeType] || content[:mime_type],
+                  text: content[:text]
+                )
+              end
+            end
+          end
+        end
+      end
+    end
+
+    # Create dynamic Prompt classes from prompt definitions
+    def create_prompt_classes(prompts)
+      prompts.map do |prompt_def|
+        # Create a new class that extends MCP::Prompt
+        Class.new(MCP::Prompt) do
+          @prompt_def = prompt_def
+
+          class << self
+            attr_reader :prompt_def
+
+            def name
+              @prompt_def.name
+            end
+
+            def description
+              @prompt_def.description
+            end
+
+            def arguments
+              @prompt_def.arguments || []
+            end
+
+            def get(**args)
+              result = @prompt_def.generator.call(args)
+
+              # Convert to MCP format
+              {
+                messages: result[:messages].map do |msg|
+                  {
+                    role: msg[:role],
+                    content: msg[:content]
+                  }
+                end
+              }
+            end
+          end
+        end
+      end
+    end
+
     def convert_input_schema(schema)
       # If it's already a proper JSON schema, return it
       if schema.is_a?(Hash) && schema[:type] && schema[:properties]

data/lib/claude_agent_sdk/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module ClaudeAgentSDK
-  VERSION = '0.1.4'
+  VERSION = '0.2.1'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: claude-agent-sdk
 version: !ruby/object:Gem::Version
-  version: 0.1.4
+  version: 0.2.1
 platform: ruby
 authors:
 - Community Contributors
@@ -23,6 +23,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: mcp
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.4'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.4'
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -114,7 +128,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: 3.0.0
+      version: 3.2.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="