ox-ai-workers 1.0.2.1 → 1.0.3.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 33177f5c822272383d00d96fadd552db0b966af7ddb2ec494538ca8cace7fadf
-  data.tar.gz: e3f756943f16fb8be2823f493156e261eb2d4d02a99d9c0373e8a3ed606c8726
+  metadata.gz: 63333753af17a6c29abe4896073180e09fd70f13c64f234b347abba88805ca22
+  data.tar.gz: 60c2cff72eca13c1ecb23fc9b8b0a26bf3ccdfc2d5da26395ba7db059ede19cc
 SHA512:
-  metadata.gz: a64203d85443d9b812c2d36aa0718283fb21fd784043a4d1346ec96a3386bec3766f4ed127d41c1c6f149af7abb4af6241fad089caf72b058037b7bc323b08cb
-  data.tar.gz: f0e1ec3b3be53fbe00f31dddc6e16e6a3befb7457d307cbc9906160b02e94bc66d8d18ec1c0095c2e6a53c9d48d4f256515da6ac0ea1ae4bb327ab07fb0772a0
+  metadata.gz: 9452ff80ce8e7eedd97bd5ee9b3e536d5e8ec596cf81ff7981a092a5d4e7d08eb9cd221735b57d8f5988fa6f27eaef5503f16e40a60aa8af9d45bb44dfa7a2f4
+  data.tar.gz: 4927f961c0036849a20ca6365e87aaa035ca618918cf8bbe7261f5052d95e0552e251c3b9e51179af0f677950e78ff489d18bfe2d794c5fa9af2880787702e47

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
@@ -267,7 +280,7 @@ module OxAiWorkers
         type: 'file',
         file: {
           filename:,
-          file_data: Base64.strict_encode64(pdf)
+          file_data: "data:application/pdf;base64,#{Base64.strict_encode64(pdf)}"
         }
       }
 

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.1'
 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.1
 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