omniai-google 1.6.1 → 1.6.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 76a81bb43234ee957a4eeaad25a4aada75d776d2f2a510c65fbabed5ecb41954
-  data.tar.gz: 21dafc2e2b81cfa2dce6e74ce8a093812a0299cde5837ac8425c5ef46020aa45
+  metadata.gz: b2626eef567a04e92b73eaf64a39dc274c2651b479810cd62a94be1a920cf25f
+  data.tar.gz: 8c968e5ca2328ce77e604f91c36607c22ef6572bf2e7633505316fdd6bab93db
 SHA512:
-  metadata.gz: 3c3595de2ad04e982aa0f7ca062db4ccf9f7bf500ba05edaad988e9b3379c5183a585ab877eba3c150794584007a9caae9ed279300ace51f71a7f223fe32ec5c
-  data.tar.gz: 2b4046f363a0f8ccd4817f6266e92db2a86e43b93c9622710dab614e7413a02ec4f8c29eeefd905f1cac77fd2e48f2368fcd8389a7a2daa0aaf77490a2688047
+  metadata.gz: c104b92ca09ae4c29285b2ac35e7badd23d2c54e4cb7189b0f1fb2fa1d0ad0cac4770e6c8ceeaaf5468225db52913458e6d4b6b94a0054f5c58735b01a78911d
+  data.tar.gz: a1037ddc9ccc58f84dfba6a094d0c7e1481dbba103e4e4d9f16dfbfb21def59fee61e4c463e96e3807c0763db4920f84934133b66090c9a8d575ac20c4225913

data/lib/omniai/google/chat.rb

@@ -23,6 +23,8 @@ module OmniAI
         GEMINI_FLASH = GEMINI_1_5_FLASH
       end
 
+      DEFAULT_MODEL = Model::GEMINI_PRO
+
       TEXT_SERIALIZER = lambda do |content, *|
         { text: content.text }
       end
@@ -34,9 +36,10 @@ module OmniAI
       #   MESSAGE_SERIALIZER.call(message)
       MESSAGE_SERIALIZER = lambda do |message, context:|
         parts = message.content.is_a?(String) ? [Text.new(message.content)] : message.content
+        role = message.system? ? Role::USER : message.role
 
         {
-          role: message.role,
+          role:,
           parts: parts.map { |part| part.serialize(context:) },
         }
       end

data/lib/omniai/google/client.rb

@@ -45,7 +45,7 @@ module OmniAI
 
       # @raise [OmniAI::Error]
       #
-      # @param messages [String, Array, Hash]
+      # @param messages [String] optional
       # @param model [String] optional
       # @param format [Symbol] optional :text or :json
       # @param temperature [Float, nil] optional
@@ -56,7 +56,7 @@ module OmniAI
       # @yieldparam prompt [OmniAI::Chat::Prompt]
       #
       # @return [OmniAI::Chat::Completion]
-      def chat(messages, model: Chat::Model::GEMINI_PRO, temperature: nil, format: nil, stream: nil, tools: nil, &)
+      def chat(messages = nil, model: Chat::DEFAULT_MODEL, temperature: nil, format: nil, stream: nil, tools: nil, &)
         Chat.process!(messages, model:, temperature:, format:, stream:, tools:, client: self, &)
       end
     end

data/lib/omniai/google/version.rb

@@ -2,6 +2,6 @@
 
 module OmniAI
   module Google
-    VERSION = '1.6.1'
+    VERSION = '1.6.3'
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: omniai-google
 version: !ruby/object:Gem::Version
-  version: 1.6.1
+  version: 1.6.3
 platform: ruby
 authors:
 - Kevin Sylvestre