rasti-ai 1.1.0 → 1.2.1

data/lib/rasti/ai/tool_serializer.rb

@@ -0,0 +1,112 @@
+module Rasti
+  module AI
+    class ToolSerializer
+      class << self
+
+        def serialize(tool_class)
+          serialization = {
+            name: serialize_name(tool_class)
+          }
+
+          serialization[:description] = normalize_description(tool_class.description) if tool_class.respond_to? :description
+
+          serialization[:inputSchema] = serialize_form(tool_class.form) if tool_class.respond_to? :form
+
+          serialization
+          
+        rescue => ex
+          raise Errors::ToolSerializationError.new(tool_class), cause: ex
+        end
+
+        private
+
+        def serialize_name(tool_class)
+          Inflecto.underscore Inflecto.demodulize(tool_class.name)
+        end
+
+        def serialize_form(form_class)
+          serialized_attributes = form_class.attributes.each_with_object({}) do |attribute, hash|
+            hash[attribute.name] = serialize_attribute attribute
+          end
+
+          serialization = {
+            type: 'object',
+            properties: serialized_attributes
+          }
+
+          required_attributes = form_class.attributes.select { |a| a.option(:required) }
+
+          serialization[:required] = required_attributes.map(&:name) unless required_attributes.empty?
+
+          serialization
+        end
+
+        def serialize_attribute(attribute)
+          serialization = {}
+          
+          if attribute.option(:description)
+            serialization[:description] = normalize_description attribute.option(:description)
+          end
+          
+          type_serialization = serialize_type attribute.type
+          serialization.merge! type_serialization
+
+          if attribute.type.is_a? Types::Enum
+            values = "#{type_serialization[:enum].join(', ')}"
+            if serialization[:description]
+              serialization[:description] += " (#{values})"
+            else
+              serialization[:description] = values
+            end
+          end
+
+          serialization
+        end
+
+        def serialize_type(type)
+          if type == Types::String
+            {type: 'string'}
+
+          elsif type == Types::Integer
+            {type: 'integer'}
+
+          elsif type == Types::Float
+            {type: 'number'}
+
+          elsif type == Types::Boolean
+            {type: 'boolean'}
+
+          elsif type.is_a? Types::Time
+            {
+              type: 'string',
+              format: 'date'
+            }
+
+          elsif type.is_a? Types::Enum
+            {
+              type: 'string',
+              enum: type.values
+            }
+
+          elsif type.is_a? Types::Array
+            {
+              type: 'array',
+              items: serialize_type(type.type)
+            }
+
+          elsif type.is_a? Types::Model
+            serialize_form(type.model)
+
+          else
+            raise "Type not serializable #{type}"
+          end
+        end
+
+        def normalize_description(description)
+          description.split("\n").map(&:strip).join(' ').strip
+        end
+        
+      end
+    end
+  end
+end

data/lib/rasti/ai/version.rb

@@ -1,5 +1,5 @@
 module Rasti
   module AI
-    VERSION = '1.1.0'
+    VERSION = '1.2.1'
   end
 end

data/lib/rasti/ai.rb

@@ -17,6 +17,10 @@ module Rasti
 
     attr_config :logger, Logger.new(STDOUT)
 
+    attr_config :http_connect_timeout, 60
+    attr_config :http_read_timeout, 60
+    attr_config :http_max_retries, 3
+
     attr_config :openai_api_key, ENV['OPENAI_API_KEY']
     attr_config :openai_default_model, ENV['OPENAI_DEFAULT_MODEL']
 

data/rasti-ai.gemspec

@@ -24,9 +24,11 @@ Gem::Specification.new do |spec|
   spec.add_runtime_dependency 'class_config', '~> 0.0'
 
   spec.add_development_dependency 'rake', '~> 12.0'
+  spec.add_development_dependency 'rack-test', '~> 2.0'
   spec.add_development_dependency 'minitest', '~> 5.0', '< 5.11'
   spec.add_development_dependency 'minitest-colorin', '~> 0.1'
   spec.add_development_dependency 'minitest-line', '~> 0.6'
+  spec.add_development_dependency 'minitest-extended_assertions', '~> 1.0'
   spec.add_development_dependency 'simplecov', '~> 0.12'
   spec.add_development_dependency 'pry-nav', '~> 0.2'
   spec.add_development_dependency 'webmock', '~> 3.0'

data/spec/mcp/client_spec.rb

@@ -0,0 +1,256 @@
+require 'minitest_helper'
+
+describe Rasti::AI::MCP::Client do
+
+  let(:mcp_url) { 'https://mcp.server.ai/mcp' }
+
+  let(:client) { Rasti::AI::MCP::Client.new(url: mcp_url) }
+
+  let(:client_with_allowed_tools) { Rasti::AI::MCP::Client.new(url: mcp_url, allowed_tools: ['sum_tool']) }
+
+  def stub_mcp_request(method, params:{}, result:{}, error:nil)
+    request_body = {
+      jsonrpc: '2.0',
+      method: method,
+      params: params
+    }
+
+    response_body = {
+      jsonrpc: '2.0'
+    }
+
+    if error
+      response_body[:error] = error
+    else
+      response_body[:result] = result
+    end
+
+    stub_request(:post, mcp_url)
+      .with(
+        body: JSON.dump(request_body),
+        headers: {'Content-Type' => 'application/json'}
+      )
+      .to_return(
+        body: JSON.dump(response_body),
+        headers: {'Content-Type' => 'application/json'}
+      )
+  end
+
+  describe 'List tools' do
+
+    it 'Returns all tools when no allowed_tools' do
+      tools = [
+        {
+          'name' => 'hello_world_tool',
+          'description' => 'Hello World'
+        },
+        {
+          'name' => 'sum_tool',
+          'description' => 'Sum two numbers'
+        }
+      ]
+
+      stub_mcp_request 'tools/list', result: {'tools' => tools}
+
+      result = client.list_tools
+
+      assert_equal tools, result
+    end
+
+    it 'Returns empty list when no tools available' do
+      stub_mcp_request 'tools/list', result: {'tools' => []}
+
+      result = client.list_tools
+
+      assert_equal [], result
+    end
+
+    it 'Filters tools by allowed_tools' do
+      tools = [
+        {
+          'name' => 'hello_world_tool',
+          'description' => 'Hello World'
+        },
+        {
+          'name' => 'sum_tool',
+          'description' => 'Sum two numbers'
+        },
+        {
+          'name' => 'multiply_tool',
+          'description' => 'Multiply'
+        }
+      ]
+
+      stub_mcp_request 'tools/list', result: {'tools' => tools}
+
+      result = client_with_allowed_tools.list_tools
+
+      expected = [
+        {
+          'name' => 'sum_tool',
+          'description' => 'Sum two numbers'
+        }
+      ]
+
+      assert_equal expected, result
+    end
+
+    it 'Returns empty when no tools match allowed_tools' do
+      tools = [
+        {
+          'name' => 'hello_world_tool',
+          'description' => 'Hello World'
+        },
+        {
+          'name' => 'multiply_tool',
+          'description' => 'Multiply'
+        }
+      ]
+
+      stub_mcp_request 'tools/list', result: {'tools' => tools}
+
+      result = client_with_allowed_tools.list_tools
+
+      assert_equal [], result
+    end
+
+  end
+
+  describe 'Call tool' do
+
+    it 'Executes tool with arguments' do
+      params = {
+        name: 'sum_tool',
+        arguments: {
+          number_a: 1,
+          number_b: 2
+        }
+      }
+
+      result_content = {
+        'content' => [
+          {
+            'type' => 'text',
+            'text' => '{"result":3.0}'
+          }
+        ]
+      }
+
+      stub_mcp_request 'tools/call', params: params, result: result_content
+
+      result = client.call_tool 'sum_tool', number_a: 1, number_b: 2
+
+      assert_equal '{"type":"text","text":"{\"result\":3.0}"}', result
+    end
+
+    it 'Executes tool without arguments' do
+      params = {
+        name: 'hello_world_tool',
+        arguments: {}
+      }
+
+      result_content = {
+        'content' => [
+          {
+            'type' => 'text',
+            'text' => '{"text":"Hello world"}'
+          }
+        ]
+      }
+
+      stub_mcp_request 'tools/call', params: params, result: result_content
+
+      result = client.call_tool 'hello_world_tool'
+
+      assert_equal '{"type":"text","text":"{\"text\":\"Hello world\"}"}', result
+    end
+
+    it 'Raises error when tool not in allowed_tools' do
+      error = assert_raises RuntimeError do
+        client_with_allowed_tools.call_tool 'hello_world_tool'
+      end
+
+      assert_equal 'Invalid tool: hello_world_tool', error.message
+    end
+
+    it 'Allows tool call when in allowed_tools' do
+      params = {
+        name: 'sum_tool',
+        arguments: {
+          number_a: 5,
+          number_b: 3
+        }
+      }
+
+      result_content = {
+        'content' => [
+          {
+            'type' => 'text',
+            'text' => '{"result":8.0}'
+          }
+        ]
+      }
+
+      stub_mcp_request 'tools/call', params: params, result: result_content
+
+      result = client_with_allowed_tools.call_tool 'sum_tool', number_a: 5, number_b: 3
+
+      assert_equal '{"type":"text","text":"{\"result\":8.0}"}', result
+    end
+
+  end
+
+  describe 'Error handling' do
+
+    it 'Raises error when MCP returns error' do
+      error_response = {
+        'code' => -32603,
+        'message' => 'Tool not found'
+      }
+
+      params = {
+        name: 'nonexistent',
+        arguments: {}
+      }
+
+      stub_mcp_request 'tools/call', params: params, error: error_response
+
+      error = assert_raises RuntimeError do
+        client.call_tool 'nonexistent'
+      end
+
+      assert_equal 'MCP Error: Tool not found', error.message
+    end
+
+    it 'Raises error when result is missing' do
+      stub_request(:post, mcp_url)
+        .to_return(
+          status: 200,
+          body: JSON.dump({ jsonrpc: '2.0' }),
+          headers: { 'Content-Type' => 'application/json' }
+        )
+
+      error = assert_raises RuntimeError do
+        client.list_tools
+      end
+
+      assert_equal 'MCP Error: invalid result', error.message
+    end
+
+    it 'Raises error when response is invalid JSON' do
+      stub_request(:post, mcp_url)
+        .to_return(
+          status: 200,
+          body: 'invalid json{',
+          headers: {'Content-Type' => 'application/json'}
+        )
+
+      error = assert_raises JSON::ParserError do
+        client.list_tools
+      end
+
+      assert_match /unexpected token/, error.message
+    end
+
+  end
+end