ox-ai-workers 0.9.3.1 → 0.9.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b0e2cfa9b2f611dc00729dfff984e91b4b3977703d91f360f80ff4625013cccc
-  data.tar.gz: fe2e37c5fc822a1a661f36b421f511f24c4979962ab4d67e79912d0101ace732
+  metadata.gz: c008f2f4f413251ffa88e00935e118c9844b2e67ab5088ca1091f2b7ac64ac77
+  data.tar.gz: a9b4cf53a19322a10db55ffa480d1e9bfe3a587411398ca5797058f5941c6ee8
 SHA512:
-  metadata.gz: ab9b3e707484c1607ad46ba9cecaedc19d9e673aa4e3a17564402e5b80e8d63617a62842176eb01aea46e29e98b9b2c6e63861cb5ad82470581fc2ba379606a6
-  data.tar.gz: ceba0647dbbcc2a1cde6b34963d7f3a7f6d9d0c299826bd06891c03d6501c3edb6f3e9eecc3f84c155f61bc70deb613897c8c6ae7e62c41084784b3e5688c0b8
+  metadata.gz: ac9053e4092115ef23a2f39d834c511c91d9594d8fc8ba37af95652917879ff5cd390581d2bb6d37d184af61d2809a9292c33375c8bd3d06eccb4b84a29ad2ff
+  data.tar.gz: 907e52e93352af84f3bc7b38ef02180dc1a4850f645bb40c7e36d371e85c1a8cdb00e60aa9805d91834b01437c3ae2ff5eb60746e379ec0bfae65a147b627c5c

data/CHANGELOG.md

@@ -1,3 +1,16 @@
+
+## [0.9.5] - 2025-05-10
+
+- Added `call_stack` for `Iterator` and `ModuleRequest`
+- Added `stop_double_calls` for `Iterator` and `ModuleRequest`
+
+## [0.9.4] - 2025-05-08
+
+- Added `stability_images` model
+- Added `openai_gpt_image` model
+- Added `openai_dalle3` model
+- Added `access_token_stability` for configuration
+
 ## [0.9.3] - 2025-05-06
 
 - Changed `max_tokens` to `max_completion_tokens` in `params`

data/README.md

@@ -154,14 +154,25 @@ steps << "Step 4. When the solution is ready, notify about it and wait for the u
 # To retain the locale if you have assistants in different languages in your project.
 store_locale # Optional
 
+tool = MyTool.new
+
 @iterator = OxAiWorkers::Iterator.new(
   worker: init_worker(delayed: delayed, model: model),
   role: 'You are a software agent inside my computer',
-  tools: [MyTool.new],
+  tools: [tool],
   locale: @locale || I18n.locale,
   steps: steps,
   # def_except: [:outer_voice], # It's except steps with that functions
   # def_only: [:inner_monologue, :outer_voice], # Use it only with your steps
+  # Forced Function: Uses call_stack parameter to force the model to call functions in this exact order, one at a time
+  # call_stack: [ 
+  #   OxAiWorkers::Iterator.full_function_name(:outer_voice),
+  #   tool.full_function_name(:func1)
+  # ],
+  # Stop Double Calls: Uses stop_double_calls parameter to prevent the model from calling the same function twice in a row
+  # stop_double_calls: [
+  #   tool.full_function_name(:func1)
+  # ],
   on_inner_monologue: ->(text:) { puts "monologue: #{text}".colorize(:yellow) },
   on_outer_voice: ->(text:) { puts "voice: #{text}".colorize(:green) }
 )
@@ -438,6 +449,86 @@ OxAiWorkers provides several specialized assistant types:
   localizer.task = "Translate my application's interface"
   ```
 
+- **Painter**: Image generation and manipulation
+
+  ```ruby
+  painter = OxAiWorkers::Assistant::Painter.new
+  # or Set working directory to save generated images as files
+  # painter = OxAiWorkers::Assistant::Painter.new(current_dir: Dir.pwd)
+  painter.task = "Create an image of a sunset over mountains"
+  ```
+
+- **Orchestrator**: Coordinates multiple assistants to work together on complex tasks
+
+  ```ruby
+  orchestrator = OxAiWorkers::Assistant::Orchestrator.new(
+    workflow: 'Development team creates an application and tests it.'
+  )
+  orchestrator.add_assistant(OxAiWorkers::Assistant::Coder.new)
+  orchestrator.add_assistant(OxAiWorkers::Assistant::Sysop.new)
+  orchestrator.add_assistant(OxAiWorkers::Assistant::Localizer.new)
+  orchestrator.task = "Create a hello world application in C, save it to hello_world.c, compile, run, and verify it works."
+  ```
+
+### Available Tools
+
+OxAiWorkers provides several specialized tools to extend functionality:
+
+- **Pixels**: Image generation and manipulation tool
+
+  ```ruby
+  # Initialize with worker and optional parameters
+  pixels = OxAiWorkers::Tool::Pixels.new(
+    worker: worker,                 # Required: Request or DelayedRequest instance
+    current_dir: Dir.pwd,           # Optional: Directory to save generated images
+    image_model: 'dall-e-3',        # Optional: 'dall-e-3' or 'gpt-image-1'
+    only: [:generate_image]         # Optional: Limit available functions
+  )
+  ```
+
+  Provides functions for generating images with customizable parameters like size and quality, with ability to save generated images to disk.
+
+- **Pipeline**: Assistant coordination and communication tool
+
+  ```ruby
+  # Initialize with optional parameters
+  pipeline = OxAiWorkers::Tool::Pipeline.new(
+    on_message: ->(text:) { puts text } # Optional: Message handler callback
+  )
+  
+  # Add assistants to the pipeline
+  pipeline.add_assistant(OxAiWorkers::Assistant::Coder.new)
+  pipeline.add_assistant(OxAiWorkers::Assistant::Sysop.new)
+  ```
+
+  Enables communication between multiple assistants, maintaining message context and facilitating collaborative problem-solving.
+
+- **Eval**: Code execution tool
+
+  ```ruby
+  # Initialize with optional parameters
+  eval_tool = OxAiWorkers::Tool::Eval.new(
+    only: [:ruby, :sh],             # Optional: Limit available functions
+    current_dir: Dir.pwd            # Optional: Directory to execute commands in
+  )
+  ```
+
+  Allows execution of Ruby code and shell commands, with directory context support.
+
+- **FileSystem**: File operations tool
+
+  ```ruby
+  # Initialize with optional parameters
+  file_system = OxAiWorkers::Tool::FileSystem.new(
+    current_dir: Dir.pwd,           # Optional: Base directory for operations
+    only: [:list_directory, :read_file, :write_to_file] # Optional: Limit available functions
+  )
+  ```
+
+  Provides functions for listing directory contents, reading from files, and writing to files with support for relative paths.
+
+Additional tools like Database and Converter are available for specialized tasks and can be integrated using the same pattern.
+
 ### Implementing Your Own Assistant
 
 Create custom assistants by inheriting from existing ones or composing with the Iterator:

data/config/locales/en.oxaiworkers.tool.yml

@@ -40,8 +40,8 @@ en:
         assistant_info: "ID: **%{id}**, Title: %{title}\nRole: %{role}\nDescription: %{description}\nCapabilities: %{capabilities}"
       pixels:
         generate_image:
-          description: 'Image Generation Tool: Creates an image based on the provided text prompt. If a file name is provided, the image will be saved in png format. In any case, the url of the generated image will be returned.'
-          prompt: 'Text description of the image to generate'
-          size: 'Size of the generated image (1024x1792, 1792x1024, 1024x1024)'
-          quality: 'Quality of the generated image (standard, hd)'
+          description: 'Image Generation Tool: Creates an image based on the provided text prompt on **English language**.'
+          prompt: 'Text description of the image to generate on **English language**'
+          size: 'Size of the generated image'
+          quality: 'Quality of the generated image'
           file_name: 'File name for saving the generated image in png format. For example: image.png'

data/config/locales/ru.oxaiworkers.tool.yml

@@ -40,12 +40,12 @@ ru:
         assistant_info: "ID: **%{id}**, Название: %{title}\nРоль: %{role}\nОписание: %{description}\nВозможности: %{capabilities}"
       pixels:
         generate_image:
-          description: 'Инструмент генерации изображений: Создает изображение на основе предоставленного текстового запроса. Если указать имя файла, изображение будет сохранено в формате png. В любом случае будет возвращен url изображения.'
-          prompt: 'Текстовое описание изображения для генерации'
+          description: 'Инструмент генерации изображений: Создает изображение на основе предоставленного текстового запроса на **английском языке**.'
+          prompt: 'Текстовое описание изображения для генерации строго **на английском** языке'
           size: 'Размер генерируемого изображения'
           quality: 'Качество генерируемого изображения'
           file_name: 'Имя файла для сохранения изображения в формате png. Например: image.png'
         edit_image:
-          description: 'Инструмент редактирования изображений: Редактирует изображение на основе предоставленного текстового запроса. Если указать имя файла, изображение будет сохранено в формате png. В любом случае будет возвращен url изображения.'
+          description: 'Инструмент редактирования изображений: Редактирует изображение на основе предоставленного текстового запроса.'
           input_image: 'URL или путь к изображению для редактирования'
           prompt: 'Текстовое описание действий для редактирования изображения'

data/lib/ox-ai-workers.rb

@@ -36,12 +36,17 @@ require_relative 'oxaiworkers/assistant/localizer'
 require_relative 'oxaiworkers/assistant/orchestrator'
 require_relative 'oxaiworkers/assistant/painter'
 
-require_relative 'oxaiworkers/models/module_base'
+require_relative 'oxaiworkers/models/llm_base'
 require_relative 'oxaiworkers/models/openai_max'
 require_relative 'oxaiworkers/models/openai_mini'
 require_relative 'oxaiworkers/models/openai_nano'
 require_relative 'oxaiworkers/models/deepseek_max'
 
+require_relative 'oxaiworkers/models/images_base'
+require_relative 'oxaiworkers/models/stability_images'
+require_relative 'oxaiworkers/models/openai_gpt_image'
+require_relative 'oxaiworkers/models/openai_dalle3'
+
 require_relative 'oxaiworkers/engine' if defined?(Rails)
 
 module OxAiWorkers
@@ -52,7 +57,8 @@ module OxAiWorkers
   class ConfigurationError < Error; end
 
   class Configuration
-    attr_accessor :max_tokens, :temperature, :wait_for_complete, :access_token_deepseek, :access_token_openai
+    attr_accessor :max_tokens, :temperature, :wait_for_complete, :access_token_deepseek, :access_token_openai,
+                  :access_token_stability
 
     def initialize
       @max_tokens = DEFAULT_MAX_TOKEN
@@ -61,6 +67,7 @@ module OxAiWorkers
 
       @access_token_deepseek = nil
       @access_token_openai = nil
+      @access_token_stability = nil
 
       [Array, NilClass, String, Symbol, Hash].each do |c|
         c.send(:include, OxAiWorkers::PresentCompat) unless c.method_defined?(:present?)

data/lib/oxaiworkers/assistant/painter.rb

@@ -5,7 +5,7 @@ module OxAiWorkers
     class Painter
       include OxAiWorkers::Assistant::ModuleBase
 
-      def initialize(current_dir: nil, delayed: false, model: nil)
+      def initialize(current_dir: nil, delayed: false, model: nil, image_model: nil)
         store_locale
         @current_dir = current_dir
 
@@ -17,10 +17,13 @@ module OxAiWorkers
           @title = I18n.t('oxaiworkers.assistant.painter.title')
         end
 
+        worker = init_worker(delayed:, model:)
+        image_model ||= OxAiWorkers::Models::OpenaiDalle3.new
+
         @iterator = Iterator.new(
-          worker: init_worker(delayed:, model:),
+          worker:,
           role: @role,
-          tools: [Tool::Pixels.new(worker: init_worker(delayed: false, model:), current_dir:)],
+          tools: [Tool::Pixels.new(worker: image_model, current_dir:)],
           locale: @locale,
           on_inner_monologue: ->(text:) { puts "monologue: #{text}".colorize(:yellow) },
           on_outer_voice: ->(text:) { puts "voice: #{text}".colorize(:green) }

data/lib/oxaiworkers/iterator.rb

@@ -8,10 +8,12 @@ module OxAiWorkers
     include OxAiWorkers::LoadI18n
 
     attr_accessor :worker, :role, :messages, :context, :tools, :queue, :monologue, :tasks,
-                  :on_inner_monologue, :on_outer_voice, :on_finish, :def_except, :def_only
+                  :on_inner_monologue, :on_outer_voice, :on_finish, :def_except, :def_only,
+                  :call_stack, :stop_double_calls
 
     def initialize(worker:, role: nil, tools: [], on_inner_monologue: nil, on_outer_voice: nil,
-                   on_finish: nil, steps: nil, def_except: [], def_only: nil, locale: nil)
+                   on_finish: nil, steps: nil, def_except: [], def_only: nil, locale: nil,
+                   call_stack: nil, stop_double_calls: [])
 
       @locale = locale || I18n.locale
 
@@ -41,6 +43,18 @@ module OxAiWorkers
       @on_outer_voice = on_outer_voice
       @on_finish = on_finish
 
+      if call_stack&.any?
+        if available_defs.include?(:inner_monologue) && !call_stack.include?(OxAiWorkers::Iterator.full_function_name(:inner_monologue))
+          # Add inner_monologue first
+          @call_stack = [OxAiWorkers::Iterator.full_function_name(:inner_monologue)] + call_stack
+        end
+        # Add finish_it last
+        @call_stack = call_stack + [OxAiWorkers::Iterator.full_function_name(:finish_it)]
+      end
+
+      @stop_double_calls = [OxAiWorkers::Iterator.full_function_name(:inner_monologue),
+                            OxAiWorkers::Iterator.full_function_name(:outer_voice)] + stop_double_calls
+
       cleanup
 
       super()
@@ -99,6 +113,9 @@ module OxAiWorkers
     end
 
     def rebuild_worker
+      @worker.last_call = nil
+      @worker.call_stack = @call_stack.dup
+      @worker.stop_double_calls = @stop_double_calls
       @worker.messages = []
       @worker.append(role: :system, content: @role) if @role.present?
 

data/lib/oxaiworkers/models/deepseek_max.rb

@@ -2,9 +2,7 @@
 
 module OxAiWorkers
   module Models
-    class DeepseekMax
-      include OxAiWorkers::Models::ModuleBase
-
+    class DeepseekMax < LLMBase
       def initialize(uri_base: nil, api_key: nil, model: nil, max_tokens: nil, temperature: nil, frequency_penalty: nil)
         @model = model || 'deepseek-chat'
         @uri_base = uri_base || 'https://api.deepseek.com/'

data/lib/oxaiworkers/models/images_base.rb

@@ -0,0 +1,11 @@
+module OxAiWorkers
+  module Models
+    class ImagesBase
+      attr_accessor :client, :options, :qualities, :sizes, :result
+
+      def initialize(options:)
+        @options = options
+      end
+    end
+  end
+end

data/lib/oxaiworkers/models/{module_base.rb → llm_base.rb}

@@ -2,7 +2,7 @@
 
 module OxAiWorkers
   module Models
-    module ModuleBase
+    class LLMBase
       attr_accessor :uri_base, :api_key, :model, :max_tokens, :temperature, :frequency_penalty
 
       def initialize(uri_base:, api_key:, model:, max_tokens: nil, temperature: nil, frequency_penalty: nil)

data/lib/oxaiworkers/models/openai_dalle3.rb

@@ -0,0 +1,47 @@
+module OxAiWorkers
+  module Models
+    class OpenaiDalle3 < ImagesBase
+      def initialize(api_key: nil, options: {})
+        @sizes = %w[1024x1024 1024x1792 1792x1024]
+        @qualities = %w[standard hd]
+        api_key ||= OxAiWorkers.configuration.access_token_openai
+
+        @client = OpenAI::Client.new(access_token: api_key)
+        super(options:)
+      end
+
+      def generate(prompt:, size: nil, quality: nil)
+        puts "OpenaiDalle3: #{prompt}"
+
+        size ||= @sizes.first
+        quality ||= @qualities.first
+
+        options = {
+          prompt:,
+          model: 'dall-e-3',
+          size:,
+          quality:
+        }
+        options.merge!(@options)
+
+        response = @client.images.generate(
+          parameters: options
+        )
+
+        url = response.dig('data', 0, 'url')
+        b64_json = response.dig('data', 0, 'b64_json')
+        revised_prompt = response.dig('data', 0, 'revised_prompt')
+
+        @result = "revised_prompt: #{revised_prompt}\n\n"
+        if url.present?
+          @result += "url: #{url}\n\n"
+          URI.open(url).read
+        elsif b64_json.present?
+          Base64.decode64(b64_json)
+        else
+          nil
+        end
+      end
+    end
+  end
+end

data/lib/oxaiworkers/models/openai_gpt_image.rb

@@ -0,0 +1,37 @@
+module OxAiWorkers
+  module Models
+    class OpenaiGptImage < ImagesBase
+      def initialize(api_key: nil, options: {})
+        @sizes = %w[auto 1024x1024 1536x1024 1024x1536]
+        @qualities = %w[auto low medium high]
+        api_key ||= OxAiWorkers.configuration.access_token_openai
+
+        @client = OpenAI::Client.new(access_token: api_key)
+        super(options:)
+      end
+
+      def generate(prompt:, size: nil, quality: nil)
+        puts "OpenaiGptImage: #{prompt}"
+
+        size ||= @sizes.first
+        quality ||= @qualities.first
+
+        options = {
+          prompt:,
+          model: 'gpt-image-1',
+          size:,
+          quality:
+        }
+        options.merge!(@options)
+
+        response = @client.images.generate(
+          parameters: options
+        )
+
+        b64_json = response.dig('data', 0, 'b64_json')
+
+        Base64.decode64(b64_json)
+      end
+    end
+  end
+end

data/lib/oxaiworkers/models/openai_max.rb

@@ -2,9 +2,7 @@
 
 module OxAiWorkers
   module Models
-    class OpenaiMax
-      include OxAiWorkers::Models::ModuleBase
-
+    class OpenaiMax < LLMBase
       def initialize(uri_base: nil, api_key: nil, model: nil, max_tokens: nil, temperature: nil, frequency_penalty: nil)
         @model = model || 'gpt-4.1'
         @api_key = api_key || OxAiWorkers.configuration.access_token_openai

data/lib/oxaiworkers/models/stability_images.rb

@@ -0,0 +1,32 @@
+module OxAiWorkers
+  module Models
+    class StabilityImages < ImagesBase
+      def initialize(api_key: nil, timeout: 300, options: {})
+        @sizes = %w[auto]
+        @qualities = %w[auto]
+        api_key ||= OxAiWorkers.configuration.access_token_stability
+
+        @client = StabilitySDK::Client.new(api_key:, timeout:)
+        super(options:)
+      end
+
+      def generate(prompt:, size: nil, quality: nil)
+        puts "StabilityImages: #{prompt}"
+        options = {
+          engine_id: 'stable-diffusion-xl-1024-v1-0'
+        }
+        options.merge!(@options)
+        binary = nil
+        @client.generate(prompt, options) do |answer|
+          answer.artifacts.each do |artifact|
+            next unless artifact.type == :ARTIFACT_IMAGE
+
+            binary = artifact.binary
+          end
+        end
+
+        binary
+      end
+    end
+  end
+end

data/lib/oxaiworkers/module_request.rb

@@ -3,15 +3,17 @@
 module OxAiWorkers
   class ModuleRequest
     attr_accessor :result, :client, :messages, :model, :custom_id, :tools, :errors,
-                  :tool_calls_raw, :tool_calls, :is_truncated, :finish_reason, :on_stream_proc
+                  :tool_calls_raw, :tool_calls, :is_truncated, :finish_reason, 
+                  :on_stream_proc, :call_stack, :last_call, :stop_double_calls
 
-    def initialize_requests(model:, on_stream: nil)
+    def initialize_requests(model:, on_stream: nil, call_stack: nil)
       @custom_id = SecureRandom.uuid
       @model = model
       @client = nil
       @is_truncated = false
       @finish_reason = nil
       @on_stream_proc = on_stream
+      @call_stack = call_stack
 
       if @model.api_key.nil?
         error_text = "#{@model.model} access token missing!"
@@ -34,6 +36,7 @@ module OxAiWorkers
       @tool_calls_raw = nil
       @is_truncated = false
       @finish_reason = nil
+      @last_call = nil
     end
 
     def append(role: nil, content: nil, messages: nil)
@@ -50,8 +53,17 @@ module OxAiWorkers
         frequency_penalty: @model.frequency_penalty
       }
       if @tools.present?
-        parameters[:tools] = @tools
-        parameters[:tool_choice] = 'required'
+        parameters[:tools] = @tools.reject do |f|
+          tool_name = f[:function][:name]
+          tool_name == @last_call && @stop_double_calls.include?(tool_name)
+        end
+        if @call_stack&.any?
+          func1 = @call_stack.first
+          @call_stack = @call_stack.drop(1)
+          parameters[:tool_choice] = { type: 'function', function: { name: func1 } }
+        else
+          parameters[:tool_choice] = 'required'
+        end
       end
       if @on_stream_proc.present?
         parameters[:stream] = @on_stream_proc
@@ -134,6 +146,7 @@ module OxAiWorkers
           name: function['name'].split('__').last,
           args: args
         }
+        @last_call = function['name']
       end
     end
   end

data/lib/oxaiworkers/tool/pixels.rb

@@ -8,22 +8,9 @@ module OxAiWorkers
       include OxAiWorkers::DependencyHelper
       include OxAiWorkers::LoadI18n
 
-      attr_accessor :worker, :url, :current_dir, :image_model, :mask
-
-      MODELS = {
-        'dall-e-3' => {
-          'model' => 'dall-e-3',
-          'size' => %w[1024x1024 1024x1792 1792x1024],
-          'quality' => %w[standard hd]
-        },
-        'gpt-image-1' => {
-          'model' => 'gpt-image-1',
-          'size' => %w[1024x1024 1024x1792 1792x1024],
-          'quality' => %w[auto low medium high]
-        }
-      }
-
-      def initialize(worker:, current_dir: nil, only: nil, image_model: 'dall-e-3', mask: nil)
+      attr_accessor :worker, :current_dir
+
+      def initialize(worker:, current_dir: nil, only: nil)
         store_locale
 
         init_white_list_with only
@@ -31,15 +18,19 @@ module OxAiWorkers
         define_function :generate_image, description: I18n.t('oxaiworkers.tool.pixels.generate_image.description') do
           property :prompt, type: 'string', description: I18n.t('oxaiworkers.tool.pixels.generate_image.prompt'),
                             required: true
-          property :size, type: 'string', description: I18n.t('oxaiworkers.tool.pixels.generate_image.size'),
-                          enum: MODELS[image_model]['size']
+          if worker.sizes.length > 1
+            property :size, type: 'string', description: I18n.t('oxaiworkers.tool.pixels.generate_image.size'),
+                            enum: worker.sizes
+          end
           if current_dir.present?
             property :file_name, type: 'string',
                                  description: I18n.t('oxaiworkers.tool.pixels.generate_image.file_name'),
                                  required: true
           end
-          property :quality, type: 'string', description: I18n.t('oxaiworkers.tool.pixels.generate_image.quality'),
-                             enum: MODELS[image_model]['quality']
+          if worker.qualities.length > 1
+            property :quality, type: 'string', description: I18n.t('oxaiworkers.tool.pixels.generate_image.quality'),
+                               enum: worker.qualities
+          end
         end
 
         # define_function :edit_image, description: I18n.t('oxaiworkers.tool.pixels.edit_image.description') do
@@ -56,32 +47,18 @@ module OxAiWorkers
 
         @worker = worker
         @current_dir = current_dir
-        @image_model = MODELS[image_model]
-        @mask = mask
       end
 
       def generate_image(prompt:, file_name: nil, size: nil, quality: nil)
-        puts "generate_image: #{prompt}"
-
-        size ||= @image_model['size'].first
-        quality ||= @image_model['quality'].first
-
-        response = @worker.client.images.generate(
-          parameters: {
-            prompt:,
-            model: @image_model['model'],
-            size:,
-            quality:
-          }
-        )
+        binary = @worker.generate(prompt:, size:, quality:)
 
-        @url = response.dig('data', 0, 'url')
-        revised_prompt = response.dig('data', 0, 'revised_prompt')
         if file_name.present?
-          path = save_generated_image(file_name:)
-          "url: #{@url}\nfile_name: #{path}\n\nrevised_prompt: #{revised_prompt}"
+          path = save_generated_image(file_name:, binary:)
+          "Successfully generated image. file_name: #{path}\n\n#{@worker.result}"
+        elsif @worker.result.present?
+          @worker.result
         else
-          "url: #{@url}\n\nrevised_prompt: #{revised_prompt}"
+          'file_name not set for OxAiWorkers::Tool::Pixels. Please set file name first.'
         end
       end
 
@@ -109,20 +86,24 @@ module OxAiWorkers
         end
       end
 
-      def save_generated_image(file_name:)
+      def save_generated_image(file_name:, binary:)
         unless @current_dir.present?
           return 'Current directory not set for OxAiWorkers::Tool::Pixels. Please set current directory first.'
         end
 
+        return 'File name not set for OxAiWorkers::Tool::Pixels. Please set file name first.' unless file_name.present?
+
         # Ensure current_dir exists
         FileUtils.mkdir_p(@current_dir) unless Dir.exist?(@current_dir)
 
         path = File.join(@current_dir, file_name)
 
         File.open(path, 'wb') do |file|
-          file.write(URI.open(@url).read)
+          file.write(binary)
         end
 
+        puts "Successfully saved image. file_name: #{path}"
+
         file_name
       end
     end

data/lib/oxaiworkers/version.rb

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ox-ai-workers
 version: !ruby/object:Gem::Version
-  version: 0.9.3.1
+  version: 0.9.5
 platform: ruby
 authors:
 - Denis Smolev
@@ -155,10 +155,14 @@ files:
 - lib/oxaiworkers/iterator.rb
 - lib/oxaiworkers/load_i18n.rb
 - lib/oxaiworkers/models/deepseek_max.rb
-- lib/oxaiworkers/models/module_base.rb
+- lib/oxaiworkers/models/images_base.rb
+- lib/oxaiworkers/models/llm_base.rb
+- lib/oxaiworkers/models/openai_dalle3.rb
+- lib/oxaiworkers/models/openai_gpt_image.rb
 - lib/oxaiworkers/models/openai_max.rb
 - lib/oxaiworkers/models/openai_mini.rb
 - lib/oxaiworkers/models/openai_nano.rb
+- lib/oxaiworkers/models/stability_images.rb
 - lib/oxaiworkers/module_request.rb
 - lib/oxaiworkers/present_compat.rb
 - lib/oxaiworkers/request.rb