gen-ai 0.2.2 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2b25d130c100818a73d652eaf9b585a4d024f6134a66de3f20eb5dd6e3d2212a
-  data.tar.gz: 7d971d676915f71edf74fec57cd7cb59b2d2c6607d7ed818d05d07b19f8cdecc
+  metadata.gz: 1bbe9213620c2cb3e76d8b77bf895408a3e48398ac6bddf2a9cad20e5933fee4
+  data.tar.gz: 479fe26279cff268b640aba8afd100422f1c2548b0842543081c829637bbf8dd
 SHA512:
-  metadata.gz: 98554f5f32925f6ea70ea78e8c462d3d24949fb265fed69ae5f73fc722206c455e81bec75a879aa21bf841b372518f6ffce5497ec3c82bf60260b4d4c8ed2525
-  data.tar.gz: 905b2aa2b483711de399aa166dc5c83f9e7a482a1649f77281c995adbcf0f3dc9847b3ecbc0bdc21028a0f79f4ffb770c0a09dbad6a02d2a191d206dfeb43712
+  metadata.gz: dacb5b6e3a5b56e3d2dc0833b9169ef77146decd34010a20945f69a974a273c5a3e7316893c7c0a2809b6e96f0113b976cd9d3dad036060c6a054ca6a9170831
+  data.tar.gz: c0fd1e20f18d89c828b32a76f8c43af78fa7b0e3730f8172256e193e65eeec5a4eba26b32bb08ddcf9f2147196c2ba20d37c1eba6fdbbbea8f91787319723f79

data/README.md

@@ -20,23 +20,29 @@ If bundler is not being used to manage dependencies, install the gem by executin
 
 ## Usage
 
+Require it in you code:
+
+```ruby
+require 'gen_ai'
+```
+
 ### Feature support
-✅ - Supported  |  ❌ - Not supported  |  🛠️ - Work in progress
+
+✅ - Supported | ❌ - Not supported | 🛠️ - Work in progress
 
 Language models capabilities
 
 | Provider         | Embedding | Completion | Conversation | Sentiment | Summarization |
-| ---------------- | --------- | :--------: | :----------: | :-------: | :-----------: |
-| **OpenAI**       | ✅        | ✅         | ✅            | 🛠️        | 🛠️             |
-| **Google Palm2** | ✅        | ✅         | ✅            | 🛠️        | 🛠️             |
-
+| ---------------- | :-------: | :--------: | :----------: | :-------: | :-----------: |
+| **OpenAI**       |    ✅     |     ✅     |      ✅      |    🛠️     |      🛠️       |
+| **Google Palm2** |    ✅     |     ✅     |      ✅      |    🛠️     |      🛠️       |
 
 Image generation model capabilities
 
-| Provider         | Generate  | Variations | Edit         | Upscale   |
-| ---------------- | --------- | :--------: | :----------: | :-------: |
-| **OpenAI**       | ✅        | ✅         | ✅            | ❌        |
-| **StabilityAI**  | ✅        | ❌         | ✅            | ✅        |
+| Provider        | Generate | Variations | Edit | Upscale |
+| --------------- | :------: | :--------: | :--: | :-----: |
+| **OpenAI**      |    ✅    |     ✅     |  ✅  |   ❌    |
+| **StabilityAI** |    ✅    |     ❌     |  ✅  |   ✅    |
 
 ### Language
 
@@ -116,15 +122,18 @@ result = model.generate('A painting of a dog')
 # => #<GenAI::Result:0x0000000110be6f20...>
 
 result.value
-# => Base64 encoded image
+# => image binary
+
+result.value(:base64)
+# => image in base64
 
 # Save image to file
 File.open('dog.jpg', 'wb') do |f|
-  f.write(Base64.decode64(result.value))
+  f.write(result.value)
 end
 ```
-![dog](https://github.com/alchaplinsky/gen-ai/assets/695947/27a2af5d-530b-4966-94e8-6cdf628b6cac)
 
+![dog](https://github.com/alchaplinsky/gen-ai/assets/695947/27a2af5d-530b-4966-94e8-6cdf628b6cac)
 
 Get more **variations** of the same image
 
@@ -133,16 +142,19 @@ result = model.variations('./dog.jpg')
 # => #<GenAI::Result:0x0000000116a1ec50...>
 
 result.value
-# => Base64 encoded image
+# => image binary
+
+result.value(:base64)
+# => image in base64
 
 # Save image to file
 File.open('dog_variation.jpg', 'wb') do |f|
-  f.write(Base64.decode64(result.value))
+  f.write(result.value)
 end
 
 ```
-![dog_variation](https://github.com/alchaplinsky/gen-ai/assets/695947/977f5238-0114-4085-8e61-8f8b356ce308)
 
+![dog_variation](https://github.com/alchaplinsky/gen-ai/assets/695947/977f5238-0114-4085-8e61-8f8b356ce308)
 
 **Editing** existing images with additional prompt
 
@@ -151,18 +163,20 @@ result = model.edit('./llama.jpg', 'A cute llama wearing a beret', mask: './mask
 # => #<GenAI::Result:0x0000000116a1ec50...>
 
 result.value
-# => Base64 encoded image
+# => image binary
+
+result.value(:base64)
+# => image in base64
 
 # Save image to file
 File.open('dog_edited.jpg', 'wb') do |f|
-  f.write(Base64.decode64(result.value))
+  f.write(result.value)
 end
 ```
 
 ![llama](https://github.com/alchaplinsky/gen-ai/assets/695947/9c862c6c-428e-463c-b935-ca749a6a33df)
 ![llama_edited](https://github.com/alchaplinsky/gen-ai/assets/695947/070d8e6a-07a0-4ed2-826f-8b9aabd183ae)
 
-
 ## Development
 
 After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.

data/lib/gen/ai.rb

@@ -0,0 +1,3 @@
+# frozen_string_literal: true
+
+require_relative '../gen_ai'

data/lib/gen_ai/image/open_ai.rb

@@ -1,9 +1,12 @@
 # frozen_string_literal: true
 
+require 'base64'
+
 module GenAI
   class Image
     class OpenAI < Base
       DEFAULT_SIZE = '256x256'
+      DEFAULT_MODEL = 'dall-e-3'
       RESPONSE_FORMAT = 'b64_json'
 
       def initialize(token:, options: {})
@@ -19,8 +22,8 @@ module GenAI
 
         build_result(
           raw: response,
-          model: 'dall-e',
-          parsed: response['data'].map { |datum| datum[RESPONSE_FORMAT] }
+          model: parameters[:model],
+          parsed: parse_response_data(response['data'])
         )
       end
 
@@ -31,8 +34,8 @@ module GenAI
 
         build_result(
           raw: response,
-          model: 'dall-e',
-          parsed: response['data'].map { |datum| datum[RESPONSE_FORMAT] }
+          model: parameters[:model],
+          parsed: parse_response_data(response['data'])
         )
       end
 
@@ -43,8 +46,8 @@ module GenAI
 
         build_result(
           raw: response,
-          model: 'dall-e',
-          parsed: response['data'].map { |datum| datum[RESPONSE_FORMAT] }
+          model: parameters[:model],
+          parsed: parse_response_data(response['data'])
         )
       end
 
@@ -54,6 +57,7 @@ module GenAI
         {
           prompt: prompt,
           size: options.delete(:size) || DEFAULT_SIZE,
+          model: options.delete(:model) || DEFAULT_MODEL,
           response_format: options.delete(:response_format) || RESPONSE_FORMAT
         }.merge(options)
       end
@@ -62,6 +66,7 @@ module GenAI
         {
           image: image,
           size: options.delete(:size) || DEFAULT_SIZE,
+          model: 'dall-e-2', # variation is only available on dall-e-2
           response_format: options.delete(:response_format) || RESPONSE_FORMAT
         }.merge(options)
       end
@@ -71,9 +76,14 @@ module GenAI
           image: image,
           prompt: prompt,
           size: options.delete(:size) || DEFAULT_SIZE,
+          model: 'dall-e-2', # edit is only available on dall-e-2
           response_format: options.delete(:response_format) || RESPONSE_FORMAT
         }.merge(options)
       end
+
+      def parse_response_data(data)
+        data.map { |datum| Base64.decode64(datum[RESPONSE_FORMAT]) }
+      end
     end
   end
 end

data/lib/gen_ai/image/stability_ai.rb

@@ -23,7 +23,7 @@ module GenAI
         build_result(
           raw: response,
           model: model,
-          parsed: response['artifacts'].map { |artifact| artifact['base64'] }
+          parsed: parse_response_data(response['artifacts'])
         )
       end
 
@@ -36,7 +36,7 @@ module GenAI
         build_result(
           raw: response,
           model: model,
-          parsed: response['artifacts'].map { |artifact| artifact['base64'] }
+          parsed: parse_response_data(response['artifacts'])
         )
       end
 
@@ -49,7 +49,7 @@ module GenAI
         build_result(
           raw: response,
           model: model,
-          parsed: response['artifacts'].map { |artifact| artifact['base64'] }
+          parsed: parse_response_data(response['artifacts'])
         )
       end
 
@@ -91,6 +91,10 @@ module GenAI
         size = options.delete(:size) || DEFAULT_SIZE
         size.split('x').map(&:to_i)
       end
+
+      def parse_response_data(data)
+        data.map { |artifact| Base64.decode64(artifact['base64']) }
+      end
     end
   end
 end

data/lib/gen_ai/result.rb

@@ -11,8 +11,15 @@ module GenAI
       @values = values
     end
 
-    def value
-      values.first
+    def value(format = :raw)
+      case format
+      when :raw
+        values.first
+      when :base64
+        Base64.encode64(values.first)
+      else
+        raise "Unsupported format: #{format}"
+      end
     end
 
     def prompt_tokens

data/lib/gen_ai/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module GenAI
-  VERSION = '0.2.2'
+  VERSION = '0.3.0'
 end

data/lib/gen_ai.rb

@@ -8,6 +8,7 @@ loader.inflector.inflect(
   'open_ai' => 'OpenAI',
   'stability_ai' => 'StabilityAI'
 )
+loader.ignore("#{__dir__}/gen")
 loader.setup
 
 module GenAI

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: gen-ai
 version: !ruby/object:Gem::Version
-  version: 0.2.2
+  version: 0.3.0
 platform: ruby
 authors:
 - Alex Chaplinsky
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-10-29 00:00:00.000000000 Z
+date: 2023-11-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday
@@ -96,6 +96,7 @@ files:
 - README.md
 - Rakefile
 - gen-ai.gemspec
+- lib/gen/ai.rb
 - lib/gen_ai.rb
 - lib/gen_ai/api/client.rb
 - lib/gen_ai/base.rb