ox-ai-workers 0.9.2 → 0.9.3.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 716fe34f8c0ee4841a0788a04febee87e885af815d30cf533e46d55c4fa76e54
-  data.tar.gz: ee4e89522ba6b8ae4484298e810b321af3f7a4d15bbd70dd0b2118dfd070b26c
+  metadata.gz: b0e2cfa9b2f611dc00729dfff984e91b4b3977703d91f360f80ff4625013cccc
+  data.tar.gz: fe2e37c5fc822a1a661f36b421f511f24c4979962ab4d67e79912d0101ace732
 SHA512:
-  metadata.gz: e17ecef126df8779d99e3b4de595cd590c3abf6e24e74a85dcbaefaf6daade38541348fe12477b029ea94de2c2cb29b312be8ee7543a154b575bcd32a876b4f2
-  data.tar.gz: 0fbd6beee90840ba1c3490edcc7f2e670881d3b81765c42b1829109f456a77e0120f153f444e0ac85e3201a9e451015b9601d021f845ec608365234c71852d22
+  metadata.gz: ab9b3e707484c1607ad46ba9cecaedc19d9e673aa4e3a17564402e5b80e8d63617a62842176eb01aea46e29e98b9b2c6e63861cb5ad82470581fc2ba379606a6
+  data.tar.gz: ceba0647dbbcc2a1cde6b34963d7f3a7f6d9d0c299826bd06891c03d6501c3edb6f3e9eecc3f84c155f61bc70deb613897c8c6ae7e62c41084784b3e5688c0b8

data/CHANGELOG.md

@@ -1,6 +1,8 @@
-## [0.9.2] - 2025-05-06
+## [0.9.3] - 2025-05-06
 
 - Changed `max_tokens` to `max_completion_tokens` in `params`
+- Added `frequency_penalty` for configuration
+- Added `edit_image` for `Pixels` tool
 
 ## [0.9.1] - 2025-05-04
 

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

@@ -42,6 +42,10 @@ ru:
         generate_image:
           description: 'Инструмент генерации изображений: Создает изображение на основе предоставленного текстового запроса. Если указать имя файла, изображение будет сохранено в формате png. В любом случае будет возвращен url изображения.'
           prompt: 'Текстовое описание изображения для генерации'
-          size: 'Размер генерируемого изображения (1024x1792, 1792x1024, 1024x1024)'
-          quality: 'Качество генерируемого изображения (standard, hd)'
+          size: 'Размер генерируемого изображения'
+          quality: 'Качество генерируемого изображения'
           file_name: 'Имя файла для сохранения изображения в формате png. Например: image.png'
+        edit_image:
+          description: 'Инструмент редактирования изображений: Редактирует изображение на основе предоставленного текстового запроса. Если указать имя файла, изображение будет сохранено в формате png. В любом случае будет возвращен url изображения.'
+          input_image: 'URL или путь к изображению для редактирования'
+          prompt: 'Текстовое описание действий для редактирования изображения'

data/lib/oxaiworkers/models/deepseek_max.rb

@@ -5,11 +5,11 @@ module OxAiWorkers
     class DeepseekMax
       include OxAiWorkers::Models::ModuleBase
 
-      def initialize(uri_base: nil, api_key: nil, model: nil, max_tokens: nil, temperature: nil)
+      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/'
         @api_key = api_key || OxAiWorkers.configuration.access_token_deepseek
-        super(uri_base: @uri_base, api_key: @api_key, model: @model, max_tokens:, temperature:)
+        super(uri_base: @uri_base, api_key: @api_key, model: @model, max_tokens:, temperature:, frequency_penalty:)
       end
     end
   end

data/lib/oxaiworkers/models/module_base.rb

@@ -3,11 +3,12 @@
 module OxAiWorkers
   module Models
     module ModuleBase
-      attr_accessor :uri_base, :api_key, :model, :max_tokens, :temperature
+      attr_accessor :uri_base, :api_key, :model, :max_tokens, :temperature, :frequency_penalty
 
-      def initialize(uri_base:, api_key:, model:, max_tokens: nil, temperature: nil)
+      def initialize(uri_base:, api_key:, model:, max_tokens: nil, temperature: nil, frequency_penalty: nil)
         @max_tokens = max_tokens || OxAiWorkers.configuration.max_tokens
         @temperature = temperature || OxAiWorkers.configuration.temperature
+        @frequency_penalty = frequency_penalty || 0
         @api_key = api_key
         @uri_base = uri_base
         @model = model

data/lib/oxaiworkers/models/openai_max.rb

@@ -5,10 +5,10 @@ module OxAiWorkers
     class OpenaiMax
       include OxAiWorkers::Models::ModuleBase
 
-      def initialize(uri_base: nil, api_key: nil, model: nil, max_tokens: nil, temperature: nil)
+      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
-        super(uri_base:, api_key: @api_key, model: @model, max_tokens:, temperature:)
+        super(uri_base:, api_key: @api_key, model: @model, max_tokens:, temperature:, frequency_penalty:)
       end
     end
   end

data/lib/oxaiworkers/models/openai_mini.rb

@@ -3,9 +3,9 @@
 module OxAiWorkers
   module Models
     class OpenaiMini < OpenaiMax
-      def initialize(uri_base: nil, api_key: nil, model: nil, max_tokens: nil, temperature: nil)
+      def initialize(uri_base: nil, api_key: nil, model: nil, max_tokens: nil, temperature: nil, frequency_penalty: nil)
         @model = model || 'gpt-4.1-mini'
-        super(uri_base:, api_key:, model: @model, max_tokens:, temperature:)
+        super(uri_base:, api_key:, model: @model, max_tokens:, temperature:, frequency_penalty:)
       end
     end
   end

data/lib/oxaiworkers/models/openai_nano.rb

@@ -3,9 +3,9 @@
 module OxAiWorkers
   module Models
     class OpenaiNano < OpenaiMax
-      def initialize(uri_base: nil, api_key: nil, model: nil, max_tokens: nil, temperature: nil)
+      def initialize(uri_base: nil, api_key: nil, model: nil, max_tokens: nil, temperature: nil, frequency_penalty: nil)
         @model = model || 'gpt-4.1-nano'
-        super(uri_base:, api_key:, model: @model, max_tokens:, temperature:)
+        super(uri_base:, api_key:, model: @model, max_tokens:, temperature:, frequency_penalty:)
       end
     end
   end

data/lib/oxaiworkers/module_request.rb

@@ -46,7 +46,8 @@ module OxAiWorkers
         model: @model.model,
         messages: @messages,
         temperature: @model.temperature,
-        max_completion_tokens: @model.max_tokens
+        max_completion_tokens: @model.max_tokens,
+        frequency_penalty: @model.frequency_penalty
       }
       if @tools.present?
         parameters[:tools] = @tools

data/lib/oxaiworkers/tool/pixels.rb

@@ -8,9 +8,22 @@ module OxAiWorkers
       include OxAiWorkers::DependencyHelper
       include OxAiWorkers::LoadI18n
 
-      attr_accessor :worker, :url, :current_dir
+      attr_accessor :worker, :url, :current_dir, :image_model, :mask
 
-      def initialize(worker:, current_dir: nil, only: nil)
+      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)
         store_locale
 
         init_white_list_with only
@@ -19,27 +32,44 @@ module OxAiWorkers
           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: %w[1024x1792 1792x1024 1024x1024]
+                          enum: MODELS[image_model]['size']
           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: %w[standard hd]
+                             enum: MODELS[image_model]['quality']
         end
 
+        # define_function :edit_image, description: I18n.t('oxaiworkers.tool.pixels.edit_image.description') do
+        #   property :input_image, type: 'string', description: I18n.t('oxaiworkers.tool.pixels.edit_image.input_image'),
+        #                          required: true
+        #   property :prompt, type: 'string', description: I18n.t('oxaiworkers.tool.pixels.edit_image.prompt'),
+        #                     required: true
+        #   if current_dir.present?
+        #     property :output_file_name, type: 'string',
+        #                                 description: I18n.t('oxaiworkers.tool.pixels.generate_image.file_name'),
+        #                                 required: true
+        #   end
+        # end
+
         @worker = worker
         @current_dir = current_dir
+        @image_model = MODELS[image_model]
+        @mask = mask
       end
 
-      def generate_image(prompt:, file_name: nil, size: '1024x1792', quality: 'standard')
+      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: 'dall-e-3',
+            model: @image_model['model'],
             size:,
             quality:
           }
@@ -55,6 +85,30 @@ module OxAiWorkers
         end
       end
 
+      def edit_image(input_image:, prompt:, output_file_name: nil, size: nil, mask: nil)
+        size ||= @image_model['size'].first
+        mask ||= @mask
+
+        response = @worker.client.images.edit(
+          parameters: {
+            image: input_image,
+            model: @image_model['model'],
+            prompt:,
+            size:,
+            mask:
+          }
+        )
+
+        @url = response.dig('data', 0, 'url')
+        revised_prompt = response.dig('data', 0, 'revised_prompt')
+        if output_file_name.present?
+          path = save_generated_image(file_name: output_file_name)
+          "url: #{@url}\nfile_name: #{path}\n\nrevised_prompt: #{revised_prompt}"
+        else
+          "url: #{@url}\n\nrevised_prompt: #{revised_prompt}"
+        end
+      end
+
       def save_generated_image(file_name:)
         unless @current_dir.present?
           return 'Current directory not set for OxAiWorkers::Tool::Pixels. Please set current directory first.'

data/lib/oxaiworkers/version.rb

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ox-ai-workers
 version: !ruby/object:Gem::Version
-  version: 0.9.2
+  version: 0.9.3.1
 platform: ruby
 authors:
 - Denis Smolev