aws-sdk-bedrockagentruntime 1.3.0 → 1.5.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.
Files changed (10) hide show
  1. checksums.yaml +4 -4
  2. data/CHANGELOG.md +10 -0
  3. data/VERSION +1 -1
  4. data/lib/aws-sdk-bedrockagentruntime/client.rb +144 -17
  5. data/lib/aws-sdk-bedrockagentruntime/client_api.rb +56 -0
  6. data/lib/aws-sdk-bedrockagentruntime/types.rb +477 -25
  7. data/lib/aws-sdk-bedrockagentruntime.rb +1 -1
  8. data/sig/client.rbs +111 -0
  9. data/sig/types.rbs +64 -3
  10. metadata +2 -2
data/lib/aws-sdk-bedrockagentruntime.rb CHANGED
@@ -53,6 +53,6 @@ require_relative 'aws-sdk-bedrockagentruntime/event_streams'
53
53
  # @!group service
54
54
  module Aws::BedrockAgentRuntime
55
55
 
56
- GEM_VERSION = '1.3.0'
56
+ GEM_VERSION = '1.5.0'
57
57
 
58
58
  end
data/sig/client.rbs CHANGED
@@ -107,6 +107,59 @@ module Aws
107
107
  ?next_token: ::String,
108
108
  ?retrieval_configuration: {
109
109
  vector_search_configuration: {
110
+ filter: {
111
+ and_all: Array[
112
+ untyped,
113
+ ]?,
114
+ equals: {
115
+ key: ::String,
116
+ value: {
117
+ }
118
+ }?,
119
+ greater_than: {
120
+ key: ::String,
121
+ value: {
122
+ }
123
+ }?,
124
+ greater_than_or_equals: {
125
+ key: ::String,
126
+ value: {
127
+ }
128
+ }?,
129
+ in: {
130
+ key: ::String,
131
+ value: {
132
+ }
133
+ }?,
134
+ less_than: {
135
+ key: ::String,
136
+ value: {
137
+ }
138
+ }?,
139
+ less_than_or_equals: {
140
+ key: ::String,
141
+ value: {
142
+ }
143
+ }?,
144
+ not_equals: {
145
+ key: ::String,
146
+ value: {
147
+ }
148
+ }?,
149
+ not_in: {
150
+ key: ::String,
151
+ value: {
152
+ }
153
+ }?,
154
+ or_all: Array[
155
+ untyped,
156
+ ]?,
157
+ starts_with: {
158
+ key: ::String,
159
+ value: {
160
+ }
161
+ }?
162
+ }?,
110
163
  number_of_results: ::Integer?,
111
164
  override_search_type: ("HYBRID" | "SEMANTIC")?
112
165
  }
@@ -130,10 +183,68 @@ module Aws
130
183
  },
131
184
  ?retrieve_and_generate_configuration: {
132
185
  knowledge_base_configuration: {
186
+ generation_configuration: {
187
+ prompt_template: {
188
+ text_prompt_template: ::String?
189
+ }?
190
+ }?,
133
191
  knowledge_base_id: ::String,
134
192
  model_arn: ::String,
135
193
  retrieval_configuration: {
136
194
  vector_search_configuration: {
195
+ filter: {
196
+ and_all: Array[
197
+ untyped,
198
+ ]?,
199
+ equals: {
200
+ key: ::String,
201
+ value: {
202
+ }
203
+ }?,
204
+ greater_than: {
205
+ key: ::String,
206
+ value: {
207
+ }
208
+ }?,
209
+ greater_than_or_equals: {
210
+ key: ::String,
211
+ value: {
212
+ }
213
+ }?,
214
+ in: {
215
+ key: ::String,
216
+ value: {
217
+ }
218
+ }?,
219
+ less_than: {
220
+ key: ::String,
221
+ value: {
222
+ }
223
+ }?,
224
+ less_than_or_equals: {
225
+ key: ::String,
226
+ value: {
227
+ }
228
+ }?,
229
+ not_equals: {
230
+ key: ::String,
231
+ value: {
232
+ }
233
+ }?,
234
+ not_in: {
235
+ key: ::String,
236
+ value: {
237
+ }
238
+ }?,
239
+ or_all: Array[
240
+ untyped,
241
+ ]?,
242
+ starts_with: {
243
+ key: ::String,
244
+ value: {
245
+ }
246
+ }?
247
+ }?,
137
248
  number_of_results: ::Integer?,
138
249
  override_search_type: ("HYBRID" | "SEMANTIC")?
139
250
  }
data/sig/types.rbs CHANGED
@@ -65,6 +65,12 @@ module Aws::BedrockAgentRuntime
65
65
  SENSITIVE: [:failure_reason]
66
66
  end
67
67
 
68
+ class FilterAttribute
69
+ attr_accessor key: ::String
70
+ attr_accessor value: untyped
71
+ SENSITIVE: []
72
+ end
73
+
68
74
  class FinalResponse
69
75
  attr_accessor text: ::String
70
76
  SENSITIVE: [:text]
@@ -75,6 +81,11 @@ module Aws::BedrockAgentRuntime
75
81
  SENSITIVE: [:text_response_part]
76
82
  end
77
83
 
84
+ class GenerationConfiguration
85
+ attr_accessor prompt_template: Types::PromptTemplate
86
+ SENSITIVE: []
87
+ end
88
+
78
89
  class InferenceConfiguration
79
90
  attr_accessor maximum_length: ::Integer
80
91
  attr_accessor stop_sequences: ::Array[::String]
@@ -140,11 +151,13 @@ module Aws::BedrockAgentRuntime
140
151
  class KnowledgeBaseRetrievalResult
141
152
  attr_accessor content: Types::RetrievalResultContent
142
153
  attr_accessor location: Types::RetrievalResultLocation
154
+ attr_accessor metadata: ::Hash[::String, untyped]
143
155
  attr_accessor score: ::Float
144
- SENSITIVE: [:content, :location]
156
+ SENSITIVE: [:content, :location, :metadata]
145
157
  end
146
158
 
147
159
  class KnowledgeBaseRetrieveAndGenerateConfiguration
160
+ attr_accessor generation_configuration: Types::GenerationConfiguration
148
161
  attr_accessor knowledge_base_id: ::String
149
162
  attr_accessor model_arn: ::String
150
163
  attr_accessor retrieval_configuration: Types::KnowledgeBaseRetrievalConfiguration
@@ -152,9 +165,10 @@ module Aws::BedrockAgentRuntime
152
165
  end
153
166
 
154
167
  class KnowledgeBaseVectorSearchConfiguration
168
+ attr_accessor filter: Types::RetrievalFilter
155
169
  attr_accessor number_of_results: ::Integer
156
170
  attr_accessor override_search_type: ("HYBRID" | "SEMANTIC")
157
- SENSITIVE: []
171
+ SENSITIVE: [:filter]
158
172
  end
159
173
 
160
174
  class ModelInvocationInput
@@ -263,6 +277,11 @@ module Aws::BedrockAgentRuntime
263
277
  end
264
278
  end
265
279
 
280
+ class PromptTemplate
281
+ attr_accessor text_prompt_template: ::String
282
+ SENSITIVE: [:text_prompt_template]
283
+ end
284
+
266
285
  class Rationale
267
286
  attr_accessor text: ::String
268
287
  attr_accessor trace_id: ::String
@@ -286,6 +305,47 @@ module Aws::BedrockAgentRuntime
286
305
  SENSITIVE: []
287
306
  end
288
307
 
308
+ class RetrievalFilter
309
+ attr_accessor and_all: ::Array[Types::RetrievalFilter]
310
+ attr_accessor equals: Types::FilterAttribute
311
+ attr_accessor greater_than: Types::FilterAttribute
312
+ attr_accessor greater_than_or_equals: Types::FilterAttribute
313
+ attr_accessor in: Types::FilterAttribute
314
+ attr_accessor less_than: Types::FilterAttribute
315
+ attr_accessor less_than_or_equals: Types::FilterAttribute
316
+ attr_accessor not_equals: Types::FilterAttribute
317
+ attr_accessor not_in: Types::FilterAttribute
318
+ attr_accessor or_all: ::Array[Types::RetrievalFilter]
319
+ attr_accessor starts_with: Types::FilterAttribute
320
+ attr_accessor unknown: untyped
321
+ SENSITIVE: []
322
+
323
+ class AndAll < RetrievalFilter
324
+ end
325
+ class Equals < RetrievalFilter
326
+ end
327
+ class GreaterThan < RetrievalFilter
328
+ end
329
+ class GreaterThanOrEquals < RetrievalFilter
330
+ end
331
+ class In < RetrievalFilter
332
+ end
333
+ class LessThan < RetrievalFilter
334
+ end
335
+ class LessThanOrEquals < RetrievalFilter
336
+ end
337
+ class NotEquals < RetrievalFilter
338
+ end
339
+ class NotIn < RetrievalFilter
340
+ end
341
+ class OrAll < RetrievalFilter
342
+ end
343
+ class StartsWith < RetrievalFilter
344
+ end
345
+ class Unknown < RetrievalFilter
346
+ end
347
+ end
348
+
289
349
  class RetrievalResultContent
290
350
  attr_accessor text: ::String
291
351
  SENSITIVE: []
@@ -355,7 +415,8 @@ module Aws::BedrockAgentRuntime
355
415
  class RetrievedReference
356
416
  attr_accessor content: Types::RetrievalResultContent
357
417
  attr_accessor location: Types::RetrievalResultLocation
358
- SENSITIVE: [:content, :location]
418
+ attr_accessor metadata: ::Hash[::String, untyped]
419
+ SENSITIVE: [:content, :location, :metadata]
359
420
  end
360
421
 
361
422
  class ServiceQuotaExceededException
metadata CHANGED
@@ -1,14 +1,14 @@
1
1
  --- !ruby/object:Gem::Specification
2
2
  name: aws-sdk-bedrockagentruntime
3
3
  version: !ruby/object:Gem::Version
4
- version: 1.3.0
4
+ version: 1.5.0
5
5
  platform: ruby
6
6
  authors:
7
7
  - Amazon Web Services
8
8
  autorequire:
9
9
  bindir: bin
10
10
  cert_chain: []
11
- date: 2024-03-08 00:00:00.000000000 Z
11
+ date: 2024-03-27 00:00:00.000000000 Z
12
12
  dependencies:
13
13
  - !ruby/object:Gem::Dependency
14
14
  name: aws-sdk-core