gen-ai 0.3.0 → 0.3.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1bbe9213620c2cb3e76d8b77bf895408a3e48398ac6bddf2a9cad20e5933fee4
-  data.tar.gz: 479fe26279cff268b640aba8afd100422f1c2548b0842543081c829637bbf8dd
+  metadata.gz: 6d0af08a39e94ab978107c2fafcbabf8c40906237212af8b85a8f14667892d43
+  data.tar.gz: 0b9392b7eab8ac6421a83ccaf0e81f658f46e050fbb8f5c34b30382cf542e564
 SHA512:
-  metadata.gz: dacb5b6e3a5b56e3d2dc0833b9169ef77146decd34010a20945f69a974a273c5a3e7316893c7c0a2809b6e96f0113b976cd9d3dad036060c6a054ca6a9170831
-  data.tar.gz: c0fd1e20f18d89c828b32a76f8c43af78fa7b0e3730f8172256e193e65eeec5a4eba26b32bb08ddcf9f2147196c2ba20d37c1eba6fdbbbea8f91787319723f79
+  metadata.gz: 94def9a001633716757f550d6fc5cc601de10c53d6d542b401261fb55a90e21d43a35ac9257e9a9144cca049d4cc88734b7caee9313bccd6fa2cd905f74d2a68
+  data.tar.gz: 29047daf653faaa79cae27548aeb36620cb8cf8d26f634e9504356f6aa6d2768a9eefa68383f362f2c969c5e1c1ef13fe7d4bfa751a5cb31c2c46ff7dee9d3b2

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/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module GenAI
-  VERSION = '0.3.0'
+  VERSION = '0.3.1'
 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.1
 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