gen-ai 0.2.3 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e6fbef446e1383e14734c390b0e342433991858e95a083e0aaf9a31f844f4503
-  data.tar.gz: e8bda40ea6377b4ada7175bcabae430681ac6d707c75cce745e4a079ce17dba9
+  metadata.gz: 1bbe9213620c2cb3e76d8b77bf895408a3e48398ac6bddf2a9cad20e5933fee4
+  data.tar.gz: 479fe26279cff268b640aba8afd100422f1c2548b0842543081c829637bbf8dd
 SHA512:
-  metadata.gz: ade0a401f9c6cb8fdb1c1de1fd5e31b52a79ffc69af2267644f4ffe6f2bf8892d3f92bcf1a2e2aebfd1ea0790dab1dd3a8c5b42fa59e870bc0cbab13766ce1b9
-  data.tar.gz: 39826a4c7f854b2a0ffd24010306e612b2cad1465e60719ba518f5b2ee43a17338c268684e615b376388f280dd20740f750fc6c8e5b49dfbc822c1f2873ba9f6
+  metadata.gz: dacb5b6e3a5b56e3d2dc0833b9169ef77146decd34010a20945f69a974a273c5a3e7316893c7c0a2809b6e96f0113b976cd9d3dad036060c6a054ca6a9170831
+  data.tar.gz: c0fd1e20f18d89c828b32a76f8c43af78fa7b0e3730f8172256e193e65eeec5a4eba26b32bb08ddcf9f2147196c2ba20d37c1eba6fdbbbea8f91787319723f79

data/README.md

@@ -21,29 +21,28 @@ 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
 
@@ -123,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
 
@@ -140,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
 
@@ -158,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/image/open_ai.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require 'base64'
+
 module GenAI
   class Image
     class OpenAI < Base
@@ -21,7 +23,7 @@ module GenAI
         build_result(
           raw: response,
           model: parameters[:model],
-          parsed: response['data'].map { |datum| datum[RESPONSE_FORMAT] }
+          parsed: parse_response_data(response['data'])
         )
       end
 
@@ -32,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
 
@@ -44,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
 
@@ -64,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
@@ -73,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.3'
+  VERSION = '0.3.0'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: gen-ai
 version: !ruby/object:Gem::Version
-  version: 0.2.3
+  version: 0.3.0
 platform: ruby
 authors:
 - Alex Chaplinsky
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-11-14 00:00:00.000000000 Z
+date: 2023-11-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday