rasti-ai 1.1.0 → 1.2.1

data/spec/mcp/server_spec.rb

@@ -0,0 +1,372 @@
+require 'minitest_helper'
+
+describe Rasti::AI::MCP::Server do
+
+  class HelloWorldTool < Rasti::AI::Tool
+    def self.description
+      'Hello World'
+    end
+
+    def execute(form)
+      {text: 'Hello world'}
+    end
+  end
+
+  class SumTool < Rasti::AI::Tool
+    class Form < Rasti::Form
+      attribute :number_a, Rasti::Types::Float
+      attribute :number_b, Rasti::Types::Float
+    end
+
+    def execute(form)
+      {result: form.number_a + form.number_b}
+    end
+  end
+
+  class ErrorTool < Rasti::AI::Tool
+    def execute(form)
+      raise "Unexpected tool error"
+    end
+  end
+
+  include Rack::Test::Methods
+
+  let(:app) do
+    app = lambda { |env| [200, {'Content-Type' => 'text/plain'}, ['App response']] }
+    Rasti::AI::MCP::Server.new app
+  end
+
+  before do
+    Rasti::AI::MCP::Server.clear_tools
+    Rasti::AI::MCP::Server.restore_default_configuration
+  end
+
+  def post_mcp_request(method, params={})
+    request_data = {
+      jsonrpc: '2.0',
+      id: 1,
+      method: method
+    }
+    request_data[:params] = params unless params.empty?
+
+    post Rasti::AI::MCP::Server.relative_path, JSON.dump(request_data), CONTENT_TYPE: 'application/json'
+  end
+
+  def assert_jsonrpc_success(expected_result)
+    expected_response = {
+      jsonrpc: '2.0',
+      id: 1,
+      result: expected_result
+    }
+    
+    assert_equal 200, last_response.status
+    assert_equal 'application/json', last_response.content_type
+    assert_equal_json JSON.dump(expected_response), last_response.body
+  end
+
+  def assert_jsonrpc_error(expected_error)
+    expected_response = {
+      jsonrpc: '2.0',
+      id: 1,
+      error: expected_error
+    }
+    
+    assert_equal 200, last_response.status
+    assert_equal 'application/json', last_response.content_type
+    assert_equal_json JSON.dump(expected_response), last_response.body
+  end
+
+  describe 'Tool registration' do
+
+    it 'Register tool' do
+      Rasti::AI::MCP::Server.register_tool HelloWorldTool.new
+      Rasti::AI::MCP::Server.register_tool SumTool.new
+      
+      serializations = Rasti::AI::MCP::Server.tools_serializations
+
+      expeted_serializations = [
+        Rasti::AI::ToolSerializer.serialize(HelloWorldTool),
+        Rasti::AI::ToolSerializer.serialize(SumTool)
+      ]
+
+      assert_equal expeted_serializations, serializations
+    end
+
+    it 'Register duplicate tool raises error' do
+      Rasti::AI::MCP::Server.register_tool HelloWorldTool.new
+      
+      error = assert_raises RuntimeError do
+        Rasti::AI::MCP::Server.register_tool HelloWorldTool.new
+      end
+      
+      assert_equal 'Tool hello_world_tool already exist', error.message
+    end
+
+    it 'Tools serializations empty when no tools' do
+      assert_empty Rasti::AI::MCP::Server.tools_serializations
+    end
+
+    it 'Call tool executes registered tool' do
+      Rasti::AI::MCP::Server.register_tool HelloWorldTool.new
+      
+      result = Rasti::AI::MCP::Server.call_tool 'hello_world_tool'
+      
+      assert_equal '{"text":"Hello world"}', result
+    end
+
+    it 'Call tool not found raises error' do
+      error = assert_raises RuntimeError do
+        Rasti::AI::MCP::Server.call_tool 'non_existent', {}
+      end
+      
+      assert_equal 'Tool non_existent not found', error.message
+    end
+
+  end
+
+  describe 'Initialize request' do
+
+    it 'Returns protocol version and capabilities' do
+      post_mcp_request 'initialize'
+
+      expected_result = {
+        protocolVersion: '2024-11-05',
+        capabilities: {
+          tools: {
+            list: true,
+            call: true
+          }
+        },
+        serverInfo: {
+          name: 'MCP Server',
+          version: '1.0.0'
+        }
+      }
+
+      assert_jsonrpc_success expected_result
+    end
+
+    it 'Custom server configuration' do
+      Rasti::AI::MCP::Server.configure do |config|
+        config.server_name = 'Custom MCP Server'
+        config.server_version = '2.0.0'
+      end
+      
+      post_mcp_request 'initialize'
+
+      expected_result = {
+        protocolVersion: '2024-11-05',
+        capabilities: {
+          tools: {
+            list: true,
+            call: true
+          }
+        },
+        serverInfo: {
+          name: 'Custom MCP Server',
+          version: '2.0.0'
+        }
+      }
+
+      assert_jsonrpc_success expected_result
+    end
+
+  end
+
+  describe 'Tools list request' do
+
+    it 'Returns empty list when no tools' do
+      post_mcp_request 'tools/list'
+
+      expected_result = {
+        tools: []
+      }
+
+      assert_jsonrpc_success expected_result
+    end
+
+    it 'Returns registered tools' do
+      Rasti::AI::MCP::Server.register_tool HelloWorldTool.new
+      Rasti::AI::MCP::Server.register_tool SumTool.new
+
+      post_mcp_request 'tools/list'
+
+      expected_result = {
+        tools: [
+          Rasti::AI::ToolSerializer.serialize(HelloWorldTool),
+          Rasti::AI::ToolSerializer.serialize(SumTool)
+        ]
+      }
+
+      assert_jsonrpc_success expected_result
+    end
+
+  end
+
+  describe 'Tools call request' do
+
+    before do
+      Rasti::AI::MCP::Server.register_tool HelloWorldTool.new
+      Rasti::AI::MCP::Server.register_tool SumTool.new
+    end
+
+    it 'Executes tool with arguments' do
+      params = {
+        name: 'sum_tool',
+        arguments: {
+          number_a: 1,
+          number_b: 2
+        }
+      }
+
+      post_mcp_request 'tools/call', params
+
+      expected_result = {
+        content: [
+          {
+            type: 'text',
+            text: '{"result":3.0}'
+          }
+        ]
+      }
+
+      assert_jsonrpc_success expected_result
+    end
+
+    it 'Executes tool without arguments' do
+      params = {
+        name: 'hello_world_tool'
+      }
+
+      post_mcp_request 'tools/call', params
+
+      expected_result = {
+        content: [
+          {
+            type: 'text',
+            text: '{"text":"Hello world"}'
+          }
+        ]
+      }
+
+      assert_jsonrpc_success expected_result
+    end
+
+    it 'Tool call with nonexistent tool returns error' do
+      params = {
+        name: 'nonexistent'
+      }
+
+      post_mcp_request 'tools/call', params
+
+      expected_error = {
+        code: -32603,
+        message: 'Tool nonexistent not found'
+      }
+
+      assert_jsonrpc_error expected_error
+    end
+
+    it 'Tool execution error returns error response' do
+      Rasti::AI::MCP::Server.register_tool ErrorTool.new
+
+      params = {
+        name: 'error_tool'
+      }
+
+      post_mcp_request 'tools/call', params
+
+      expected_error = {
+        code: -32603,
+        message: 'Unexpected tool error'
+      }
+
+      assert_jsonrpc_error expected_error
+    end
+
+  end
+
+  describe 'Error handling' do
+
+    it 'Method not found' do
+      post_mcp_request 'invalid/method'
+
+      expected_error = {
+        code: -32601,
+        message: 'Method not found'
+      }
+
+      assert_jsonrpc_error expected_error      
+    end
+
+    it 'Invalid JSON returns 400' do
+      post '/mcp', 'invalid json{'
+      
+      assert_equal 400, last_response.status
+      assert_equal 'application/json', last_response.content_type
+
+      json_response = JSON.parse last_response.body
+      assert_equal -32700, json_response['error']['code']
+      assert_match /unexpected token/, json_response['error']['message']
+    end
+
+    it 'Unhandled exception returns 500' do
+      app.stub :handle_initialize, ->(_) { raise 'Unexpected server error' } do
+        post_mcp_request 'initialize'
+
+        expected_response = {
+          jsonrpc: '2.0',
+          id: nil,
+          error: {
+            code: -32603,
+            message: 'Unexpected server error'
+          }
+        }
+        
+        assert_equal 500, last_response.status
+        assert_equal 'application/json', last_response.content_type
+        assert_equal_json JSON.dump(expected_response), last_response.body
+      end
+    end
+
+  end
+
+  describe 'Middleware behavior' do
+
+    it 'Non MCP requests passed to app' do
+      get '/other/path'
+      
+      assert_equal 200, last_response.status
+      assert_equal 'App response', last_response.body
+    end
+
+    it 'GET requests to MCP path passed to app' do
+      get '/mcp'
+      
+      assert_equal 200, last_response.status
+      assert_equal 'App response', last_response.body
+    end
+
+    it 'POST to different path passed to app' do
+      post '/other/path'
+
+      assert_equal 200, last_response.status
+      assert_equal 'App response', last_response.body
+    end
+
+    it 'Custom relative path' do
+      Rasti::AI::MCP::Server.configure do |config|
+        config.relative_path = '/path/too/custom_mcp'
+      end
+
+      post '/mcp'
+      assert_equal 200, last_response.status
+      assert_equal 'App response', last_response.body 
+      
+      post_mcp_request 'tools/list'
+      assert_jsonrpc_success tools: []
+    end
+
+  end
+
+end

data/spec/minitest_helper.rb

@@ -1,7 +1,9 @@
 require 'coverage_helper'
 require 'minitest/autorun'
 require 'minitest/colorin'
+require 'minitest/extended_assertions'
 require 'webmock/minitest'
+require 'rack/test'
 require 'pry-nav'
 require 'rasti-ai'
 require 'securerandom'
@@ -13,6 +15,8 @@ require_relative 'support/helpers/resources'
 Rasti::AI.configure do |config|
   config.logger.level = Logger::FATAL
 
+  config.http_max_retries = 1
+
   config.openai_api_key = 'test_api_key'
   config.openai_default_model = 'gpt-test'
 end

data/spec/open_ai/assistant_spec.rb

@@ -163,11 +163,18 @@ describe Rasti::AI::OpenAI::Assistant do
     end
 
     def stub_client_request(role:, content:, response:, tools:[])
+      serialized_tools = tools.map do |tool|
+        {
+          type: 'function',
+          function: Rasti::AI::ToolSerializer.serialize(tool.class)
+        }
+      end
+      
       client.expect :chat_completions, response do |params|
         last_message = params[:messages].last
         last_message[:role] == role &&
           last_message[:content] == content &&
-          params[:tools] == tools.map { |t| Rasti::AI::OpenAI::ToolSerializer.serialize t.class }
+          params[:tools] == serialized_tools
       end
     end
 
@@ -295,4 +302,4 @@ describe Rasti::AI::OpenAI::Assistant do
 
   end
 
-end
+end

data/spec/{open_ai/tool_serializer_spec.rb → tool_serializer_spec.rb}

@@ -1,8 +1,8 @@
 require 'minitest_helper'
 
-describe Rasti::AI::OpenAI::ToolSerializer do
+describe Rasti::AI::ToolSerializer do
 
-  let(:serializer) { Rasti::AI::OpenAI::ToolSerializer }
+  let(:serializer) { Rasti::AI::ToolSerializer }
 
   def build_tool_class(form_class=nil)
     tool_class = Minitest::Mock.new
@@ -13,15 +13,12 @@ describe Rasti::AI::OpenAI::ToolSerializer do
 
   def build_serializaton(param_name:, param_type:)
     {
-      type: 'function',
-      function: {
-        name: 'call_custom_function',
-        parameters: {
-          type: 'object',
-          properties: {
-            param_name.to_sym => {
-              type: param_type
-            }
+      name: 'call_custom_function',
+      inputSchema: {
+        type: 'object',
+        properties: {
+          param_name.to_sym => {
+            type: param_type
           }
         }
       }
@@ -34,10 +31,7 @@ describe Rasti::AI::OpenAI::ToolSerializer do
     serialization = serializer.serialize tool_class
 
     expeted_serialization = {
-      type: 'function',
-      function: {
-        name: 'call_custom_function'
-      }
+      name: 'call_custom_function'
     }
 
     assert_equal expeted_serialization, serialization
@@ -52,11 +46,8 @@ describe Rasti::AI::OpenAI::ToolSerializer do
     serialization = serializer.serialize tool_class
 
     expeted_serialization = {
-      type: 'function',
-      function: {
-        name: 'call_custom_function',
-        description: 'Call custom function without arguments',
-      }
+      name: 'call_custom_function',
+      description: 'Call custom function without arguments',
     }
 
     assert_equal expeted_serialization, serialization
@@ -132,7 +123,7 @@ describe Rasti::AI::OpenAI::ToolSerializer do
         serialization = serializer.serialize tool_class
 
         expeted_serialization = build_serializaton param_name: 'timestamp', param_type: 'string'
-        expeted_serialization[:function][:parameters][:properties][:timestamp][:format] = 'date'
+        expeted_serialization[:inputSchema][:properties][:timestamp][:format] = 'date'
 
         assert_equal expeted_serialization, serialization
 
@@ -147,7 +138,8 @@ describe Rasti::AI::OpenAI::ToolSerializer do
         serialization = serializer.serialize tool_class
 
         expeted_serialization = build_serializaton param_name: 'option', param_type: 'string'
-        expeted_serialization[:function][:parameters][:properties][:option][:enum] = ['option_1', 'option_2']
+        expeted_serialization[:inputSchema][:properties][:option][:enum] = ['option_1', 'option_2']
+        expeted_serialization[:inputSchema][:properties][:option][:description] = 'option_1, option_2'
 
         assert_equal expeted_serialization, serialization
 
@@ -163,7 +155,7 @@ describe Rasti::AI::OpenAI::ToolSerializer do
         serialization = serializer.serialize tool_class
 
         expeted_serialization = build_serializaton param_name: 'form', param_type: 'object'
-        expeted_serialization[:function][:parameters][:properties][:form][:properties] = {
+        expeted_serialization[:inputSchema][:properties][:form][:properties] = {
           text: {type: 'string'},
           int: {type: 'integer'}
         }
@@ -183,7 +175,7 @@ describe Rasti::AI::OpenAI::ToolSerializer do
           serialization = serializer.serialize tool_class
 
           expeted_serialization = build_serializaton param_name: 'texts', param_type: 'array'
-          expeted_serialization[:function][:parameters][:properties][:texts][:items] = {type: 'string'}
+          expeted_serialization[:inputSchema][:properties][:texts][:items] = {type: 'string'}
 
           assert_equal expeted_serialization, serialization
 
@@ -198,7 +190,7 @@ describe Rasti::AI::OpenAI::ToolSerializer do
           serialization = serializer.serialize tool_class
 
           expeted_serialization = build_serializaton param_name: 'numbers', param_type: 'array'
-          expeted_serialization[:function][:parameters][:properties][:numbers][:items] = {type: 'number'}
+          expeted_serialization[:inputSchema][:properties][:numbers][:items] = {type: 'number'}
 
           assert_equal expeted_serialization, serialization
 
@@ -214,7 +206,7 @@ describe Rasti::AI::OpenAI::ToolSerializer do
           serialization = serializer.serialize tool_class
 
           expeted_serialization = build_serializaton param_name: 'forms', param_type: 'array'
-          expeted_serialization[:function][:parameters][:properties][:forms][:items] = {
+          expeted_serialization[:inputSchema][:properties][:forms][:items] = {
             type: 'object',
             properties: {
               text: {type: 'string'},
@@ -242,11 +234,11 @@ describe Rasti::AI::OpenAI::ToolSerializer do
       serialization = serializer.serialize tool_class
 
       expeted_serialization = build_serializaton param_name: 'form', param_type: 'object'
-      expeted_serialization[:function][:parameters][:properties] = {
+      expeted_serialization[:inputSchema][:properties] = {
         text: {type: 'string'},
         int: {type: 'integer'}
       }
-      expeted_serialization[:function][:parameters][:required] = [:text]
+      expeted_serialization[:inputSchema][:required] = [:text]
 
       assert_equal expeted_serialization, serialization
 
@@ -264,7 +256,7 @@ describe Rasti::AI::OpenAI::ToolSerializer do
       serialization = serializer.serialize tool_class
 
       expeted_serialization = build_serializaton param_name: 'form', param_type: 'object'
-      expeted_serialization[:function][:parameters][:properties] = {
+      expeted_serialization[:inputSchema][:properties] = {
         text: {
           description: 'Text param',
           type: 'string'

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rasti-ai
 version: !ruby/object:Gem::Version
-  version: 1.1.0
+  version: 1.2.1
 platform: ruby
 authors:
 - Gabriel Naiman
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-06-27 00:00:00.000000000 Z
+date: 2025-12-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: multi_require
@@ -80,6 +80,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '12.0'
+- !ruby/object:Gem::Dependency
+  name: rack-test
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
 - !ruby/object:Gem::Dependency
   name: minitest
   requirement: !ruby/object:Gem::Requirement
@@ -128,6 +142,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '0.6'
+- !ruby/object:Gem::Dependency
+  name: minitest-extended_assertions
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
 - !ruby/object:Gem::Dependency
   name: simplecov
   requirement: !ruby/object:Gem::Requirement
@@ -188,24 +216,31 @@ files:
 - lib/rasti-ai.rb
 - lib/rasti/ai.rb
 - lib/rasti/ai/errors.rb
+- lib/rasti/ai/mcp/client.rb
+- lib/rasti/ai/mcp/errors.rb
+- lib/rasti/ai/mcp/server.rb
 - lib/rasti/ai/open_ai/assistant.rb
 - lib/rasti/ai/open_ai/assistant_state.rb
 - lib/rasti/ai/open_ai/client.rb
 - lib/rasti/ai/open_ai/roles.rb
-- lib/rasti/ai/open_ai/tool_serializer.rb
+- lib/rasti/ai/tool.rb
+- lib/rasti/ai/tool_serializer.rb
 - lib/rasti/ai/version.rb
+- log/.gitkeep
 - rasti-ai.gemspec
 - spec/coverage_helper.rb
+- spec/mcp/client_spec.rb
+- spec/mcp/server_spec.rb
 - spec/minitest_helper.rb
 - spec/open_ai/assistant_spec.rb
 - spec/open_ai/client_spec.rb
-- spec/open_ai/tool_serializer_spec.rb
 - spec/resources/open_ai/basic_request.json
 - spec/resources/open_ai/basic_response.json
 - spec/resources/open_ai/tool_request.json
 - spec/resources/open_ai/tool_response.json
 - spec/support/helpers/erb.rb
 - spec/support/helpers/resources.rb
+- spec/tool_serializer_spec.rb
 homepage: https://github.com/gabynaiman/rasti-ai
 licenses:
 - MIT
@@ -231,13 +266,15 @@ specification_version: 4
 summary: AI for apps
 test_files:
 - spec/coverage_helper.rb
+- spec/mcp/client_spec.rb
+- spec/mcp/server_spec.rb
 - spec/minitest_helper.rb
 - spec/open_ai/assistant_spec.rb
 - spec/open_ai/client_spec.rb
-- spec/open_ai/tool_serializer_spec.rb
 - spec/resources/open_ai/basic_request.json
 - spec/resources/open_ai/basic_response.json
 - spec/resources/open_ai/tool_request.json
 - spec/resources/open_ai/tool_response.json
 - spec/support/helpers/erb.rb
 - spec/support/helpers/resources.rb
+- spec/tool_serializer_spec.rb