aws-sdk-bedrockagentruntime 1.2.0 → 1.4.0
Sign up to get free protection for your applications and to get access to all the features.
- checksums.yaml +4 -4
- data/CHANGELOG.md +10 -0
- data/VERSION +1 -1
- data/lib/aws-sdk-bedrockagentruntime/client.rb +240 -179
- data/lib/aws-sdk-bedrockagentruntime/client_api.rb +61 -51
- data/lib/aws-sdk-bedrockagentruntime/event_streams.rb +24 -24
- data/lib/aws-sdk-bedrockagentruntime/types.rb +859 -374
- data/lib/aws-sdk-bedrockagentruntime.rb +1 -1
- data/sig/client.rbs +24 -19
- data/sig/types.rbs +62 -51
- metadata +2 -2
@@ -37,6 +37,7 @@ module Aws::BedrockAgentRuntime
|
|
37
37
|
FinalResponse = Shapes::StructureShape.new(name: 'FinalResponse')
|
38
38
|
FinalResponseString = Shapes::StringShape.new(name: 'FinalResponseString')
|
39
39
|
GeneratedResponsePart = Shapes::StructureShape.new(name: 'GeneratedResponsePart')
|
40
|
+
GenerationConfiguration = Shapes::StructureShape.new(name: 'GenerationConfiguration')
|
40
41
|
InferenceConfiguration = Shapes::StructureShape.new(name: 'InferenceConfiguration')
|
41
42
|
InputText = Shapes::StringShape.new(name: 'InputText')
|
42
43
|
InternalServerException = Shapes::StructureShape.new(name: 'InternalServerException')
|
@@ -77,6 +78,7 @@ module Aws::BedrockAgentRuntime
|
|
77
78
|
PreProcessingParsedResponse = Shapes::StructureShape.new(name: 'PreProcessingParsedResponse')
|
78
79
|
PreProcessingTrace = Shapes::UnionShape.new(name: 'PreProcessingTrace')
|
79
80
|
PromptSessionAttributesMap = Shapes::MapShape.new(name: 'PromptSessionAttributesMap')
|
81
|
+
PromptTemplate = Shapes::StructureShape.new(name: 'PromptTemplate')
|
80
82
|
PromptText = Shapes::StringShape.new(name: 'PromptText')
|
81
83
|
PromptType = Shapes::StringShape.new(name: 'PromptType')
|
82
84
|
Rationale = Shapes::StructureShape.new(name: 'Rationale')
|
@@ -113,6 +115,7 @@ module Aws::BedrockAgentRuntime
|
|
113
115
|
StopSequences = Shapes::ListShape.new(name: 'StopSequences')
|
114
116
|
String = Shapes::StringShape.new(name: 'String')
|
115
117
|
Temperature = Shapes::FloatShape.new(name: 'Temperature')
|
118
|
+
TextPromptTemplate = Shapes::StringShape.new(name: 'TextPromptTemplate')
|
116
119
|
TextResponsePart = Shapes::StructureShape.new(name: 'TextResponsePart')
|
117
120
|
ThrottlingException = Shapes::StructureShape.new(name: 'ThrottlingException')
|
118
121
|
TopK = Shapes::IntegerShape.new(name: 'TopK')
|
@@ -129,10 +132,10 @@ module Aws::BedrockAgentRuntime
|
|
129
132
|
AccessDeniedException.struct_class = Types::AccessDeniedException
|
130
133
|
|
131
134
|
ActionGroupInvocationInput.add_member(:action_group_name, Shapes::ShapeRef.new(shape: ActionGroupName, location_name: "actionGroupName"))
|
132
|
-
ActionGroupInvocationInput.add_member(:verb, Shapes::ShapeRef.new(shape: Verb, location_name: "verb"))
|
133
135
|
ActionGroupInvocationInput.add_member(:api_path, Shapes::ShapeRef.new(shape: ApiPath, location_name: "apiPath"))
|
134
136
|
ActionGroupInvocationInput.add_member(:parameters, Shapes::ShapeRef.new(shape: Parameters, location_name: "parameters"))
|
135
137
|
ActionGroupInvocationInput.add_member(:request_body, Shapes::ShapeRef.new(shape: RequestBody, location_name: "requestBody"))
|
138
|
+
ActionGroupInvocationInput.add_member(:verb, Shapes::ShapeRef.new(shape: Verb, location_name: "verb"))
|
136
139
|
ActionGroupInvocationInput.struct_class = Types::ActionGroupInvocationInput
|
137
140
|
|
138
141
|
ActionGroupInvocationOutput.add_member(:text, Shapes::ShapeRef.new(shape: ActionGroupOutputString, location_name: "text"))
|
@@ -161,8 +164,8 @@ module Aws::BedrockAgentRuntime
|
|
161
164
|
DependencyFailedException.add_member(:resource_name, Shapes::ShapeRef.new(shape: NonBlankString, location_name: "resourceName"))
|
162
165
|
DependencyFailedException.struct_class = Types::DependencyFailedException
|
163
166
|
|
164
|
-
FailureTrace.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))
|
165
167
|
FailureTrace.add_member(:failure_reason, Shapes::ShapeRef.new(shape: FailureReasonString, location_name: "failureReason"))
|
168
|
+
FailureTrace.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))
|
166
169
|
FailureTrace.struct_class = Types::FailureTrace
|
167
170
|
|
168
171
|
FinalResponse.add_member(:text, Shapes::ShapeRef.new(shape: FinalResponseString, location_name: "text"))
|
@@ -171,29 +174,32 @@ module Aws::BedrockAgentRuntime
|
|
171
174
|
GeneratedResponsePart.add_member(:text_response_part, Shapes::ShapeRef.new(shape: TextResponsePart, location_name: "textResponsePart"))
|
172
175
|
GeneratedResponsePart.struct_class = Types::GeneratedResponsePart
|
173
176
|
|
174
|
-
|
175
|
-
|
176
|
-
|
177
|
+
GenerationConfiguration.add_member(:prompt_template, Shapes::ShapeRef.new(shape: PromptTemplate, location_name: "promptTemplate"))
|
178
|
+
GenerationConfiguration.struct_class = Types::GenerationConfiguration
|
179
|
+
|
177
180
|
InferenceConfiguration.add_member(:maximum_length, Shapes::ShapeRef.new(shape: MaximumLength, location_name: "maximumLength"))
|
178
181
|
InferenceConfiguration.add_member(:stop_sequences, Shapes::ShapeRef.new(shape: StopSequences, location_name: "stopSequences"))
|
182
|
+
InferenceConfiguration.add_member(:temperature, Shapes::ShapeRef.new(shape: Temperature, location_name: "temperature"))
|
183
|
+
InferenceConfiguration.add_member(:top_k, Shapes::ShapeRef.new(shape: TopK, location_name: "topK"))
|
184
|
+
InferenceConfiguration.add_member(:top_p, Shapes::ShapeRef.new(shape: TopP, location_name: "topP"))
|
179
185
|
InferenceConfiguration.struct_class = Types::InferenceConfiguration
|
180
186
|
|
181
187
|
InternalServerException.add_member(:message, Shapes::ShapeRef.new(shape: NonBlankString, location_name: "message"))
|
182
188
|
InternalServerException.struct_class = Types::InternalServerException
|
183
189
|
|
184
|
-
InvocationInput.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))
|
185
|
-
InvocationInput.add_member(:invocation_type, Shapes::ShapeRef.new(shape: InvocationType, location_name: "invocationType"))
|
186
190
|
InvocationInput.add_member(:action_group_invocation_input, Shapes::ShapeRef.new(shape: ActionGroupInvocationInput, location_name: "actionGroupInvocationInput"))
|
191
|
+
InvocationInput.add_member(:invocation_type, Shapes::ShapeRef.new(shape: InvocationType, location_name: "invocationType"))
|
187
192
|
InvocationInput.add_member(:knowledge_base_lookup_input, Shapes::ShapeRef.new(shape: KnowledgeBaseLookupInput, location_name: "knowledgeBaseLookupInput"))
|
193
|
+
InvocationInput.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))
|
188
194
|
InvocationInput.struct_class = Types::InvocationInput
|
189
195
|
|
190
|
-
InvokeAgentRequest.add_member(:session_state, Shapes::ShapeRef.new(shape: SessionState, location_name: "sessionState"))
|
191
|
-
InvokeAgentRequest.add_member(:agent_id, Shapes::ShapeRef.new(shape: AgentId, required: true, location: "uri", location_name: "agentId"))
|
192
196
|
InvokeAgentRequest.add_member(:agent_alias_id, Shapes::ShapeRef.new(shape: AgentAliasId, required: true, location: "uri", location_name: "agentAliasId"))
|
193
|
-
InvokeAgentRequest.add_member(:
|
194
|
-
InvokeAgentRequest.add_member(:end_session, Shapes::ShapeRef.new(shape: Boolean, location_name: "endSession"))
|
197
|
+
InvokeAgentRequest.add_member(:agent_id, Shapes::ShapeRef.new(shape: AgentId, required: true, location: "uri", location_name: "agentId"))
|
195
198
|
InvokeAgentRequest.add_member(:enable_trace, Shapes::ShapeRef.new(shape: Boolean, location_name: "enableTrace"))
|
199
|
+
InvokeAgentRequest.add_member(:end_session, Shapes::ShapeRef.new(shape: Boolean, location_name: "endSession"))
|
196
200
|
InvokeAgentRequest.add_member(:input_text, Shapes::ShapeRef.new(shape: InputText, required: true, location_name: "inputText"))
|
201
|
+
InvokeAgentRequest.add_member(:session_id, Shapes::ShapeRef.new(shape: SessionId, required: true, location: "uri", location_name: "sessionId"))
|
202
|
+
InvokeAgentRequest.add_member(:session_state, Shapes::ShapeRef.new(shape: SessionState, location_name: "sessionState"))
|
197
203
|
InvokeAgentRequest.struct_class = Types::InvokeAgentRequest
|
198
204
|
|
199
205
|
InvokeAgentResponse.add_member(:completion, Shapes::ShapeRef.new(shape: ResponseStream, required: true, eventstream: true, location_name: "completion"))
|
@@ -203,8 +209,8 @@ module Aws::BedrockAgentRuntime
|
|
203
209
|
InvokeAgentResponse[:payload] = :completion
|
204
210
|
InvokeAgentResponse[:payload_member] = InvokeAgentResponse.member(:completion)
|
205
211
|
|
206
|
-
KnowledgeBaseLookupInput.add_member(:text, Shapes::ShapeRef.new(shape: KnowledgeBaseLookupInputString, location_name: "text"))
|
207
212
|
KnowledgeBaseLookupInput.add_member(:knowledge_base_id, Shapes::ShapeRef.new(shape: TraceKnowledgeBaseId, location_name: "knowledgeBaseId"))
|
213
|
+
KnowledgeBaseLookupInput.add_member(:text, Shapes::ShapeRef.new(shape: KnowledgeBaseLookupInputString, location_name: "text"))
|
208
214
|
KnowledgeBaseLookupInput.struct_class = Types::KnowledgeBaseLookupInput
|
209
215
|
|
210
216
|
KnowledgeBaseLookupOutput.add_member(:retrieved_references, Shapes::ShapeRef.new(shape: RetrievedReferences, location_name: "retrievedReferences"))
|
@@ -223,6 +229,7 @@ module Aws::BedrockAgentRuntime
|
|
223
229
|
|
224
230
|
KnowledgeBaseRetrievalResults.member = Shapes::ShapeRef.new(shape: KnowledgeBaseRetrievalResult)
|
225
231
|
|
232
|
+
KnowledgeBaseRetrieveAndGenerateConfiguration.add_member(:generation_configuration, Shapes::ShapeRef.new(shape: GenerationConfiguration, location_name: "generationConfiguration"))
|
226
233
|
KnowledgeBaseRetrieveAndGenerateConfiguration.add_member(:knowledge_base_id, Shapes::ShapeRef.new(shape: KnowledgeBaseId, required: true, location_name: "knowledgeBaseId"))
|
227
234
|
KnowledgeBaseRetrieveAndGenerateConfiguration.add_member(:model_arn, Shapes::ShapeRef.new(shape: BedrockModelArn, required: true, location_name: "modelArn"))
|
228
235
|
KnowledgeBaseRetrieveAndGenerateConfiguration.add_member(:retrieval_configuration, Shapes::ShapeRef.new(shape: KnowledgeBaseRetrievalConfiguration, location_name: "retrievalConfiguration"))
|
@@ -232,32 +239,32 @@ module Aws::BedrockAgentRuntime
|
|
232
239
|
KnowledgeBaseVectorSearchConfiguration.add_member(:override_search_type, Shapes::ShapeRef.new(shape: SearchType, location_name: "overrideSearchType"))
|
233
240
|
KnowledgeBaseVectorSearchConfiguration.struct_class = Types::KnowledgeBaseVectorSearchConfiguration
|
234
241
|
|
235
|
-
ModelInvocationInput.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))
|
236
|
-
ModelInvocationInput.add_member(:text, Shapes::ShapeRef.new(shape: PromptText, location_name: "text"))
|
237
|
-
ModelInvocationInput.add_member(:type, Shapes::ShapeRef.new(shape: PromptType, location_name: "type"))
|
238
242
|
ModelInvocationInput.add_member(:inference_configuration, Shapes::ShapeRef.new(shape: InferenceConfiguration, location_name: "inferenceConfiguration"))
|
239
243
|
ModelInvocationInput.add_member(:override_lambda, Shapes::ShapeRef.new(shape: LambdaArn, location_name: "overrideLambda"))
|
240
|
-
ModelInvocationInput.add_member(:prompt_creation_mode, Shapes::ShapeRef.new(shape: CreationMode, location_name: "promptCreationMode"))
|
241
244
|
ModelInvocationInput.add_member(:parser_mode, Shapes::ShapeRef.new(shape: CreationMode, location_name: "parserMode"))
|
245
|
+
ModelInvocationInput.add_member(:prompt_creation_mode, Shapes::ShapeRef.new(shape: CreationMode, location_name: "promptCreationMode"))
|
246
|
+
ModelInvocationInput.add_member(:text, Shapes::ShapeRef.new(shape: PromptText, location_name: "text"))
|
247
|
+
ModelInvocationInput.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))
|
248
|
+
ModelInvocationInput.add_member(:type, Shapes::ShapeRef.new(shape: PromptType, location_name: "type"))
|
242
249
|
ModelInvocationInput.struct_class = Types::ModelInvocationInput
|
243
250
|
|
244
|
-
Observation.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))
|
245
|
-
Observation.add_member(:type, Shapes::ShapeRef.new(shape: Type, location_name: "type"))
|
246
251
|
Observation.add_member(:action_group_invocation_output, Shapes::ShapeRef.new(shape: ActionGroupInvocationOutput, location_name: "actionGroupInvocationOutput"))
|
247
|
-
Observation.add_member(:knowledge_base_lookup_output, Shapes::ShapeRef.new(shape: KnowledgeBaseLookupOutput, location_name: "knowledgeBaseLookupOutput"))
|
248
252
|
Observation.add_member(:final_response, Shapes::ShapeRef.new(shape: FinalResponse, location_name: "finalResponse"))
|
253
|
+
Observation.add_member(:knowledge_base_lookup_output, Shapes::ShapeRef.new(shape: KnowledgeBaseLookupOutput, location_name: "knowledgeBaseLookupOutput"))
|
249
254
|
Observation.add_member(:reprompt_response, Shapes::ShapeRef.new(shape: RepromptResponse, location_name: "repromptResponse"))
|
255
|
+
Observation.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))
|
256
|
+
Observation.add_member(:type, Shapes::ShapeRef.new(shape: Type, location_name: "type"))
|
250
257
|
Observation.struct_class = Types::Observation
|
251
258
|
|
252
|
-
OrchestrationTrace.add_member(:rationale, Shapes::ShapeRef.new(shape: Rationale, location_name: "rationale"))
|
253
259
|
OrchestrationTrace.add_member(:invocation_input, Shapes::ShapeRef.new(shape: InvocationInput, location_name: "invocationInput"))
|
254
|
-
OrchestrationTrace.add_member(:observation, Shapes::ShapeRef.new(shape: Observation, location_name: "observation"))
|
255
260
|
OrchestrationTrace.add_member(:model_invocation_input, Shapes::ShapeRef.new(shape: ModelInvocationInput, location_name: "modelInvocationInput"))
|
261
|
+
OrchestrationTrace.add_member(:observation, Shapes::ShapeRef.new(shape: Observation, location_name: "observation"))
|
262
|
+
OrchestrationTrace.add_member(:rationale, Shapes::ShapeRef.new(shape: Rationale, location_name: "rationale"))
|
256
263
|
OrchestrationTrace.add_member(:unknown, Shapes::ShapeRef.new(shape: nil, location_name: 'unknown'))
|
257
|
-
OrchestrationTrace.add_member_subclass(:rationale, Types::OrchestrationTrace::Rationale)
|
258
264
|
OrchestrationTrace.add_member_subclass(:invocation_input, Types::OrchestrationTrace::InvocationInput)
|
259
|
-
OrchestrationTrace.add_member_subclass(:observation, Types::OrchestrationTrace::Observation)
|
260
265
|
OrchestrationTrace.add_member_subclass(:model_invocation_input, Types::OrchestrationTrace::ModelInvocationInput)
|
266
|
+
OrchestrationTrace.add_member_subclass(:observation, Types::OrchestrationTrace::Observation)
|
267
|
+
OrchestrationTrace.add_member_subclass(:rationale, Types::OrchestrationTrace::Rationale)
|
261
268
|
OrchestrationTrace.add_member_subclass(:unknown, Types::OrchestrationTrace::Unknown)
|
262
269
|
OrchestrationTrace.struct_class = Types::OrchestrationTrace
|
263
270
|
|
@@ -268,12 +275,12 @@ module Aws::BedrockAgentRuntime
|
|
268
275
|
|
269
276
|
Parameters.member = Shapes::ShapeRef.new(shape: Parameter)
|
270
277
|
|
271
|
-
PayloadPart.add_member(:bytes, Shapes::ShapeRef.new(shape: PartBody, location_name: "bytes"))
|
272
278
|
PayloadPart.add_member(:attribution, Shapes::ShapeRef.new(shape: Attribution, location_name: "attribution"))
|
279
|
+
PayloadPart.add_member(:bytes, Shapes::ShapeRef.new(shape: PartBody, location_name: "bytes"))
|
273
280
|
PayloadPart.struct_class = Types::PayloadPart
|
274
281
|
|
275
|
-
PostProcessingModelInvocationOutput.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))
|
276
282
|
PostProcessingModelInvocationOutput.add_member(:parsed_response, Shapes::ShapeRef.new(shape: PostProcessingParsedResponse, location_name: "parsedResponse"))
|
283
|
+
PostProcessingModelInvocationOutput.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))
|
277
284
|
PostProcessingModelInvocationOutput.struct_class = Types::PostProcessingModelInvocationOutput
|
278
285
|
|
279
286
|
PostProcessingParsedResponse.add_member(:text, Shapes::ShapeRef.new(shape: OutputString, location_name: "text"))
|
@@ -287,12 +294,12 @@ module Aws::BedrockAgentRuntime
|
|
287
294
|
PostProcessingTrace.add_member_subclass(:unknown, Types::PostProcessingTrace::Unknown)
|
288
295
|
PostProcessingTrace.struct_class = Types::PostProcessingTrace
|
289
296
|
|
290
|
-
PreProcessingModelInvocationOutput.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))
|
291
297
|
PreProcessingModelInvocationOutput.add_member(:parsed_response, Shapes::ShapeRef.new(shape: PreProcessingParsedResponse, location_name: "parsedResponse"))
|
298
|
+
PreProcessingModelInvocationOutput.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))
|
292
299
|
PreProcessingModelInvocationOutput.struct_class = Types::PreProcessingModelInvocationOutput
|
293
300
|
|
294
|
-
PreProcessingParsedResponse.add_member(:rationale, Shapes::ShapeRef.new(shape: RationaleString, location_name: "rationale"))
|
295
301
|
PreProcessingParsedResponse.add_member(:is_valid, Shapes::ShapeRef.new(shape: Boolean, location_name: "isValid"))
|
302
|
+
PreProcessingParsedResponse.add_member(:rationale, Shapes::ShapeRef.new(shape: RationaleString, location_name: "rationale"))
|
296
303
|
PreProcessingParsedResponse.struct_class = Types::PreProcessingParsedResponse
|
297
304
|
|
298
305
|
PreProcessingTrace.add_member(:model_invocation_input, Shapes::ShapeRef.new(shape: ModelInvocationInput, location_name: "modelInvocationInput"))
|
@@ -306,12 +313,15 @@ module Aws::BedrockAgentRuntime
|
|
306
313
|
PromptSessionAttributesMap.key = Shapes::ShapeRef.new(shape: String)
|
307
314
|
PromptSessionAttributesMap.value = Shapes::ShapeRef.new(shape: String)
|
308
315
|
|
309
|
-
|
316
|
+
PromptTemplate.add_member(:text_prompt_template, Shapes::ShapeRef.new(shape: TextPromptTemplate, location_name: "textPromptTemplate"))
|
317
|
+
PromptTemplate.struct_class = Types::PromptTemplate
|
318
|
+
|
310
319
|
Rationale.add_member(:text, Shapes::ShapeRef.new(shape: RationaleString, location_name: "text"))
|
320
|
+
Rationale.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))
|
311
321
|
Rationale.struct_class = Types::Rationale
|
312
322
|
|
313
|
-
RepromptResponse.add_member(:text, Shapes::ShapeRef.new(shape: String, location_name: "text"))
|
314
323
|
RepromptResponse.add_member(:source, Shapes::ShapeRef.new(shape: Source, location_name: "source"))
|
324
|
+
RepromptResponse.add_member(:text, Shapes::ShapeRef.new(shape: String, location_name: "text"))
|
315
325
|
RepromptResponse.struct_class = Types::RepromptResponse
|
316
326
|
|
317
327
|
RequestBody.add_member(:content, Shapes::ShapeRef.new(shape: ContentMap, location_name: "content"))
|
@@ -320,31 +330,31 @@ module Aws::BedrockAgentRuntime
|
|
320
330
|
ResourceNotFoundException.add_member(:message, Shapes::ShapeRef.new(shape: NonBlankString, location_name: "message"))
|
321
331
|
ResourceNotFoundException.struct_class = Types::ResourceNotFoundException
|
322
332
|
|
333
|
+
ResponseStream.add_member(:access_denied_exception, Shapes::ShapeRef.new(shape: AccessDeniedException, location_name: "accessDeniedException"))
|
334
|
+
ResponseStream.add_member(:bad_gateway_exception, Shapes::ShapeRef.new(shape: BadGatewayException, location_name: "badGatewayException"))
|
323
335
|
ResponseStream.add_member(:chunk, Shapes::ShapeRef.new(shape: PayloadPart, event: true, location_name: "chunk"))
|
324
|
-
ResponseStream.add_member(:
|
336
|
+
ResponseStream.add_member(:conflict_exception, Shapes::ShapeRef.new(shape: ConflictException, location_name: "conflictException"))
|
337
|
+
ResponseStream.add_member(:dependency_failed_exception, Shapes::ShapeRef.new(shape: DependencyFailedException, location_name: "dependencyFailedException"))
|
325
338
|
ResponseStream.add_member(:internal_server_exception, Shapes::ShapeRef.new(shape: InternalServerException, location_name: "internalServerException"))
|
326
|
-
ResponseStream.add_member(:validation_exception, Shapes::ShapeRef.new(shape: ValidationException, location_name: "validationException"))
|
327
339
|
ResponseStream.add_member(:resource_not_found_exception, Shapes::ShapeRef.new(shape: ResourceNotFoundException, location_name: "resourceNotFoundException"))
|
328
340
|
ResponseStream.add_member(:service_quota_exceeded_exception, Shapes::ShapeRef.new(shape: ServiceQuotaExceededException, location_name: "serviceQuotaExceededException"))
|
329
341
|
ResponseStream.add_member(:throttling_exception, Shapes::ShapeRef.new(shape: ThrottlingException, location_name: "throttlingException"))
|
330
|
-
ResponseStream.add_member(:
|
331
|
-
ResponseStream.add_member(:
|
332
|
-
ResponseStream.add_member(:dependency_failed_exception, Shapes::ShapeRef.new(shape: DependencyFailedException, location_name: "dependencyFailedException"))
|
333
|
-
ResponseStream.add_member(:bad_gateway_exception, Shapes::ShapeRef.new(shape: BadGatewayException, location_name: "badGatewayException"))
|
342
|
+
ResponseStream.add_member(:trace, Shapes::ShapeRef.new(shape: TracePart, event: true, location_name: "trace"))
|
343
|
+
ResponseStream.add_member(:validation_exception, Shapes::ShapeRef.new(shape: ValidationException, location_name: "validationException"))
|
334
344
|
ResponseStream.struct_class = Types::ResponseStream
|
335
345
|
|
336
346
|
RetrievalResultContent.add_member(:text, Shapes::ShapeRef.new(shape: String, required: true, location_name: "text"))
|
337
347
|
RetrievalResultContent.struct_class = Types::RetrievalResultContent
|
338
348
|
|
339
|
-
RetrievalResultLocation.add_member(:type, Shapes::ShapeRef.new(shape: RetrievalResultLocationType, required: true, location_name: "type"))
|
340
349
|
RetrievalResultLocation.add_member(:s3_location, Shapes::ShapeRef.new(shape: RetrievalResultS3Location, location_name: "s3Location"))
|
350
|
+
RetrievalResultLocation.add_member(:type, Shapes::ShapeRef.new(shape: RetrievalResultLocationType, required: true, location_name: "type"))
|
341
351
|
RetrievalResultLocation.struct_class = Types::RetrievalResultLocation
|
342
352
|
|
343
353
|
RetrievalResultS3Location.add_member(:uri, Shapes::ShapeRef.new(shape: String, location_name: "uri"))
|
344
354
|
RetrievalResultS3Location.struct_class = Types::RetrievalResultS3Location
|
345
355
|
|
346
|
-
RetrieveAndGenerateConfiguration.add_member(:type, Shapes::ShapeRef.new(shape: RetrieveAndGenerateType, required: true, location_name: "type"))
|
347
356
|
RetrieveAndGenerateConfiguration.add_member(:knowledge_base_configuration, Shapes::ShapeRef.new(shape: KnowledgeBaseRetrieveAndGenerateConfiguration, location_name: "knowledgeBaseConfiguration"))
|
357
|
+
RetrieveAndGenerateConfiguration.add_member(:type, Shapes::ShapeRef.new(shape: RetrieveAndGenerateType, required: true, location_name: "type"))
|
348
358
|
RetrieveAndGenerateConfiguration.struct_class = Types::RetrieveAndGenerateConfiguration
|
349
359
|
|
350
360
|
RetrieveAndGenerateInput.add_member(:text, Shapes::ShapeRef.new(shape: RetrieveAndGenerateInputTextString, required: true, location_name: "text"))
|
@@ -353,28 +363,28 @@ module Aws::BedrockAgentRuntime
|
|
353
363
|
RetrieveAndGenerateOutput.add_member(:text, Shapes::ShapeRef.new(shape: String, required: true, location_name: "text"))
|
354
364
|
RetrieveAndGenerateOutput.struct_class = Types::RetrieveAndGenerateOutput
|
355
365
|
|
356
|
-
RetrieveAndGenerateRequest.add_member(:session_id, Shapes::ShapeRef.new(shape: SessionId, location_name: "sessionId"))
|
357
366
|
RetrieveAndGenerateRequest.add_member(:input, Shapes::ShapeRef.new(shape: RetrieveAndGenerateInput, required: true, location_name: "input"))
|
358
367
|
RetrieveAndGenerateRequest.add_member(:retrieve_and_generate_configuration, Shapes::ShapeRef.new(shape: RetrieveAndGenerateConfiguration, location_name: "retrieveAndGenerateConfiguration"))
|
359
368
|
RetrieveAndGenerateRequest.add_member(:session_configuration, Shapes::ShapeRef.new(shape: RetrieveAndGenerateSessionConfiguration, location_name: "sessionConfiguration"))
|
369
|
+
RetrieveAndGenerateRequest.add_member(:session_id, Shapes::ShapeRef.new(shape: SessionId, location_name: "sessionId"))
|
360
370
|
RetrieveAndGenerateRequest.struct_class = Types::RetrieveAndGenerateRequest
|
361
371
|
|
362
|
-
RetrieveAndGenerateResponse.add_member(:session_id, Shapes::ShapeRef.new(shape: SessionId, required: true, location_name: "sessionId"))
|
363
|
-
RetrieveAndGenerateResponse.add_member(:output, Shapes::ShapeRef.new(shape: RetrieveAndGenerateOutput, required: true, location_name: "output"))
|
364
372
|
RetrieveAndGenerateResponse.add_member(:citations, Shapes::ShapeRef.new(shape: Citations, location_name: "citations"))
|
373
|
+
RetrieveAndGenerateResponse.add_member(:output, Shapes::ShapeRef.new(shape: RetrieveAndGenerateOutput, required: true, location_name: "output"))
|
374
|
+
RetrieveAndGenerateResponse.add_member(:session_id, Shapes::ShapeRef.new(shape: SessionId, required: true, location_name: "sessionId"))
|
365
375
|
RetrieveAndGenerateResponse.struct_class = Types::RetrieveAndGenerateResponse
|
366
376
|
|
367
377
|
RetrieveAndGenerateSessionConfiguration.add_member(:kms_key_arn, Shapes::ShapeRef.new(shape: KmsKeyArn, required: true, location_name: "kmsKeyArn"))
|
368
378
|
RetrieveAndGenerateSessionConfiguration.struct_class = Types::RetrieveAndGenerateSessionConfiguration
|
369
379
|
|
370
380
|
RetrieveRequest.add_member(:knowledge_base_id, Shapes::ShapeRef.new(shape: KnowledgeBaseId, required: true, location: "uri", location_name: "knowledgeBaseId"))
|
371
|
-
RetrieveRequest.add_member(:retrieval_query, Shapes::ShapeRef.new(shape: KnowledgeBaseQuery, required: true, location_name: "retrievalQuery"))
|
372
|
-
RetrieveRequest.add_member(:retrieval_configuration, Shapes::ShapeRef.new(shape: KnowledgeBaseRetrievalConfiguration, location_name: "retrievalConfiguration"))
|
373
381
|
RetrieveRequest.add_member(:next_token, Shapes::ShapeRef.new(shape: NextToken, location_name: "nextToken"))
|
382
|
+
RetrieveRequest.add_member(:retrieval_configuration, Shapes::ShapeRef.new(shape: KnowledgeBaseRetrievalConfiguration, location_name: "retrievalConfiguration"))
|
383
|
+
RetrieveRequest.add_member(:retrieval_query, Shapes::ShapeRef.new(shape: KnowledgeBaseQuery, required: true, location_name: "retrievalQuery"))
|
374
384
|
RetrieveRequest.struct_class = Types::RetrieveRequest
|
375
385
|
|
376
|
-
RetrieveResponse.add_member(:retrieval_results, Shapes::ShapeRef.new(shape: KnowledgeBaseRetrievalResults, required: true, location_name: "retrievalResults"))
|
377
386
|
RetrieveResponse.add_member(:next_token, Shapes::ShapeRef.new(shape: NextToken, location_name: "nextToken"))
|
387
|
+
RetrieveResponse.add_member(:retrieval_results, Shapes::ShapeRef.new(shape: KnowledgeBaseRetrievalResults, required: true, location_name: "retrievalResults"))
|
378
388
|
RetrieveResponse.struct_class = Types::RetrieveResponse
|
379
389
|
|
380
390
|
RetrievedReference.add_member(:content, Shapes::ShapeRef.new(shape: RetrievalResultContent, location_name: "content"))
|
@@ -389,37 +399,37 @@ module Aws::BedrockAgentRuntime
|
|
389
399
|
SessionAttributesMap.key = Shapes::ShapeRef.new(shape: String)
|
390
400
|
SessionAttributesMap.value = Shapes::ShapeRef.new(shape: String)
|
391
401
|
|
392
|
-
SessionState.add_member(:session_attributes, Shapes::ShapeRef.new(shape: SessionAttributesMap, location_name: "sessionAttributes"))
|
393
402
|
SessionState.add_member(:prompt_session_attributes, Shapes::ShapeRef.new(shape: PromptSessionAttributesMap, location_name: "promptSessionAttributes"))
|
403
|
+
SessionState.add_member(:session_attributes, Shapes::ShapeRef.new(shape: SessionAttributesMap, location_name: "sessionAttributes"))
|
394
404
|
SessionState.struct_class = Types::SessionState
|
395
405
|
|
396
|
-
Span.add_member(:start, Shapes::ShapeRef.new(shape: SpanStartInteger, location_name: "start"))
|
397
406
|
Span.add_member(:end, Shapes::ShapeRef.new(shape: SpanEndInteger, location_name: "end"))
|
407
|
+
Span.add_member(:start, Shapes::ShapeRef.new(shape: SpanStartInteger, location_name: "start"))
|
398
408
|
Span.struct_class = Types::Span
|
399
409
|
|
400
410
|
StopSequences.member = Shapes::ShapeRef.new(shape: String)
|
401
411
|
|
402
|
-
TextResponsePart.add_member(:text, Shapes::ShapeRef.new(shape: String, location_name: "text"))
|
403
412
|
TextResponsePart.add_member(:span, Shapes::ShapeRef.new(shape: Span, location_name: "span"))
|
413
|
+
TextResponsePart.add_member(:text, Shapes::ShapeRef.new(shape: String, location_name: "text"))
|
404
414
|
TextResponsePart.struct_class = Types::TextResponsePart
|
405
415
|
|
406
416
|
ThrottlingException.add_member(:message, Shapes::ShapeRef.new(shape: NonBlankString, location_name: "message"))
|
407
417
|
ThrottlingException.struct_class = Types::ThrottlingException
|
408
418
|
|
409
|
-
Trace.add_member(:
|
419
|
+
Trace.add_member(:failure_trace, Shapes::ShapeRef.new(shape: FailureTrace, location_name: "failureTrace"))
|
410
420
|
Trace.add_member(:orchestration_trace, Shapes::ShapeRef.new(shape: OrchestrationTrace, location_name: "orchestrationTrace"))
|
411
421
|
Trace.add_member(:post_processing_trace, Shapes::ShapeRef.new(shape: PostProcessingTrace, location_name: "postProcessingTrace"))
|
412
|
-
Trace.add_member(:
|
422
|
+
Trace.add_member(:pre_processing_trace, Shapes::ShapeRef.new(shape: PreProcessingTrace, location_name: "preProcessingTrace"))
|
413
423
|
Trace.add_member(:unknown, Shapes::ShapeRef.new(shape: nil, location_name: 'unknown'))
|
414
|
-
Trace.add_member_subclass(:
|
424
|
+
Trace.add_member_subclass(:failure_trace, Types::Trace::FailureTrace)
|
415
425
|
Trace.add_member_subclass(:orchestration_trace, Types::Trace::OrchestrationTrace)
|
416
426
|
Trace.add_member_subclass(:post_processing_trace, Types::Trace::PostProcessingTrace)
|
417
|
-
Trace.add_member_subclass(:
|
427
|
+
Trace.add_member_subclass(:pre_processing_trace, Types::Trace::PreProcessingTrace)
|
418
428
|
Trace.add_member_subclass(:unknown, Types::Trace::Unknown)
|
419
429
|
Trace.struct_class = Types::Trace
|
420
430
|
|
421
|
-
TracePart.add_member(:agent_id, Shapes::ShapeRef.new(shape: AgentId, location_name: "agentId"))
|
422
431
|
TracePart.add_member(:agent_alias_id, Shapes::ShapeRef.new(shape: AgentAliasId, location_name: "agentAliasId"))
|
432
|
+
TracePart.add_member(:agent_id, Shapes::ShapeRef.new(shape: AgentId, location_name: "agentId"))
|
423
433
|
TracePart.add_member(:session_id, Shapes::ShapeRef.new(shape: SessionId, location_name: "sessionId"))
|
424
434
|
TracePart.add_member(:trace, Shapes::ShapeRef.new(shape: Trace, location_name: "trace"))
|
425
435
|
TracePart.struct_class = Types::TracePart
|
@@ -15,20 +15,28 @@ module Aws::BedrockAgentRuntime
|
|
15
15
|
@event_emitter = Aws::EventEmitter.new
|
16
16
|
end
|
17
17
|
|
18
|
+
def on_access_denied_exception_event(&block)
|
19
|
+
@event_emitter.on(:access_denied_exception, block) if block_given?
|
20
|
+
end
|
21
|
+
|
22
|
+
def on_bad_gateway_exception_event(&block)
|
23
|
+
@event_emitter.on(:bad_gateway_exception, block) if block_given?
|
24
|
+
end
|
25
|
+
|
18
26
|
def on_chunk_event(&block)
|
19
27
|
@event_emitter.on(:chunk, block) if block_given?
|
20
28
|
end
|
21
29
|
|
22
|
-
def
|
23
|
-
@event_emitter.on(:
|
30
|
+
def on_conflict_exception_event(&block)
|
31
|
+
@event_emitter.on(:conflict_exception, block) if block_given?
|
24
32
|
end
|
25
33
|
|
26
|
-
def
|
27
|
-
@event_emitter.on(:
|
34
|
+
def on_dependency_failed_exception_event(&block)
|
35
|
+
@event_emitter.on(:dependency_failed_exception, block) if block_given?
|
28
36
|
end
|
29
37
|
|
30
|
-
def
|
31
|
-
@event_emitter.on(:
|
38
|
+
def on_internal_server_exception_event(&block)
|
39
|
+
@event_emitter.on(:internal_server_exception, block) if block_given?
|
32
40
|
end
|
33
41
|
|
34
42
|
def on_resource_not_found_exception_event(&block)
|
@@ -43,20 +51,12 @@ module Aws::BedrockAgentRuntime
|
|
43
51
|
@event_emitter.on(:throttling_exception, block) if block_given?
|
44
52
|
end
|
45
53
|
|
46
|
-
def
|
47
|
-
@event_emitter.on(:
|
48
|
-
end
|
49
|
-
|
50
|
-
def on_conflict_exception_event(&block)
|
51
|
-
@event_emitter.on(:conflict_exception, block) if block_given?
|
52
|
-
end
|
53
|
-
|
54
|
-
def on_dependency_failed_exception_event(&block)
|
55
|
-
@event_emitter.on(:dependency_failed_exception, block) if block_given?
|
54
|
+
def on_trace_event(&block)
|
55
|
+
@event_emitter.on(:trace, block) if block_given?
|
56
56
|
end
|
57
57
|
|
58
|
-
def
|
59
|
-
@event_emitter.on(:
|
58
|
+
def on_validation_exception_event(&block)
|
59
|
+
@event_emitter.on(:validation_exception, block) if block_given?
|
60
60
|
end
|
61
61
|
|
62
62
|
def on_error_event(&block)
|
@@ -72,17 +72,17 @@ module Aws::BedrockAgentRuntime
|
|
72
72
|
end
|
73
73
|
|
74
74
|
def on_event(&block)
|
75
|
+
on_access_denied_exception_event(&block)
|
76
|
+
on_bad_gateway_exception_event(&block)
|
75
77
|
on_chunk_event(&block)
|
76
|
-
|
78
|
+
on_conflict_exception_event(&block)
|
79
|
+
on_dependency_failed_exception_event(&block)
|
77
80
|
on_internal_server_exception_event(&block)
|
78
|
-
on_validation_exception_event(&block)
|
79
81
|
on_resource_not_found_exception_event(&block)
|
80
82
|
on_service_quota_exceeded_exception_event(&block)
|
81
83
|
on_throttling_exception_event(&block)
|
82
|
-
|
83
|
-
|
84
|
-
on_dependency_failed_exception_event(&block)
|
85
|
-
on_bad_gateway_exception_event(&block)
|
84
|
+
on_trace_event(&block)
|
85
|
+
on_validation_exception_event(&block)
|
86
86
|
on_error_event(&block)
|
87
87
|
on_initial_response_event(&block)
|
88
88
|
on_unknown_event(&block)
|