omniai-google 1.6.0 → 1.6.1
Sign up to get free protection for your applications and to get access to all the features.
- checksums.yaml +4 -4
- data/lib/omniai/google/client.rb +5 -2
- data/lib/omniai/google/version.rb +1 -1
- metadata +1 -1
checksums.yaml
CHANGED
@@ -1,7 +1,7 @@
|
|
1
1
|
---
|
2
2
|
SHA256:
|
3
|
-
metadata.gz:
|
4
|
-
data.tar.gz:
|
3
|
+
metadata.gz: 76a81bb43234ee957a4eeaad25a4aada75d776d2f2a510c65fbabed5ecb41954
|
4
|
+
data.tar.gz: 21dafc2e2b81cfa2dce6e74ce8a093812a0299cde5837ac8425c5ef46020aa45
|
5
5
|
SHA512:
|
6
|
-
metadata.gz:
|
7
|
-
data.tar.gz:
|
6
|
+
metadata.gz: 3c3595de2ad04e982aa0f7ca062db4ccf9f7bf500ba05edaad988e9b3379c5183a585ab877eba3c150794584007a9caae9ed279300ace51f71a7f223fe32ec5c
|
7
|
+
data.tar.gz: 2b4046f363a0f8ccd4817f6266e92db2a86e43b93c9622710dab614e7413a02ec4f8c29eeefd905f1cac77fd2e48f2368fcd8389a7a2daa0aaf77490a2688047
|
data/lib/omniai/google/client.rb
CHANGED
@@ -52,9 +52,12 @@ module OmniAI
|
|
52
52
|
# @param stream [Proc, nil] optional
|
53
53
|
# @param tools [Array<OmniAI::Chat::Tool>, nil] optional
|
54
54
|
#
|
55
|
+
# @yield [prompt] optional
|
56
|
+
# @yieldparam prompt [OmniAI::Chat::Prompt]
|
57
|
+
#
|
55
58
|
# @return [OmniAI::Chat::Completion]
|
56
|
-
def chat(messages, model: Chat::Model::GEMINI_PRO, temperature: nil, format: nil, stream: nil, tools: nil)
|
57
|
-
Chat.process!(messages, model:, temperature:, format:, stream:, tools:, client: self)
|
59
|
+
def chat(messages, model: Chat::Model::GEMINI_PRO, temperature: nil, format: nil, stream: nil, tools: nil, &)
|
60
|
+
Chat.process!(messages, model:, temperature:, format:, stream:, tools:, client: self, &)
|
58
61
|
end
|
59
62
|
end
|
60
63
|
end
|