gen-ai 0.2.3 → 0.3.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e6fbef446e1383e14734c390b0e342433991858e95a083e0aaf9a31f844f4503
-  data.tar.gz: e8bda40ea6377b4ada7175bcabae430681ac6d707c75cce745e4a079ce17dba9
+  metadata.gz: 6d0af08a39e94ab978107c2fafcbabf8c40906237212af8b85a8f14667892d43
+  data.tar.gz: 0b9392b7eab8ac6421a83ccaf0e81f658f46e050fbb8f5c34b30382cf542e564
 SHA512:
-  metadata.gz: ade0a401f9c6cb8fdb1c1de1fd5e31b52a79ffc69af2267644f4ffe6f2bf8892d3f92bcf1a2e2aebfd1ea0790dab1dd3a8c5b42fa59e870bc0cbab13766ce1b9
-  data.tar.gz: 39826a4c7f854b2a0ffd24010306e612b2cad1465e60719ba518f5b2ee43a17338c268684e615b376388f280dd20740f750fc6c8e5b49dfbc822c1f2873ba9f6
+  metadata.gz: 94def9a001633716757f550d6fc5cc601de10c53d6d542b401261fb55a90e21d43a35ac9257e9a9144cca049d4cc88734b7caee9313bccd6fa2cd905f74d2a68
+  data.tar.gz: 29047daf653faaa79cae27548aeb36620cb8cf8d26f634e9504356f6aa6d2768a9eefa68383f362f2c969c5e1c1ef13fe7d4bfa751a5cb31c2c46ff7dee9d3b2

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/language/google_palm.rb

@@ -55,7 +55,7 @@ module GenAI
       def build_chat_options(message, context, history, examples, options)
         {
           model: options.delete(:model) || CHAT_COMPLETION_MODEL,
-          messages: history.append({ author: DEFAULT_ROLE, content: message }),
+          messages: history.append(build_message(message, history)),
           examples: compose_examples(examples),
           context: context
         }.merge(options)
@@ -87,6 +87,14 @@ module GenAI
         object.respond_to?(:to_ary) ? object.to_ary || [object] : [object]
       end
 
+      def build_message(message, history)
+        if message.is_a?(String)
+          { author: history.dig(0, :role) || DEFAULT_ROLE, content: message }
+        else
+          message
+        end
+      end
+
       def extract_embeddings(responses)
         responses.map { |response| response.dig('embedding', 'value') }
       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.1'
 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.1
 platform: ruby
 authors:
 - Alex Chaplinsky
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-11-14 00:00:00.000000000 Z
+date: 2024-01-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday