aws-sdk-bedrockagentruntime 1.5.0 → 1.6.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 24e10fc42b8ea0b4e35d44ca0c691949feaab6fe28ebe2a12c290ea005730c27
-  data.tar.gz: 7239ae58b7fc4a5343de154490f9c81eb7011fdd9a1be9472fc64575f6b4f803
+  metadata.gz: c4fadd76b1f390cc94a0a3234d392089d15516b65e24c52663bcc7b2e3e5d8db
+  data.tar.gz: 3c18e9019f15b5e7db08da1c15c566cce6c2b74e348c7e73df7afd2f26007dcb
 SHA512:
-  metadata.gz: 3f4d8ef61cdf0882cea1ab50df77212c91dda6f4a9457ad6632702c328f21fa20c1ff2dcff52544e935f747bb7b973c65f68ee40d77c63342c14f079599be7d7
-  data.tar.gz: 7b77085b8fe83e0f72e6276a82403b2a891515a51c3f2957860d516cb4b9657ea8be881250cda8abe01cd5417ef783b2ee499b8a84484bf72ca3f82f2e483ae6
+  metadata.gz: 56d4583de4da4c430fb9fdc74651c40e2c609505886ab67e5f101a6abff02d1c0408f542d924d6d1fcaeb6775146fab10b77fb9caba0df3f54587891d2d09ae6
+  data.tar.gz: f54537daf902f208c5df229c62472aa9b29d544a003b0cae55177a30eebb0465e35a87d8259a462c0e2f5c89c28413d350b7dfc05712b51b127f5323b4f48851

data/CHANGELOG.md

@@ -1,6 +1,11 @@
 Unreleased Changes
 ------------------
 
+1.6.0 (2024-04-23)
+------------------
+
+* Feature - This release introduces zero-setup file upload support for the RetrieveAndGenerate API. This allows you to chat with your data without setting up a Knowledge Base.
+
 1.5.0 (2024-03-27)
 ------------------
 

data/VERSION

@@ -1 +1 @@
-1.5.0
+1.6.0

data/lib/aws-sdk-bedrockagentruntime/client.rb

@@ -399,7 +399,8 @@ module Aws::BedrockAgentRuntime
 
     # @!group API Operations
 
-    # Sends a prompt for the agent to process and respond to.
+    # Sends a prompt for the agent to process and respond to. Use return
+    # control event type for function calling.
     #
     # <note markdown="1"> The CLI doesn't support `InvokeAgent`.
     #
@@ -416,8 +417,10 @@ module Aws::BedrockAgentRuntime
     #
     # * End a conversation by setting `endSession` to `true`.
     #
-    # * Include attributes for the session or prompt in the `sessionState`
-    #   object.
+    # * In the `sessionState` object, you can include attributes for the
+    #   session or prompt or parameters returned from the action group.
+    #
+    # * Use return control event type for function calling.
     #
     # The response is returned in the `bytes` field of the `chunk` object.
     #
@@ -450,7 +453,7 @@ module Aws::BedrockAgentRuntime
     # @option params [Boolean] :end_session
     #   Specifies whether to end the session with the agent or not.
     #
-    # @option params [required, String] :input_text
+    # @option params [String] :input_text
     #   The prompt text to send the agent.
     #
     # @option params [required, String] :session_id
@@ -535,6 +538,9 @@ module Aws::BedrockAgentRuntime
     #       handler.on_resource_not_found_exception_event do |event|
     #         event # => Aws::BedrockAgentRuntime::Types::resourceNotFoundException
     #       end
+    #       handler.on_return_control_event do |event|
+    #         event # => Aws::BedrockAgentRuntime::Types::returnControl
+    #       end
     #       handler.on_service_quota_exceeded_exception_event do |event|
     #         event # => Aws::BedrockAgentRuntime::Types::serviceQuotaExceededException
     #       end
@@ -575,6 +581,9 @@ module Aws::BedrockAgentRuntime
     #       stream.on_resource_not_found_exception_event do |event|
     #         event # => Aws::BedrockAgentRuntime::Types::resourceNotFoundException
     #       end
+    #       stream.on_return_control_event do |event|
+    #         event # => Aws::BedrockAgentRuntime::Types::returnControl
+    #       end
     #       stream.on_service_quota_exceeded_exception_event do |event|
     #         event # => Aws::BedrockAgentRuntime::Types::serviceQuotaExceededException
     #       end
@@ -615,6 +624,9 @@ module Aws::BedrockAgentRuntime
     #       handler.on_resource_not_found_exception_event do |event|
     #         event # => Aws::BedrockAgentRuntime::Types::resourceNotFoundException
     #       end
+    #       handler.on_return_control_event do |event|
+    #         event # => Aws::BedrockAgentRuntime::Types::returnControl
+    #       end
     #       handler.on_service_quota_exceeded_exception_event do |event|
     #         event # => Aws::BedrockAgentRuntime::Types::serviceQuotaExceededException
     #       end
@@ -652,12 +664,39 @@ module Aws::BedrockAgentRuntime
     #     agent_id: "AgentId", # required
     #     enable_trace: false,
     #     end_session: false,
-    #     input_text: "InputText", # required
+    #     input_text: "InputText",
     #     session_id: "SessionId", # required
     #     session_state: {
+    #       invocation_id: "String",
     #       prompt_session_attributes: {
     #         "String" => "String",
     #       },
+    #       return_control_invocation_results: [
+    #         {
+    #           api_result: {
+    #             action_group: "String", # required
+    #             api_path: "ApiPath",
+    #             http_method: "String",
+    #             http_status_code: 1,
+    #             response_body: {
+    #               "String" => {
+    #                 body: "String",
+    #               },
+    #             },
+    #             response_state: "FAILURE", # accepts FAILURE, REPROMPT
+    #           },
+    #           function_result: {
+    #             action_group: "String", # required
+    #             function: "String",
+    #             response_body: {
+    #               "String" => {
+    #                 body: "String",
+    #               },
+    #             },
+    #             response_state: "FAILURE", # accepts FAILURE, REPROMPT
+    #           },
+    #         },
+    #       ],
     #       session_attributes: {
     #         "String" => "String",
     #       },
@@ -668,7 +707,7 @@ module Aws::BedrockAgentRuntime
     #
     #   All events are available at resp.completion:
     #   resp.completion #=> Enumerator
-    #   resp.completion.event_types #=> [:access_denied_exception, :bad_gateway_exception, :chunk, :conflict_exception, :dependency_failed_exception, :internal_server_exception, :resource_not_found_exception, :service_quota_exceeded_exception, :throttling_exception, :trace, :validation_exception]
+    #   resp.completion.event_types #=> [:access_denied_exception, :bad_gateway_exception, :chunk, :conflict_exception, :dependency_failed_exception, :internal_server_exception, :resource_not_found_exception, :return_control, :service_quota_exceeded_exception, :throttling_exception, :trace, :validation_exception]
     #
     #   For :access_denied_exception event available at #on_access_denied_exception_event callback and response eventstream enumerator:
     #   event.message #=> String
@@ -702,6 +741,28 @@ module Aws::BedrockAgentRuntime
     #   For :resource_not_found_exception event available at #on_resource_not_found_exception_event callback and response eventstream enumerator:
     #   event.message #=> String
     #
+    #   For :return_control event available at #on_return_control_event callback and response eventstream enumerator:
+    #   event.invocation_id #=> String
+    #   event.invocation_inputs #=> Array
+    #   event.invocation_inputs[0].api_invocation_input.action_group #=> String
+    #   event.invocation_inputs[0].api_invocation_input.api_path #=> String
+    #   event.invocation_inputs[0].api_invocation_input.http_method #=> String
+    #   event.invocation_inputs[0].api_invocation_input.parameters #=> Array
+    #   event.invocation_inputs[0].api_invocation_input.parameters[0].name #=> String
+    #   event.invocation_inputs[0].api_invocation_input.parameters[0].type #=> String
+    #   event.invocation_inputs[0].api_invocation_input.parameters[0].value #=> String
+    #   event.invocation_inputs[0].api_invocation_input.request_body.content #=> Hash
+    #   event.invocation_inputs[0].api_invocation_input.request_body.content["String"].properties #=> Array
+    #   event.invocation_inputs[0].api_invocation_input.request_body.content["String"].properties[0].name #=> String
+    #   event.invocation_inputs[0].api_invocation_input.request_body.content["String"].properties[0].type #=> String
+    #   event.invocation_inputs[0].api_invocation_input.request_body.content["String"].properties[0].value #=> String
+    #   event.invocation_inputs[0].function_invocation_input.action_group #=> String
+    #   event.invocation_inputs[0].function_invocation_input.function #=> String
+    #   event.invocation_inputs[0].function_invocation_input.parameters #=> Array
+    #   event.invocation_inputs[0].function_invocation_input.parameters[0].name #=> String
+    #   event.invocation_inputs[0].function_invocation_input.parameters[0].type #=> String
+    #   event.invocation_inputs[0].function_invocation_input.parameters[0].value #=> String
+    #
     #   For :service_quota_exceeded_exception event available at #on_service_quota_exceeded_exception_event callback and response eventstream enumerator:
     #   event.message #=> String
     #
@@ -711,11 +772,13 @@ module Aws::BedrockAgentRuntime
     #   For :trace event available at #on_trace_event callback and response eventstream enumerator:
     #   event.agent_alias_id #=> String
     #   event.agent_id #=> String
+    #   event.agent_version #=> String
     #   event.session_id #=> String
     #   event.trace.failure_trace.failure_reason #=> String
     #   event.trace.failure_trace.trace_id #=> String
     #   event.trace.orchestration_trace.invocation_input.action_group_invocation_input.action_group_name #=> String
     #   event.trace.orchestration_trace.invocation_input.action_group_invocation_input.api_path #=> String
+    #   event.trace.orchestration_trace.invocation_input.action_group_invocation_input.function #=> String
     #   event.trace.orchestration_trace.invocation_input.action_group_invocation_input.parameters #=> Array
     #   event.trace.orchestration_trace.invocation_input.action_group_invocation_input.parameters[0].name #=> String
     #   event.trace.orchestration_trace.invocation_input.action_group_invocation_input.parameters[0].type #=> String
@@ -973,6 +1036,27 @@ module Aws::BedrockAgentRuntime
     #       text: "RetrieveAndGenerateInputTextString", # required
     #     },
     #     retrieve_and_generate_configuration: {
+    #       external_sources_configuration: {
+    #         generation_configuration: {
+    #           prompt_template: {
+    #             text_prompt_template: "TextPromptTemplate",
+    #           },
+    #         },
+    #         model_arn: "BedrockModelArn", # required
+    #         sources: [ # required
+    #           {
+    #             byte_content: {
+    #               content_type: "ContentType", # required
+    #               data: "data", # required
+    #               identifier: "Identifier", # required
+    #             },
+    #             s3_location: {
+    #               uri: "S3Uri", # required
+    #             },
+    #             source_type: "S3", # required, accepts S3, BYTE_CONTENT
+    #           },
+    #         ],
+    #       },
     #       knowledge_base_configuration: {
     #         generation_configuration: {
     #           prompt_template: {
@@ -1045,7 +1129,7 @@ module Aws::BedrockAgentRuntime
     #           },
     #         },
     #       },
-    #       type: "KNOWLEDGE_BASE", # required, accepts KNOWLEDGE_BASE
+    #       type: "KNOWLEDGE_BASE", # required, accepts KNOWLEDGE_BASE, EXTERNAL_SOURCES
     #     },
     #     session_configuration: {
     #       kms_key_arn: "KmsKeyArn", # required
@@ -1089,7 +1173,7 @@ module Aws::BedrockAgentRuntime
         params: params,
         config: config)
       context[:gem_name] = 'aws-sdk-bedrockagentruntime'
-      context[:gem_version] = '1.5.0'
+      context[:gem_version] = '1.6.0'
       Seahorse::Client::Request.new(handlers, context)
     end
 

data/lib/aws-sdk-bedrockagentruntime/client_api.rb

@@ -20,18 +20,34 @@ module Aws::BedrockAgentRuntime
     ActionGroupOutputString = Shapes::StringShape.new(name: 'ActionGroupOutputString')
     AgentAliasId = Shapes::StringShape.new(name: 'AgentAliasId')
     AgentId = Shapes::StringShape.new(name: 'AgentId')
+    AgentVersion = Shapes::StringShape.new(name: 'AgentVersion')
+    ApiContentMap = Shapes::MapShape.new(name: 'ApiContentMap')
+    ApiInvocationInput = Shapes::StructureShape.new(name: 'ApiInvocationInput')
+    ApiParameter = Shapes::StructureShape.new(name: 'ApiParameter')
+    ApiParameters = Shapes::ListShape.new(name: 'ApiParameters')
     ApiPath = Shapes::StringShape.new(name: 'ApiPath')
+    ApiRequestBody = Shapes::StructureShape.new(name: 'ApiRequestBody')
+    ApiResult = Shapes::StructureShape.new(name: 'ApiResult')
     Attribution = Shapes::StructureShape.new(name: 'Attribution')
     BadGatewayException = Shapes::StructureShape.new(name: 'BadGatewayException')
     BedrockModelArn = Shapes::StringShape.new(name: 'BedrockModelArn')
     Boolean = Shapes::BooleanShape.new(name: 'Boolean')
+    ByteContentBlob = Shapes::BlobShape.new(name: 'ByteContentBlob')
+    ByteContentDoc = Shapes::StructureShape.new(name: 'ByteContentDoc')
     Citation = Shapes::StructureShape.new(name: 'Citation')
     Citations = Shapes::ListShape.new(name: 'Citations')
     ConflictException = Shapes::StructureShape.new(name: 'ConflictException')
+    ContentBody = Shapes::StructureShape.new(name: 'ContentBody')
     ContentMap = Shapes::MapShape.new(name: 'ContentMap')
+    ContentType = Shapes::StringShape.new(name: 'ContentType')
     CreationMode = Shapes::StringShape.new(name: 'CreationMode')
     DependencyFailedException = Shapes::StructureShape.new(name: 'DependencyFailedException')
     Double = Shapes::FloatShape.new(name: 'Double')
+    ExternalSource = Shapes::StructureShape.new(name: 'ExternalSource')
+    ExternalSourceType = Shapes::StringShape.new(name: 'ExternalSourceType')
+    ExternalSources = Shapes::ListShape.new(name: 'ExternalSources')
+    ExternalSourcesGenerationConfiguration = Shapes::StructureShape.new(name: 'ExternalSourcesGenerationConfiguration')
+    ExternalSourcesRetrieveAndGenerateConfiguration = Shapes::StructureShape.new(name: 'ExternalSourcesRetrieveAndGenerateConfiguration')
     FailureReasonString = Shapes::StringShape.new(name: 'FailureReasonString')
     FailureTrace = Shapes::StructureShape.new(name: 'FailureTrace')
     FilterAttribute = Shapes::StructureShape.new(name: 'FilterAttribute')
@@ -39,12 +55,22 @@ module Aws::BedrockAgentRuntime
     FilterValue = Shapes::DocumentShape.new(name: 'FilterValue', document: true)
     FinalResponse = Shapes::StructureShape.new(name: 'FinalResponse')
     FinalResponseString = Shapes::StringShape.new(name: 'FinalResponseString')
+    Function = Shapes::StringShape.new(name: 'Function')
+    FunctionInvocationInput = Shapes::StructureShape.new(name: 'FunctionInvocationInput')
+    FunctionParameter = Shapes::StructureShape.new(name: 'FunctionParameter')
+    FunctionParameters = Shapes::ListShape.new(name: 'FunctionParameters')
+    FunctionResult = Shapes::StructureShape.new(name: 'FunctionResult')
     GeneratedResponsePart = Shapes::StructureShape.new(name: 'GeneratedResponsePart')
     GenerationConfiguration = Shapes::StructureShape.new(name: 'GenerationConfiguration')
+    Identifier = Shapes::StringShape.new(name: 'Identifier')
     InferenceConfiguration = Shapes::StructureShape.new(name: 'InferenceConfiguration')
     InputText = Shapes::StringShape.new(name: 'InputText')
+    Integer = Shapes::IntegerShape.new(name: 'Integer')
     InternalServerException = Shapes::StructureShape.new(name: 'InternalServerException')
     InvocationInput = Shapes::StructureShape.new(name: 'InvocationInput')
+    InvocationInputMember = Shapes::UnionShape.new(name: 'InvocationInputMember')
+    InvocationInputs = Shapes::ListShape.new(name: 'InvocationInputs')
+    InvocationResultMember = Shapes::UnionShape.new(name: 'InvocationResultMember')
     InvocationType = Shapes::StringShape.new(name: 'InvocationType')
     InvokeAgentRequest = Shapes::StructureShape.new(name: 'InvokeAgentRequest')
     InvokeAgentResponse = Shapes::StructureShape.new(name: 'InvokeAgentResponse')
@@ -71,6 +97,7 @@ module Aws::BedrockAgentRuntime
     OrchestrationTrace = Shapes::UnionShape.new(name: 'OrchestrationTrace')
     OutputString = Shapes::StringShape.new(name: 'OutputString')
     Parameter = Shapes::StructureShape.new(name: 'Parameter')
+    ParameterList = Shapes::ListShape.new(name: 'ParameterList')
     Parameters = Shapes::ListShape.new(name: 'Parameters')
     PartBody = Shapes::BlobShape.new(name: 'PartBody')
     PayloadPart = Shapes::StructureShape.new(name: 'PayloadPart')
@@ -84,11 +111,14 @@ module Aws::BedrockAgentRuntime
     PromptTemplate = Shapes::StructureShape.new(name: 'PromptTemplate')
     PromptText = Shapes::StringShape.new(name: 'PromptText')
     PromptType = Shapes::StringShape.new(name: 'PromptType')
+    PropertyParameters = Shapes::StructureShape.new(name: 'PropertyParameters')
     Rationale = Shapes::StructureShape.new(name: 'Rationale')
     RationaleString = Shapes::StringShape.new(name: 'RationaleString')
     RepromptResponse = Shapes::StructureShape.new(name: 'RepromptResponse')
     RequestBody = Shapes::StructureShape.new(name: 'RequestBody')
     ResourceNotFoundException = Shapes::StructureShape.new(name: 'ResourceNotFoundException')
+    ResponseBody = Shapes::MapShape.new(name: 'ResponseBody')
+    ResponseState = Shapes::StringShape.new(name: 'ResponseState')
     ResponseStream = Shapes::StructureShape.new(name: 'ResponseStream')
     RetrievalFilter = Shapes::UnionShape.new(name: 'RetrievalFilter')
     RetrievalFilterList = Shapes::ListShape.new(name: 'RetrievalFilterList')
@@ -111,6 +141,10 @@ module Aws::BedrockAgentRuntime
     RetrieveResponse = Shapes::StructureShape.new(name: 'RetrieveResponse')
     RetrievedReference = Shapes::StructureShape.new(name: 'RetrievedReference')
     RetrievedReferences = Shapes::ListShape.new(name: 'RetrievedReferences')
+    ReturnControlInvocationResults = Shapes::ListShape.new(name: 'ReturnControlInvocationResults')
+    ReturnControlPayload = Shapes::StructureShape.new(name: 'ReturnControlPayload')
+    S3ObjectDoc = Shapes::StructureShape.new(name: 'S3ObjectDoc')
+    S3Uri = Shapes::StringShape.new(name: 'S3Uri')
     SearchType = Shapes::StringShape.new(name: 'SearchType')
     ServiceQuotaExceededException = Shapes::StructureShape.new(name: 'ServiceQuotaExceededException')
     SessionAttributesMap = Shapes::MapShape.new(name: 'SessionAttributesMap')
@@ -141,6 +175,7 @@ module Aws::BedrockAgentRuntime
 
     ActionGroupInvocationInput.add_member(:action_group_name, Shapes::ShapeRef.new(shape: ActionGroupName, location_name: "actionGroupName"))
     ActionGroupInvocationInput.add_member(:api_path, Shapes::ShapeRef.new(shape: ApiPath, location_name: "apiPath"))
+    ActionGroupInvocationInput.add_member(:function, Shapes::ShapeRef.new(shape: Function, location_name: "function"))
     ActionGroupInvocationInput.add_member(:parameters, Shapes::ShapeRef.new(shape: Parameters, location_name: "parameters"))
     ActionGroupInvocationInput.add_member(:request_body, Shapes::ShapeRef.new(shape: RequestBody, location_name: "requestBody"))
     ActionGroupInvocationInput.add_member(:verb, Shapes::ShapeRef.new(shape: Verb, location_name: "verb"))
@@ -149,6 +184,34 @@ module Aws::BedrockAgentRuntime
     ActionGroupInvocationOutput.add_member(:text, Shapes::ShapeRef.new(shape: ActionGroupOutputString, location_name: "text"))
     ActionGroupInvocationOutput.struct_class = Types::ActionGroupInvocationOutput
 
+    ApiContentMap.key = Shapes::ShapeRef.new(shape: String)
+    ApiContentMap.value = Shapes::ShapeRef.new(shape: PropertyParameters)
+
+    ApiInvocationInput.add_member(:action_group, Shapes::ShapeRef.new(shape: String, required: true, location_name: "actionGroup"))
+    ApiInvocationInput.add_member(:api_path, Shapes::ShapeRef.new(shape: ApiPath, location_name: "apiPath"))
+    ApiInvocationInput.add_member(:http_method, Shapes::ShapeRef.new(shape: String, location_name: "httpMethod"))
+    ApiInvocationInput.add_member(:parameters, Shapes::ShapeRef.new(shape: ApiParameters, location_name: "parameters"))
+    ApiInvocationInput.add_member(:request_body, Shapes::ShapeRef.new(shape: ApiRequestBody, location_name: "requestBody"))
+    ApiInvocationInput.struct_class = Types::ApiInvocationInput
+
+    ApiParameter.add_member(:name, Shapes::ShapeRef.new(shape: String, location_name: "name"))
+    ApiParameter.add_member(:type, Shapes::ShapeRef.new(shape: String, location_name: "type"))
+    ApiParameter.add_member(:value, Shapes::ShapeRef.new(shape: String, location_name: "value"))
+    ApiParameter.struct_class = Types::ApiParameter
+
+    ApiParameters.member = Shapes::ShapeRef.new(shape: ApiParameter)
+
+    ApiRequestBody.add_member(:content, Shapes::ShapeRef.new(shape: ApiContentMap, location_name: "content"))
+    ApiRequestBody.struct_class = Types::ApiRequestBody
+
+    ApiResult.add_member(:action_group, Shapes::ShapeRef.new(shape: String, required: true, location_name: "actionGroup"))
+    ApiResult.add_member(:api_path, Shapes::ShapeRef.new(shape: ApiPath, location_name: "apiPath"))
+    ApiResult.add_member(:http_method, Shapes::ShapeRef.new(shape: String, location_name: "httpMethod"))
+    ApiResult.add_member(:http_status_code, Shapes::ShapeRef.new(shape: Integer, location_name: "httpStatusCode"))
+    ApiResult.add_member(:response_body, Shapes::ShapeRef.new(shape: ResponseBody, location_name: "responseBody"))
+    ApiResult.add_member(:response_state, Shapes::ShapeRef.new(shape: ResponseState, location_name: "responseState"))
+    ApiResult.struct_class = Types::ApiResult
+
     Attribution.add_member(:citations, Shapes::ShapeRef.new(shape: Citations, location_name: "citations"))
     Attribution.struct_class = Types::Attribution
 
@@ -156,6 +219,11 @@ module Aws::BedrockAgentRuntime
     BadGatewayException.add_member(:resource_name, Shapes::ShapeRef.new(shape: NonBlankString, location_name: "resourceName"))
     BadGatewayException.struct_class = Types::BadGatewayException
 
+    ByteContentDoc.add_member(:content_type, Shapes::ShapeRef.new(shape: ContentType, required: true, location_name: "contentType"))
+    ByteContentDoc.add_member(:data, Shapes::ShapeRef.new(shape: ByteContentBlob, required: true, location_name: "data"))
+    ByteContentDoc.add_member(:identifier, Shapes::ShapeRef.new(shape: Identifier, required: true, location_name: "identifier"))
+    ByteContentDoc.struct_class = Types::ByteContentDoc
+
     Citation.add_member(:generated_response_part, Shapes::ShapeRef.new(shape: GeneratedResponsePart, location_name: "generatedResponsePart"))
     Citation.add_member(:retrieved_references, Shapes::ShapeRef.new(shape: RetrievedReferences, location_name: "retrievedReferences"))
     Citation.struct_class = Types::Citation
@@ -165,6 +233,9 @@ module Aws::BedrockAgentRuntime
     ConflictException.add_member(:message, Shapes::ShapeRef.new(shape: NonBlankString, location_name: "message"))
     ConflictException.struct_class = Types::ConflictException
 
+    ContentBody.add_member(:body, Shapes::ShapeRef.new(shape: String, location_name: "body"))
+    ContentBody.struct_class = Types::ContentBody
+
     ContentMap.key = Shapes::ShapeRef.new(shape: String)
     ContentMap.value = Shapes::ShapeRef.new(shape: Parameters)
 
@@ -172,6 +243,21 @@ module Aws::BedrockAgentRuntime
     DependencyFailedException.add_member(:resource_name, Shapes::ShapeRef.new(shape: NonBlankString, location_name: "resourceName"))
     DependencyFailedException.struct_class = Types::DependencyFailedException
 
+    ExternalSource.add_member(:byte_content, Shapes::ShapeRef.new(shape: ByteContentDoc, location_name: "byteContent"))
+    ExternalSource.add_member(:s3_location, Shapes::ShapeRef.new(shape: S3ObjectDoc, location_name: "s3Location"))
+    ExternalSource.add_member(:source_type, Shapes::ShapeRef.new(shape: ExternalSourceType, required: true, location_name: "sourceType"))
+    ExternalSource.struct_class = Types::ExternalSource
+
+    ExternalSources.member = Shapes::ShapeRef.new(shape: ExternalSource)
+
+    ExternalSourcesGenerationConfiguration.add_member(:prompt_template, Shapes::ShapeRef.new(shape: PromptTemplate, location_name: "promptTemplate"))
+    ExternalSourcesGenerationConfiguration.struct_class = Types::ExternalSourcesGenerationConfiguration
+
+    ExternalSourcesRetrieveAndGenerateConfiguration.add_member(:generation_configuration, Shapes::ShapeRef.new(shape: ExternalSourcesGenerationConfiguration, location_name: "generationConfiguration"))
+    ExternalSourcesRetrieveAndGenerateConfiguration.add_member(:model_arn, Shapes::ShapeRef.new(shape: BedrockModelArn, required: true, location_name: "modelArn"))
+    ExternalSourcesRetrieveAndGenerateConfiguration.add_member(:sources, Shapes::ShapeRef.new(shape: ExternalSources, required: true, location_name: "sources"))
+    ExternalSourcesRetrieveAndGenerateConfiguration.struct_class = Types::ExternalSourcesRetrieveAndGenerateConfiguration
+
     FailureTrace.add_member(:failure_reason, Shapes::ShapeRef.new(shape: FailureReasonString, location_name: "failureReason"))
     FailureTrace.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))
     FailureTrace.struct_class = Types::FailureTrace
@@ -183,6 +269,24 @@ module Aws::BedrockAgentRuntime
     FinalResponse.add_member(:text, Shapes::ShapeRef.new(shape: FinalResponseString, location_name: "text"))
     FinalResponse.struct_class = Types::FinalResponse
 
+    FunctionInvocationInput.add_member(:action_group, Shapes::ShapeRef.new(shape: String, required: true, location_name: "actionGroup"))
+    FunctionInvocationInput.add_member(:function, Shapes::ShapeRef.new(shape: String, location_name: "function"))
+    FunctionInvocationInput.add_member(:parameters, Shapes::ShapeRef.new(shape: FunctionParameters, location_name: "parameters"))
+    FunctionInvocationInput.struct_class = Types::FunctionInvocationInput
+
+    FunctionParameter.add_member(:name, Shapes::ShapeRef.new(shape: String, location_name: "name"))
+    FunctionParameter.add_member(:type, Shapes::ShapeRef.new(shape: String, location_name: "type"))
+    FunctionParameter.add_member(:value, Shapes::ShapeRef.new(shape: String, location_name: "value"))
+    FunctionParameter.struct_class = Types::FunctionParameter
+
+    FunctionParameters.member = Shapes::ShapeRef.new(shape: FunctionParameter)
+
+    FunctionResult.add_member(:action_group, Shapes::ShapeRef.new(shape: String, required: true, location_name: "actionGroup"))
+    FunctionResult.add_member(:function, Shapes::ShapeRef.new(shape: String, location_name: "function"))
+    FunctionResult.add_member(:response_body, Shapes::ShapeRef.new(shape: ResponseBody, location_name: "responseBody"))
+    FunctionResult.add_member(:response_state, Shapes::ShapeRef.new(shape: ResponseState, location_name: "responseState"))
+    FunctionResult.struct_class = Types::FunctionResult
+
     GeneratedResponsePart.add_member(:text_response_part, Shapes::ShapeRef.new(shape: TextResponsePart, location_name: "textResponsePart"))
     GeneratedResponsePart.struct_class = Types::GeneratedResponsePart
 
@@ -205,11 +309,29 @@ module Aws::BedrockAgentRuntime
     InvocationInput.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))
     InvocationInput.struct_class = Types::InvocationInput
 
+    InvocationInputMember.add_member(:api_invocation_input, Shapes::ShapeRef.new(shape: ApiInvocationInput, location_name: "apiInvocationInput"))
+    InvocationInputMember.add_member(:function_invocation_input, Shapes::ShapeRef.new(shape: FunctionInvocationInput, location_name: "functionInvocationInput"))
+    InvocationInputMember.add_member(:unknown, Shapes::ShapeRef.new(shape: nil, location_name: 'unknown'))
+    InvocationInputMember.add_member_subclass(:api_invocation_input, Types::InvocationInputMember::ApiInvocationInput)
+    InvocationInputMember.add_member_subclass(:function_invocation_input, Types::InvocationInputMember::FunctionInvocationInput)
+    InvocationInputMember.add_member_subclass(:unknown, Types::InvocationInputMember::Unknown)
+    InvocationInputMember.struct_class = Types::InvocationInputMember
+
+    InvocationInputs.member = Shapes::ShapeRef.new(shape: InvocationInputMember)
+
+    InvocationResultMember.add_member(:api_result, Shapes::ShapeRef.new(shape: ApiResult, location_name: "apiResult"))
+    InvocationResultMember.add_member(:function_result, Shapes::ShapeRef.new(shape: FunctionResult, location_name: "functionResult"))
+    InvocationResultMember.add_member(:unknown, Shapes::ShapeRef.new(shape: nil, location_name: 'unknown'))
+    InvocationResultMember.add_member_subclass(:api_result, Types::InvocationResultMember::ApiResult)
+    InvocationResultMember.add_member_subclass(:function_result, Types::InvocationResultMember::FunctionResult)
+    InvocationResultMember.add_member_subclass(:unknown, Types::InvocationResultMember::Unknown)
+    InvocationResultMember.struct_class = Types::InvocationResultMember
+
     InvokeAgentRequest.add_member(:agent_alias_id, Shapes::ShapeRef.new(shape: AgentAliasId, required: true, location: "uri", location_name: "agentAliasId"))
     InvokeAgentRequest.add_member(:agent_id, Shapes::ShapeRef.new(shape: AgentId, required: true, location: "uri", location_name: "agentId"))
     InvokeAgentRequest.add_member(:enable_trace, Shapes::ShapeRef.new(shape: Boolean, location_name: "enableTrace"))
     InvokeAgentRequest.add_member(:end_session, Shapes::ShapeRef.new(shape: Boolean, location_name: "endSession"))
-    InvokeAgentRequest.add_member(:input_text, Shapes::ShapeRef.new(shape: InputText, required: true, location_name: "inputText"))
+    InvokeAgentRequest.add_member(:input_text, Shapes::ShapeRef.new(shape: InputText, location_name: "inputText"))
     InvokeAgentRequest.add_member(:session_id, Shapes::ShapeRef.new(shape: SessionId, required: true, location: "uri", location_name: "sessionId"))
     InvokeAgentRequest.add_member(:session_state, Shapes::ShapeRef.new(shape: SessionState, location_name: "sessionState"))
     InvokeAgentRequest.struct_class = Types::InvokeAgentRequest
@@ -287,6 +409,8 @@ module Aws::BedrockAgentRuntime
     Parameter.add_member(:value, Shapes::ShapeRef.new(shape: String, location_name: "value"))
     Parameter.struct_class = Types::Parameter
 
+    ParameterList.member = Shapes::ShapeRef.new(shape: Parameter)
+
     Parameters.member = Shapes::ShapeRef.new(shape: Parameter)
 
     PayloadPart.add_member(:attribution, Shapes::ShapeRef.new(shape: Attribution, location_name: "attribution"))
@@ -330,6 +454,9 @@ module Aws::BedrockAgentRuntime
     PromptTemplate.add_member(:text_prompt_template, Shapes::ShapeRef.new(shape: TextPromptTemplate, location_name: "textPromptTemplate"))
     PromptTemplate.struct_class = Types::PromptTemplate
 
+    PropertyParameters.add_member(:properties, Shapes::ShapeRef.new(shape: ParameterList, location_name: "properties"))
+    PropertyParameters.struct_class = Types::PropertyParameters
+
     Rationale.add_member(:text, Shapes::ShapeRef.new(shape: RationaleString, location_name: "text"))
     Rationale.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))
     Rationale.struct_class = Types::Rationale
@@ -344,6 +471,9 @@ module Aws::BedrockAgentRuntime
     ResourceNotFoundException.add_member(:message, Shapes::ShapeRef.new(shape: NonBlankString, location_name: "message"))
     ResourceNotFoundException.struct_class = Types::ResourceNotFoundException
 
+    ResponseBody.key = Shapes::ShapeRef.new(shape: String)
+    ResponseBody.value = Shapes::ShapeRef.new(shape: ContentBody)
+
     ResponseStream.add_member(:access_denied_exception, Shapes::ShapeRef.new(shape: AccessDeniedException, location_name: "accessDeniedException"))
     ResponseStream.add_member(:bad_gateway_exception, Shapes::ShapeRef.new(shape: BadGatewayException, location_name: "badGatewayException"))
     ResponseStream.add_member(:chunk, Shapes::ShapeRef.new(shape: PayloadPart, event: true, location_name: "chunk"))
@@ -351,6 +481,7 @@ module Aws::BedrockAgentRuntime
     ResponseStream.add_member(:dependency_failed_exception, Shapes::ShapeRef.new(shape: DependencyFailedException, location_name: "dependencyFailedException"))
     ResponseStream.add_member(:internal_server_exception, Shapes::ShapeRef.new(shape: InternalServerException, location_name: "internalServerException"))
     ResponseStream.add_member(:resource_not_found_exception, Shapes::ShapeRef.new(shape: ResourceNotFoundException, location_name: "resourceNotFoundException"))
+    ResponseStream.add_member(:return_control, Shapes::ShapeRef.new(shape: ReturnControlPayload, event: true, location_name: "returnControl"))
     ResponseStream.add_member(:service_quota_exceeded_exception, Shapes::ShapeRef.new(shape: ServiceQuotaExceededException, location_name: "serviceQuotaExceededException"))
     ResponseStream.add_member(:throttling_exception, Shapes::ShapeRef.new(shape: ThrottlingException, location_name: "throttlingException"))
     ResponseStream.add_member(:trace, Shapes::ShapeRef.new(shape: TracePart, event: true, location_name: "trace"))
@@ -398,6 +529,7 @@ module Aws::BedrockAgentRuntime
     RetrievalResultS3Location.add_member(:uri, Shapes::ShapeRef.new(shape: String, location_name: "uri"))
     RetrievalResultS3Location.struct_class = Types::RetrievalResultS3Location
 
+    RetrieveAndGenerateConfiguration.add_member(:external_sources_configuration, Shapes::ShapeRef.new(shape: ExternalSourcesRetrieveAndGenerateConfiguration, location_name: "externalSourcesConfiguration"))
     RetrieveAndGenerateConfiguration.add_member(:knowledge_base_configuration, Shapes::ShapeRef.new(shape: KnowledgeBaseRetrieveAndGenerateConfiguration, location_name: "knowledgeBaseConfiguration"))
     RetrieveAndGenerateConfiguration.add_member(:type, Shapes::ShapeRef.new(shape: RetrieveAndGenerateType, required: true, location_name: "type"))
     RetrieveAndGenerateConfiguration.struct_class = Types::RetrieveAndGenerateConfiguration
@@ -439,13 +571,24 @@ module Aws::BedrockAgentRuntime
 
     RetrievedReferences.member = Shapes::ShapeRef.new(shape: RetrievedReference)
 
+    ReturnControlInvocationResults.member = Shapes::ShapeRef.new(shape: InvocationResultMember)
+
+    ReturnControlPayload.add_member(:invocation_id, Shapes::ShapeRef.new(shape: String, location_name: "invocationId"))
+    ReturnControlPayload.add_member(:invocation_inputs, Shapes::ShapeRef.new(shape: InvocationInputs, location_name: "invocationInputs"))
+    ReturnControlPayload.struct_class = Types::ReturnControlPayload
+
+    S3ObjectDoc.add_member(:uri, Shapes::ShapeRef.new(shape: S3Uri, required: true, location_name: "uri"))
+    S3ObjectDoc.struct_class = Types::S3ObjectDoc
+
     ServiceQuotaExceededException.add_member(:message, Shapes::ShapeRef.new(shape: NonBlankString, location_name: "message"))
     ServiceQuotaExceededException.struct_class = Types::ServiceQuotaExceededException
 
     SessionAttributesMap.key = Shapes::ShapeRef.new(shape: String)
     SessionAttributesMap.value = Shapes::ShapeRef.new(shape: String)
 
+    SessionState.add_member(:invocation_id, Shapes::ShapeRef.new(shape: String, location_name: "invocationId"))
     SessionState.add_member(:prompt_session_attributes, Shapes::ShapeRef.new(shape: PromptSessionAttributesMap, location_name: "promptSessionAttributes"))
+    SessionState.add_member(:return_control_invocation_results, Shapes::ShapeRef.new(shape: ReturnControlInvocationResults, location_name: "returnControlInvocationResults"))
     SessionState.add_member(:session_attributes, Shapes::ShapeRef.new(shape: SessionAttributesMap, location_name: "sessionAttributes"))
     SessionState.struct_class = Types::SessionState
 
@@ -476,6 +619,7 @@ module Aws::BedrockAgentRuntime
 
     TracePart.add_member(:agent_alias_id, Shapes::ShapeRef.new(shape: AgentAliasId, location_name: "agentAliasId"))
     TracePart.add_member(:agent_id, Shapes::ShapeRef.new(shape: AgentId, location_name: "agentId"))
+    TracePart.add_member(:agent_version, Shapes::ShapeRef.new(shape: AgentVersion, location_name: "agentVersion"))
     TracePart.add_member(:session_id, Shapes::ShapeRef.new(shape: SessionId, location_name: "sessionId"))
     TracePart.add_member(:trace, Shapes::ShapeRef.new(shape: Trace, location_name: "trace"))
     TracePart.struct_class = Types::TracePart

data/lib/aws-sdk-bedrockagentruntime/event_streams.rb

@@ -43,6 +43,10 @@ module Aws::BedrockAgentRuntime
         @event_emitter.on(:resource_not_found_exception, block) if block_given?
       end
 
+      def on_return_control_event(&block)
+        @event_emitter.on(:return_control, block) if block_given?
+      end
+
       def on_service_quota_exceeded_exception_event(&block)
         @event_emitter.on(:service_quota_exceeded_exception, block) if block_given?
       end
@@ -79,6 +83,7 @@ module Aws::BedrockAgentRuntime
         on_dependency_failed_exception_event(&block)
         on_internal_server_exception_event(&block)
         on_resource_not_found_exception_event(&block)
+        on_return_control_event(&block)
         on_service_quota_exceeded_exception_event(&block)
         on_throttling_exception_event(&block)
         on_trace_event(&block)