ox-ai-workers 1.0.2.1 → 1.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 33177f5c822272383d00d96fadd552db0b966af7ddb2ec494538ca8cace7fadf
-  data.tar.gz: e3f756943f16fb8be2823f493156e261eb2d4d02a99d9c0373e8a3ed606c8726
+  metadata.gz: 838a2c53f4ce2423cd215d9cda21c76fe5870f709e528b3de491666b25b7b516
+  data.tar.gz: 7c55a9076dcbd3f4e21222a5b2300d1677e08fcdfe0d9ae0f186c88b0206585b
 SHA512:
-  metadata.gz: a64203d85443d9b812c2d36aa0718283fb21fd784043a4d1346ec96a3386bec3766f4ed127d41c1c6f149af7abb4af6241fad089caf72b058037b7bc323b08cb
-  data.tar.gz: f0e1ec3b3be53fbe00f31dddc6e16e6a3befb7457d307cbc9906160b02e94bc66d8d18ec1c0095c2e6a53c9d48d4f256515da6ac0ea1ae4bb327ab07fb0772a0
+  metadata.gz: f642f834e477f074fffda948b505bcce54381369391e05f3ca54acb0943d87e3dc78ff050c2f9b8cb507ad2f3754fd86355dd0e32495e9bc00a04cea95d1b0d6
+  data.tar.gz: 55464fe188ebea0a2c73a07c1c91e381cec4d16faa24755b803c527671db6185b95fc022104db5670d2a79d65cc6ad7c28b5992655c5926a5da64d3ea3a9fd63

data/lib/ox-ai-workers.rb

@@ -61,7 +61,7 @@ module OxAiWorkers
 
   class Configuration
     attr_accessor :max_tokens, :temperature, :wait_for_complete, :access_token_deepseek, :access_token_openai,
-                  :access_token_stability, :access_token_wolfram
+                  :access_token_stability, :access_token_wolfram, :access_token_anthropic
 
     def initialize
       @max_tokens = DEFAULT_MAX_TOKEN
@@ -72,6 +72,7 @@ module OxAiWorkers
       @access_token_openai = nil
       @access_token_stability = nil
       @access_token_wolfram = nil
+      @access_token_anthropic = nil
 
       [Array, NilClass, String, Symbol, Hash].each do |c|
         c.send(:include, OxAiWorkers::PresentCompat) unless c.method_defined?(:present?)

data/lib/oxaiworkers/iterator.rb

@@ -217,7 +217,7 @@ module OxAiWorkers
       end
 
       if @worker.tool_calls.present?
-        @queue << { role: :assistant, content: @worker.tool_calls_raw.to_s }
+        # @queue << { role: :assistant, content: @worker.tool_calls_raw.to_s }
         @worker.tool_calls.each do |external_call|
           tool = ([self] + @tools).select do |t|
             tool_name = t.respond_to?(:tool_name) ? t.tool_name : t.class.tool_name
@@ -226,13 +226,26 @@ module OxAiWorkers
           next if tool.nil?
 
           @call_id += 1
+          # Add tool call message in the correct format
           out = tool.send(external_call[:name], **external_call[:args])
-          @queue << { role: :assistant,
-                      content: "Tool call #{external_call[:name]} completed." }
+          @queue << {
+            role: :assistant,
+            tool_calls: [{
+              id: "call_#{@call_id}",
+              type: 'function',
+              function: {
+                name: external_call[:name],
+                arguments: external_call[:args].to_json
+              }
+            }]
+          }
           @queue << if out.present?
-                      { role: :tool, content: out, tool_call_id: "call_#{@call_id}" }
+                      { role: :tool,
+                        content: out,
+                        tool_call_id: "call_#{@call_id}" }
                     else
-                      { role: :tool, content: "Tool call #{external_call[:name]} successful.",
+                      { role: :tool,
+                        content: "Tool call #{external_call[:name]} successful.",
                         tool_call_id: "call_#{@call_id}" }
                     end
         end

data/lib/oxaiworkers/models/anthropic_max.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+module OxAiWorkers
+  module Models
+    class AnthropicMax < LLMBase
+      def initialize(uri_base: nil, api_key: nil, model: nil, max_tokens: nil, temperature: nil, frequency_penalty: nil)
+        @model = model || 'claude-3-7-sonnet-latest'
+        @uri_base = uri_base || 'https://api.anthropic.com/v1/'
+        @api_key = api_key || OxAiWorkers.configuration.access_token_anthropic
+        super(uri_base:, api_key: @api_key, model: @model, max_tokens:, temperature:, frequency_penalty:)
+      end
+    end
+  end
+end

data/lib/oxaiworkers/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module OxAiWorkers
-  VERSION = '1.0.2.1'
+  VERSION = '1.0.3'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ox-ai-workers
 version: !ruby/object:Gem::Version
-  version: 1.0.2.1
+  version: 1.0.3
 platform: ruby
 authors:
 - Denis Smolev
@@ -157,6 +157,7 @@ files:
 - lib/oxaiworkers/image_iterator.rb
 - lib/oxaiworkers/iterator.rb
 - lib/oxaiworkers/load_i18n.rb
+- lib/oxaiworkers/models/anthropic_max.rb
 - lib/oxaiworkers/models/deepseek_max.rb
 - lib/oxaiworkers/models/images_base.rb
 - lib/oxaiworkers/models/llm_base.rb