rasti-ai 1.0.0

data/spec/open_ai/tool_serializer_spec.rb

@@ -0,0 +1,297 @@
+require 'minitest_helper'
+
+describe Rasti::AI::OpenAI::ToolSerializer do
+
+  let(:serializer) { Rasti::AI::OpenAI::ToolSerializer }
+
+  def build_tool_class(form_class=nil)
+    tool_class = Minitest::Mock.new
+    tool_class.expect :name, 'CallCustomFunction'
+    tool_class.expect :form, form_class if form_class
+    tool_class
+  end
+
+  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
+            }
+          }
+        }
+      }
+    }
+  end
+
+  it 'Basic tool' do
+    tool_class = build_tool_class
+
+    serialization = serializer.serialize tool_class
+
+    expeted_serialization = {
+      type: 'function',
+      function: {
+        name: 'call_custom_function'
+      }
+    }
+
+    assert_equal expeted_serialization, serialization
+
+    tool_class.verify
+  end
+
+  it 'With description' do
+    tool_class = build_tool_class
+    tool_class.expect :description, 'Call custom function without arguments'
+
+    serialization = serializer.serialize tool_class
+
+    expeted_serialization = {
+      type: 'function',
+      function: {
+        name: 'call_custom_function',
+        description: 'Call custom function without arguments',
+      }
+    }
+
+    assert_equal expeted_serialization, serialization
+
+    tool_class.verify
+  end
+
+  describe 'Arguments' do
+
+    describe 'Types' do
+
+      it 'String' do
+        form_class = Rasti::Form[text: Rasti::Types::String]
+
+        tool_class = build_tool_class form_class
+
+        serialization = serializer.serialize tool_class
+
+        expeted_serialization = build_serializaton param_name: 'text', param_type: 'string'
+
+        assert_equal expeted_serialization, serialization
+
+        tool_class.verify
+      end
+
+      it 'Integer' do
+        form_class = Rasti::Form[int: Rasti::Types::Integer]
+
+        tool_class = build_tool_class form_class
+
+        serialization = serializer.serialize tool_class
+
+        expeted_serialization = build_serializaton param_name: 'int', param_type: 'integer'
+
+        assert_equal expeted_serialization, serialization
+
+        tool_class.verify
+      end
+
+      it 'Number' do
+        form_class = Rasti::Form[number: Rasti::Types::Float]
+
+        tool_class = build_tool_class form_class
+
+        serialization = serializer.serialize tool_class
+
+        expeted_serialization = build_serializaton param_name: 'number', param_type: 'number'
+
+        assert_equal expeted_serialization, serialization
+
+        tool_class.verify
+      end
+
+      it 'Boolean' do
+        form_class = Rasti::Form[bool: Rasti::Types::Boolean]
+
+        tool_class = build_tool_class form_class
+
+        serialization = serializer.serialize tool_class
+
+        expeted_serialization = build_serializaton param_name: 'bool', param_type: 'boolean'
+
+        assert_equal expeted_serialization, serialization
+
+        tool_class.verify
+      end
+
+      it 'Time' do
+        form_class = Rasti::Form[timestamp: Rasti::Types::Time['%Y-%m-%dT%H:%M:%S%z']]
+
+        tool_class = build_tool_class form_class
+
+        serialization = serializer.serialize tool_class
+
+        expeted_serialization = build_serializaton param_name: 'timestamp', param_type: 'string'
+        expeted_serialization[:function][:parameters][:properties][:timestamp][:format] = 'date'
+
+        assert_equal expeted_serialization, serialization
+
+        tool_class.verify
+      end
+
+      it 'Enum' do
+        form_class = Rasti::Form[option: Rasti::Types::Enum['option_1', 'option_2']]
+
+        tool_class = build_tool_class form_class
+
+        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']
+
+        assert_equal expeted_serialization, serialization
+
+        tool_class.verify
+      end
+
+      it 'Object' do
+        inner_form_class = Rasti::Form[text: Rasti::Types::String, int: Rasti::Types::Integer]
+        form_class = Rasti::Form[form: Rasti::Types::Model[inner_form_class]]
+
+        tool_class = build_tool_class form_class
+
+        serialization = serializer.serialize tool_class
+
+        expeted_serialization = build_serializaton param_name: 'form', param_type: 'object'
+        expeted_serialization[:function][:parameters][:properties][:form][:properties] = {
+          text: {type: 'string'},
+          int: {type: 'integer'}
+        }
+
+        assert_equal expeted_serialization, serialization
+
+        tool_class.verify
+      end
+
+      describe 'Array' do
+
+        it 'String' do
+          form_class = Rasti::Form[texts: Rasti::Types::Array[Rasti::Types::String]]
+
+          tool_class = build_tool_class form_class
+
+          serialization = serializer.serialize tool_class
+
+          expeted_serialization = build_serializaton param_name: 'texts', param_type: 'array'
+          expeted_serialization[:function][:parameters][:properties][:texts][:items] = {type: 'string'}
+
+          assert_equal expeted_serialization, serialization
+
+          tool_class.verify
+        end
+
+        it 'Number' do
+          form_class = Rasti::Form[numbers: Rasti::Types::Array[Rasti::Types::Float]]
+
+          tool_class = build_tool_class form_class
+
+          serialization = serializer.serialize tool_class
+
+          expeted_serialization = build_serializaton param_name: 'numbers', param_type: 'array'
+          expeted_serialization[:function][:parameters][:properties][:numbers][:items] = {type: 'number'}
+
+          assert_equal expeted_serialization, serialization
+
+          tool_class.verify
+        end
+
+        it 'Object' do
+          inner_form_class = Rasti::Form[text: Rasti::Types::String, int: Rasti::Types::Integer]
+          form_class = Rasti::Form[forms: Rasti::Types::Array[Rasti::Types::Model[inner_form_class]]]
+
+          tool_class = build_tool_class form_class
+
+          serialization = serializer.serialize tool_class
+
+          expeted_serialization = build_serializaton param_name: 'forms', param_type: 'array'
+          expeted_serialization[:function][:parameters][:properties][:forms][:items] = {
+            type: 'object',
+            properties: {
+              text: {type: 'string'},
+              int: {type: 'integer'}
+            }
+          }
+
+          assert_equal expeted_serialization, serialization
+
+          tool_class.verify
+        end
+
+      end
+
+    end
+
+    it 'Required' do
+      form_class = Class.new Rasti::Form do
+        attribute :text, Rasti::Types::String, required: true
+        attribute :int, Rasti::Types::Integer, required: false
+      end
+
+      tool_class = build_tool_class form_class
+
+      serialization = serializer.serialize tool_class
+
+      expeted_serialization = build_serializaton param_name: 'form', param_type: 'object'
+      expeted_serialization[:function][:parameters][:properties] = {
+        text: {type: 'string'},
+        int: {type: 'integer'}
+      }
+      expeted_serialization[:function][:parameters][:required] = [:text]
+
+      assert_equal expeted_serialization, serialization
+
+      tool_class.verify
+    end
+
+    it 'Description' do
+      form_class = Class.new Rasti::Form do
+        attribute :text, Rasti::Types::String, description: 'Text param'
+        attribute :int, Rasti::Types::Integer, description: 'Int param'
+      end
+
+      tool_class = build_tool_class form_class
+
+      serialization = serializer.serialize tool_class
+
+      expeted_serialization = build_serializaton param_name: 'form', param_type: 'object'
+      expeted_serialization[:function][:parameters][:properties] = {
+        text: {
+          description: 'Text param',
+          type: 'string'
+        },
+        int: {
+          description: 'Int param',
+          type: 'integer'
+        }
+      }
+
+      assert_equal expeted_serialization, serialization
+
+      tool_class.verify
+    end
+
+  end
+
+  it 'Invalid tool' do
+    form_class = Rasti::Form[obj: Object]
+    tool_class = build_tool_class form_class
+
+    error = assert_raises(Rasti::AI::Errors::ToolSerializationError) do
+      serializer.serialize tool_class
+    end
+
+    assert_equal "Tool serialization error: #{tool_class}", error.message
+    assert_equal 'Type not serializable Object', error.cause.message
+  end
+
+end

data/spec/resources/open_ai/basic_request.json

@@ -0,0 +1 @@
+{"model":"<%= model %>","messages":[{"role":"user","content":"<%= prompt %>"}],"tools":[],"tool_choice":"none"}

data/spec/resources/open_ai/basic_response.json

@@ -0,0 +1,36 @@
+{
+  "id": "chatcmpl-BPudorJPTVaFGNU9thj1e81Vatlpt",
+  "object": "chat.completion",
+  "created": 1745515628,
+  "model": "gpt-4o-mini-2024-07-18",
+  "choices": [
+    {
+      "index": 0,
+      "message": {
+        "role": "assistant",
+        "content": "<%= content %>",
+        "refusal": null,
+        "annotations": []
+      },
+      "logprobs": null,
+      "finish_reason": "stop"
+    }
+  ],
+  "usage": {
+    "prompt_tokens": 27,
+    "completion_tokens": 229,
+    "total_tokens": 256,
+    "prompt_tokens_details": {
+      "cached_tokens": 0,
+      "audio_tokens": 0
+    },
+    "completion_tokens_details": {
+      "reasoning_tokens": 0,
+      "audio_tokens": 0,
+      "accepted_prediction_tokens": 0,
+      "rejected_prediction_tokens": 0
+    }
+  },
+  "service_tier": "default",
+  "system_fingerprint": "fp_dbaca60df0"
+}

data/spec/resources/open_ai/tool_request.json

@@ -0,0 +1 @@
+{"model":"<%= model %>","messages":[{"role":"user","content":"<%= prompt %>"}],"tools":<%= JSON.dump(tools) %>,"tool_choice":"auto"}

data/spec/resources/open_ai/tool_response.json

@@ -0,0 +1,46 @@
+{
+  "id": "chatcmpl-BPy9KtqtdrccgC5dTTjaCmq2KKFS6",
+  "object": "chat.completion",
+  "created": 1745529114,
+  "model": "gpt-4o-mini-2024-07-18",
+  "choices": [
+    {
+      "index": 0,
+      "message": {
+        "role": "assistant",
+        "content": null,
+        "tool_calls": [
+          {
+            "id": "call_0D9Fo0O3TEGNMlFm7KllmKLK",
+            "type": "function",
+            "function": {
+              "name": "<%= name %>",
+              "arguments": "<%= JSON.dump(arguments).gsub('"', '\\"') %>"
+            }
+          }
+        ],
+        "refusal": null,
+        "annotations": []
+      },
+      "logprobs": null,
+      "finish_reason": "tool_calls"
+    }
+  ],
+  "usage": {
+    "prompt_tokens": 83,
+    "completion_tokens": 23,
+    "total_tokens": 106,
+    "prompt_tokens_details": {
+      "cached_tokens": 0,
+      "audio_tokens": 0
+    },
+    "completion_tokens_details": {
+      "reasoning_tokens": 0,
+      "audio_tokens": 0,
+      "accepted_prediction_tokens": 0,
+      "rejected_prediction_tokens": 0
+    }
+  },
+  "service_tier": "default",
+  "system_fingerprint": "fp_0392822090"
+}

data/spec/support/helpers/erb.rb

@@ -0,0 +1,17 @@
+module Support
+  module Helpers
+    module ERB
+
+      def erb(template, variables={})
+        variables_binding = binding.tap do |b|
+          variables.each do |key, value|
+            b.local_variable_set key, value
+          end
+        end
+
+        ::ERB.new(template).result variables_binding
+      end
+
+    end
+  end
+end

data/spec/support/helpers/resources.rb

@@ -0,0 +1,23 @@
+module Support
+  module Helpers
+    module Resources
+
+      RESOURCES_PATH = File.expand_path('../../../resources', __FILE__)
+
+      def resource_path(relative_path)
+        File.join RESOURCES_PATH, relative_path
+      end
+
+      def read_resource(relative_path, variables={})
+        filename = resource_path relative_path
+        content = File.read filename
+        erb content, variables
+      end
+
+      def read_json_resource(relative_path, variables={})
+        JSON.parse read_resource(relative_path, variables)
+      end
+
+    end
+  end
+end

metadata

@@ -0,0 +1,243 @@
+--- !ruby/object:Gem::Specification
+name: rasti-ai
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+platform: ruby
+authors:
+- Gabriel Naiman
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2025-04-28 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: multi_require
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+- !ruby/object:Gem::Dependency
+  name: rasti-form
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '6.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '6.0'
+- !ruby/object:Gem::Dependency
+  name: inflecto
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.0'
+- !ruby/object:Gem::Dependency
+  name: class_config
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '12.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '12.0'
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.0'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '5.11'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.0'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '5.11'
+- !ruby/object:Gem::Dependency
+  name: minitest-colorin
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.1'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.1'
+- !ruby/object:Gem::Dependency
+  name: minitest-line
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.6'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.6'
+- !ruby/object:Gem::Dependency
+  name: simplecov
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.12'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.12'
+- !ruby/object:Gem::Dependency
+  name: pry-nav
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.2'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.2'
+- !ruby/object:Gem::Dependency
+  name: webmock
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+description: AI for apps
+email:
+- gabynaiman@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".github/workflows/ci.yml"
+- ".gitignore"
+- ".ruby-gemset"
+- ".ruby-version"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- lib/rasti-ai.rb
+- lib/rasti/ai.rb
+- lib/rasti/ai/errors.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/version.rb
+- rasti-ai.gemspec
+- spec/coverage_helper.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
+homepage: https://github.com/gabynaiman/rasti-ai
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.3.26
+signing_key: 
+specification_version: 4
+summary: AI for apps
+test_files:
+- spec/coverage_helper.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