lib/gen_ai/image.rb in gen-ai-0.3.2 vs lib/gen_ai/image.rb in gen-ai-0.4.0.alpha.1

@@ -1,31 +1,17 @@
 # frozen_string_literal: true
 
 module GenAI
   class Image
+    extend Forwardable
+
+    def_delegators :@client, :generate, :variations, :edit, :upscale
+
     def initialize(provider, token, options: {})
       build_client(provider, token, options)
     end
 
-    def generate(prompt, options = {})
-      client.generate(prompt, options)
-    end
-
-    def variations(image, options = {})
-      client.variations(image, options)
-    end
-
-    def edit(image, prompt, options = {})
-      client.edit(image, prompt, options)
-    end
-
-    def upscale(image, options = {})
-      client.upscale(image, options)
-    end
-
     private
-
-    attr_reader :client
 
     def build_client(provider, token, options)
       klass = GenAI::Image.constants.find do |const|
         const.to_s.downcase == provider.to_s.downcase.gsub('_', '')
       end