ai-chat 0.6.0 → 0.6.1

data/ai-chat.gemspec

@@ -2,7 +2,7 @@
 
 Gem::Specification.new do |spec|
   spec.name = "ai-chat"
-  spec.version = "0.6.0"
+  spec.version = "0.6.1"
   spec.authors = ["Raghu Betina", "Jelani Woods"]
   spec.email = ["raghu@firstdraft.com", "jelani@firstdraft.com"]
   spec.homepage = "https://github.com/firstdraft/ai-chat"
@@ -19,7 +19,7 @@ Gem::Specification.new do |spec|
   }
 
   spec.required_ruby_version = ">= 3.2"
-  spec.add_runtime_dependency "openai", "~> 0.43"
+  spec.add_runtime_dependency "openai", "~> 0.59"
   spec.add_runtime_dependency "marcel", "~> 1.0"
   spec.add_runtime_dependency "base64", "~> 0.1", "> 0.1.1"
   spec.add_runtime_dependency "json", "~> 2.0"

data/lib/ai/chat.rb

@@ -18,8 +18,8 @@ module AI
   # :reek:IrresponsibleModule
   class Chat
     # :reek:Attribute
-    attr_accessor :background, :code_interpreter, :conversation_id, :image_generation, :image_folder, :messages, :model, :reasoning_effort, :web_search
-    attr_reader :client, :last_response_id, :proxy, :schema, :schema_file, :verbosity
+    attr_accessor :background, :code_interpreter, :conversation_id, :image_folder, :messages, :model, :reasoning_effort, :web_search
+    attr_reader :client, :image_generation, :last_response_id, :proxy, :schema, :schema_file, :verbosity
 
     BASE_PROXY_URL = "https://prepend.me/api.openai.com/v1"
     PROXY_ENV = "AICHAT_PROXY"
@@ -205,6 +205,19 @@ module AI
       end
     end
 
+    def image_generation=(value)
+      case value
+      when true
+        @image_generation = true
+      when false, nil
+        @image_generation = false
+      when Hash
+        @image_generation = value.transform_keys(&:to_sym)
+      else
+        raise ArgumentError, "Invalid image_generation value: #{value.inspect}. Must be true, false, or a Hash of tool options (e.g., { size: \"1536x1024\", quality: \"low\" })."
+      end
+    end
+
     def last
       messages.last
     end
@@ -523,7 +536,8 @@ module AI
         tools_list << {type: "web_search"}
       end
       if image_generation
-        tools_list << {type: "image_generation"}
+        options = image_generation.is_a?(Hash) ? image_generation : {}
+        tools_list << options.merge(type: "image_generation")
       end
       if code_interpreter
         tools_list << {type: "code_interpreter", container: {type: "auto"}}
@@ -577,7 +591,8 @@ module AI
           result = output.result
           image_data = Base64.strict_decode64(result)
 
-          filename = "#{(index + 1).to_s.rjust(3, "0")}.png"
+          extension = image_extension_for(image_data)
+          filename = "#{(index + 1).to_s.rjust(3, "0")}.#{extension}"
           file_path = File.join(subfolder_path, filename)
 
           File.binwrite(file_path, image_data)
@@ -589,6 +604,18 @@ module AI
       image_filenames
     end
 
+    IMAGE_EXTENSIONS_BY_MIME_TYPE = {
+      "image/png" => "png",
+      "image/jpeg" => "jpg",
+      "image/webp" => "webp"
+    }.freeze
+
+    # :reek:UtilityFunction
+    def image_extension_for(bytes)
+      mime_type = Marcel::MimeType.for(StringIO.new(bytes))
+      IMAGE_EXTENSIONS_BY_MIME_TYPE.fetch(mime_type, "png")
+    end
+
     def create_images_folder(response_id)
       # ISO 8601 basic format with centisecond precision
       timestamp = Time.now.strftime("%Y%m%dT%H%M%S%2N")

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ai-chat
 version: !ruby/object:Gem::Version
-  version: 0.6.0
+  version: 0.6.1
 platform: ruby
 authors:
 - Raghu Betina
@@ -16,14 +16,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.43'
+        version: '0.59'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.43'
+        version: '0.59'
 - !ruby/object:Gem::Dependency
   name: marcel
   requirement: !ruby/object:Gem::Requirement