ruby_llm 0.1.0.pre43 → 0.1.0.pre44

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b2433d0b669ef8bd499507edf33ed672d011fc1c95c2de0febd4e83d2593e0b8
-  data.tar.gz: a818a1c59220404c29c0e562c473480ac630a6fd53fadee65ab102b8955e9ec5
+  metadata.gz: 519b87700f2ba3d5aeb8ea3a87d0881f4ef58f974991647b7af4c9f138cbf3d2
+  data.tar.gz: 553b6441ad59fe5efe6d51374103690101c2cb8d072f91bf2a850d84b4934601
 SHA512:
-  metadata.gz: afa3882a1850692e63f16dbc8c379a9aa71d704096f1eb2d5d88aeefa0c9c17f2585e8e215be0ed126caa0ec6bb7d07b4c12b5b6eaa5a1cf69a054b976b0d9e8
-  data.tar.gz: a8089658eab75b8200f51d39568dc876cdcc2d61d6581dbd1e6c211f487fa127c9300e0fcbb318f958b6b109f5fb43c2f7f6824acd2f426e4a0a9ae2a7023c63
+  metadata.gz: e52cb96479a0f4443521bf3b5b2e98c882fb1c404e3d1527c74f638137bf56c2cb9442cc6e30ef197157fbdaa7dc189b7f6331a31e6f31ddfa28a926035d6578
+  data.tar.gz: b1d7948851e6c1ffa5257e214f01e6d6efaf64d32d98dcc66ec8f5d6776df232a92b3b26ff1de78889f4195ff27e41dc7375a4b2530dc1b97edc8fd57bb2c835

data/.rspec_status

@@ -35,7 +35,7 @@ example_id                                         | status | run_time        |
 ./spec/ruby_llm/embeddings_spec.rb[1:1:2:1]        | passed | 0.65614 seconds |
 ./spec/ruby_llm/embeddings_spec.rb[1:1:2:2]        | passed | 2.16 seconds    |
 ./spec/ruby_llm/error_handling_spec.rb[1:1]        | passed | 0.29366 seconds |
-./spec/ruby_llm/image_generation_spec.rb[1:1:1]    | passed | 11.61 seconds   |
-./spec/ruby_llm/image_generation_spec.rb[1:1:2]    | passed | 17.63 seconds   |
-./spec/ruby_llm/image_generation_spec.rb[1:1:3]    | passed | 8.77 seconds    |
-./spec/ruby_llm/image_generation_spec.rb[1:1:4]    | passed | 0.00319 seconds |
+./spec/ruby_llm/image_generation_spec.rb[1:1:1]    | passed | 24.16 seconds   |
+./spec/ruby_llm/image_generation_spec.rb[1:1:2]    | passed | 14.81 seconds   |
+./spec/ruby_llm/image_generation_spec.rb[1:1:3]    | passed | 9.17 seconds    |
+./spec/ruby_llm/image_generation_spec.rb[1:1:4]    | passed | 0.00083 seconds |

data/lib/ruby_llm/image.rb

@@ -3,16 +3,39 @@
 module RubyLLM
   # Represents a generated image from an AI model.
   # Provides an interface to image generation capabilities
-  # from providers like DALL-E.
+  # from providers like DALL-E and Gemini's Imagen.
   class Image
-    attr_reader :url, :revised_prompt, :model_id
+    attr_reader :url, :data, :mime_type, :revised_prompt, :model_id
 
-    def initialize(url:, revised_prompt: nil, model_id: nil)
+    def initialize(url: nil, data: nil, mime_type: nil, revised_prompt: nil, model_id: nil)
       @url = url
+      @data = data
+      @mime_type = mime_type
       @revised_prompt = revised_prompt
       @model_id = model_id
     end
 
+    def base64?
+      !@data.nil?
+    end
+
+    # Returns the raw binary image data regardless of source
+    def to_blob
+      if base64?
+        Base64.decode64(@data)
+      else
+        # Use Faraday instead of URI.open for better security
+        response = Faraday.get(@url)
+        response.body
+      end
+    end
+
+    # Saves the image to a file path
+    def save(path)
+      File.binwrite(File.expand_path(path), to_blob)
+      path
+    end
+
     def self.paint(prompt, model: nil, size: '1024x1024')
       model_id = model || RubyLLM.config.default_image_model
       Models.find(model_id) # Validate model exists

data/lib/ruby_llm/provider.rb

@@ -158,6 +158,16 @@ module RubyLLM
       end
     end
 
+    def parse_data_uri(uri)
+      if uri&.start_with?('data:')
+        match = uri.match(/\Adata:([^;]+);base64,(.+)\z/)
+        return { mime_type: match[1], data: match[2] } if match
+      end
+
+      # If it's not a data URI, return nil
+      nil
+    end
+
     class << self
       def extended(base)
         base.extend(Methods)

data/lib/ruby_llm/providers/gemini/images.rb

@@ -37,12 +37,13 @@ module RubyLLM
             raise Error, 'Unexpected response format from Gemini image generation API'
           end
 
-          # Handle response with base64 encoded image data
-          image_url = "data:#{image_data['mimeType'] || 'image/png'};base64,#{image_data['bytesBase64Encoded']}"
+          # Extract mime type and base64 data
+          mime_type = image_data['mimeType'] || 'image/png'
+          base64_data = image_data['bytesBase64Encoded']
+
           Image.new(
-            url: image_url,
-            revised_prompt: '', # Imagen doesn't return revised prompts
-            model_id: ''
+            data: base64_data,
+            mime_type: mime_type
           )
         end
       end

data/lib/ruby_llm/providers/openai/images.rb

@@ -26,6 +26,7 @@ module RubyLLM
 
           Image.new(
             url: image_data['url'],
+            mime_type: 'image/png', # DALL-E typically returns PNGs
             revised_prompt: image_data['revised_prompt'],
             model_id: data['model']
           )

data/lib/ruby_llm/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RubyLLM
-  VERSION = '0.1.0.pre43'
+  VERSION = '0.1.0.pre44'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby_llm
 version: !ruby/object:Gem::Version
-  version: 0.1.0.pre43
+  version: 0.1.0.pre44
 platform: ruby
 authors:
 - Carmine Paolino