rasti-ai 1.0.1 → 1.2.0

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.0.1
+  version: 1.2.0
 platform: ruby
 authors:
 - Gabriel Naiman
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-05-26 00:00:00.000000000 Z
+date: 2025-11-18 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

data/lib/rasti/ai/open_ai/tool_serializer.rb

@@ -1,111 +0,0 @@
-module Rasti
-  module AI
-    module OpenAI
-      class ToolSerializer
-        class << self
-
-          def serialize(tool_class)
-            {
-              type: 'function',
-              function: serialize_function(tool_class)
-            }
-
-          rescue => ex
-            raise Errors::ToolSerializationError.new(tool_class), cause: ex
-          end
-
-          private
-
-          def serialize_function(tool_class)
-            serialization = {
-              name: serialize_name(tool_class)
-            }
-
-            serialization[:description] = normalize_description(tool_class.description) if tool_class.respond_to? :description
-
-            serialization[:parameters] = serialize_form(tool_class.form) if tool_class.respond_to? :form
-
-            serialization
-          end
-
-          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
-
-            serialization.merge! serialize_type(attribute.type)
-
-            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
-end