rasti-ai 1.0.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: e5f39be89f25780d7f9b33adf06b12109c9ddb482c2aa85220d400cfb945c000
+  data.tar.gz: e4ce69118debdf636aab53cf8fa25cfd233633ac5c499f9f0bc1a4b5bc05294f
+SHA512:
+  metadata.gz: 2f19e650f0908142aedbc09b9d8d372b24da94979f23464d79de17f8c44cd2c3ed86c666dfc3dc46066bf800ecd572c16cc2b3c321692601d29c76c89a38170e
+  data.tar.gz: a9da7e1802d9877e1171eb9cfa5b602ad642de251131ee6cdbe0bf77cbf810a24817af8846e246a2a57cf47191dc00542d60b2591ded65ef87860e3ec3104d63

data/.github/workflows/ci.yml

@@ -0,0 +1,44 @@
+name: CI
+
+on:
+  push:
+    branches: [ '**' ]
+  pull_request:
+    branches: [ '**' ]
+
+jobs:
+  test:
+
+    name: Tests
+    runs-on: ubuntu-22.04
+    strategy:
+      matrix:
+        ruby-version: ['2.3', '2.4', '2.5', '2.6', '2.7', '3.0', 'jruby-9.2.9.0']
+
+    steps:
+    - uses: actions/checkout@v3
+
+    - name: Set up Ruby
+      uses: ruby/setup-ruby@v1
+      with:
+        ruby-version: ${{ matrix.ruby-version }}
+        bundler-cache: false
+
+    - name: Install native dependencies for Ruby 3.0
+      if: ${{ startsWith(matrix.ruby-version, '3.') }}
+      run: |
+        sudo apt-get update
+        sudo apt-get install -y libcurl4-openssl-dev
+
+    - name: Configure bundler
+      if: ${{ startsWith(matrix.ruby-version, '3.') }}
+      run: |
+        bundle config set --local force_ruby_platform true
+        bundle install
+
+    - name: Install dependencies for other Ruby versions
+      if: ${{ !startsWith(matrix.ruby-version, '3.') }}
+      run: bundle install
+
+    - name: Run tests
+      run: bundle exec rake

data/.gitignore

@@ -0,0 +1,10 @@
+/.bundle/
+/.yardoc
+/Gemfile.lock
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/
+/.ruby-env

data/.ruby-gemset

@@ -0,0 +1 @@
+rasti-ai

data/.ruby-version

@@ -0,0 +1 @@
+ruby-2.3.8

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in rasti-ai.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2025 Gabriel Naiman
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

data/README.md

@@ -0,0 +1,107 @@
+# Rasti::AI
+
+[![Gem Version](https://badge.fury.io/rb/rasti-ai.svg)](https://rubygems.org/gems/rasti-ai)
+[![CI](https://github.com/gabynaiman/rasti-ai/actions/workflows/ci.yml/badge.svg)](https://github.com/gabynaiman/rasti-ai/actions/workflows/ci.yml)
+
+AI for apps
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'rasti-ai'
+```
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install rasti-ai
+
+## Usage
+
+### Configuration
+```ruby
+Rasti::AI.configure do |config|
+  config.logger = Logger.new 'log/development.log'
+  config.openai_api_key = 'abcd12345' # Default ENV['OPENAI_API_KEY']
+  config.openai_default_model = 'gpt-4o-mini' # Default ENV['OPENAI_DEFAULT_MODEL']
+end
+```
+
+### Open AI
+
+#### Assistant
+```ruby
+assistant = Rasti::AI::OpenAI::Assistant.new
+assistant.call 'who is the best player' # => 'The best player is Lionel Messi'
+```
+
+#### Tools
+```ruby
+class GetCurrentTime
+  def call(params={})
+    Time.now.iso8601
+  end
+end
+
+class GetCurrentWeather
+  def self.form
+    Rasti::Form[location: Rasti::Types::String]
+  end
+
+  def call(params={})
+    response = HTTP.get "https://api.wheater.com/?location=#{params['location']}"
+    response.body.to_s
+  end
+end
+
+tools = [
+  GetCurrentTime.new,
+  GetCurrentWeather.new
+]
+
+assistant = Rasti::AI::OpenAI::Assistant.new tools: tools
+
+assistant.call 'what time is it' # => 'The current time is 3:03 PM on April 28, 2025.'
+
+assistant.call 'what is the weather in Buenos Aires' # => 'In Buenos Aires it is 15 degrees'
+```
+
+#### Context and state
+```ruby
+state = Rasti::AI::OpenAI::AssistantState.new context: 'Act as sports journalist'
+
+assistant = Rasti::AI::OpenAI::Assistant.new state: state
+
+assistant.call 'who is the best player'
+
+state.messages
+# [
+#   {
+#     role: 'system',
+#     content: 'Act as sports journalist'
+#   },
+#   {
+#     role: 'user',
+#     content: 'who is the best player'
+#   },
+#   {
+#     role: 'assistant',
+#     content: 'The best player is Lionel Messi'
+#   }
+# ]
+```
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/gabynaiman/rasti-ai.
+
+
+## License
+
+The gem is available as open source under the terms of the [MIT License](http://opensource.org/licenses/MIT).
+

data/Rakefile

@@ -0,0 +1,25 @@
+require 'bundler/gem_tasks'
+require 'rake/testtask'
+
+Rake::TestTask.new(:spec) do |t|
+  t.libs << 'spec'
+  t.libs << 'lib'
+  t.pattern = ENV['DIR'] ? File.join(ENV['DIR'], '**', '*_spec.rb') : 'spec/**/*_spec.rb'
+  t.verbose = false
+  t.warning = false
+  t.loader = nil if ENV['TEST']
+  ENV['TEST'], ENV['LINE'] = ENV['TEST'].split(':') if ENV['TEST'] && !ENV['LINE']
+  t.options = ''
+  t.options << "--name=/#{ENV['NAME']}/ " if ENV['NAME']
+  t.options << "-l #{ENV['LINE']} " if ENV['LINE'] && ENV['TEST']
+end
+
+task default: :spec
+
+desc 'Pry console'
+task :console do
+  require 'rasti-ai'
+  require 'pry'
+  ARGV.clear
+  Pry.start
+end

data/lib/rasti/ai/errors.rb

@@ -0,0 +1,39 @@
+module Rasti
+  module AI
+    module Errors
+
+      class RequestFail < StandardError
+
+        attr_reader :url, :body, :response
+
+        def initialize(url, body, response)
+          @url = url
+          @body = body
+          @response = response
+        end
+
+        def message
+          "Request fail\nRequest: #{url}\n#{JSON.pretty_generate(body)}\nResponse: #{response.code}\n#{response.body}"
+        end
+
+      end
+
+      class ToolSerializationError < StandardError
+
+        def initialize(tool_class)
+          super "Tool serialization error: #{tool_class}"
+        end
+
+      end
+
+      class UndefinedTool < StandardError
+
+        def initialize(tool_name)
+          super "Undefined tool #{tool_name}"
+        end
+
+      end
+
+    end
+  end
+end

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

@@ -0,0 +1,96 @@
+module Rasti
+  module AI
+    module OpenAI
+      class Assistant
+
+        attr_reader :state
+
+        def initialize(client:nil, state:nil, model:nil, tools:[], logger:nil)
+          @client = client || Client.new
+          @state = state || AssistantState.new
+          @model = model
+          @tools = {}
+          @serialized_tools = []
+          @logger = logger || Rasti::AI.logger
+
+          tools.each do |tool|
+            serialization = ToolSerializer.serialize tool.class
+            @tools[serialization[:function][:name]] = tool
+            @serialized_tools << serialization
+          end
+        end
+
+        def call(prompt)
+          messages << {
+            role: Roles::USER,
+            content: prompt
+          }
+
+          loop do
+            response = client.chat_completions messages: messages,
+                                               model: model,
+                                               tools: serialized_tools
+
+            choice = response['choices'][0]['message']
+
+            if choice['tool_calls']
+              messages << {
+                role: Roles::ASSISTANT,
+                tool_calls: choice['tool_calls']
+              }
+
+              choice['tool_calls'].each do |tool_call|
+                name = tool_call['function']['name']
+                args = JSON.parse tool_call['function']['arguments']
+
+                result = call_tool name, args
+
+                messages << {
+                  role: Roles::TOOL,
+                  tool_call_id: tool_call['id'],
+                  content: result
+                }
+              end
+            else
+              messages << {
+                role: Roles::ASSISTANT,
+                content: choice['content']
+              }
+
+              return choice['content']
+            end
+          end
+        end
+
+        private
+
+        attr_reader :client, :model, :tools, :serialized_tools, :logger
+
+        def messages
+          state.messages
+        end
+
+        def call_tool(name, args)
+          raise Errors::UndefinedTool.new(name) unless tools.key? name
+
+          key = "#{name} -> #{args}"
+
+          state.fetch(key) do
+            logger.info(self.class) { "Calling function #{name} with #{args}" }
+
+            result = tools[name].call args
+
+            logger.info(self.class) { "Function result: #{result}" }
+
+            result
+          end
+
+        rescue => ex
+          logger.warn(self.class) { "Function failed: #{ex.message}\n#{ex.backtrace.join("\n")}" }
+          "Error: #{ex.message}"
+        end
+
+      end
+    end
+  end
+end

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

@@ -0,0 +1,27 @@
+module Rasti
+  module AI
+    module OpenAI
+      class AssistantState
+
+        attr_reader :messages
+
+        def initialize(context:nil)
+          @messages = []
+          @cache = {}
+
+          messages << {role: Roles::SYSTEM, content: context} if context
+        end
+
+        def fetch(key, &block)
+          cache[key] = block.call unless cache.key? key
+          cache[key]
+        end
+
+        private
+
+        attr_reader :cache
+
+      end
+    end
+  end
+end

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

@@ -0,0 +1,57 @@
+module Rasti
+  module AI
+    module OpenAI
+      class Client
+
+        BASE_URL = 'https://api.openai.com/v1'.freeze
+
+        def initialize(api_key:nil, logger:nil)
+          @api_key = api_key || Rasti::AI.openai_api_key
+          @logger = logger || Rasti::AI.logger
+        end
+
+        def chat_completions(messages:, model:nil, tools:[])
+          body = {
+            model: model || Rasti::AI.openai_default_model,
+            messages: messages,
+            tools: tools,
+            tool_choice: tools.empty? ? 'none' : 'auto'
+          }
+
+          post '/chat/completions', body
+        end
+
+        private
+
+        attr_reader :api_key, :logger
+
+        def post(relative_url, body)
+          url = "#{BASE_URL}#{relative_url}"
+          uri = URI(url)
+
+          logger.info(self.class) { "POST #{url}" }
+          logger.debug(self.class) { JSON.pretty_generate(body) }
+
+          request = Net::HTTP::Post.new uri
+          request['Authorization'] = "Bearer #{api_key}"
+          request['Content-Type'] = 'application/json'
+          request.body = JSON.dump(body)
+
+          http = Net::HTTP.new uri.host, uri.port
+          http.use_ssl = uri.scheme == 'https'
+
+          response = http.request request
+
+          logger.info(self.class) { "Response #{response.code}" }
+          logger.debug(self.class) { response.body }
+
+          raise Errors::RequestFail.new(url, body, response) unless response.is_a? Net::HTTPSuccess
+
+          JSON.parse response.body
+        end
+
+      end
+    end
+
+  end
+end

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

@@ -0,0 +1,14 @@
+module Rasti
+  module AI
+    module OpenAI
+      module Roles
+
+        ASSISTANT = 'assistant'.freeze
+        SYSTEM = 'system'.freeze
+        TOOL = 'tool'.freeze
+        USER = 'user'.freeze
+
+      end
+    end
+  end
+end

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

@@ -0,0 +1,111 @@
+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

data/lib/rasti/ai/version.rb

@@ -0,0 +1,5 @@
+module Rasti
+  module AI
+    VERSION = '1.0.0'
+  end
+end

data/lib/rasti/ai.rb

@@ -0,0 +1,24 @@
+require 'multi_require'
+require 'rasti-form'
+require 'class_config'
+require 'inflecto'
+require 'net/http'
+require 'uri'
+require 'json'
+require 'logger'
+
+module Rasti
+  module AI
+
+    extend MultiRequire
+    extend ClassConfig
+
+    require_relative_pattern 'ai/**/*'
+
+    attr_config :logger, Logger.new(STDOUT)
+
+    attr_config :openai_api_key, ENV['OPENAI_API_KEY']
+    attr_config :openai_default_model, ENV['OPENAI_DEFAULT_MODEL']
+
+  end
+end

data/lib/rasti-ai.rb

@@ -0,0 +1 @@
+require_relative 'rasti/ai'

data/rasti-ai.gemspec

@@ -0,0 +1,33 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'rasti/ai/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = 'rasti-ai'
+  spec.version       = Rasti::AI::VERSION
+  spec.authors       = ['Gabriel Naiman']
+  spec.email         = ['gabynaiman@gmail.com']
+  spec.summary       = 'AI for apps'
+  spec.description   = 'AI for apps'
+  spec.homepage      = 'https://github.com/gabynaiman/rasti-ai'
+  spec.license       = 'MIT'
+
+  spec.files         = `git ls-files -z`.split("\x0")
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ['lib']
+
+  spec.add_runtime_dependency 'multi_require', '~> 1.0'
+  spec.add_runtime_dependency 'rasti-form', '~> 6.0'
+  spec.add_runtime_dependency 'inflecto', '~> 0.0'
+  spec.add_runtime_dependency 'class_config', '~> 0.0'
+
+  spec.add_development_dependency 'rake', '~> 12.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 'simplecov', '~> 0.12'
+  spec.add_development_dependency 'pry-nav', '~> 0.2'
+  spec.add_development_dependency 'webmock', '~> 3.0'
+end

data/spec/coverage_helper.rb

@@ -0,0 +1,2 @@
+require 'simplecov'
+SimpleCov.start

data/spec/minitest_helper.rb

@@ -0,0 +1,26 @@
+require 'coverage_helper'
+require 'minitest/autorun'
+require 'minitest/colorin'
+require 'webmock/minitest'
+require 'pry-nav'
+require 'rasti-ai'
+require 'securerandom'
+
+require_relative 'support/helpers/erb'
+require_relative 'support/helpers/resources'
+
+
+Rasti::AI.configure do |config|
+  config.logger.level = Logger::FATAL
+
+  config.openai_api_key = 'test_api_key'
+  config.openai_default_model = 'gpt-test'
+end
+
+
+class Minitest::Test
+
+  include Support::Helpers::ERB
+  include Support::Helpers::Resources
+
+end