omniai-google 1.6.1 → 1.6.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 76a81bb43234ee957a4eeaad25a4aada75d776d2f2a510c65fbabed5ecb41954
-  data.tar.gz: 21dafc2e2b81cfa2dce6e74ce8a093812a0299cde5837ac8425c5ef46020aa45
+  metadata.gz: dbef0413582b7fa949e9458e1cde8d18d2a41aeb65120c5eeebf5335c90c708b
+  data.tar.gz: e5b25bb1c028defefb39ae7c97973df08d63a9391168025a379604e8321cb418
 SHA512:
-  metadata.gz: 3c3595de2ad04e982aa0f7ca062db4ccf9f7bf500ba05edaad988e9b3379c5183a585ab877eba3c150794584007a9caae9ed279300ace51f71a7f223fe32ec5c
-  data.tar.gz: 2b4046f363a0f8ccd4817f6266e92db2a86e43b93c9622710dab614e7413a02ec4f8c29eeefd905f1cac77fd2e48f2368fcd8389a7a2daa0aaf77490a2688047
+  metadata.gz: a3ddf7b5f4a8c4ea865978ebf60eca20848b15bea2660c2bf136634ef9578ae6887f7c6a4b2a471ab526403e5d3ba94f16298f980360edd40671008bbd555ea2
+  data.tar.gz: 7d49c733cb51aa17ed3dd7ccca7702aaeacd818623c2a227e4363ce675ed6aa42ae6cde3a1754011273d9ab31db41b72e1c2c98555740a00bc75cb29bfb9652b

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

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.2'
   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.2
 platform: ruby
 authors:
 - Kevin Sylvestre