rasti-ai 2.0.2 → 3.0.0

data/spec/minitest_helper.rb

@@ -22,6 +22,9 @@ Rasti::AI.configure do |config|
 
   config.gemini_api_key = 'test_gemini_api_key'
   config.gemini_default_model = 'gemini-test'
+
+  config.anthropic_api_key = 'test_anthropic_api_key'
+  config.anthropic_default_model = 'claude-test'
 end
 
 
@@ -32,6 +35,32 @@ class Minitest::Test
 
 end
 
+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 self.description
+    'Sum two numbers'
+  end
+
+  def execute(form)
+    {result: form.number_a + form.number_b}
+  end
+end
+
+
 class GoalsByPlayer
   def self.form
     Rasti::Form[player: Rasti::Types::String, team: Rasti::Types::String]

data/spec/open_ai/assistant_spec.rb

@@ -35,15 +35,16 @@ describe Rasti::AI::OpenAI::Assistant do
     it 'Client' do
       client_arguments = [
         {
-          model: nil,
-          tools: [],
+          model:            nil,
+          tools:            [],
           messages: [
             {
-              role: Rasti::AI::OpenAI::Roles::USER,
+              role:    Rasti::AI::OpenAI::Roles::USER,
               content: question
             }
           ],
-          response_format: nil
+          response_format:  nil,
+          reasoning_effort: nil
         }
       ]
 
@@ -111,6 +112,21 @@ describe Rasti::AI::OpenAI::Assistant do
       assert_equal answer, response
     end
 
+    it 'Thinking' do
+      body = read_json_resource('open_ai/basic_request.json', model: Rasti::AI.openai_default_model, prompt: question)
+      body['reasoning_effort'] = 'medium'
+
+      stub_request(:post, api_url)
+        .with(body: JSON.dump(body))
+        .to_return(body: read_resource('open_ai/basic_response.json', content: answer))
+
+      assistant = Rasti::AI::OpenAI::Assistant.new thinking: 'medium'
+
+      response = assistant.call question
+
+      assert_equal answer, response
+    end
+
     it 'JSON Schema' do
       json_schema = {answer: 'Response answer'}
       json_answer = "{\\\"answer\\\": \\\"#{answer}\\\"}"

data/spec/resources/anthropic/basic_request.json

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

data/spec/resources/anthropic/basic_response.json

@@ -0,0 +1,20 @@
+{
+  "id": "msg_01XFDUDYJgAACzvnptvVoYEL",
+  "type": "message",
+  "role": "assistant",
+  "content": [
+    {
+      "type": "text",
+      "text": "<%= content %>"
+    }
+  ],
+  "model": "claude-test",
+  "stop_reason": "end_turn",
+  "stop_sequence": null,
+  "usage": {
+    "input_tokens": 25,
+    "output_tokens": 11,
+    "cache_creation_input_tokens": 0,
+    "cache_read_input_tokens": 0
+  }
+}

data/spec/resources/anthropic/tool_request.json

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

data/spec/resources/anthropic/tool_response.json

@@ -0,0 +1,22 @@
+{
+  "id": "msg_01HqBZkZSz4zKijpCKB8AMKM",
+  "type": "message",
+  "role": "assistant",
+  "content": [
+    {
+      "type": "tool_use",
+      "id": "toolu_01A09q90qw90lq917835lq9",
+      "name": "<%= name %>",
+      "input": <%= JSON.dump(arguments) %>
+    }
+  ],
+  "model": "claude-test",
+  "stop_reason": "tool_use",
+  "stop_sequence": null,
+  "usage": {
+    "input_tokens": 82,
+    "output_tokens": 29,
+    "cache_creation_input_tokens": 0,
+    "cache_read_input_tokens": 0
+  }
+}

data/spec/tool_serializer_spec.rb

@@ -274,16 +274,41 @@ describe Rasti::AI::ToolSerializer do
 
   end
 
-  it 'Invalid tool' do
+  it 'Custom registered type' do
+    custom_type = Class.new
+    Rasti::Model::Schema.register_type_serializer(custom_type) { {type: :string} }
+
+    form_class = Rasti::Form[value: custom_type]
+    tool_class = build_tool_class form_class
+
+    serialization = serializer.serialize tool_class
+
+    expected_serialization = build_serializaton param_name: 'value', param_type: 'string'
+
+    assert_equal expected_serialization, serialization
+
+    tool_class.verify
+  end
+
+  it 'Unknown type serializes without constraints' 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
+    serialization = serializer.serialize tool_class
 
-    assert_equal "Tool serialization error: #{tool_class}", error.message
-    assert_equal 'Type not serializable Object', error.cause.message
+    expected_serialization = {
+      name: 'call_custom_function',
+      inputSchema: {
+        type: 'object',
+        properties: {
+          obj: {}
+        }
+      }
+    }
+
+    assert_equal expected_serialization, serialization
+
+    tool_class.verify
   end
 
 end

data/tasks/assistant.rake

@@ -0,0 +1,94 @@
+# frozen_string_literal: true
+
+require 'fileutils'
+require 'logger'
+
+# Interactive CLI to chat with an AI assistant backed by a public weather MCP server.
+#
+# The Pipeworx weather MCP is free, requires no auth key, and exposes:
+#   get_weather    — current conditions for any city or lat/lon
+#   get_forecast   — daily forecast up to 16 days ahead
+#   get_historical — historical weather back to 1940
+# Docs: https://pipeworx.io
+#
+# Usage:
+#   rake assistant:openai      # requires OPENAI_API_KEY    (+ optionally OPENAI_DEFAULT_MODEL)
+#   rake assistant:gemini      # requires GEMINI_API_KEY    (+ optionally GEMINI_DEFAULT_MODEL)
+#   rake assistant:anthropic   # requires ANTHROPIC_API_KEY (+ optionally ANTHROPIC_DEFAULT_MODEL)
+
+WEATHER_MCP_URL = 'https://gateway.pipeworx.io/weather/mcp'.freeze
+
+PROVIDERS = {
+  'openai'    => {key: 'OPENAI_API_KEY',    klass: -> { Rasti::AI::OpenAI::Assistant }},
+  'gemini'    => {key: 'GEMINI_API_KEY',    klass: -> { Rasti::AI::Gemini::Assistant }},
+  'anthropic' => {key: 'ANTHROPIC_API_KEY', klass: -> { Rasti::AI::Anthropic::Assistant }}
+}.freeze
+
+def build_weather_mcp
+  Rasti::AI::MCP::Client.new(
+    url: WEATHER_MCP_URL,
+    allowed_tools: ['get_weather', 'get_forecast', 'get_historical']
+  )
+end
+
+def print_banner(provider, tool_names)
+  puts
+  puts '=' * 60
+  puts "  Rasti AI Chat — #{provider}"
+  puts "  MCP: #{WEATHER_MCP_URL}"
+  puts "  Tools: #{tool_names.join(', ')}"
+  puts '=' * 60
+  puts "  Type your message and press Enter."
+  puts "  Type 'exit' or press Ctrl+C to quit."
+  puts '=' * 60
+  puts
+end
+
+def chat_loop(assistant)
+  loop do
+    print 'You: '
+    $stdout.flush
+
+    input = $stdin.gets&.chomp
+    break if input.nil? || input.strip.downcase == 'exit'
+    next  if input.strip.empty?
+
+    begin
+      response = assistant.call(input.strip)
+      puts
+      puts "Assistant: #{response}"
+      puts
+    rescue Interrupt
+      break
+    rescue => e
+      puts "\n[Error] #{e.message}\n"
+    end
+  end
+
+  puts "\nGoodbye!"
+end
+
+def start_assistant(provider, env_key, assistant_klass)
+  require 'rasti-ai'
+
+  abort "[Error] #{env_key} is not set." unless ENV[env_key]
+
+  FileUtils.mkdir_p 'log'
+  Rasti::AI.configure { |c| c.logger = Logger.new("log/#{provider}.log") }
+
+  mcp = build_weather_mcp
+  print_banner provider.capitalize, mcp.list_tools.map { |t| t['name'] }
+
+  chat_loop assistant_klass.call.new(mcp_servers: {weather: mcp})
+end
+
+namespace :assistant do
+
+  PROVIDERS.each do |provider, config|
+    desc "Chat with #{provider.capitalize} assistant using Pipeworx weather MCP (requires #{config[:key]})"
+    task provider.to_sym do
+      start_assistant(provider, config[:key], config[:klass])
+    end
+  end
+
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rasti-ai
 version: !ruby/object:Gem::Version
-  version: 2.0.2
+  version: 3.0.0
 platform: ruby
 authors:
 - Gabriel Naiman
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2026-03-16 00:00:00.000000000 Z
+date: 2026-06-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: multi_require
@@ -94,6 +94,26 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '12.0'
+- !ruby/object:Gem::Dependency
+  name: rack
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '1.3'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '1.3'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '3'
 - !ruby/object:Gem::Dependency
   name: rack-test
   requirement: !ruby/object:Gem::Requirement
@@ -188,14 +208,14 @@ dependencies:
   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
@@ -223,12 +243,16 @@ files:
 - ".gitignore"
 - ".ruby-gemset"
 - ".ruby-version"
+- AGENTS.md
 - Gemfile
 - LICENSE.txt
 - README.md
 - Rakefile
 - lib/rasti-ai.rb
 - lib/rasti/ai.rb
+- lib/rasti/ai/anthropic/assistant.rb
+- lib/rasti/ai/anthropic/client.rb
+- lib/rasti/ai/anthropic/roles.rb
 - lib/rasti/ai/assistant.rb
 - lib/rasti/ai/assistant_state.rb
 - lib/rasti/ai/client.rb
@@ -237,8 +261,9 @@ files:
 - lib/rasti/ai/gemini/client.rb
 - lib/rasti/ai/gemini/roles.rb
 - lib/rasti/ai/mcp/client.rb
-- lib/rasti/ai/mcp/errors.rb
+- lib/rasti/ai/mcp/constants.rb
 - lib/rasti/ai/mcp/server.rb
+- lib/rasti/ai/mcp/tools_registry.rb
 - lib/rasti/ai/open_ai/assistant.rb
 - lib/rasti/ai/open_ai/client.rb
 - lib/rasti/ai/open_ai/roles.rb
@@ -248,14 +273,21 @@ files:
 - lib/rasti/ai/version.rb
 - log/.gitkeep
 - rasti-ai.gemspec
+- spec/anthropic/assistant_spec.rb
+- spec/anthropic/client_spec.rb
 - spec/coverage_helper.rb
 - spec/gemini/assistant_spec.rb
 - spec/gemini/client_spec.rb
 - spec/mcp/client_spec.rb
 - spec/mcp/server_spec.rb
+- spec/mcp/tools_registry_spec.rb
 - spec/minitest_helper.rb
 - spec/open_ai/assistant_spec.rb
 - spec/open_ai/client_spec.rb
+- spec/resources/anthropic/basic_request.json
+- spec/resources/anthropic/basic_response.json
+- spec/resources/anthropic/tool_request.json
+- spec/resources/anthropic/tool_response.json
 - spec/resources/gemini/basic_request.json
 - spec/resources/gemini/basic_response.json
 - spec/resources/gemini/tool_request.json
@@ -267,6 +299,7 @@ files:
 - spec/support/helpers/erb.rb
 - spec/support/helpers/resources.rb
 - spec/tool_serializer_spec.rb
+- tasks/assistant.rake
 homepage: https://github.com/gabynaiman/rasti-ai
 licenses:
 - MIT
@@ -279,7 +312,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: '0'
+      version: '2.3'
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
@@ -291,14 +324,21 @@ signing_key:
 specification_version: 4
 summary: AI for apps
 test_files:
+- spec/anthropic/assistant_spec.rb
+- spec/anthropic/client_spec.rb
 - spec/coverage_helper.rb
 - spec/gemini/assistant_spec.rb
 - spec/gemini/client_spec.rb
 - spec/mcp/client_spec.rb
 - spec/mcp/server_spec.rb
+- spec/mcp/tools_registry_spec.rb
 - spec/minitest_helper.rb
 - spec/open_ai/assistant_spec.rb
 - spec/open_ai/client_spec.rb
+- spec/resources/anthropic/basic_request.json
+- spec/resources/anthropic/basic_response.json
+- spec/resources/anthropic/tool_request.json
+- spec/resources/anthropic/tool_response.json
 - spec/resources/gemini/basic_request.json
 - spec/resources/gemini/basic_response.json
 - spec/resources/gemini/tool_request.json