aws-sdk-bedrockagentruntime 1.5.0 → 1.7.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.
@@ -20,18 +20,34 @@ module Aws::BedrockAgentRuntime
20
20
  ActionGroupOutputString = Shapes::StringShape.new(name: 'ActionGroupOutputString')
21
21
  AgentAliasId = Shapes::StringShape.new(name: 'AgentAliasId')
22
22
  AgentId = Shapes::StringShape.new(name: 'AgentId')
23
+ AgentVersion = Shapes::StringShape.new(name: 'AgentVersion')
24
+ ApiContentMap = Shapes::MapShape.new(name: 'ApiContentMap')
25
+ ApiInvocationInput = Shapes::StructureShape.new(name: 'ApiInvocationInput')
26
+ ApiParameter = Shapes::StructureShape.new(name: 'ApiParameter')
27
+ ApiParameters = Shapes::ListShape.new(name: 'ApiParameters')
23
28
  ApiPath = Shapes::StringShape.new(name: 'ApiPath')
29
+ ApiRequestBody = Shapes::StructureShape.new(name: 'ApiRequestBody')
30
+ ApiResult = Shapes::StructureShape.new(name: 'ApiResult')
24
31
  Attribution = Shapes::StructureShape.new(name: 'Attribution')
25
32
  BadGatewayException = Shapes::StructureShape.new(name: 'BadGatewayException')
26
33
  BedrockModelArn = Shapes::StringShape.new(name: 'BedrockModelArn')
27
34
  Boolean = Shapes::BooleanShape.new(name: 'Boolean')
35
+ ByteContentBlob = Shapes::BlobShape.new(name: 'ByteContentBlob')
36
+ ByteContentDoc = Shapes::StructureShape.new(name: 'ByteContentDoc')
28
37
  Citation = Shapes::StructureShape.new(name: 'Citation')
29
38
  Citations = Shapes::ListShape.new(name: 'Citations')
30
39
  ConflictException = Shapes::StructureShape.new(name: 'ConflictException')
40
+ ContentBody = Shapes::StructureShape.new(name: 'ContentBody')
31
41
  ContentMap = Shapes::MapShape.new(name: 'ContentMap')
42
+ ContentType = Shapes::StringShape.new(name: 'ContentType')
32
43
  CreationMode = Shapes::StringShape.new(name: 'CreationMode')
33
44
  DependencyFailedException = Shapes::StructureShape.new(name: 'DependencyFailedException')
34
45
  Double = Shapes::FloatShape.new(name: 'Double')
46
+ ExternalSource = Shapes::StructureShape.new(name: 'ExternalSource')
47
+ ExternalSourceType = Shapes::StringShape.new(name: 'ExternalSourceType')
48
+ ExternalSources = Shapes::ListShape.new(name: 'ExternalSources')
49
+ ExternalSourcesGenerationConfiguration = Shapes::StructureShape.new(name: 'ExternalSourcesGenerationConfiguration')
50
+ ExternalSourcesRetrieveAndGenerateConfiguration = Shapes::StructureShape.new(name: 'ExternalSourcesRetrieveAndGenerateConfiguration')
35
51
  FailureReasonString = Shapes::StringShape.new(name: 'FailureReasonString')
36
52
  FailureTrace = Shapes::StructureShape.new(name: 'FailureTrace')
37
53
  FilterAttribute = Shapes::StructureShape.new(name: 'FilterAttribute')
@@ -39,12 +55,22 @@ module Aws::BedrockAgentRuntime
39
55
  FilterValue = Shapes::DocumentShape.new(name: 'FilterValue', document: true)
40
56
  FinalResponse = Shapes::StructureShape.new(name: 'FinalResponse')
41
57
  FinalResponseString = Shapes::StringShape.new(name: 'FinalResponseString')
58
+ Function = Shapes::StringShape.new(name: 'Function')
59
+ FunctionInvocationInput = Shapes::StructureShape.new(name: 'FunctionInvocationInput')
60
+ FunctionParameter = Shapes::StructureShape.new(name: 'FunctionParameter')
61
+ FunctionParameters = Shapes::ListShape.new(name: 'FunctionParameters')
62
+ FunctionResult = Shapes::StructureShape.new(name: 'FunctionResult')
42
63
  GeneratedResponsePart = Shapes::StructureShape.new(name: 'GeneratedResponsePart')
43
64
  GenerationConfiguration = Shapes::StructureShape.new(name: 'GenerationConfiguration')
65
+ Identifier = Shapes::StringShape.new(name: 'Identifier')
44
66
  InferenceConfiguration = Shapes::StructureShape.new(name: 'InferenceConfiguration')
45
67
  InputText = Shapes::StringShape.new(name: 'InputText')
68
+ Integer = Shapes::IntegerShape.new(name: 'Integer')
46
69
  InternalServerException = Shapes::StructureShape.new(name: 'InternalServerException')
47
70
  InvocationInput = Shapes::StructureShape.new(name: 'InvocationInput')
71
+ InvocationInputMember = Shapes::UnionShape.new(name: 'InvocationInputMember')
72
+ InvocationInputs = Shapes::ListShape.new(name: 'InvocationInputs')
73
+ InvocationResultMember = Shapes::UnionShape.new(name: 'InvocationResultMember')
48
74
  InvocationType = Shapes::StringShape.new(name: 'InvocationType')
49
75
  InvokeAgentRequest = Shapes::StructureShape.new(name: 'InvokeAgentRequest')
50
76
  InvokeAgentResponse = Shapes::StructureShape.new(name: 'InvokeAgentResponse')
@@ -71,6 +97,7 @@ module Aws::BedrockAgentRuntime
71
97
  OrchestrationTrace = Shapes::UnionShape.new(name: 'OrchestrationTrace')
72
98
  OutputString = Shapes::StringShape.new(name: 'OutputString')
73
99
  Parameter = Shapes::StructureShape.new(name: 'Parameter')
100
+ ParameterList = Shapes::ListShape.new(name: 'ParameterList')
74
101
  Parameters = Shapes::ListShape.new(name: 'Parameters')
75
102
  PartBody = Shapes::BlobShape.new(name: 'PartBody')
76
103
  PayloadPart = Shapes::StructureShape.new(name: 'PayloadPart')
@@ -84,11 +111,14 @@ module Aws::BedrockAgentRuntime
84
111
  PromptTemplate = Shapes::StructureShape.new(name: 'PromptTemplate')
85
112
  PromptText = Shapes::StringShape.new(name: 'PromptText')
86
113
  PromptType = Shapes::StringShape.new(name: 'PromptType')
114
+ PropertyParameters = Shapes::StructureShape.new(name: 'PropertyParameters')
87
115
  Rationale = Shapes::StructureShape.new(name: 'Rationale')
88
116
  RationaleString = Shapes::StringShape.new(name: 'RationaleString')
89
117
  RepromptResponse = Shapes::StructureShape.new(name: 'RepromptResponse')
90
118
  RequestBody = Shapes::StructureShape.new(name: 'RequestBody')
91
119
  ResourceNotFoundException = Shapes::StructureShape.new(name: 'ResourceNotFoundException')
120
+ ResponseBody = Shapes::MapShape.new(name: 'ResponseBody')
121
+ ResponseState = Shapes::StringShape.new(name: 'ResponseState')
92
122
  ResponseStream = Shapes::StructureShape.new(name: 'ResponseStream')
93
123
  RetrievalFilter = Shapes::UnionShape.new(name: 'RetrievalFilter')
94
124
  RetrievalFilterList = Shapes::ListShape.new(name: 'RetrievalFilterList')
@@ -111,6 +141,10 @@ module Aws::BedrockAgentRuntime
111
141
  RetrieveResponse = Shapes::StructureShape.new(name: 'RetrieveResponse')
112
142
  RetrievedReference = Shapes::StructureShape.new(name: 'RetrievedReference')
113
143
  RetrievedReferences = Shapes::ListShape.new(name: 'RetrievedReferences')
144
+ ReturnControlInvocationResults = Shapes::ListShape.new(name: 'ReturnControlInvocationResults')
145
+ ReturnControlPayload = Shapes::StructureShape.new(name: 'ReturnControlPayload')
146
+ S3ObjectDoc = Shapes::StructureShape.new(name: 'S3ObjectDoc')
147
+ S3Uri = Shapes::StringShape.new(name: 'S3Uri')
114
148
  SearchType = Shapes::StringShape.new(name: 'SearchType')
115
149
  ServiceQuotaExceededException = Shapes::StructureShape.new(name: 'ServiceQuotaExceededException')
116
150
  SessionAttributesMap = Shapes::MapShape.new(name: 'SessionAttributesMap')
@@ -141,6 +175,7 @@ module Aws::BedrockAgentRuntime
141
175
 
142
176
  ActionGroupInvocationInput.add_member(:action_group_name, Shapes::ShapeRef.new(shape: ActionGroupName, location_name: "actionGroupName"))
143
177
  ActionGroupInvocationInput.add_member(:api_path, Shapes::ShapeRef.new(shape: ApiPath, location_name: "apiPath"))
178
+ ActionGroupInvocationInput.add_member(:function, Shapes::ShapeRef.new(shape: Function, location_name: "function"))
144
179
  ActionGroupInvocationInput.add_member(:parameters, Shapes::ShapeRef.new(shape: Parameters, location_name: "parameters"))
145
180
  ActionGroupInvocationInput.add_member(:request_body, Shapes::ShapeRef.new(shape: RequestBody, location_name: "requestBody"))
146
181
  ActionGroupInvocationInput.add_member(:verb, Shapes::ShapeRef.new(shape: Verb, location_name: "verb"))
@@ -149,6 +184,34 @@ module Aws::BedrockAgentRuntime
149
184
  ActionGroupInvocationOutput.add_member(:text, Shapes::ShapeRef.new(shape: ActionGroupOutputString, location_name: "text"))
150
185
  ActionGroupInvocationOutput.struct_class = Types::ActionGroupInvocationOutput
151
186
 
187
+ ApiContentMap.key = Shapes::ShapeRef.new(shape: String)
188
+ ApiContentMap.value = Shapes::ShapeRef.new(shape: PropertyParameters)
189
+
190
+ ApiInvocationInput.add_member(:action_group, Shapes::ShapeRef.new(shape: String, required: true, location_name: "actionGroup"))
191
+ ApiInvocationInput.add_member(:api_path, Shapes::ShapeRef.new(shape: ApiPath, location_name: "apiPath"))
192
+ ApiInvocationInput.add_member(:http_method, Shapes::ShapeRef.new(shape: String, location_name: "httpMethod"))
193
+ ApiInvocationInput.add_member(:parameters, Shapes::ShapeRef.new(shape: ApiParameters, location_name: "parameters"))
194
+ ApiInvocationInput.add_member(:request_body, Shapes::ShapeRef.new(shape: ApiRequestBody, location_name: "requestBody"))
195
+ ApiInvocationInput.struct_class = Types::ApiInvocationInput
196
+
197
+ ApiParameter.add_member(:name, Shapes::ShapeRef.new(shape: String, location_name: "name"))
198
+ ApiParameter.add_member(:type, Shapes::ShapeRef.new(shape: String, location_name: "type"))
199
+ ApiParameter.add_member(:value, Shapes::ShapeRef.new(shape: String, location_name: "value"))
200
+ ApiParameter.struct_class = Types::ApiParameter
201
+
202
+ ApiParameters.member = Shapes::ShapeRef.new(shape: ApiParameter)
203
+
204
+ ApiRequestBody.add_member(:content, Shapes::ShapeRef.new(shape: ApiContentMap, location_name: "content"))
205
+ ApiRequestBody.struct_class = Types::ApiRequestBody
206
+
207
+ ApiResult.add_member(:action_group, Shapes::ShapeRef.new(shape: String, required: true, location_name: "actionGroup"))
208
+ ApiResult.add_member(:api_path, Shapes::ShapeRef.new(shape: ApiPath, location_name: "apiPath"))
209
+ ApiResult.add_member(:http_method, Shapes::ShapeRef.new(shape: String, location_name: "httpMethod"))
210
+ ApiResult.add_member(:http_status_code, Shapes::ShapeRef.new(shape: Integer, location_name: "httpStatusCode"))
211
+ ApiResult.add_member(:response_body, Shapes::ShapeRef.new(shape: ResponseBody, location_name: "responseBody"))
212
+ ApiResult.add_member(:response_state, Shapes::ShapeRef.new(shape: ResponseState, location_name: "responseState"))
213
+ ApiResult.struct_class = Types::ApiResult
214
+
152
215
  Attribution.add_member(:citations, Shapes::ShapeRef.new(shape: Citations, location_name: "citations"))
153
216
  Attribution.struct_class = Types::Attribution
154
217
 
@@ -156,6 +219,11 @@ module Aws::BedrockAgentRuntime
156
219
  BadGatewayException.add_member(:resource_name, Shapes::ShapeRef.new(shape: NonBlankString, location_name: "resourceName"))
157
220
  BadGatewayException.struct_class = Types::BadGatewayException
158
221
 
222
+ ByteContentDoc.add_member(:content_type, Shapes::ShapeRef.new(shape: ContentType, required: true, location_name: "contentType"))
223
+ ByteContentDoc.add_member(:data, Shapes::ShapeRef.new(shape: ByteContentBlob, required: true, location_name: "data"))
224
+ ByteContentDoc.add_member(:identifier, Shapes::ShapeRef.new(shape: Identifier, required: true, location_name: "identifier"))
225
+ ByteContentDoc.struct_class = Types::ByteContentDoc
226
+
159
227
  Citation.add_member(:generated_response_part, Shapes::ShapeRef.new(shape: GeneratedResponsePart, location_name: "generatedResponsePart"))
160
228
  Citation.add_member(:retrieved_references, Shapes::ShapeRef.new(shape: RetrievedReferences, location_name: "retrievedReferences"))
161
229
  Citation.struct_class = Types::Citation
@@ -165,6 +233,9 @@ module Aws::BedrockAgentRuntime
165
233
  ConflictException.add_member(:message, Shapes::ShapeRef.new(shape: NonBlankString, location_name: "message"))
166
234
  ConflictException.struct_class = Types::ConflictException
167
235
 
236
+ ContentBody.add_member(:body, Shapes::ShapeRef.new(shape: String, location_name: "body"))
237
+ ContentBody.struct_class = Types::ContentBody
238
+
168
239
  ContentMap.key = Shapes::ShapeRef.new(shape: String)
169
240
  ContentMap.value = Shapes::ShapeRef.new(shape: Parameters)
170
241
 
@@ -172,6 +243,21 @@ module Aws::BedrockAgentRuntime
172
243
  DependencyFailedException.add_member(:resource_name, Shapes::ShapeRef.new(shape: NonBlankString, location_name: "resourceName"))
173
244
  DependencyFailedException.struct_class = Types::DependencyFailedException
174
245
 
246
+ ExternalSource.add_member(:byte_content, Shapes::ShapeRef.new(shape: ByteContentDoc, location_name: "byteContent"))
247
+ ExternalSource.add_member(:s3_location, Shapes::ShapeRef.new(shape: S3ObjectDoc, location_name: "s3Location"))
248
+ ExternalSource.add_member(:source_type, Shapes::ShapeRef.new(shape: ExternalSourceType, required: true, location_name: "sourceType"))
249
+ ExternalSource.struct_class = Types::ExternalSource
250
+
251
+ ExternalSources.member = Shapes::ShapeRef.new(shape: ExternalSource)
252
+
253
+ ExternalSourcesGenerationConfiguration.add_member(:prompt_template, Shapes::ShapeRef.new(shape: PromptTemplate, location_name: "promptTemplate"))
254
+ ExternalSourcesGenerationConfiguration.struct_class = Types::ExternalSourcesGenerationConfiguration
255
+
256
+ ExternalSourcesRetrieveAndGenerateConfiguration.add_member(:generation_configuration, Shapes::ShapeRef.new(shape: ExternalSourcesGenerationConfiguration, location_name: "generationConfiguration"))
257
+ ExternalSourcesRetrieveAndGenerateConfiguration.add_member(:model_arn, Shapes::ShapeRef.new(shape: BedrockModelArn, required: true, location_name: "modelArn"))
258
+ ExternalSourcesRetrieveAndGenerateConfiguration.add_member(:sources, Shapes::ShapeRef.new(shape: ExternalSources, required: true, location_name: "sources"))
259
+ ExternalSourcesRetrieveAndGenerateConfiguration.struct_class = Types::ExternalSourcesRetrieveAndGenerateConfiguration
260
+
175
261
  FailureTrace.add_member(:failure_reason, Shapes::ShapeRef.new(shape: FailureReasonString, location_name: "failureReason"))
176
262
  FailureTrace.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))
177
263
  FailureTrace.struct_class = Types::FailureTrace
@@ -183,6 +269,24 @@ module Aws::BedrockAgentRuntime
183
269
  FinalResponse.add_member(:text, Shapes::ShapeRef.new(shape: FinalResponseString, location_name: "text"))
184
270
  FinalResponse.struct_class = Types::FinalResponse
185
271
 
272
+ FunctionInvocationInput.add_member(:action_group, Shapes::ShapeRef.new(shape: String, required: true, location_name: "actionGroup"))
273
+ FunctionInvocationInput.add_member(:function, Shapes::ShapeRef.new(shape: String, location_name: "function"))
274
+ FunctionInvocationInput.add_member(:parameters, Shapes::ShapeRef.new(shape: FunctionParameters, location_name: "parameters"))
275
+ FunctionInvocationInput.struct_class = Types::FunctionInvocationInput
276
+
277
+ FunctionParameter.add_member(:name, Shapes::ShapeRef.new(shape: String, location_name: "name"))
278
+ FunctionParameter.add_member(:type, Shapes::ShapeRef.new(shape: String, location_name: "type"))
279
+ FunctionParameter.add_member(:value, Shapes::ShapeRef.new(shape: String, location_name: "value"))
280
+ FunctionParameter.struct_class = Types::FunctionParameter
281
+
282
+ FunctionParameters.member = Shapes::ShapeRef.new(shape: FunctionParameter)
283
+
284
+ FunctionResult.add_member(:action_group, Shapes::ShapeRef.new(shape: String, required: true, location_name: "actionGroup"))
285
+ FunctionResult.add_member(:function, Shapes::ShapeRef.new(shape: String, location_name: "function"))
286
+ FunctionResult.add_member(:response_body, Shapes::ShapeRef.new(shape: ResponseBody, location_name: "responseBody"))
287
+ FunctionResult.add_member(:response_state, Shapes::ShapeRef.new(shape: ResponseState, location_name: "responseState"))
288
+ FunctionResult.struct_class = Types::FunctionResult
289
+
186
290
  GeneratedResponsePart.add_member(:text_response_part, Shapes::ShapeRef.new(shape: TextResponsePart, location_name: "textResponsePart"))
187
291
  GeneratedResponsePart.struct_class = Types::GeneratedResponsePart
188
292
 
@@ -205,11 +309,29 @@ module Aws::BedrockAgentRuntime
205
309
  InvocationInput.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))
206
310
  InvocationInput.struct_class = Types::InvocationInput
207
311
 
312
+ InvocationInputMember.add_member(:api_invocation_input, Shapes::ShapeRef.new(shape: ApiInvocationInput, location_name: "apiInvocationInput"))
313
+ InvocationInputMember.add_member(:function_invocation_input, Shapes::ShapeRef.new(shape: FunctionInvocationInput, location_name: "functionInvocationInput"))
314
+ InvocationInputMember.add_member(:unknown, Shapes::ShapeRef.new(shape: nil, location_name: 'unknown'))
315
+ InvocationInputMember.add_member_subclass(:api_invocation_input, Types::InvocationInputMember::ApiInvocationInput)
316
+ InvocationInputMember.add_member_subclass(:function_invocation_input, Types::InvocationInputMember::FunctionInvocationInput)
317
+ InvocationInputMember.add_member_subclass(:unknown, Types::InvocationInputMember::Unknown)
318
+ InvocationInputMember.struct_class = Types::InvocationInputMember
319
+
320
+ InvocationInputs.member = Shapes::ShapeRef.new(shape: InvocationInputMember)
321
+
322
+ InvocationResultMember.add_member(:api_result, Shapes::ShapeRef.new(shape: ApiResult, location_name: "apiResult"))
323
+ InvocationResultMember.add_member(:function_result, Shapes::ShapeRef.new(shape: FunctionResult, location_name: "functionResult"))
324
+ InvocationResultMember.add_member(:unknown, Shapes::ShapeRef.new(shape: nil, location_name: 'unknown'))
325
+ InvocationResultMember.add_member_subclass(:api_result, Types::InvocationResultMember::ApiResult)
326
+ InvocationResultMember.add_member_subclass(:function_result, Types::InvocationResultMember::FunctionResult)
327
+ InvocationResultMember.add_member_subclass(:unknown, Types::InvocationResultMember::Unknown)
328
+ InvocationResultMember.struct_class = Types::InvocationResultMember
329
+
208
330
  InvokeAgentRequest.add_member(:agent_alias_id, Shapes::ShapeRef.new(shape: AgentAliasId, required: true, location: "uri", location_name: "agentAliasId"))
209
331
  InvokeAgentRequest.add_member(:agent_id, Shapes::ShapeRef.new(shape: AgentId, required: true, location: "uri", location_name: "agentId"))
210
332
  InvokeAgentRequest.add_member(:enable_trace, Shapes::ShapeRef.new(shape: Boolean, location_name: "enableTrace"))
211
333
  InvokeAgentRequest.add_member(:end_session, Shapes::ShapeRef.new(shape: Boolean, location_name: "endSession"))
212
- InvokeAgentRequest.add_member(:input_text, Shapes::ShapeRef.new(shape: InputText, required: true, location_name: "inputText"))
334
+ InvokeAgentRequest.add_member(:input_text, Shapes::ShapeRef.new(shape: InputText, location_name: "inputText"))
213
335
  InvokeAgentRequest.add_member(:session_id, Shapes::ShapeRef.new(shape: SessionId, required: true, location: "uri", location_name: "sessionId"))
214
336
  InvokeAgentRequest.add_member(:session_state, Shapes::ShapeRef.new(shape: SessionState, location_name: "sessionState"))
215
337
  InvokeAgentRequest.struct_class = Types::InvokeAgentRequest
@@ -287,6 +409,8 @@ module Aws::BedrockAgentRuntime
287
409
  Parameter.add_member(:value, Shapes::ShapeRef.new(shape: String, location_name: "value"))
288
410
  Parameter.struct_class = Types::Parameter
289
411
 
412
+ ParameterList.member = Shapes::ShapeRef.new(shape: Parameter)
413
+
290
414
  Parameters.member = Shapes::ShapeRef.new(shape: Parameter)
291
415
 
292
416
  PayloadPart.add_member(:attribution, Shapes::ShapeRef.new(shape: Attribution, location_name: "attribution"))
@@ -330,6 +454,9 @@ module Aws::BedrockAgentRuntime
330
454
  PromptTemplate.add_member(:text_prompt_template, Shapes::ShapeRef.new(shape: TextPromptTemplate, location_name: "textPromptTemplate"))
331
455
  PromptTemplate.struct_class = Types::PromptTemplate
332
456
 
457
+ PropertyParameters.add_member(:properties, Shapes::ShapeRef.new(shape: ParameterList, location_name: "properties"))
458
+ PropertyParameters.struct_class = Types::PropertyParameters
459
+
333
460
  Rationale.add_member(:text, Shapes::ShapeRef.new(shape: RationaleString, location_name: "text"))
334
461
  Rationale.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))
335
462
  Rationale.struct_class = Types::Rationale
@@ -344,6 +471,9 @@ module Aws::BedrockAgentRuntime
344
471
  ResourceNotFoundException.add_member(:message, Shapes::ShapeRef.new(shape: NonBlankString, location_name: "message"))
345
472
  ResourceNotFoundException.struct_class = Types::ResourceNotFoundException
346
473
 
474
+ ResponseBody.key = Shapes::ShapeRef.new(shape: String)
475
+ ResponseBody.value = Shapes::ShapeRef.new(shape: ContentBody)
476
+
347
477
  ResponseStream.add_member(:access_denied_exception, Shapes::ShapeRef.new(shape: AccessDeniedException, location_name: "accessDeniedException"))
348
478
  ResponseStream.add_member(:bad_gateway_exception, Shapes::ShapeRef.new(shape: BadGatewayException, location_name: "badGatewayException"))
349
479
  ResponseStream.add_member(:chunk, Shapes::ShapeRef.new(shape: PayloadPart, event: true, location_name: "chunk"))
@@ -351,6 +481,7 @@ module Aws::BedrockAgentRuntime
351
481
  ResponseStream.add_member(:dependency_failed_exception, Shapes::ShapeRef.new(shape: DependencyFailedException, location_name: "dependencyFailedException"))
352
482
  ResponseStream.add_member(:internal_server_exception, Shapes::ShapeRef.new(shape: InternalServerException, location_name: "internalServerException"))
353
483
  ResponseStream.add_member(:resource_not_found_exception, Shapes::ShapeRef.new(shape: ResourceNotFoundException, location_name: "resourceNotFoundException"))
484
+ ResponseStream.add_member(:return_control, Shapes::ShapeRef.new(shape: ReturnControlPayload, event: true, location_name: "returnControl"))
354
485
  ResponseStream.add_member(:service_quota_exceeded_exception, Shapes::ShapeRef.new(shape: ServiceQuotaExceededException, location_name: "serviceQuotaExceededException"))
355
486
  ResponseStream.add_member(:throttling_exception, Shapes::ShapeRef.new(shape: ThrottlingException, location_name: "throttlingException"))
356
487
  ResponseStream.add_member(:trace, Shapes::ShapeRef.new(shape: TracePart, event: true, location_name: "trace"))
@@ -398,6 +529,7 @@ module Aws::BedrockAgentRuntime
398
529
  RetrievalResultS3Location.add_member(:uri, Shapes::ShapeRef.new(shape: String, location_name: "uri"))
399
530
  RetrievalResultS3Location.struct_class = Types::RetrievalResultS3Location
400
531
 
532
+ RetrieveAndGenerateConfiguration.add_member(:external_sources_configuration, Shapes::ShapeRef.new(shape: ExternalSourcesRetrieveAndGenerateConfiguration, location_name: "externalSourcesConfiguration"))
401
533
  RetrieveAndGenerateConfiguration.add_member(:knowledge_base_configuration, Shapes::ShapeRef.new(shape: KnowledgeBaseRetrieveAndGenerateConfiguration, location_name: "knowledgeBaseConfiguration"))
402
534
  RetrieveAndGenerateConfiguration.add_member(:type, Shapes::ShapeRef.new(shape: RetrieveAndGenerateType, required: true, location_name: "type"))
403
535
  RetrieveAndGenerateConfiguration.struct_class = Types::RetrieveAndGenerateConfiguration
@@ -439,13 +571,24 @@ module Aws::BedrockAgentRuntime
439
571
 
440
572
  RetrievedReferences.member = Shapes::ShapeRef.new(shape: RetrievedReference)
441
573
 
574
+ ReturnControlInvocationResults.member = Shapes::ShapeRef.new(shape: InvocationResultMember)
575
+
576
+ ReturnControlPayload.add_member(:invocation_id, Shapes::ShapeRef.new(shape: String, location_name: "invocationId"))
577
+ ReturnControlPayload.add_member(:invocation_inputs, Shapes::ShapeRef.new(shape: InvocationInputs, location_name: "invocationInputs"))
578
+ ReturnControlPayload.struct_class = Types::ReturnControlPayload
579
+
580
+ S3ObjectDoc.add_member(:uri, Shapes::ShapeRef.new(shape: S3Uri, required: true, location_name: "uri"))
581
+ S3ObjectDoc.struct_class = Types::S3ObjectDoc
582
+
442
583
  ServiceQuotaExceededException.add_member(:message, Shapes::ShapeRef.new(shape: NonBlankString, location_name: "message"))
443
584
  ServiceQuotaExceededException.struct_class = Types::ServiceQuotaExceededException
444
585
 
445
586
  SessionAttributesMap.key = Shapes::ShapeRef.new(shape: String)
446
587
  SessionAttributesMap.value = Shapes::ShapeRef.new(shape: String)
447
588
 
589
+ SessionState.add_member(:invocation_id, Shapes::ShapeRef.new(shape: String, location_name: "invocationId"))
448
590
  SessionState.add_member(:prompt_session_attributes, Shapes::ShapeRef.new(shape: PromptSessionAttributesMap, location_name: "promptSessionAttributes"))
591
+ SessionState.add_member(:return_control_invocation_results, Shapes::ShapeRef.new(shape: ReturnControlInvocationResults, location_name: "returnControlInvocationResults"))
449
592
  SessionState.add_member(:session_attributes, Shapes::ShapeRef.new(shape: SessionAttributesMap, location_name: "sessionAttributes"))
450
593
  SessionState.struct_class = Types::SessionState
451
594
 
@@ -476,6 +619,7 @@ module Aws::BedrockAgentRuntime
476
619
 
477
620
  TracePart.add_member(:agent_alias_id, Shapes::ShapeRef.new(shape: AgentAliasId, location_name: "agentAliasId"))
478
621
  TracePart.add_member(:agent_id, Shapes::ShapeRef.new(shape: AgentId, location_name: "agentId"))
622
+ TracePart.add_member(:agent_version, Shapes::ShapeRef.new(shape: AgentVersion, location_name: "agentVersion"))
479
623
  TracePart.add_member(:session_id, Shapes::ShapeRef.new(shape: SessionId, location_name: "sessionId"))
480
624
  TracePart.add_member(:trace, Shapes::ShapeRef.new(shape: Trace, location_name: "trace"))
481
625
  TracePart.struct_class = Types::TracePart
@@ -43,6 +43,10 @@ module Aws::BedrockAgentRuntime
43
43
  @event_emitter.on(:resource_not_found_exception, block) if block_given?
44
44
  end
45
45
 
46
+ def on_return_control_event(&block)
47
+ @event_emitter.on(:return_control, block) if block_given?
48
+ end
49
+
46
50
  def on_service_quota_exceeded_exception_event(&block)
47
51
  @event_emitter.on(:service_quota_exceeded_exception, block) if block_given?
48
52
  end
@@ -79,6 +83,7 @@ module Aws::BedrockAgentRuntime
79
83
  on_dependency_failed_exception_event(&block)
80
84
  on_internal_server_exception_event(&block)
81
85
  on_resource_not_found_exception_event(&block)
86
+ on_return_control_event(&block)
82
87
  on_service_quota_exceeded_exception_event(&block)
83
88
  on_throttling_exception_event(&block)
84
89
  on_trace_event(&block)