omniai-google 1.6.0 → 1.6.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3a64561af96984da60b42099f415cdb90d738757f6108af74b035e1a2e5d34ce
-  data.tar.gz: 837b82e0ec5e45b3300858591f1a60ffd2783d16d0c5b1ad8280dd9585d944de
+  metadata.gz: dbef0413582b7fa949e9458e1cde8d18d2a41aeb65120c5eeebf5335c90c708b
+  data.tar.gz: e5b25bb1c028defefb39ae7c97973df08d63a9391168025a379604e8321cb418
 SHA512:
-  metadata.gz: abb56f1dc0d7ddb3efcd79ffcd132c45c85e3ca5fc43f72bfc8fd94b535f0bf76bb32946e1dca7e55cb180bc8d7492ecc116f16cb7fd8b454825f38395c92f22
-  data.tar.gz: 9baa74ce37c0a0c7a20b9f36e15d205a89df9556042d17e999f622c86e505f1f5ace1b143091f346120045102e62a1b4bf34d522d4c90b179aec8773f34b8c78
+  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,16 +45,19 @@ 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
       # @param stream [Proc, nil] optional
       # @param tools [Array<OmniAI::Chat::Tool>, nil] optional
       #
+      # @yield [prompt] optional
+      # @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)
-        Chat.process!(messages, model:, temperature:, format:, stream:, tools:, client: self)
+      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
   end

data/lib/omniai/google/version.rb

@@ -2,6 +2,6 @@
 
 module OmniAI
   module Google
-    VERSION = '1.6.0'
+    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.0
+  version: 1.6.2
 platform: ruby
 authors:
 - Kevin Sylvestre