omniai-google 1.8.1 → 1.8.3
Sign up to get free protection for your applications and to get access to all the features.
checksums.yaml
CHANGED
@@ -1,7 +1,7 @@
|
|
1
1
|
---
|
2
2
|
SHA256:
|
3
|
-
metadata.gz:
|
4
|
-
data.tar.gz:
|
3
|
+
metadata.gz: '05953e1154c7f24ee6d502ea7cb45661c64c0633ccf42ca507a3574374912a71'
|
4
|
+
data.tar.gz: 6e01a56c6c1f41bf2129a9aefe053aae33bb3ea6d63beb6b05fb5922ee7b9baf
|
5
5
|
SHA512:
|
6
|
-
metadata.gz:
|
7
|
-
data.tar.gz:
|
6
|
+
metadata.gz: 3c1cf6d1e36143ad8cfa154b03cace02a4f02e7514778512b3310133a9716ca3291dd7c20b924c475a3006786bfb1bf4830f7d455787c10b18a99ffc66def9e8
|
7
|
+
data.tar.gz: 196f1458b212e5809fdb10c6b2a5a1c7ca7dfa443ef9eed384ef148afad43d56591486e1cb838f35455cd00ecbc0f847a0f0b8ce328848bbb0d5fc8720700d0b
|
@@ -14,7 +14,7 @@ module OmniAI
|
|
14
14
|
|
15
15
|
{
|
16
16
|
candidates:,
|
17
|
-
usage_metadata
|
17
|
+
usageMetadata: usage_metadata,
|
18
18
|
}
|
19
19
|
end
|
20
20
|
|
@@ -23,7 +23,7 @@ module OmniAI
|
|
23
23
|
# @return [OmniAI::Chat::Payload]
|
24
24
|
def self.deserialize(data, context:)
|
25
25
|
choices = data['candidates'].map { |candidate| OmniAI::Chat::Choice.deserialize(candidate, context:) }
|
26
|
-
usage = OmniAI::Chat::Usage.deserialize(data['
|
26
|
+
usage = OmniAI::Chat::Usage.deserialize(data['usageMetadata'], context:) if data['usageMetadata']
|
27
27
|
|
28
28
|
OmniAI::Chat::Payload.new(choices:, usage:)
|
29
29
|
end
|
@@ -9,18 +9,18 @@ module OmniAI
|
|
9
9
|
# @return [Hash]
|
10
10
|
def self.serialize(usage, *)
|
11
11
|
{
|
12
|
-
|
13
|
-
|
14
|
-
|
12
|
+
promptTokenCount: usage.input_tokens,
|
13
|
+
candidatesTokenCount: usage.output_tokens,
|
14
|
+
totalTokenCount: usage.total_tokens,
|
15
15
|
}
|
16
16
|
end
|
17
17
|
|
18
18
|
# @param data [Hash]
|
19
19
|
# @return [OmniAI::Chat::Usage]
|
20
20
|
def self.deserialize(data, *)
|
21
|
-
input_tokens = data['
|
22
|
-
output_tokens = data['
|
23
|
-
total_tokens = data['
|
21
|
+
input_tokens = data['promptTokenCount']
|
22
|
+
output_tokens = data['candidatesTokenCount']
|
23
|
+
total_tokens = data['totalTokenCount']
|
24
24
|
OmniAI::Chat::Usage.new(input_tokens:, output_tokens:, total_tokens:)
|
25
25
|
end
|
26
26
|
end
|
data/lib/omniai/google/chat.rb
CHANGED
data/lib/omniai/google/config.rb
CHANGED
metadata
CHANGED
@@ -1,14 +1,14 @@
|
|
1
1
|
--- !ruby/object:Gem::Specification
|
2
2
|
name: omniai-google
|
3
3
|
version: !ruby/object:Gem::Version
|
4
|
-
version: 1.8.
|
4
|
+
version: 1.8.3
|
5
5
|
platform: ruby
|
6
6
|
authors:
|
7
7
|
- Kevin Sylvestre
|
8
8
|
autorequire:
|
9
9
|
bindir: exe
|
10
10
|
cert_chain: []
|
11
|
-
date: 2024-
|
11
|
+
date: 2024-10-22 00:00:00.000000000 Z
|
12
12
|
dependencies:
|
13
13
|
- !ruby/object:Gem::Dependency
|
14
14
|
name: event_stream_parser
|