Imageable -> Blobbable

Author: josefarias

lib/ruby_llm/blobbable.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+module RubyLLM
+  # Mixin for classes that can be used to save a blob to a file
+  module Blobbable
+    def save(path)
+      if (blob = to_blob)
+        File.binwrite(File.expand_path(path), blob)
+        path
+      end
+    end
+
+    def to_blob
+      raise NotImplementedError, 'to_blob is not implemented'
+    end
+  end
+end

lib/ruby_llm/deferred_image.rb

@@ -3,7 +3,7 @@
 module RubyLLM
   # Represents an image that may not yet be fully generated.
   class DeferredImage
-    include Imageable
+    include Blobbable
 
     attr_reader :provider_instance, :url
 
@@ -12,10 +12,6 @@ def initialize(url:, provider_instance:)
       @provider_instance = provider_instance
     end
 
-    def deferred?
-      true
-    end
-
     def to_blob
       provider_instance.fetch_image_blob(url)
     end

lib/ruby_llm/image.rb

@@ -3,7 +3,7 @@
 module RubyLLM
   # Represents a generated image from an AI model.
   class Image
-    include Imageable
+    include Blobbable
 
     attr_reader :url, :data, :mime_type, :revised_prompt, :model_id
 

lib/ruby_llm/imageable.rb

@@ -75,7 +75,7 @@ def save_and_verify_deferred_image(image)
     it 'google/imagen-4-ultra, an official replicate model, can paint images' do
       image = RubyLLM.paint('a siamese cat', model: 'google/imagen-4-ultra', output_format: 'png')
 
-      expect(image.deferred?).to be(true)
+      expect(image).to be_a(RubyLLM::DeferredImage)
 
       save_and_verify_deferred_image image
     end
@@ -87,15 +87,15 @@ def save_and_verify_deferred_image(image)
                                              image_input: [ref_img1, ref_img2],
                                              output_format: 'png')
 
-      expect(image.deferred?).to be(true)
+      expect(image).to be_a(RubyLLM::DeferredImage)
 
       save_and_verify_deferred_image image
     end
 
     it 'prunaai/hidream-l1-fast, an unofficial replicate model, can paint images' do
       image = RubyLLM.paint('a siamese cat', model: 'prunaai/hidream-l1-fast', output_format: 'png')
 
-      expect(image.deferred?).to be(true)
+      expect(image).to be_a(RubyLLM::DeferredImage)
 
       save_and_verify_deferred_image image
     end