omniai-google 1.8.3 → 1.9.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
checksums.yaml CHANGED
@@ -1,7 +1,7 @@
1
1
  ---
2
2
  SHA256:
3
- metadata.gz: '05953e1154c7f24ee6d502ea7cb45661c64c0633ccf42ca507a3574374912a71'
4
- data.tar.gz: 6e01a56c6c1f41bf2129a9aefe053aae33bb3ea6d63beb6b05fb5922ee7b9baf
3
+ metadata.gz: 18c29d5c7580f38ae07eb596186fea540ccd1e179dccd906b3fb1473c187546c
4
+ data.tar.gz: 187dc0cffd36841469b5fd998216f3f3fa2a56b1cce7bd7d51c9ececa12784e9
5
5
  SHA512:
6
- metadata.gz: 3c1cf6d1e36143ad8cfa154b03cace02a4f02e7514778512b3310133a9716ca3291dd7c20b924c475a3006786bfb1bf4830f7d455787c10b18a99ffc66def9e8
7
- data.tar.gz: 196f1458b212e5809fdb10c6b2a5a1c7ca7dfa443ef9eed384ef148afad43d56591486e1cb838f35455cd00ecbc0f847a0f0b8ce328848bbb0d5fc8720700d0b
6
+ metadata.gz: 0ddd16af090b9692ff68f7d816e7d153759dd65793d7b816ae40e5ce195c0341f4c6959df4a1bfc09a9452df07bd6fe9633ccc7b46a89e2513c3a417c9f12ad0
7
+ data.tar.gz: 48c1d3a78de86387e9a95716ece83d27cb4c128bacc43285ed3830d210f2769516bfd69e9212f66caaf0e38dfe77ddcabd67d0005b4d21eeed41094de60e1590
@@ -25,6 +25,8 @@ module OmniAI
25
25
 
26
26
  DEFAULT_MODEL = Model::GEMINI_PRO
27
27
 
28
+ JSON_MIME_TYPE = 'application/json'
29
+
28
30
  # @return [Context]
29
31
  CONTEXT = Context.build do |context|
30
32
  context.serializers[:text] = TextSerializer.method(:serialize)
@@ -98,9 +100,20 @@ module OmniAI
98
100
 
99
101
  # @return [Hash]
100
102
  def generation_config
101
- return unless @temperature
103
+ response_mime_type = (JSON_MIME_TYPE if json_mime_type?)
104
+
105
+ return unless @temperature || response_mime_type
106
+
107
+ {
108
+ temperature: @temperature,
109
+ responseMimeType: response_mime_type,
110
+ }.compact
111
+ end
102
112
 
103
- { temperature: @temperature }.compact
113
+ # Checks if setting a jsonMimeType is supported
114
+ # @return [Boolean]
115
+ def json_mime_type?
116
+ @client.version == OmniAI::Google::Config::Version::BETA && @format.eql?(:json)
104
117
  end
105
118
 
106
119
  # @return [String]
@@ -2,6 +2,6 @@
2
2
 
3
3
  module OmniAI
4
4
  module Google
5
- VERSION = '1.8.3'
5
+ VERSION = '1.9.0'
6
6
  end
7
7
  end
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.3
4
+ version: 1.9.0
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-10-22 00:00:00.000000000 Z
11
+ date: 2024-10-30 00:00:00.000000000 Z
12
12
  dependencies:
13
13
  - !ruby/object:Gem::Dependency
14
14
  name: event_stream_parser
@@ -92,7 +92,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
92
92
  requirements:
93
93
  - - ">="
94
94
  - !ruby/object:Gem::Version
95
- version: 3.3.0
95
+ version: 3.2.0
96
96
  required_rubygems_version: !ruby/object:Gem::Requirement
97
97
  requirements:
98
98
  - - ">="