gen-ai 0.3.0 → 0.3.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1bbe9213620c2cb3e76d8b77bf895408a3e48398ac6bddf2a9cad20e5933fee4
-  data.tar.gz: 479fe26279cff268b640aba8afd100422f1c2548b0842543081c829637bbf8dd
+  metadata.gz: 17dcb3c412f63caf6c69bd68617658683828aab1d5554daf8943a78ebf73eb0f
+  data.tar.gz: 23a6f04f28ac6535ac7b2ab965390d40fbdc92d4daf08ceba4ed5fe59b55ab74
 SHA512:
-  metadata.gz: dacb5b6e3a5b56e3d2dc0833b9169ef77146decd34010a20945f69a974a273c5a3e7316893c7c0a2809b6e96f0113b976cd9d3dad036060c6a054ca6a9170831
-  data.tar.gz: c0fd1e20f18d89c828b32a76f8c43af78fa7b0e3730f8172256e193e65eeec5a4eba26b32bb08ddcf9f2147196c2ba20d37c1eba6fdbbbea8f91787319723f79
+  metadata.gz: 179541fd7a5c4b3e15b589e101fe4daf0f442d4b7ecc4af3b63f3c826bd14ac5a4386269c1c39a7f2e33583050c65ab5b41b726f21c2355e695d718e8aedf044
+  data.tar.gz: cdd5b4223b1d756c6b1812ce8bd1d8b182ff029af68ab024b0f160d3315ff0917b616587e1efaf393d11516859bab762596c22241230b091f166c7eef186ac39

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, messages)
+        if message.is_a?(String)
+          { author: messages.dig(0, :author) || DEFAULT_ROLE, content: message }
+        else
+          message
+        end
+      end
+
       def extract_embeddings(responses)
         responses.map { |response| response.dig('embedding', 'value') }
       end

data/lib/gen_ai/version.rb

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

metadata

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