aws-sdk-bedrockagentruntime 1.2.0 → 1.3.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.
@@ -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)