omniai-google 2.3.4 → 2.6.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: ce7a07b793b56721b69b9094b640ffa834cf89e4c1eeae7845b3d86d9d41fbac
4
- data.tar.gz: 1915096495222231feffd39887ffe253b54348d9339763bc54f6f54beebdbb9a
3
+ metadata.gz: bb7d9c573b2db16b5cf0ec251d1d35920d40a8e2ba1a717d50c4001360f05cf0
4
+ data.tar.gz: 33d8b5af813b07fc0eb1dc8f0bc56b6c1a2033caf9a1a97da3e33ad9b0a14854
5
5
  SHA512:
6
- metadata.gz: e39f1dcd34cb6b7a19d69d3d940e5a8e3737917bfcd80e833f2753b2d6ebc6cbd5b1431199c739beb931937acfee1139ef8e13140e3e0520cd49b72ed703ae22
7
- data.tar.gz: 4e87328e756c6c7c7fc61e0bcb52ec466ea550ff5fdde3e93c467f2c5a9fc316227e6fc9d06ea7e91e0ba921221f586690831468b2b83359156b7c9b71f69559
6
+ metadata.gz: 6c7ffd48b86af5d634d24795126bbc5ea98284c515f0c5d9de13f48df8ffef98367589cc4c11d062c21f68682fb5ff36b8a8942a82a32f9a515321dc5cb15fc6
7
+ data.tar.gz: a26d1c8f7348fd97fe4a21234a67387ab11ce5b39c96ba191c6c4abd3d6f61d8ead3f6f16f322b615949f54282ef578baa2e80d072a8c12f7a112b31b31694de
@@ -23,9 +23,13 @@ module OmniAI
23
23
  GEMINI_FLASH = GEMINI_2_5_FLASH
24
24
  end
25
25
 
26
- DEFAULT_MODEL = Model::GEMINI_PRO
26
+ DEFAULT_MODEL = Model::GEMINI_2_5_FLASH
27
27
 
28
- JSON_MIME_TYPE = "application/json"
28
+ module ResponseMimeType
29
+ JSON = "application/json"
30
+ TEXT = "text/plain"
31
+ SCHEMA = JSON
32
+ end
29
33
 
30
34
  # @return [Context]
31
35
  CONTEXT = Context.build do |context|
@@ -100,14 +104,22 @@ module OmniAI
100
104
 
101
105
  # @return [Hash]
102
106
  def generation_config
103
- response_mime_type = (JSON_MIME_TYPE if @format.eql?(:json))
104
-
105
- return unless @temperature || response_mime_type
106
-
107
- {
108
- temperature: @temperature,
109
- responseMimeType: response_mime_type,
110
- }.compact
107
+ data =
108
+ case @format
109
+ when :json then { responseMimeType: ResponseMimeType::JSON }
110
+ when :text then { responseMimeType: ResponseMimeType::TEXT }
111
+ when OmniAI::Schema::Format
112
+ {
113
+ responseMimeType: ResponseMimeType::SCHEMA,
114
+ responseSchema: @format.schema.serialize,
115
+ }
116
+ else {}
117
+ end
118
+
119
+ data[:temperature] = @temperature if @temperature
120
+
121
+ data = data.compact
122
+ data unless data.empty?
111
123
  end
112
124
 
113
125
  # @return [String]
@@ -2,6 +2,6 @@
2
2
 
3
3
  module OmniAI
4
4
  module Google
5
- VERSION = "2.3.4"
5
+ VERSION = "2.6.0"
6
6
  end
7
7
  end
metadata CHANGED
@@ -1,13 +1,13 @@
1
1
  --- !ruby/object:Gem::Specification
2
2
  name: omniai-google
3
3
  version: !ruby/object:Gem::Version
4
- version: 2.3.4
4
+ version: 2.6.0
5
5
  platform: ruby
6
6
  authors:
7
7
  - Kevin Sylvestre
8
8
  bindir: exe
9
9
  cert_chain: []
10
- date: 2025-05-05 00:00:00.000000000 Z
10
+ date: 2025-05-08 00:00:00.000000000 Z
11
11
  dependencies:
12
12
  - !ruby/object:Gem::Dependency
13
13
  name: event_stream_parser
@@ -43,14 +43,14 @@ dependencies:
43
43
  requirements:
44
44
  - - "~>"
45
45
  - !ruby/object:Gem::Version
46
- version: '2.3'
46
+ version: '2.6'
47
47
  type: :runtime
48
48
  prerelease: false
49
49
  version_requirements: !ruby/object:Gem::Requirement
50
50
  requirements:
51
51
  - - "~>"
52
52
  - !ruby/object:Gem::Version
53
- version: '2.3'
53
+ version: '2.6'
54
54
  - !ruby/object:Gem::Dependency
55
55
  name: zeitwerk
56
56
  requirement: !ruby/object:Gem::Requirement