generate_image 1.1.1 → 1.1.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 00f69d4b0c88b6e09af4a1cdd8b766041c0d014a4777368b791fb9acc6af82ec
-  data.tar.gz: d78c09e191db7a1be962f4b698c4682f10cc24108cb1309770fbdf5d96918df8
+  metadata.gz: 9b49d016f61d888fd44be3bf340bba2613f7df0587922edf96ace80ef3d0c183
+  data.tar.gz: b05ca85ee8e2c925e1a19fbbaf493fc38e5804d62a26ec81409ed58f21e06f23
 SHA512:
-  metadata.gz: 662d9656c7d83a67633dbea6995502729ff238f393d0924dccecebdd6f89abab3f6f2fadc810327a804c663792f431b7b98839135941b420049936604f1cbcae
-  data.tar.gz: f9b12463cd29f9ca602ef7e0d742aa93ca5a74431ad8e8a4ec7a5d877d7b0f998bca2d0096874aafc144861efa283555d4eef63aa42032b9d6f6ad60e990d782
+  metadata.gz: 635b4d2702eef6145cd4f9be6ca51e8a3bff5a2d2cdfbf0993f7617a259c50dd71713477647b2de318abc7e8ce4cbf2038b7d4a3efecd5d628491fcb32141cff
+  data.tar.gz: ccf51aeffc3cc9571935710af9693e9e0ee5ec62c8591705244dbd31a51440412a1cf5b25d5367f4fa74778d93c2713b6768fb47ee4ef6d37debad3c133d6538

data/lib/generate_image/version.rb

@@ -1,3 +1,3 @@
 module GenerateImage
-  VERSION = "1.1.1"
+  VERSION = "1.1.2.1"
 end

data/lib/generate_image.rb

@@ -6,31 +6,24 @@ module GenerateImage
 
   class Client
     API_ENDPOINT = 'https://api.openai.com/v1/images/generations'
-    IMAGE_MODEL_NAME = 'image-alpha-001'
-    TEXT_MODEL_NAME = 'text-davinci-002'
 
-    attr_accessor :api_key
-
-    def initialize(api_key = nil)
-      @api_key = api_key || ENV['DALL_E_API_KEY']
-      raise StandardError, "API Key not set" unless @api_key
+    def initialize(api_key)
+      @api_key = api_key
     end
 
     def generate_image(text, options = {})
+      unless @api_key
+        raise StandardError, "API Key not set"
+      end
       uri = URI(API_ENDPOINT)
       request = Net::HTTP::Post.new(uri)
       request['Content-Type'] = 'application/json'
-      request['Authorization'] = "Bearer #{@api_key}"
+      request['Authorization'] = "Bearer #{API_KEY}"
       request.body = {
-        model: options[:model] || IMAGE_MODEL_NAME,
         prompt: text,
-        num_images: options[:num_images] || 1,
+        n: options[:num_images] || 1,
         size: options[:size] || '512x512',
         response_format: options[:response_format] || 'url',
-        style: options[:style] || nil,
-        scale: options[:scale] || 1,
-        seed: options[:seed] || nil,
-        quality: options[:quality] || 80
       }.to_json
 
       response = Net::HTTP.start(uri.hostname, uri.port, use_ssl: true) do |http|
@@ -49,6 +42,5 @@ module GenerateImage
         raise RequestFailed, "Failed to generate image. Response code: #{response.code}. Response body: #{response.body}"
       end
     end
-
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: generate_image
 version: !ruby/object:Gem::Version
-  version: 1.1.1
+  version: 1.1.2.1
 platform: ruby
 authors:
 - AMQOR Merouane