aws-sdk-bedrockagentruntime 1.2.0 → 1.3.0

Sign up to get free protection for your applications and to get access to all the features.
@@ -129,10 +129,10 @@ module Aws::BedrockAgentRuntime
129
129
  AccessDeniedException.struct_class = Types::AccessDeniedException
130
130
 
131
131
  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
132
  ActionGroupInvocationInput.add_member(:api_path, Shapes::ShapeRef.new(shape: ApiPath, location_name: "apiPath"))
134
133
  ActionGroupInvocationInput.add_member(:parameters, Shapes::ShapeRef.new(shape: Parameters, location_name: "parameters"))
135
134
  ActionGroupInvocationInput.add_member(:request_body, Shapes::ShapeRef.new(shape: RequestBody, location_name: "requestBody"))
135
+ ActionGroupInvocationInput.add_member(:verb, Shapes::ShapeRef.new(shape: Verb, location_name: "verb"))
136
136
  ActionGroupInvocationInput.struct_class = Types::ActionGroupInvocationInput
137
137
 
138
138
  ActionGroupInvocationOutput.add_member(:text, Shapes::ShapeRef.new(shape: ActionGroupOutputString, location_name: "text"))
@@ -161,8 +161,8 @@ module Aws::BedrockAgentRuntime
161
161
  DependencyFailedException.add_member(:resource_name, Shapes::ShapeRef.new(shape: NonBlankString, location_name: "resourceName"))
162
162
  DependencyFailedException.struct_class = Types::DependencyFailedException
163
163
 
164
- FailureTrace.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))
165
164
  FailureTrace.add_member(:failure_reason, Shapes::ShapeRef.new(shape: FailureReasonString, location_name: "failureReason"))
165
+ FailureTrace.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))
166
166
  FailureTrace.struct_class = Types::FailureTrace
167
167
 
168
168
  FinalResponse.add_member(:text, Shapes::ShapeRef.new(shape: FinalResponseString, location_name: "text"))
@@ -171,29 +171,29 @@ module Aws::BedrockAgentRuntime
171
171
  GeneratedResponsePart.add_member(:text_response_part, Shapes::ShapeRef.new(shape: TextResponsePart, location_name: "textResponsePart"))
172
172
  GeneratedResponsePart.struct_class = Types::GeneratedResponsePart
173
173
 
174
- InferenceConfiguration.add_member(:temperature, Shapes::ShapeRef.new(shape: Temperature, location_name: "temperature"))
175
- InferenceConfiguration.add_member(:top_p, Shapes::ShapeRef.new(shape: TopP, location_name: "topP"))
176
- InferenceConfiguration.add_member(:top_k, Shapes::ShapeRef.new(shape: TopK, location_name: "topK"))
177
174
  InferenceConfiguration.add_member(:maximum_length, Shapes::ShapeRef.new(shape: MaximumLength, location_name: "maximumLength"))
178
175
  InferenceConfiguration.add_member(:stop_sequences, Shapes::ShapeRef.new(shape: StopSequences, location_name: "stopSequences"))
176
+ InferenceConfiguration.add_member(:temperature, Shapes::ShapeRef.new(shape: Temperature, location_name: "temperature"))
177
+ InferenceConfiguration.add_member(:top_k, Shapes::ShapeRef.new(shape: TopK, location_name: "topK"))
178
+ InferenceConfiguration.add_member(:top_p, Shapes::ShapeRef.new(shape: TopP, location_name: "topP"))
179
179
  InferenceConfiguration.struct_class = Types::InferenceConfiguration
180
180
 
181
181
  InternalServerException.add_member(:message, Shapes::ShapeRef.new(shape: NonBlankString, location_name: "message"))
182
182
  InternalServerException.struct_class = Types::InternalServerException
183
183
 
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
184
  InvocationInput.add_member(:action_group_invocation_input, Shapes::ShapeRef.new(shape: ActionGroupInvocationInput, location_name: "actionGroupInvocationInput"))
185
+ InvocationInput.add_member(:invocation_type, Shapes::ShapeRef.new(shape: InvocationType, location_name: "invocationType"))
187
186
  InvocationInput.add_member(:knowledge_base_lookup_input, Shapes::ShapeRef.new(shape: KnowledgeBaseLookupInput, location_name: "knowledgeBaseLookupInput"))
187
+ InvocationInput.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))
188
188
  InvocationInput.struct_class = Types::InvocationInput
189
189
 
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
190
  InvokeAgentRequest.add_member(:agent_alias_id, Shapes::ShapeRef.new(shape: AgentAliasId, required: true, location: "uri", location_name: "agentAliasId"))
193
- InvokeAgentRequest.add_member(:session_id, Shapes::ShapeRef.new(shape: SessionId, required: true, location: "uri", location_name: "sessionId"))
194
- InvokeAgentRequest.add_member(:end_session, Shapes::ShapeRef.new(shape: Boolean, location_name: "endSession"))
191
+ InvokeAgentRequest.add_member(:agent_id, Shapes::ShapeRef.new(shape: AgentId, required: true, location: "uri", location_name: "agentId"))
195
192
  InvokeAgentRequest.add_member(:enable_trace, Shapes::ShapeRef.new(shape: Boolean, location_name: "enableTrace"))
193
+ InvokeAgentRequest.add_member(:end_session, Shapes::ShapeRef.new(shape: Boolean, location_name: "endSession"))
196
194
  InvokeAgentRequest.add_member(:input_text, Shapes::ShapeRef.new(shape: InputText, required: true, location_name: "inputText"))
195
+ InvokeAgentRequest.add_member(:session_id, Shapes::ShapeRef.new(shape: SessionId, required: true, location: "uri", location_name: "sessionId"))
196
+ InvokeAgentRequest.add_member(:session_state, Shapes::ShapeRef.new(shape: SessionState, location_name: "sessionState"))
197
197
  InvokeAgentRequest.struct_class = Types::InvokeAgentRequest
198
198
 
199
199
  InvokeAgentResponse.add_member(:completion, Shapes::ShapeRef.new(shape: ResponseStream, required: true, eventstream: true, location_name: "completion"))
@@ -203,8 +203,8 @@ module Aws::BedrockAgentRuntime
203
203
  InvokeAgentResponse[:payload] = :completion
204
204
  InvokeAgentResponse[:payload_member] = InvokeAgentResponse.member(:completion)
205
205
 
206
- KnowledgeBaseLookupInput.add_member(:text, Shapes::ShapeRef.new(shape: KnowledgeBaseLookupInputString, location_name: "text"))
207
206
  KnowledgeBaseLookupInput.add_member(:knowledge_base_id, Shapes::ShapeRef.new(shape: TraceKnowledgeBaseId, location_name: "knowledgeBaseId"))
207
+ KnowledgeBaseLookupInput.add_member(:text, Shapes::ShapeRef.new(shape: KnowledgeBaseLookupInputString, location_name: "text"))
208
208
  KnowledgeBaseLookupInput.struct_class = Types::KnowledgeBaseLookupInput
209
209
 
210
210
  KnowledgeBaseLookupOutput.add_member(:retrieved_references, Shapes::ShapeRef.new(shape: RetrievedReferences, location_name: "retrievedReferences"))
@@ -232,32 +232,32 @@ module Aws::BedrockAgentRuntime
232
232
  KnowledgeBaseVectorSearchConfiguration.add_member(:override_search_type, Shapes::ShapeRef.new(shape: SearchType, location_name: "overrideSearchType"))
233
233
  KnowledgeBaseVectorSearchConfiguration.struct_class = Types::KnowledgeBaseVectorSearchConfiguration
234
234
 
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
235
  ModelInvocationInput.add_member(:inference_configuration, Shapes::ShapeRef.new(shape: InferenceConfiguration, location_name: "inferenceConfiguration"))
239
236
  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
237
  ModelInvocationInput.add_member(:parser_mode, Shapes::ShapeRef.new(shape: CreationMode, location_name: "parserMode"))
238
+ ModelInvocationInput.add_member(:prompt_creation_mode, Shapes::ShapeRef.new(shape: CreationMode, location_name: "promptCreationMode"))
239
+ ModelInvocationInput.add_member(:text, Shapes::ShapeRef.new(shape: PromptText, location_name: "text"))
240
+ ModelInvocationInput.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))
241
+ ModelInvocationInput.add_member(:type, Shapes::ShapeRef.new(shape: PromptType, location_name: "type"))
242
242
  ModelInvocationInput.struct_class = Types::ModelInvocationInput
243
243
 
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
244
  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
245
  Observation.add_member(:final_response, Shapes::ShapeRef.new(shape: FinalResponse, location_name: "finalResponse"))
246
+ Observation.add_member(:knowledge_base_lookup_output, Shapes::ShapeRef.new(shape: KnowledgeBaseLookupOutput, location_name: "knowledgeBaseLookupOutput"))
249
247
  Observation.add_member(:reprompt_response, Shapes::ShapeRef.new(shape: RepromptResponse, location_name: "repromptResponse"))
248
+ Observation.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))
249
+ Observation.add_member(:type, Shapes::ShapeRef.new(shape: Type, location_name: "type"))
250
250
  Observation.struct_class = Types::Observation
251
251
 
252
- OrchestrationTrace.add_member(:rationale, Shapes::ShapeRef.new(shape: Rationale, location_name: "rationale"))
253
252
  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
253
  OrchestrationTrace.add_member(:model_invocation_input, Shapes::ShapeRef.new(shape: ModelInvocationInput, location_name: "modelInvocationInput"))
254
+ OrchestrationTrace.add_member(:observation, Shapes::ShapeRef.new(shape: Observation, location_name: "observation"))
255
+ OrchestrationTrace.add_member(:rationale, Shapes::ShapeRef.new(shape: Rationale, location_name: "rationale"))
256
256
  OrchestrationTrace.add_member(:unknown, Shapes::ShapeRef.new(shape: nil, location_name: 'unknown'))
257
- OrchestrationTrace.add_member_subclass(:rationale, Types::OrchestrationTrace::Rationale)
258
257
  OrchestrationTrace.add_member_subclass(:invocation_input, Types::OrchestrationTrace::InvocationInput)
259
- OrchestrationTrace.add_member_subclass(:observation, Types::OrchestrationTrace::Observation)
260
258
  OrchestrationTrace.add_member_subclass(:model_invocation_input, Types::OrchestrationTrace::ModelInvocationInput)
259
+ OrchestrationTrace.add_member_subclass(:observation, Types::OrchestrationTrace::Observation)
260
+ OrchestrationTrace.add_member_subclass(:rationale, Types::OrchestrationTrace::Rationale)
261
261
  OrchestrationTrace.add_member_subclass(:unknown, Types::OrchestrationTrace::Unknown)
262
262
  OrchestrationTrace.struct_class = Types::OrchestrationTrace
263
263
 
@@ -268,12 +268,12 @@ module Aws::BedrockAgentRuntime
268
268
 
269
269
  Parameters.member = Shapes::ShapeRef.new(shape: Parameter)
270
270
 
271
- PayloadPart.add_member(:bytes, Shapes::ShapeRef.new(shape: PartBody, location_name: "bytes"))
272
271
  PayloadPart.add_member(:attribution, Shapes::ShapeRef.new(shape: Attribution, location_name: "attribution"))
272
+ PayloadPart.add_member(:bytes, Shapes::ShapeRef.new(shape: PartBody, location_name: "bytes"))
273
273
  PayloadPart.struct_class = Types::PayloadPart
274
274
 
275
- PostProcessingModelInvocationOutput.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))
276
275
  PostProcessingModelInvocationOutput.add_member(:parsed_response, Shapes::ShapeRef.new(shape: PostProcessingParsedResponse, location_name: "parsedResponse"))
276
+ PostProcessingModelInvocationOutput.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))
277
277
  PostProcessingModelInvocationOutput.struct_class = Types::PostProcessingModelInvocationOutput
278
278
 
279
279
  PostProcessingParsedResponse.add_member(:text, Shapes::ShapeRef.new(shape: OutputString, location_name: "text"))
@@ -287,12 +287,12 @@ module Aws::BedrockAgentRuntime
287
287
  PostProcessingTrace.add_member_subclass(:unknown, Types::PostProcessingTrace::Unknown)
288
288
  PostProcessingTrace.struct_class = Types::PostProcessingTrace
289
289
 
290
- PreProcessingModelInvocationOutput.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))
291
290
  PreProcessingModelInvocationOutput.add_member(:parsed_response, Shapes::ShapeRef.new(shape: PreProcessingParsedResponse, location_name: "parsedResponse"))
291
+ PreProcessingModelInvocationOutput.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))
292
292
  PreProcessingModelInvocationOutput.struct_class = Types::PreProcessingModelInvocationOutput
293
293
 
294
- PreProcessingParsedResponse.add_member(:rationale, Shapes::ShapeRef.new(shape: RationaleString, location_name: "rationale"))
295
294
  PreProcessingParsedResponse.add_member(:is_valid, Shapes::ShapeRef.new(shape: Boolean, location_name: "isValid"))
295
+ PreProcessingParsedResponse.add_member(:rationale, Shapes::ShapeRef.new(shape: RationaleString, location_name: "rationale"))
296
296
  PreProcessingParsedResponse.struct_class = Types::PreProcessingParsedResponse
297
297
 
298
298
  PreProcessingTrace.add_member(:model_invocation_input, Shapes::ShapeRef.new(shape: ModelInvocationInput, location_name: "modelInvocationInput"))
@@ -306,12 +306,12 @@ module Aws::BedrockAgentRuntime
306
306
  PromptSessionAttributesMap.key = Shapes::ShapeRef.new(shape: String)
307
307
  PromptSessionAttributesMap.value = Shapes::ShapeRef.new(shape: String)
308
308
 
309
- Rationale.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))
310
309
  Rationale.add_member(:text, Shapes::ShapeRef.new(shape: RationaleString, location_name: "text"))
310
+ Rationale.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))
311
311
  Rationale.struct_class = Types::Rationale
312
312
 
313
- RepromptResponse.add_member(:text, Shapes::ShapeRef.new(shape: String, location_name: "text"))
314
313
  RepromptResponse.add_member(:source, Shapes::ShapeRef.new(shape: Source, location_name: "source"))
314
+ RepromptResponse.add_member(:text, Shapes::ShapeRef.new(shape: String, location_name: "text"))
315
315
  RepromptResponse.struct_class = Types::RepromptResponse
316
316
 
317
317
  RequestBody.add_member(:content, Shapes::ShapeRef.new(shape: ContentMap, location_name: "content"))
@@ -320,31 +320,31 @@ module Aws::BedrockAgentRuntime
320
320
  ResourceNotFoundException.add_member(:message, Shapes::ShapeRef.new(shape: NonBlankString, location_name: "message"))
321
321
  ResourceNotFoundException.struct_class = Types::ResourceNotFoundException
322
322
 
323
+ ResponseStream.add_member(:access_denied_exception, Shapes::ShapeRef.new(shape: AccessDeniedException, location_name: "accessDeniedException"))
324
+ ResponseStream.add_member(:bad_gateway_exception, Shapes::ShapeRef.new(shape: BadGatewayException, location_name: "badGatewayException"))
323
325
  ResponseStream.add_member(:chunk, Shapes::ShapeRef.new(shape: PayloadPart, event: true, location_name: "chunk"))
324
- ResponseStream.add_member(:trace, Shapes::ShapeRef.new(shape: TracePart, event: true, location_name: "trace"))
326
+ ResponseStream.add_member(:conflict_exception, Shapes::ShapeRef.new(shape: ConflictException, location_name: "conflictException"))
327
+ ResponseStream.add_member(:dependency_failed_exception, Shapes::ShapeRef.new(shape: DependencyFailedException, location_name: "dependencyFailedException"))
325
328
  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
329
  ResponseStream.add_member(:resource_not_found_exception, Shapes::ShapeRef.new(shape: ResourceNotFoundException, location_name: "resourceNotFoundException"))
328
330
  ResponseStream.add_member(:service_quota_exceeded_exception, Shapes::ShapeRef.new(shape: ServiceQuotaExceededException, location_name: "serviceQuotaExceededException"))
329
331
  ResponseStream.add_member(:throttling_exception, Shapes::ShapeRef.new(shape: ThrottlingException, location_name: "throttlingException"))
330
- ResponseStream.add_member(:access_denied_exception, Shapes::ShapeRef.new(shape: AccessDeniedException, location_name: "accessDeniedException"))
331
- ResponseStream.add_member(:conflict_exception, Shapes::ShapeRef.new(shape: ConflictException, location_name: "conflictException"))
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"))
332
+ ResponseStream.add_member(:trace, Shapes::ShapeRef.new(shape: TracePart, event: true, location_name: "trace"))
333
+ ResponseStream.add_member(:validation_exception, Shapes::ShapeRef.new(shape: ValidationException, location_name: "validationException"))
334
334
  ResponseStream.struct_class = Types::ResponseStream
335
335
 
336
336
  RetrievalResultContent.add_member(:text, Shapes::ShapeRef.new(shape: String, required: true, location_name: "text"))
337
337
  RetrievalResultContent.struct_class = Types::RetrievalResultContent
338
338
 
339
- RetrievalResultLocation.add_member(:type, Shapes::ShapeRef.new(shape: RetrievalResultLocationType, required: true, location_name: "type"))
340
339
  RetrievalResultLocation.add_member(:s3_location, Shapes::ShapeRef.new(shape: RetrievalResultS3Location, location_name: "s3Location"))
340
+ RetrievalResultLocation.add_member(:type, Shapes::ShapeRef.new(shape: RetrievalResultLocationType, required: true, location_name: "type"))
341
341
  RetrievalResultLocation.struct_class = Types::RetrievalResultLocation
342
342
 
343
343
  RetrievalResultS3Location.add_member(:uri, Shapes::ShapeRef.new(shape: String, location_name: "uri"))
344
344
  RetrievalResultS3Location.struct_class = Types::RetrievalResultS3Location
345
345
 
346
- RetrieveAndGenerateConfiguration.add_member(:type, Shapes::ShapeRef.new(shape: RetrieveAndGenerateType, required: true, location_name: "type"))
347
346
  RetrieveAndGenerateConfiguration.add_member(:knowledge_base_configuration, Shapes::ShapeRef.new(shape: KnowledgeBaseRetrieveAndGenerateConfiguration, location_name: "knowledgeBaseConfiguration"))
347
+ RetrieveAndGenerateConfiguration.add_member(:type, Shapes::ShapeRef.new(shape: RetrieveAndGenerateType, required: true, location_name: "type"))
348
348
  RetrieveAndGenerateConfiguration.struct_class = Types::RetrieveAndGenerateConfiguration
349
349
 
350
350
  RetrieveAndGenerateInput.add_member(:text, Shapes::ShapeRef.new(shape: RetrieveAndGenerateInputTextString, required: true, location_name: "text"))
@@ -353,28 +353,28 @@ module Aws::BedrockAgentRuntime
353
353
  RetrieveAndGenerateOutput.add_member(:text, Shapes::ShapeRef.new(shape: String, required: true, location_name: "text"))
354
354
  RetrieveAndGenerateOutput.struct_class = Types::RetrieveAndGenerateOutput
355
355
 
356
- RetrieveAndGenerateRequest.add_member(:session_id, Shapes::ShapeRef.new(shape: SessionId, location_name: "sessionId"))
357
356
  RetrieveAndGenerateRequest.add_member(:input, Shapes::ShapeRef.new(shape: RetrieveAndGenerateInput, required: true, location_name: "input"))
358
357
  RetrieveAndGenerateRequest.add_member(:retrieve_and_generate_configuration, Shapes::ShapeRef.new(shape: RetrieveAndGenerateConfiguration, location_name: "retrieveAndGenerateConfiguration"))
359
358
  RetrieveAndGenerateRequest.add_member(:session_configuration, Shapes::ShapeRef.new(shape: RetrieveAndGenerateSessionConfiguration, location_name: "sessionConfiguration"))
359
+ RetrieveAndGenerateRequest.add_member(:session_id, Shapes::ShapeRef.new(shape: SessionId, location_name: "sessionId"))
360
360
  RetrieveAndGenerateRequest.struct_class = Types::RetrieveAndGenerateRequest
361
361
 
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
362
  RetrieveAndGenerateResponse.add_member(:citations, Shapes::ShapeRef.new(shape: Citations, location_name: "citations"))
363
+ RetrieveAndGenerateResponse.add_member(:output, Shapes::ShapeRef.new(shape: RetrieveAndGenerateOutput, required: true, location_name: "output"))
364
+ RetrieveAndGenerateResponse.add_member(:session_id, Shapes::ShapeRef.new(shape: SessionId, required: true, location_name: "sessionId"))
365
365
  RetrieveAndGenerateResponse.struct_class = Types::RetrieveAndGenerateResponse
366
366
 
367
367
  RetrieveAndGenerateSessionConfiguration.add_member(:kms_key_arn, Shapes::ShapeRef.new(shape: KmsKeyArn, required: true, location_name: "kmsKeyArn"))
368
368
  RetrieveAndGenerateSessionConfiguration.struct_class = Types::RetrieveAndGenerateSessionConfiguration
369
369
 
370
370
  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
371
  RetrieveRequest.add_member(:next_token, Shapes::ShapeRef.new(shape: NextToken, location_name: "nextToken"))
372
+ RetrieveRequest.add_member(:retrieval_configuration, Shapes::ShapeRef.new(shape: KnowledgeBaseRetrievalConfiguration, location_name: "retrievalConfiguration"))
373
+ RetrieveRequest.add_member(:retrieval_query, Shapes::ShapeRef.new(shape: KnowledgeBaseQuery, required: true, location_name: "retrievalQuery"))
374
374
  RetrieveRequest.struct_class = Types::RetrieveRequest
375
375
 
376
- RetrieveResponse.add_member(:retrieval_results, Shapes::ShapeRef.new(shape: KnowledgeBaseRetrievalResults, required: true, location_name: "retrievalResults"))
377
376
  RetrieveResponse.add_member(:next_token, Shapes::ShapeRef.new(shape: NextToken, location_name: "nextToken"))
377
+ RetrieveResponse.add_member(:retrieval_results, Shapes::ShapeRef.new(shape: KnowledgeBaseRetrievalResults, required: true, location_name: "retrievalResults"))
378
378
  RetrieveResponse.struct_class = Types::RetrieveResponse
379
379
 
380
380
  RetrievedReference.add_member(:content, Shapes::ShapeRef.new(shape: RetrievalResultContent, location_name: "content"))
@@ -389,37 +389,37 @@ module Aws::BedrockAgentRuntime
389
389
  SessionAttributesMap.key = Shapes::ShapeRef.new(shape: String)
390
390
  SessionAttributesMap.value = Shapes::ShapeRef.new(shape: String)
391
391
 
392
- SessionState.add_member(:session_attributes, Shapes::ShapeRef.new(shape: SessionAttributesMap, location_name: "sessionAttributes"))
393
392
  SessionState.add_member(:prompt_session_attributes, Shapes::ShapeRef.new(shape: PromptSessionAttributesMap, location_name: "promptSessionAttributes"))
393
+ SessionState.add_member(:session_attributes, Shapes::ShapeRef.new(shape: SessionAttributesMap, location_name: "sessionAttributes"))
394
394
  SessionState.struct_class = Types::SessionState
395
395
 
396
- Span.add_member(:start, Shapes::ShapeRef.new(shape: SpanStartInteger, location_name: "start"))
397
396
  Span.add_member(:end, Shapes::ShapeRef.new(shape: SpanEndInteger, location_name: "end"))
397
+ Span.add_member(:start, Shapes::ShapeRef.new(shape: SpanStartInteger, location_name: "start"))
398
398
  Span.struct_class = Types::Span
399
399
 
400
400
  StopSequences.member = Shapes::ShapeRef.new(shape: String)
401
401
 
402
- TextResponsePart.add_member(:text, Shapes::ShapeRef.new(shape: String, location_name: "text"))
403
402
  TextResponsePart.add_member(:span, Shapes::ShapeRef.new(shape: Span, location_name: "span"))
403
+ TextResponsePart.add_member(:text, Shapes::ShapeRef.new(shape: String, location_name: "text"))
404
404
  TextResponsePart.struct_class = Types::TextResponsePart
405
405
 
406
406
  ThrottlingException.add_member(:message, Shapes::ShapeRef.new(shape: NonBlankString, location_name: "message"))
407
407
  ThrottlingException.struct_class = Types::ThrottlingException
408
408
 
409
- Trace.add_member(:pre_processing_trace, Shapes::ShapeRef.new(shape: PreProcessingTrace, location_name: "preProcessingTrace"))
409
+ Trace.add_member(:failure_trace, Shapes::ShapeRef.new(shape: FailureTrace, location_name: "failureTrace"))
410
410
  Trace.add_member(:orchestration_trace, Shapes::ShapeRef.new(shape: OrchestrationTrace, location_name: "orchestrationTrace"))
411
411
  Trace.add_member(:post_processing_trace, Shapes::ShapeRef.new(shape: PostProcessingTrace, location_name: "postProcessingTrace"))
412
- Trace.add_member(:failure_trace, Shapes::ShapeRef.new(shape: FailureTrace, location_name: "failureTrace"))
412
+ Trace.add_member(:pre_processing_trace, Shapes::ShapeRef.new(shape: PreProcessingTrace, location_name: "preProcessingTrace"))
413
413
  Trace.add_member(:unknown, Shapes::ShapeRef.new(shape: nil, location_name: 'unknown'))
414
- Trace.add_member_subclass(:pre_processing_trace, Types::Trace::PreProcessingTrace)
414
+ Trace.add_member_subclass(:failure_trace, Types::Trace::FailureTrace)
415
415
  Trace.add_member_subclass(:orchestration_trace, Types::Trace::OrchestrationTrace)
416
416
  Trace.add_member_subclass(:post_processing_trace, Types::Trace::PostProcessingTrace)
417
- Trace.add_member_subclass(:failure_trace, Types::Trace::FailureTrace)
417
+ Trace.add_member_subclass(:pre_processing_trace, Types::Trace::PreProcessingTrace)
418
418
  Trace.add_member_subclass(:unknown, Types::Trace::Unknown)
419
419
  Trace.struct_class = Types::Trace
420
420
 
421
- TracePart.add_member(:agent_id, Shapes::ShapeRef.new(shape: AgentId, location_name: "agentId"))
422
421
  TracePart.add_member(:agent_alias_id, Shapes::ShapeRef.new(shape: AgentAliasId, location_name: "agentAliasId"))
422
+ TracePart.add_member(:agent_id, Shapes::ShapeRef.new(shape: AgentId, location_name: "agentId"))
423
423
  TracePart.add_member(:session_id, Shapes::ShapeRef.new(shape: SessionId, location_name: "sessionId"))
424
424
  TracePart.add_member(:trace, Shapes::ShapeRef.new(shape: Trace, location_name: "trace"))
425
425
  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 on_trace_event(&block)
23
- @event_emitter.on(:trace, block) if block_given?
30
+ def on_conflict_exception_event(&block)
31
+ @event_emitter.on(:conflict_exception, block) if block_given?
24
32
  end
25
33
 
26
- def on_internal_server_exception_event(&block)
27
- @event_emitter.on(:internal_server_exception, block) if block_given?
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 on_validation_exception_event(&block)
31
- @event_emitter.on(:validation_exception, block) if block_given?
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 on_access_denied_exception_event(&block)
47
- @event_emitter.on(:access_denied_exception, block) if block_given?
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 on_bad_gateway_exception_event(&block)
59
- @event_emitter.on(:bad_gateway_exception, block) if block_given?
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
- on_trace_event(&block)
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
- on_access_denied_exception_event(&block)
83
- on_conflict_exception_event(&block)
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)