aws-sdk-bedrockagentruntime 1.15.0 → 1.18.0

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

@@ -41,10 +41,24 @@ module Aws::BedrockAgentRuntime
     #   The path to the API to call, based off the action group.
     #   @return [String]
     #
+    # @!attribute [rw] execution_type
+    #   How fulfillment of the action is handled. For more information, see
+    #   [Handling fulfillment of the action][1].
+    #
+    #
+    #
+    #   [1]: https://docs.aws.amazon.com/bedrock/latest/userguide/action-handle.html
+    #   @return [String]
+    #
     # @!attribute [rw] function
     #   The function in the action group to call.
     #   @return [String]
     #
+    # @!attribute [rw] invocation_id
+    #   The unique identifier of the invocation. Only returned if the
+    #   `executionType` is `RETURN_CONTROL`.
+    #   @return [String]
+    #
     # @!attribute [rw] parameters
     #   The parameters in the Lambda input event.
     #   @return [Array<Types::Parameter>]
@@ -62,7 +76,9 @@ module Aws::BedrockAgentRuntime
     class ActionGroupInvocationInput < Struct.new(
       :action_group_name,
       :api_path,
+      :execution_type,
       :function,
+      :invocation_id,
       :parameters,
       :request_body,
       :verb)
@@ -312,6 +328,28 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
+    # The property contains the file to chat with, along with its
+    # attributes.
+    #
+    # @!attribute [rw] data
+    #   The byte value of the file to attach, encoded as Base-64 string. The
+    #   maximum size of all files that is attached is 10MB. You can attach a
+    #   maximum of 5 files.
+    #   @return [String]
+    #
+    # @!attribute [rw] media_type
+    #   The MIME type of data contained in the file used for chat.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/ByteContentFile AWS API Documentation
+    #
+    class ByteContentFile < Struct.new(
+      :data,
+      :media_type)
+      SENSITIVE = [:data]
+      include Aws::Structure
+    end
+
     # An object containing a segment of the generated response that is based
     # on a source in the knowledge base, alongside information about the
     # source.
@@ -345,6 +383,55 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
+    # Contains information about the code interpreter being invoked.
+    #
+    # @!attribute [rw] code
+    #   The code for the code interpreter to use.
+    #   @return [String]
+    #
+    # @!attribute [rw] files
+    #   Files that are uploaded for code interpreter to use.
+    #   @return [Array<String>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/CodeInterpreterInvocationInput AWS API Documentation
+    #
+    class CodeInterpreterInvocationInput < Struct.new(
+      :code,
+      :files)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Contains the JSON-formatted string returned by the API invoked by the
+    # code interpreter.
+    #
+    # @!attribute [rw] execution_error
+    #   Contains the error returned from code execution.
+    #   @return [String]
+    #
+    # @!attribute [rw] execution_output
+    #   Contains the successful output returned from code execution
+    #   @return [String]
+    #
+    # @!attribute [rw] execution_timeout
+    #   Indicates if the execution of the code timed out.
+    #   @return [Boolean]
+    #
+    # @!attribute [rw] files
+    #   Contains output files, if generated by code execution.
+    #   @return [Array<String>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/CodeInterpreterInvocationOutput AWS API Documentation
+    #
+    class CodeInterpreterInvocationOutput < Struct.new(
+      :execution_error,
+      :execution_output,
+      :execution_timeout,
+      :files)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # There was a conflict performing an operation. Resolve the conflict and
     # retry your request.
     #
@@ -385,6 +472,32 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
+    # @!attribute [rw] agent_alias_id
+    #   The unique identifier of an alias of an agent.
+    #   @return [String]
+    #
+    # @!attribute [rw] agent_id
+    #   The unique identifier of the agent to which the alias belongs.
+    #   @return [String]
+    #
+    # @!attribute [rw] memory_id
+    #   The unique identifier of the memory.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/DeleteAgentMemoryRequest AWS API Documentation
+    #
+    class DeleteAgentMemoryRequest < Struct.new(
+      :agent_alias_id,
+      :agent_id,
+      :memory_id)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/DeleteAgentMemoryResponse AWS API Documentation
+    #
+    class DeleteAgentMemoryResponse < Aws::EmptyStructure; end
+
     # There was an issue with a dependency. Check the resource
     # configurations and retry the request.
     #
@@ -514,6 +627,46 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
+    # Contains intermediate response for code interpreter if any files have
+    # been generated.
+    #
+    # @!attribute [rw] files
+    #   Files containing intermediate response for the user.
+    #   @return [Array<Types::OutputFile>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/FilePart AWS API Documentation
+    #
+    class FilePart < Struct.new(
+      :files,
+      :event_type)
+      SENSITIVE = [:files]
+      include Aws::Structure
+    end
+
+    # The source file of the content contained in the wrapper object.
+    #
+    # @!attribute [rw] byte_content
+    #   The data and the text of the attached files.
+    #   @return [Types::ByteContentFile]
+    #
+    # @!attribute [rw] s3_location
+    #   The s3 location of the files to attach.
+    #   @return [Types::S3ObjectFile]
+    #
+    # @!attribute [rw] source_type
+    #   The source type of the files to attach.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/FileSource AWS API Documentation
+    #
+    class FileSource < Struct.new(
+      :byte_content,
+      :s3_location,
+      :source_type)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # Specifies the name that the metadata attribute must match and the
     # value to which to compare the value of the metadata attribute. For
     # more information, see [Query configurations][1].
@@ -560,6 +713,165 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
+    # Contains information about why a flow completed.
+    #
+    # This data type is used in the following API operations:
+    #
+    # * [InvokeFlow response][1]
+    #
+    # ^
+    #
+    #
+    #
+    # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_InvokeFlow.html#API_agent_InvokeFlow_ResponseSyntax
+    #
+    # @!attribute [rw] completion_reason
+    #   The reason that the flow completed.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/FlowCompletionEvent AWS API Documentation
+    #
+    class FlowCompletionEvent < Struct.new(
+      :completion_reason,
+      :event_type)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Contains information about an input into the prompt flow and where to
+    # send it.
+    #
+    # This data type is used in the following API operations:
+    #
+    # * [InvokeFlow request][1]
+    #
+    # ^
+    #
+    #
+    #
+    # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_InvokeFlow.html#API_agent_InvokeFlow_RequestSyntax
+    #
+    # @!attribute [rw] content
+    #   Contains information about an input into the prompt flow.
+    #   @return [Types::FlowInputContent]
+    #
+    # @!attribute [rw] node_name
+    #   The name of the flow input node that begins the prompt flow.
+    #   @return [String]
+    #
+    # @!attribute [rw] node_output_name
+    #   The name of the output from the flow input node that begins the
+    #   prompt flow.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/FlowInput AWS API Documentation
+    #
+    class FlowInput < Struct.new(
+      :content,
+      :node_name,
+      :node_output_name)
+      SENSITIVE = [:content]
+      include Aws::Structure
+    end
+
+    # Contains information about an input into the flow.
+    #
+    # This data type is used in the following API operations:
+    #
+    # * [InvokeFlow request][1]
+    #
+    # ^
+    #
+    #
+    #
+    # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_InvokeFlow.html#API_agent_InvokeFlow_RequestSyntax
+    #
+    # @note FlowInputContent is a union - when making an API calls you must set exactly one of the members.
+    #
+    # @!attribute [rw] document
+    #   The input to send to the prompt flow input node.
+    #   @return [Hash,Array,String,Numeric,Boolean]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/FlowInputContent AWS API Documentation
+    #
+    class FlowInputContent < Struct.new(
+      :document,
+      :unknown)
+      SENSITIVE = []
+      include Aws::Structure
+      include Aws::Structure::Union
+
+      class Document < FlowInputContent; end
+      class Unknown < FlowInputContent; end
+    end
+
+    # Contains information about the content in an output from prompt flow
+    # invocation.
+    #
+    # This data type is used in the following API operations:
+    #
+    # * [InvokeFlow request][1]
+    #
+    # ^
+    #
+    #
+    #
+    # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_InvokeFlow.html#API_agent_InvokeFlow_RequestSyntax
+    #
+    # @note FlowOutputContent is a union - when returned from an API call exactly one value will be set and the returned type will be a subclass of FlowOutputContent corresponding to the set member.
+    #
+    # @!attribute [rw] document
+    #   The content in the output.
+    #   @return [Hash,Array,String,Numeric,Boolean]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/FlowOutputContent AWS API Documentation
+    #
+    class FlowOutputContent < Struct.new(
+      :document,
+      :unknown)
+      SENSITIVE = []
+      include Aws::Structure
+      include Aws::Structure::Union
+
+      class Document < FlowOutputContent; end
+      class Unknown < FlowOutputContent; end
+    end
+
+    # Contains information about an output from prompt flow invoction.
+    #
+    # This data type is used in the following API operations:
+    #
+    # * [InvokeFlow response][1]
+    #
+    # ^
+    #
+    #
+    #
+    # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_InvokeFlow.html#API_agent_InvokeFlow_ResponseSyntax
+    #
+    # @!attribute [rw] content
+    #   The content in the output.
+    #   @return [Types::FlowOutputContent]
+    #
+    # @!attribute [rw] node_name
+    #   The name of the flow output node that the output is from.
+    #   @return [String]
+    #
+    # @!attribute [rw] node_type
+    #   The type of the node that the output is from.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/FlowOutputEvent AWS API Documentation
+    #
+    class FlowOutputEvent < Struct.new(
+      :content,
+      :node_name,
+      :node_type,
+      :event_type)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # Contains information about the function that the agent predicts should
     # be called.
     #
@@ -750,6 +1062,68 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
+    # @!attribute [rw] agent_alias_id
+    #   The unique identifier of an alias of an agent.
+    #   @return [String]
+    #
+    # @!attribute [rw] agent_id
+    #   The unique identifier of the agent to which the alias belongs.
+    #   @return [String]
+    #
+    # @!attribute [rw] max_items
+    #   The maximum number of items to return in the response. If the total
+    #   number of results is greater than this value, use the token returned
+    #   in the response in the `nextToken` field when making another request
+    #   to return the next batch of results.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] memory_id
+    #   The unique identifier of the memory.
+    #   @return [String]
+    #
+    # @!attribute [rw] memory_type
+    #   The type of memory.
+    #   @return [String]
+    #
+    # @!attribute [rw] next_token
+    #   If the total number of results is greater than the maxItems value
+    #   provided in the request, enter the token returned in the `nextToken`
+    #   field in the response in this field to return the next batch of
+    #   results.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/GetAgentMemoryRequest AWS API Documentation
+    #
+    class GetAgentMemoryRequest < Struct.new(
+      :agent_alias_id,
+      :agent_id,
+      :max_items,
+      :memory_id,
+      :memory_type,
+      :next_token)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] memory_contents
+    #   Contains details of the sessions stored in the memory
+    #   @return [Array<Types::Memory>]
+    #
+    # @!attribute [rw] next_token
+    #   If the total number of results is greater than the maxItems value
+    #   provided in the request, use this token when making another request
+    #   in the `nextToken` field to return the next batch of results.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/GetAgentMemoryResponse AWS API Documentation
+    #
+    class GetAgentMemoryResponse < Struct.new(
+      :memory_contents,
+      :next_token)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # Assessment details of the content analyzed by Guardrails.
     #
     # @!attribute [rw] content_policy
@@ -1020,7 +1394,7 @@ module Aws::BedrockAgentRuntime
       :input_assessments,
       :output_assessments,
       :trace_id)
-      SENSITIVE = []
+      SENSITIVE = [:input_assessments, :output_assessments]
       include Aws::Structure
     end
 
@@ -1100,7 +1474,7 @@ module Aws::BedrockAgentRuntime
     #   the following token at each point of generation. The value that you
     #   set for `Top P` determines the number of most-likely candidates from
     #   which the model chooses the next token in the sequence. For example,
-    #   if you set `topP` to 80, the model only selects the next token from
+    #   if you set `topP` to 0.8, the model only selects the next token from
     #   the top 80% of the probability distribution of next tokens.
     #   @return [Float]
     #
@@ -1116,6 +1490,30 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
+    # Contains details of the source files.
+    #
+    # @!attribute [rw] name
+    #   The name of the source file.
+    #   @return [String]
+    #
+    # @!attribute [rw] source
+    #   Specifies where the files are located.
+    #   @return [Types::FileSource]
+    #
+    # @!attribute [rw] use_case
+    #   Specifies how the source files will be used by the code interpreter.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/InputFile AWS API Documentation
+    #
+    class InputFile < Struct.new(
+      :name,
+      :source,
+      :use_case)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # An internal server error occurred. Retry your request.
     #
     # @!attribute [rw] message
@@ -1137,6 +1535,10 @@ module Aws::BedrockAgentRuntime
     #   Contains information about the action group to be invoked.
     #   @return [Types::ActionGroupInvocationInput]
     #
+    # @!attribute [rw] code_interpreter_invocation_input
+    #   Contains information about the code interpreter to be invoked.
+    #   @return [Types::CodeInterpreterInvocationInput]
+    #
     # @!attribute [rw] invocation_type
     #   Specifies whether the agent is invoking an action group or a
     #   knowledge base.
@@ -1155,6 +1557,7 @@ module Aws::BedrockAgentRuntime
     #
     class InvocationInput < Struct.new(
       :action_group_invocation_input,
+      :code_interpreter_invocation_input,
       :invocation_type,
       :knowledge_base_lookup_input,
       :trace_id)
@@ -1273,6 +1676,10 @@ module Aws::BedrockAgentRuntime
     #    </note>
     #   @return [String]
     #
+    # @!attribute [rw] memory_id
+    #   The unique identifier of the agent memory.
+    #   @return [String]
+    #
     # @!attribute [rw] session_id
     #   The unique identifier of the session. Use the same value across
     #   requests to continue the same conversation.
@@ -1300,6 +1707,7 @@ module Aws::BedrockAgentRuntime
       :enable_trace,
       :end_session,
       :input_text,
+      :memory_id,
       :session_id,
       :session_state)
       SENSITIVE = [:input_text]
@@ -1315,6 +1723,10 @@ module Aws::BedrockAgentRuntime
     #   `application/json`.
     #   @return [String]
     #
+    # @!attribute [rw] memory_id
+    #   The unique identifier of the agent memory.
+    #   @return [String]
+    #
     # @!attribute [rw] session_id
     #   The unique identifier of the session with the agent.
     #   @return [String]
@@ -1324,11 +1736,78 @@ module Aws::BedrockAgentRuntime
     class InvokeAgentResponse < Struct.new(
       :completion,
       :content_type,
+      :memory_id,
       :session_id)
       SENSITIVE = []
       include Aws::Structure
     end
 
+    # @!attribute [rw] flow_alias_identifier
+    #   The unique identifier of the flow alias.
+    #   @return [String]
+    #
+    # @!attribute [rw] flow_identifier
+    #   The unique identifier of the flow.
+    #   @return [String]
+    #
+    # @!attribute [rw] inputs
+    #   A list of objects, each containing information about an input into
+    #   the flow.
+    #   @return [Array<Types::FlowInput>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/InvokeFlowRequest AWS API Documentation
+    #
+    class InvokeFlowRequest < Struct.new(
+      :flow_alias_identifier,
+      :flow_identifier,
+      :inputs)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] response_stream
+    #   The output of the flow, returned as a stream. If there's an error,
+    #   the error is returned.
+    #   @return [Types::FlowResponseStream]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/InvokeFlowResponse AWS API Documentation
+    #
+    class InvokeFlowResponse < Struct.new(
+      :response_stream)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Configurations to apply to a knowledge base attached to the agent
+    # during query. For more information, see [Knowledge base retrieval
+    # configurations][1].
+    #
+    #
+    #
+    # [1]: https://docs.aws.amazon.com/bedrock/latest/userguide/agents-session-state.html#session-state-kb
+    #
+    # @!attribute [rw] knowledge_base_id
+    #   The unique identifier for a knowledge base attached to the agent.
+    #   @return [String]
+    #
+    # @!attribute [rw] retrieval_configuration
+    #   The configurations to apply to the knowledge base during query. For
+    #   more information, see [Query configurations][1].
+    #
+    #
+    #
+    #   [1]: https://docs.aws.amazon.com/bedrock/latest/userguide/kb-test-config.html
+    #   @return [Types::KnowledgeBaseRetrievalConfiguration]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/KnowledgeBaseConfiguration AWS API Documentation
+    #
+    class KnowledgeBaseConfiguration < Struct.new(
+      :knowledge_base_id,
+      :retrieval_configuration)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # Contains details about the knowledge base to look up and the query to
     # be made.
     #
@@ -1388,8 +1867,8 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
-    # Contains configurations for the knowledge base query and retrieval
-    # process. For more information, see [Query configurations][1].
+    # Contains configurations for knowledge base query. For more
+    # information, see [Query configurations][1].
     #
     # This data type is used in the following API operations:
     #
@@ -1481,7 +1960,7 @@ module Aws::BedrockAgentRuntime
     #
     # @!attribute [rw] generation_configuration
     #   Contains configurations for response generation based on the
-    #   knowwledge base query results.
+    #   knowledge base query results.
     #   @return [Types::GenerationConfiguration]
     #
     # @!attribute [rw] knowledge_base_id
@@ -1493,6 +1972,11 @@ module Aws::BedrockAgentRuntime
     #   The ARN of the foundation model used to generate a response.
     #   @return [String]
     #
+    # @!attribute [rw] orchestration_configuration
+    #   Settings for how the model processes the prompt prior to retrieval
+    #   and generation.
+    #   @return [Types::OrchestrationConfiguration]
+    #
     # @!attribute [rw] retrieval_configuration
     #   Contains configurations for how to retrieve and return the knowledge
     #   base query.
@@ -1504,6 +1988,7 @@ module Aws::BedrockAgentRuntime
       :generation_configuration,
       :knowledge_base_id,
       :model_arn,
+      :orchestration_configuration,
       :retrieval_configuration)
       SENSITIVE = []
       include Aws::Structure
@@ -1564,6 +2049,76 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
+    # Contains sessions summaries.
+    #
+    # @note Memory is a union - when returned from an API call exactly one value will be set and the returned type will be a subclass of Memory corresponding to the set member.
+    #
+    # @!attribute [rw] session_summary
+    #   Contains summary of a session.
+    #   @return [Types::MemorySessionSummary]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/Memory AWS API Documentation
+    #
+    class Memory < Struct.new(
+      :session_summary,
+      :unknown)
+      SENSITIVE = []
+      include Aws::Structure
+      include Aws::Structure::Union
+
+      class SessionSummary < Memory; end
+      class Unknown < Memory; end
+    end
+
+    # Contains details of a session summary.
+    #
+    # @!attribute [rw] memory_id
+    #   The unique identifier of the memory where the session summary is
+    #   stored.
+    #   @return [String]
+    #
+    # @!attribute [rw] session_expiry_time
+    #   The time when the memory duration for the session is set to end.
+    #   @return [Time]
+    #
+    # @!attribute [rw] session_id
+    #   The identifier for this session.
+    #   @return [String]
+    #
+    # @!attribute [rw] session_start_time
+    #   The start time for this session.
+    #   @return [Time]
+    #
+    # @!attribute [rw] summary_text
+    #   The summarized text for this session.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/MemorySessionSummary AWS API Documentation
+    #
+    class MemorySessionSummary < Struct.new(
+      :memory_id,
+      :session_expiry_time,
+      :session_id,
+      :session_start_time,
+      :summary_text)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Provides details of the foundation model.
+    #
+    # @!attribute [rw] usage
+    #   Contains details of the foundation model usage.
+    #   @return [Types::Usage]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/Metadata AWS API Documentation
+    #
+    class Metadata < Struct.new(
+      :usage)
+      SENSITIVE = [:usage]
+      include Aws::Structure
+    end
+
     # The input for the pre-processing step.
     #
     # * The `type` matches the agent step.
@@ -1647,6 +2202,11 @@ module Aws::BedrockAgentRuntime
     #   the action group.
     #   @return [Types::ActionGroupInvocationOutput]
     #
+    # @!attribute [rw] code_interpreter_invocation_output
+    #   Contains the JSON-formatted string returned by the API invoked by
+    #   the code interpreter.
+    #   @return [Types::CodeInterpreterInvocationOutput]
+    #
     # @!attribute [rw] final_response
     #   Contains details about the response to the user.
     #   @return [Types::FinalResponse]
@@ -1686,6 +2246,7 @@ module Aws::BedrockAgentRuntime
     #
     class Observation < Struct.new(
       :action_group_invocation_output,
+      :code_interpreter_invocation_output,
       :final_response,
       :knowledge_base_lookup_output,
       :reprompt_response,
@@ -1695,6 +2256,47 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
+    # Settings for how the model processes the prompt prior to retrieval and
+    # generation.
+    #
+    # @!attribute [rw] query_transformation_configuration
+    #   To split up the prompt and retrieve multiple sources, set the
+    #   transformation type to `QUERY_DECOMPOSITION`.
+    #   @return [Types::QueryTransformationConfiguration]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/OrchestrationConfiguration AWS API Documentation
+    #
+    class OrchestrationConfiguration < Struct.new(
+      :query_transformation_configuration)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # The foundation model output from the orchestration step.
+    #
+    # @!attribute [rw] metadata
+    #   Contains information about the foundation model output.
+    #   @return [Types::Metadata]
+    #
+    # @!attribute [rw] raw_response
+    #   Contains details of the raw response from the foundation model
+    #   output.
+    #   @return [Types::RawResponse]
+    #
+    # @!attribute [rw] trace_id
+    #   The unique identifier of the trace.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/OrchestrationModelInvocationOutput AWS API Documentation
+    #
+    class OrchestrationModelInvocationOutput < Struct.new(
+      :metadata,
+      :raw_response,
+      :trace_id)
+      SENSITIVE = [:metadata, :raw_response]
+      include Aws::Structure
+    end
+
     # Details about the orchestration step, in which the agent determines
     # the order in which actions are executed and which knowledge bases are
     # retrieved.
@@ -1722,6 +2324,11 @@ module Aws::BedrockAgentRuntime
     #   [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent_PromptOverrideConfiguration.html
     #   @return [Types::ModelInvocationInput]
     #
+    # @!attribute [rw] model_invocation_output
+    #   Contains information pertaining to the output from the foundation
+    #   model that is being invoked.
+    #   @return [Types::OrchestrationModelInvocationOutput]
+    #
     # @!attribute [rw] observation
     #   Details about the observation (the output of the action group Lambda
     #   or knowledge base) made by the agent.
@@ -1738,20 +2345,47 @@ module Aws::BedrockAgentRuntime
     class OrchestrationTrace < Struct.new(
       :invocation_input,
       :model_invocation_input,
+      :model_invocation_output,
       :observation,
       :rationale,
       :unknown)
-      SENSITIVE = [:invocation_input, :model_invocation_input, :observation, :rationale]
+      SENSITIVE = [:invocation_input, :model_invocation_input, :model_invocation_output, :observation, :rationale]
       include Aws::Structure
       include Aws::Structure::Union
 
       class InvocationInput < OrchestrationTrace; end
       class ModelInvocationInput < OrchestrationTrace; end
+      class ModelInvocationOutput < OrchestrationTrace; end
       class Observation < OrchestrationTrace; end
       class Rationale < OrchestrationTrace; end
       class Unknown < OrchestrationTrace; end
     end
 
+    # Contains details of the response from code interpreter.
+    #
+    # @!attribute [rw] bytes
+    #   The byte count of files that contains response from code
+    #   interpreter.
+    #   @return [String]
+    #
+    # @!attribute [rw] name
+    #   The name of the file containing response from code interpreter.
+    #   @return [String]
+    #
+    # @!attribute [rw] type
+    #   The type of file that contains response from the code interpreter.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/OutputFile AWS API Documentation
+    #
+    class OutputFile < Struct.new(
+      :bytes,
+      :name,
+      :type)
+      SENSITIVE = [:bytes]
+      include Aws::Structure
+    end
+
     # A parameter for the API request or function.
     #
     # @!attribute [rw] name
@@ -2010,6 +2644,21 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
+    # To split up the prompt and retrieve multiple sources, set the
+    # transformation type to `QUERY_DECOMPOSITION`.
+    #
+    # @!attribute [rw] type
+    #   The type of transformation to apply to the prompt.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/QueryTransformationConfiguration AWS API Documentation
+    #
+    class QueryTransformationConfiguration < Struct.new(
+      :type)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # Contains the reasoning, based on the input, that the agent uses to
     # justify carrying out an action group or getting information from a
     # knowledge base.
@@ -2031,6 +2680,20 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
+    # Contains the raw output from the foundation model.
+    #
+    # @!attribute [rw] content
+    #   The foundation model's raw output content.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/RawResponse AWS API Documentation
+    #
+    class RawResponse < Struct.new(
+      :content)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # Contains details about the agent's response to reprompt the input.
     #
     # @!attribute [rw] source
@@ -2257,7 +2920,7 @@ module Aws::BedrockAgentRuntime
       :starts_with,
       :string_contains,
       :unknown)
-      SENSITIVE = []
+      SENSITIVE = [:and_all, :or_all]
       include Aws::Structure
       include Aws::Structure::Union
 
@@ -2277,6 +2940,20 @@ module Aws::BedrockAgentRuntime
       class Unknown < RetrievalFilter; end
     end
 
+    # The Confluence data source location.
+    #
+    # @!attribute [rw] url
+    #   The Confluence host URL for the data source location.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/RetrievalResultConfluenceLocation AWS API Documentation
+    #
+    class RetrievalResultConfluenceLocation < Struct.new(
+      :url)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # Contains the cited text from the data source.
     #
     # This data type is used in the following API operations:
@@ -2305,7 +2982,7 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
-    # Contains information about the location of the data source.
+    # Contains information about the data source location.
     #
     # This data type is used in the following API operations:
     #
@@ -2321,24 +2998,44 @@ module Aws::BedrockAgentRuntime
     # [2]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_RetrieveAndGenerate.html#API_agent-runtime_RetrieveAndGenerate_ResponseSyntax
     # [3]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_InvokeAgent.html#API_agent-runtime_InvokeAgent_ResponseSyntax
     #
+    # @!attribute [rw] confluence_location
+    #   The Confluence data source location.
+    #   @return [Types::RetrievalResultConfluenceLocation]
+    #
     # @!attribute [rw] s3_location
-    #   Contains the S3 location of the data source.
+    #   The S3 data source location.
     #   @return [Types::RetrievalResultS3Location]
     #
+    # @!attribute [rw] salesforce_location
+    #   The Salesforce data source location.
+    #   @return [Types::RetrievalResultSalesforceLocation]
+    #
+    # @!attribute [rw] share_point_location
+    #   The SharePoint data source location.
+    #   @return [Types::RetrievalResultSharePointLocation]
+    #
     # @!attribute [rw] type
-    #   The type of the location of the data source.
+    #   The type of data source location.
     #   @return [String]
     #
+    # @!attribute [rw] web_location
+    #   The web URL/URLs data source location.
+    #   @return [Types::RetrievalResultWebLocation]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/RetrievalResultLocation AWS API Documentation
     #
     class RetrievalResultLocation < Struct.new(
+      :confluence_location,
       :s3_location,
-      :type)
+      :salesforce_location,
+      :share_point_location,
+      :type,
+      :web_location)
       SENSITIVE = []
       include Aws::Structure
     end
 
-    # Contains the S3 location of the data source.
+    # The S3 data source location.
     #
     # This data type is used in the following API operations:
     #
@@ -2355,7 +3052,7 @@ module Aws::BedrockAgentRuntime
     # [3]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_InvokeAgent.html#API_agent-runtime_InvokeAgent_ResponseSyntax
     #
     # @!attribute [rw] uri
-    #   The S3 URI of the data source.
+    #   The S3 URI for the data source location.
     #   @return [String]
     #
     # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/RetrievalResultS3Location AWS API Documentation
@@ -2366,6 +3063,48 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
+    # The Salesforce data source location.
+    #
+    # @!attribute [rw] url
+    #   The Salesforce host URL for the data source location.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/RetrievalResultSalesforceLocation AWS API Documentation
+    #
+    class RetrievalResultSalesforceLocation < Struct.new(
+      :url)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # The SharePoint data source location.
+    #
+    # @!attribute [rw] url
+    #   The SharePoint site URL for the data source location.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/RetrievalResultSharePointLocation AWS API Documentation
+    #
+    class RetrievalResultSharePointLocation < Struct.new(
+      :url)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # The web URL/URLs data source location.
+    #
+    # @!attribute [rw] url
+    #   The web URL/URLs for the data source location.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/RetrievalResultWebLocation AWS API Documentation
+    #
+    class RetrievalResultWebLocation < Struct.new(
+      :url)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # Contains details about the resource being queried.
     #
     # This data type is used in the following API operations:
@@ -2468,8 +3207,12 @@ module Aws::BedrockAgentRuntime
     #   @return [Types::RetrieveAndGenerateSessionConfiguration]
     #
     # @!attribute [rw] session_id
-    #   The unique identifier of the session. Reuse the same value to
-    #   continue the same session with the knowledge base.
+    #   The unique identifier of the session. When you first make a
+    #   `RetrieveAndGenerate` request, Amazon Bedrock automatically
+    #   generates this value. You must reuse this value for all subsequent
+    #   requests in the same conversational session. This value allows
+    #   Amazon Bedrock to maintain context and knowledge from previous
+    #   interactions. You can't explicitly set the `sessionId` yourself.
     #   @return [String]
     #
     # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/RetrieveAndGenerateRequest AWS API Documentation
@@ -2498,8 +3241,12 @@ module Aws::BedrockAgentRuntime
     #   @return [Types::RetrieveAndGenerateOutput]
     #
     # @!attribute [rw] session_id
-    #   The unique identifier of the session. Reuse the same value to
-    #   continue the same session with the knowledge base.
+    #   The unique identifier of the session. When you first make a
+    #   `RetrieveAndGenerate` request, Amazon Bedrock automatically
+    #   generates this value. You must reuse this value for all subsequent
+    #   requests in the same conversational session. This value allows
+    #   Amazon Bedrock to maintain context and knowledge from previous
+    #   interactions. You can't explicitly set the `sessionId` yourself.
     #   @return [String]
     #
     # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/RetrieveAndGenerateResponse AWS API Documentation
@@ -2679,6 +3426,20 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
+    # Contains details of the s3 object where the source file is located.
+    #
+    # @!attribute [rw] uri
+    #   The uri of the s3 object.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/S3ObjectFile AWS API Documentation
+    #
+    class S3ObjectFile < Struct.new(
+      :uri)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # The number of requests exceeds the service quota. Resubmit your
     # request later.
     #
@@ -2708,6 +3469,10 @@ module Aws::BedrockAgentRuntime
     # [2]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_InvokeAgent.html
     # [3]: https://docs.aws.amazon.com/bedrock/latest/userguide/agents-session-state.html
     #
+    # @!attribute [rw] files
+    #   Contains information about the files used by code interpreter.
+    #   @return [Array<Types::InputFile>]
+    #
     # @!attribute [rw] invocation_id
     #   The identifier of the invocation of an action. This value must match
     #   the `invocationId` returned in the `InvokeAgent` response for the
@@ -2722,6 +3487,11 @@ module Aws::BedrockAgentRuntime
     #   [2]: https://docs.aws.amazon.com/bedrock/latest/userguide/agents-session-state.html
     #   @return [String]
     #
+    # @!attribute [rw] knowledge_base_configurations
+    #   An array of configurations, each of which applies to a knowledge
+    #   base attached to the agent.
+    #   @return [Array<Types::KnowledgeBaseConfiguration>]
+    #
     # @!attribute [rw] prompt_session_attributes
     #   Contains attributes that persist across a prompt and the values of
     #   those attributes. These attributes replace the
@@ -2757,7 +3527,9 @@ module Aws::BedrockAgentRuntime
     # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/SessionState AWS API Documentation
     #
     class SessionState < Struct.new(
+      :files,
       :invocation_id,
+      :knowledge_base_configurations,
       :prompt_session_attributes,
       :return_control_invocation_results,
       :session_attributes)
@@ -3005,6 +3777,27 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
+    # Contains information of the usage of the foundation model.
+    #
+    # @!attribute [rw] input_tokens
+    #   Contains information about the input tokens from the foundation
+    #   model usage.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] output_tokens
+    #   Contains information about the output tokens from the foundation
+    #   model usage.
+    #   @return [Integer]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/Usage AWS API Documentation
+    #
+    class Usage < Struct.new(
+      :input_tokens,
+      :output_tokens)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # Input validation failed. Check your request parameters and retry the
     # request.
     #
@@ -3020,6 +3813,43 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
+    # The output of the flow.
+    #
+    # This data type is used in the following API operations:
+    #
+    # * [InvokeFlow response][1]
+    #
+    # ^
+    #
+    #
+    #
+    # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_InvokeFlow.html#API_agent_InvokeFlow_ResponseSyntax
+    #
+    # EventStream is an Enumerator of Events.
+    #  #event_types #=> Array, returns all modeled event types in the stream
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/FlowResponseStream AWS API Documentation
+    #
+    class FlowResponseStream < Enumerator
+
+      def event_types
+        [
+          :access_denied_exception,
+          :bad_gateway_exception,
+          :conflict_exception,
+          :dependency_failed_exception,
+          :flow_completion_event,
+          :flow_output_event,
+          :internal_server_exception,
+          :resource_not_found_exception,
+          :service_quota_exceeded_exception,
+          :throttling_exception,
+          :validation_exception
+        ]
+      end
+
+    end
+
     # The response from invoking the agent and associated citations and
     # trace information.
     #
@@ -3037,6 +3867,7 @@ module Aws::BedrockAgentRuntime
           :chunk,
           :conflict_exception,
           :dependency_failed_exception,
+          :files,
           :internal_server_exception,
           :resource_not_found_exception,
           :return_control,