aws-sdk-bedrockagentruntime 1.34.0 → 1.36.0

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

@@ -240,6 +240,104 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
+    # Input for an agent collaborator. The input can be text or an action
+    # invocation result.
+    #
+    # @!attribute [rw] return_control_results
+    #   An action invocation result.
+    #   @return [Types::ReturnControlResults]
+    #
+    # @!attribute [rw] text
+    #   Input text.
+    #   @return [String]
+    #
+    # @!attribute [rw] type
+    #   The input type.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/AgentCollaboratorInputPayload AWS API Documentation
+    #
+    class AgentCollaboratorInputPayload < Struct.new(
+      :return_control_results,
+      :text,
+      :type)
+      SENSITIVE = [:text]
+      include Aws::Structure
+    end
+
+    # An agent collaborator invocation input.
+    #
+    # @!attribute [rw] agent_collaborator_alias_arn
+    #   The collaborator's alias ARN.
+    #   @return [String]
+    #
+    # @!attribute [rw] agent_collaborator_name
+    #   The collaborator's name.
+    #   @return [String]
+    #
+    # @!attribute [rw] input
+    #   Text or action invocation result input for the collaborator.
+    #   @return [Types::AgentCollaboratorInputPayload]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/AgentCollaboratorInvocationInput AWS API Documentation
+    #
+    class AgentCollaboratorInvocationInput < Struct.new(
+      :agent_collaborator_alias_arn,
+      :agent_collaborator_name,
+      :input)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Output from an agent collaborator.
+    #
+    # @!attribute [rw] agent_collaborator_alias_arn
+    #   The output's agent collaborator alias ARN.
+    #   @return [String]
+    #
+    # @!attribute [rw] agent_collaborator_name
+    #   The output's agent collaborator name.
+    #   @return [String]
+    #
+    # @!attribute [rw] output
+    #   The output's output.
+    #   @return [Types::AgentCollaboratorOutputPayload]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/AgentCollaboratorInvocationOutput AWS API Documentation
+    #
+    class AgentCollaboratorInvocationOutput < Struct.new(
+      :agent_collaborator_alias_arn,
+      :agent_collaborator_name,
+      :output)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Output from an agent collaborator. The output can be text or an action
+    # invocation result.
+    #
+    # @!attribute [rw] return_control_payload
+    #   An action invocation result.
+    #   @return [Types::ReturnControlPayload]
+    #
+    # @!attribute [rw] text
+    #   Text output.
+    #   @return [String]
+    #
+    # @!attribute [rw] type
+    #   The type of output.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/AgentCollaboratorOutputPayload AWS API Documentation
+    #
+    class AgentCollaboratorOutputPayload < Struct.new(
+      :return_control_payload,
+      :text,
+      :type)
+      SENSITIVE = [:return_control_payload, :text]
+      include Aws::Structure
+    end
+
     # An event in which the prompt was analyzed in preparation for
     # optimization.
     #
@@ -277,10 +375,18 @@ module Aws::BedrockAgentRuntime
     #   Contains information about the API operation to invoke.
     #   @return [String]
     #
+    # @!attribute [rw] agent_id
+    #   The agent's ID.
+    #   @return [String]
+    #
     # @!attribute [rw] api_path
     #   The path to the API operation.
     #   @return [String]
     #
+    # @!attribute [rw] collaborator_name
+    #   The agent collaborator's name.
+    #   @return [String]
+    #
     # @!attribute [rw] http_method
     #   The HTTP method of the API operation.
     #   @return [String]
@@ -300,11 +406,13 @@ module Aws::BedrockAgentRuntime
     class ApiInvocationInput < Struct.new(
       :action_group,
       :action_invocation_type,
+      :agent_id,
       :api_path,
+      :collaborator_name,
       :http_method,
       :parameters,
       :request_body)
-      SENSITIVE = [:api_path]
+      SENSITIVE = [:api_path, :collaborator_name]
       include Aws::Structure
     end
 
@@ -386,6 +494,10 @@ module Aws::BedrockAgentRuntime
     #   The action group that the API operation belongs to.
     #   @return [String]
     #
+    # @!attribute [rw] agent_id
+    #   The agent's ID.
+    #   @return [String]
+    #
     # @!attribute [rw] api_path
     #   The path to the API operation.
     #   @return [String]
@@ -422,6 +534,7 @@ module Aws::BedrockAgentRuntime
     #
     class ApiResult < Struct.new(
       :action_group,
+      :agent_id,
       :api_path,
       :confirmation_state,
       :http_method,
@@ -468,6 +581,45 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
+    # Contains configurations for an Amazon Bedrock reranker model.
+    #
+    # @!attribute [rw] model_configuration
+    #   Contains configurations for a reranker model.
+    #   @return [Types::BedrockRerankingModelConfiguration]
+    #
+    # @!attribute [rw] number_of_results
+    #   The number of results to return after reranking.
+    #   @return [Integer]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/BedrockRerankingConfiguration AWS API Documentation
+    #
+    class BedrockRerankingConfiguration < Struct.new(
+      :model_configuration,
+      :number_of_results)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Contains configurations for a reranker model.
+    #
+    # @!attribute [rw] additional_model_request_fields
+    #   A JSON object whose keys are request fields for the model and whose
+    #   values are values for those fields.
+    #   @return [Hash<String,Hash,Array,String,Numeric,Boolean>]
+    #
+    # @!attribute [rw] model_arn
+    #   The ARN of the reranker model.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/BedrockRerankingModelConfiguration AWS API Documentation
+    #
+    class BedrockRerankingModelConfiguration < Struct.new(
+      :additional_model_request_fields,
+      :model_arn)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # This property contains the document to chat with, along with its
     # attributes.
     #
@@ -514,6 +666,27 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
+    # Details about a caller.
+    #
+    # @note Caller is a union - when returned from an API call exactly one value will be set and the returned type will be a subclass of Caller corresponding to the set member.
+    #
+    # @!attribute [rw] agent_alias_arn
+    #   The caller's agent alias ARN.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/Caller AWS API Documentation
+    #
+    class Caller < Struct.new(
+      :agent_alias_arn,
+      :unknown)
+      SENSITIVE = []
+      include Aws::Structure
+      include Aws::Structure::Union
+
+      class AgentAliasArn < Caller; end
+      class Unknown < Caller; end
+    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.
@@ -547,6 +720,21 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
+    # A citation event.
+    #
+    # @!attribute [rw] citation
+    #   The citation.
+    #   @return [Types::Citation]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/CitationEvent AWS API Documentation
+    #
+    class CitationEvent < Struct.new(
+      :citation,
+      :event_type)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # Contains information about the code interpreter being invoked.
     #
     # @!attribute [rw] code
@@ -611,6 +799,27 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
+    # A content block.
+    #
+    # @note ContentBlock is a union - when making an API calls you must set exactly one of the members.
+    #
+    # @!attribute [rw] text
+    #   The block's text.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/ContentBlock AWS API Documentation
+    #
+    class ContentBlock < Struct.new(
+      :text,
+      :unknown)
+      SENSITIVE = []
+      include Aws::Structure
+      include Aws::Structure::Union
+
+      class Text < ContentBlock; end
+      class Unknown < ContentBlock; end
+    end
+
     # Contains the body of the API response.
     #
     # This data type is used in the following API operations:
@@ -636,10 +845,24 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
+    # A conversation history.
+    #
+    # @!attribute [rw] messages
+    #   The conversation's messages.
+    #   @return [Array<Types::Message>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/ConversationHistory AWS API Documentation
+    #
+    class ConversationHistory < Struct.new(
+      :messages)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # The trace behavior for the custom orchestration.
     #
     # @!attribute [rw] event
-    #   The trace event details used with the custom orchestration.
+    #   The event details used with the custom orchestration.
     #   @return [Types::CustomOrchestrationTraceEvent]
     #
     # @!attribute [rw] trace_id
@@ -655,7 +878,9 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
-    # The event in the custom orchestration sequence.
+    # The event in the custom orchestration sequence. Events are the
+    # responses which the custom orchestration Lambda function sends as
+    # response to the agent.
     #
     # @!attribute [rw] text
     #   The text that prompted the event at this step.
@@ -824,6 +1049,22 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
+    # Contains information for a metadata field to include in or exclude
+    # from consideration when reranking.
+    #
+    # @!attribute [rw] field_name
+    #   The name of a metadata field to include in or exclude from
+    #   consideration when reranking.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/FieldForReranking AWS API Documentation
+    #
+    class FieldForReranking < Struct.new(
+      :field_name)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # Contains intermediate response for code interpreter if any files have
     # been generated.
     #
@@ -1350,6 +1591,14 @@ module Aws::BedrockAgentRuntime
     #   Contains information about the function to invoke,
     #   @return [String]
     #
+    # @!attribute [rw] agent_id
+    #   The agent's ID.
+    #   @return [String]
+    #
+    # @!attribute [rw] collaborator_name
+    #   The collaborator's name.
+    #   @return [String]
+    #
     # @!attribute [rw] function
     #   The name of the function.
     #   @return [String]
@@ -1363,9 +1612,11 @@ module Aws::BedrockAgentRuntime
     class FunctionInvocationInput < Struct.new(
       :action_group,
       :action_invocation_type,
+      :agent_id,
+      :collaborator_name,
       :function,
       :parameters)
-      SENSITIVE = []
+      SENSITIVE = [:collaborator_name]
       include Aws::Structure
     end
 
@@ -1421,6 +1672,10 @@ module Aws::BedrockAgentRuntime
     #   The action group that the function belongs to.
     #   @return [String]
     #
+    # @!attribute [rw] agent_id
+    #   The agent's ID.
+    #   @return [String]
+    #
     # @!attribute [rw] confirmation_state
     #   Contains the user confirmation information about the function that
     #   was called.
@@ -1449,6 +1704,7 @@ module Aws::BedrockAgentRuntime
     #
     class FunctionResult < Struct.new(
       :action_group,
+      :agent_id,
       :confirmation_state,
       :function,
       :response_body,
@@ -1746,6 +2002,21 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
+    # A guardrail event.
+    #
+    # @!attribute [rw] action
+    #   The guardrail action.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/GuardrailEvent AWS API Documentation
+    #
+    class GuardrailEvent < Struct.new(
+      :action,
+      :event_type)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # The managed word details for the filter in the Guardrail.
     #
     # @!attribute [rw] action
@@ -1932,6 +2203,26 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
+    # Settings for implicit filtering, where a model generates a metadata
+    # filter based on the prompt.
+    #
+    # @!attribute [rw] metadata_attributes
+    #   Metadata that can be used in a filter.
+    #   @return [Array<Types::MetadataAttributeSchema>]
+    #
+    # @!attribute [rw] model_arn
+    #   The model that generates the filter.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/ImplicitFilterConfiguration AWS API Documentation
+    #
+    class ImplicitFilterConfiguration < Struct.new(
+      :metadata_attributes,
+      :model_arn)
+      SENSITIVE = [:metadata_attributes]
+      include Aws::Structure
+    end
+
     # The configuration for inference settings when generating responses
     # using RetrieveAndGenerate.
     #
@@ -2238,6 +2529,10 @@ module Aws::BedrockAgentRuntime
     #   Contains information about the action group to be invoked.
     #   @return [Types::ActionGroupInvocationInput]
     #
+    # @!attribute [rw] agent_collaborator_invocation_input
+    #   The collaborator's invocation input.
+    #   @return [Types::AgentCollaboratorInvocationInput]
+    #
     # @!attribute [rw] code_interpreter_invocation_input
     #   Contains information about the code interpreter to be invoked.
     #   @return [Types::CodeInterpreterInvocationInput]
@@ -2260,6 +2555,7 @@ module Aws::BedrockAgentRuntime
     #
     class InvocationInput < Struct.new(
       :action_group_invocation_input,
+      :agent_collaborator_invocation_input,
       :code_interpreter_invocation_input,
       :invocation_type,
       :knowledge_base_lookup_input,
@@ -2326,6 +2622,8 @@ module Aws::BedrockAgentRuntime
     #
     # @note InvocationResultMember is a union - when making an API calls you must set exactly one of the members.
     #
+    # @note InvocationResultMember is a union - when returned from an API call exactly one value will be set and the returned type will be a subclass of InvocationResultMember corresponding to the set member.
+    #
     # @!attribute [rw] api_result
     #   The result from the API response from the action group invocation.
     #   @return [Types::ApiResult]
@@ -2402,6 +2700,10 @@ module Aws::BedrockAgentRuntime
     #   [1]: https://docs.aws.amazon.com/bedrock/latest/userguide/agents-session-state.html
     #   @return [Types::SessionState]
     #
+    # @!attribute [rw] source_arn
+    #   The ARN of the resource making the request.
+    #   @return [String]
+    #
     # @!attribute [rw] streaming_configurations
     #   Specifies the configurations for streaming.
     #   @return [Types::StreamingConfigurations]
@@ -2417,6 +2719,7 @@ module Aws::BedrockAgentRuntime
       :memory_id,
       :session_id,
       :session_state,
+      :source_arn,
       :streaming_configurations)
       SENSITIVE = [:input_text]
       include Aws::Structure
@@ -2914,6 +3217,10 @@ module Aws::BedrockAgentRuntime
     #   [1]: https://docs.aws.amazon.com/bedrock/latest/userguide/kb-test-config.html
     #   @return [Types::RetrievalFilter]
     #
+    # @!attribute [rw] implicit_filter_configuration
+    #   Settings for implicit filtering.
+    #   @return [Types::ImplicitFilterConfiguration]
+    #
     # @!attribute [rw] number_of_results
     #   The number of source chunks to retrieve.
     #   @return [Integer]
@@ -2933,12 +3240,24 @@ module Aws::BedrockAgentRuntime
     #   [1]: https://docs.aws.amazon.com/bedrock/latest/userguide/knowledge-base-test.html
     #   @return [String]
     #
+    # @!attribute [rw] reranking_configuration
+    #   Contains configurations for reranking the retrieved results. For
+    #   more information, see [Improve the relevance of query responses with
+    #   a reranker model][1].
+    #
+    #
+    #
+    #   [1]: https://docs.aws.amazon.com/bedrock/latest/userguide/rerank.html
+    #   @return [Types::VectorSearchRerankingConfiguration]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/KnowledgeBaseVectorSearchConfiguration AWS API Documentation
     #
     class KnowledgeBaseVectorSearchConfiguration < Struct.new(
       :filter,
+      :implicit_filter_configuration,
       :number_of_results,
-      :override_search_type)
+      :override_search_type,
+      :reranking_configuration)
       SENSITIVE = [:filter]
       include Aws::Structure
     end
@@ -2999,6 +3318,25 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
+    # Details about a message.
+    #
+    # @!attribute [rw] content
+    #   The message's content.
+    #   @return [Array<Types::ContentBlock>]
+    #
+    # @!attribute [rw] role
+    #   The message's role.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/Message AWS API Documentation
+    #
+    class Message < Struct.new(
+      :content,
+      :role)
+      SENSITIVE = [:content]
+      include Aws::Structure
+    end
+
     # Provides details of the foundation model.
     #
     # @!attribute [rw] usage
@@ -3013,6 +3351,52 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
+    # Details about a metadata attribute.
+    #
+    # @!attribute [rw] description
+    #   The attribute's description.
+    #   @return [String]
+    #
+    # @!attribute [rw] key
+    #   The attribute's key.
+    #   @return [String]
+    #
+    # @!attribute [rw] type
+    #   The attribute's type.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/MetadataAttributeSchema AWS API Documentation
+    #
+    class MetadataAttributeSchema < Struct.new(
+      :description,
+      :key,
+      :type)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Contains configurations for the metadata to use in reranking.
+    #
+    # @!attribute [rw] selection_mode
+    #   Specifies whether to consider all metadata when reranking, or only
+    #   the metadata that you select. If you specify `SELECTIVE`, include
+    #   the `selectiveModeConfiguration` field.
+    #   @return [String]
+    #
+    # @!attribute [rw] selective_mode_configuration
+    #   Contains configurations for the metadata fields to include or
+    #   exclude when considering reranking.
+    #   @return [Types::RerankingMetadataSelectiveModeConfiguration]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/MetadataConfigurationForReranking AWS API Documentation
+    #
+    class MetadataConfigurationForReranking < Struct.new(
+      :selection_mode,
+      :selective_mode_configuration)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # The input for the pre-processing step.
     #
     # * The `type` matches the agent step.
@@ -3027,6 +3411,10 @@ module Aws::BedrockAgentRuntime
     #
     # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent_PromptOverrideConfiguration.html
     #
+    # @!attribute [rw] foundation_model
+    #   The identifier of a foundation model.
+    #   @return [String]
+    #
     # @!attribute [rw] inference_configuration
     #   Specifications about the inference parameters that were provided
     #   alongside the prompt. These are specified in the
@@ -3077,6 +3465,7 @@ module Aws::BedrockAgentRuntime
     # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/ModelInvocationInput AWS API Documentation
     #
     class ModelInvocationInput < Struct.new(
+      :foundation_model,
       :inference_configuration,
       :override_lambda,
       :parser_mode,
@@ -3096,6 +3485,10 @@ module Aws::BedrockAgentRuntime
     #   the action group.
     #   @return [Types::ActionGroupInvocationOutput]
     #
+    # @!attribute [rw] agent_collaborator_invocation_output
+    #   A collaborator's invocation output.
+    #   @return [Types::AgentCollaboratorInvocationOutput]
+    #
     # @!attribute [rw] code_interpreter_invocation_output
     #   Contains the JSON-formatted string returned by the API invoked by
     #   the code interpreter.
@@ -3140,6 +3533,7 @@ module Aws::BedrockAgentRuntime
     #
     class Observation < Struct.new(
       :action_group_invocation_output,
+      :agent_collaborator_invocation_output,
       :code_interpreter_invocation_output,
       :final_response,
       :knowledge_base_lookup_output,
@@ -3885,6 +4279,213 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
+    # Contains information about a document to rerank. Choose the `type` to
+    # define and include the field that corresponds to the type.
+    #
+    # @!attribute [rw] json_document
+    #   Contains a JSON document to rerank.
+    #   @return [Hash,Array,String,Numeric,Boolean]
+    #
+    # @!attribute [rw] text_document
+    #   Contains information about a text document to rerank.
+    #   @return [Types::RerankTextDocument]
+    #
+    # @!attribute [rw] type
+    #   The type of document to rerank.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/RerankDocument AWS API Documentation
+    #
+    class RerankDocument < Struct.new(
+      :json_document,
+      :text_document,
+      :type)
+      SENSITIVE = [:text_document]
+      include Aws::Structure
+    end
+
+    # Contains information about a query to submit to the reranker model.
+    #
+    # @!attribute [rw] text_query
+    #   Contains information about a text query.
+    #   @return [Types::RerankTextDocument]
+    #
+    # @!attribute [rw] type
+    #   The type of the query.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/RerankQuery AWS API Documentation
+    #
+    class RerankQuery < Struct.new(
+      :text_query,
+      :type)
+      SENSITIVE = [:text_query]
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] next_token
+    #   If the total number of results was greater than could fit in a
+    #   response, a token is returned in the `nextToken` field. You can
+    #   enter that token in this field to return the next batch of results.
+    #   @return [String]
+    #
+    # @!attribute [rw] queries
+    #   An array of objects, each of which contains information about a
+    #   query to submit to the reranker model.
+    #   @return [Array<Types::RerankQuery>]
+    #
+    # @!attribute [rw] reranking_configuration
+    #   Contains configurations for reranking.
+    #   @return [Types::RerankingConfiguration]
+    #
+    # @!attribute [rw] sources
+    #   An array of objects, each of which contains information about the
+    #   sources to rerank.
+    #   @return [Array<Types::RerankSource>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/RerankRequest AWS API Documentation
+    #
+    class RerankRequest < Struct.new(
+      :next_token,
+      :queries,
+      :reranking_configuration,
+      :sources)
+      SENSITIVE = [:queries, :sources]
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] next_token
+    #   If the total number of results is greater than can fit in the
+    #   response, use this token in the `nextToken` field when making
+    #   another request to return the next batch of results.
+    #   @return [String]
+    #
+    # @!attribute [rw] results
+    #   An array of objects, each of which contains information about the
+    #   results of reranking.
+    #   @return [Array<Types::RerankResult>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/RerankResponse AWS API Documentation
+    #
+    class RerankResponse < Struct.new(
+      :next_token,
+      :results)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Contains information about a document that was reranked.
+    #
+    # @!attribute [rw] document
+    #   Contains information about the document.
+    #   @return [Types::RerankDocument]
+    #
+    # @!attribute [rw] index
+    #   The ranking of the document. The lower a number, the higher the
+    #   document is ranked.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] relevance_score
+    #   The relevance score of the document.
+    #   @return [Float]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/RerankResult AWS API Documentation
+    #
+    class RerankResult < Struct.new(
+      :document,
+      :index,
+      :relevance_score)
+      SENSITIVE = [:document]
+      include Aws::Structure
+    end
+
+    # Contains information about a source for reranking.
+    #
+    # @!attribute [rw] inline_document_source
+    #   Contains an inline definition of a source for reranking.
+    #   @return [Types::RerankDocument]
+    #
+    # @!attribute [rw] type
+    #   The type of the source.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/RerankSource AWS API Documentation
+    #
+    class RerankSource < Struct.new(
+      :inline_document_source,
+      :type)
+      SENSITIVE = [:inline_document_source]
+      include Aws::Structure
+    end
+
+    # Contains information about a text document to rerank.
+    #
+    # @!attribute [rw] text
+    #   The text of the document.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/RerankTextDocument AWS API Documentation
+    #
+    class RerankTextDocument < Struct.new(
+      :text)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Contains configurations for reranking.
+    #
+    # @!attribute [rw] bedrock_reranking_configuration
+    #   Contains configurations for an Amazon Bedrock reranker.
+    #   @return [Types::BedrockRerankingConfiguration]
+    #
+    # @!attribute [rw] type
+    #   The type of reranker that the configurations apply to.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/RerankingConfiguration AWS API Documentation
+    #
+    class RerankingConfiguration < Struct.new(
+      :bedrock_reranking_configuration,
+      :type)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Contains configurations for the metadata fields to include or exclude
+    # when considering reranking. If you include the `fieldsToExclude`
+    # field, the reranker ignores all the metadata fields that you specify.
+    # If you include the `fieldsToInclude` field, the reranker uses only the
+    # metadata fields that you specify and ignores all others. You can
+    # include only one of these fields.
+    #
+    # @note RerankingMetadataSelectiveModeConfiguration is a union - when making an API calls you must set exactly one of the members.
+    #
+    # @!attribute [rw] fields_to_exclude
+    #   An array of objects, each of which specifies a metadata field to
+    #   exclude from consideration when reranking.
+    #   @return [Array<Types::FieldForReranking>]
+    #
+    # @!attribute [rw] fields_to_include
+    #   An array of objects, each of which specifies a metadata field to
+    #   include in consideration when reranking. The remaining metadata
+    #   fields are ignored.
+    #   @return [Array<Types::FieldForReranking>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/RerankingMetadataSelectiveModeConfiguration AWS API Documentation
+    #
+    class RerankingMetadataSelectiveModeConfiguration < Struct.new(
+      :fields_to_exclude,
+      :fields_to_include,
+      :unknown)
+      SENSITIVE = [:fields_to_exclude, :fields_to_include]
+      include Aws::Structure
+      include Aws::Structure::Union
+
+      class FieldsToExclude < RerankingMetadataSelectiveModeConfiguration; end
+      class FieldsToInclude < RerankingMetadataSelectiveModeConfiguration; end
+      class Unknown < RerankingMetadataSelectiveModeConfiguration; end
+    end
+
     # The specified resource Amazon Resource Name (ARN) was not found. Check
     # the Amazon Resource Name (ARN) and try your request again.
     #
@@ -4140,6 +4741,21 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
+    # Contains information about the location of a document in a custom data
+    # source.
+    #
+    # @!attribute [rw] id
+    #   The ID of the document.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/RetrievalResultCustomDocumentLocation AWS API Documentation
+    #
+    class RetrievalResultCustomDocumentLocation < Struct.new(
+      :id)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # Contains information about the data source location.
     #
     # This data type is used in the following API operations:
@@ -4160,6 +4776,10 @@ module Aws::BedrockAgentRuntime
     #   The Confluence data source location.
     #   @return [Types::RetrievalResultConfluenceLocation]
     #
+    # @!attribute [rw] custom_document_location
+    #   Specifies the location of a document in a custom data source.
+    #   @return [Types::RetrievalResultCustomDocumentLocation]
+    #
     # @!attribute [rw] s3_location
     #   The S3 data source location.
     #   @return [Types::RetrievalResultS3Location]
@@ -4184,6 +4804,7 @@ module Aws::BedrockAgentRuntime
     #
     class RetrievalResultLocation < Struct.new(
       :confluence_location,
+      :custom_document_location,
       :s3_location,
       :salesforce_location,
       :share_point_location,
@@ -4352,6 +4973,21 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
+    # A retrieve and generate output event.
+    #
+    # @!attribute [rw] text
+    #   A text response.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/RetrieveAndGenerateOutputEvent AWS API Documentation
+    #
+    class RetrieveAndGenerateOutputEvent < Struct.new(
+      :text,
+      :event_type)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # @!attribute [rw] input
     #   Contains the query to be made to the knowledge base.
     #   @return [Types::RetrieveAndGenerateInput]
@@ -4448,6 +5084,64 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
+    # @!attribute [rw] input
+    #   Contains the query to be made to the knowledge base.
+    #   @return [Types::RetrieveAndGenerateInput]
+    #
+    # @!attribute [rw] retrieve_and_generate_configuration
+    #   Contains configurations for the knowledge base query and retrieval
+    #   process. For more information, see [Query configurations][1].
+    #
+    #
+    #
+    #   [1]: https://docs.aws.amazon.com/bedrock/latest/userguide/kb-test-config.html
+    #   @return [Types::RetrieveAndGenerateConfiguration]
+    #
+    # @!attribute [rw] session_configuration
+    #   Contains details about the session with the knowledge base.
+    #   @return [Types::RetrieveAndGenerateSessionConfiguration]
+    #
+    # @!attribute [rw] session_id
+    #   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/RetrieveAndGenerateStreamRequest AWS API Documentation
+    #
+    class RetrieveAndGenerateStreamRequest < Struct.new(
+      :input,
+      :retrieve_and_generate_configuration,
+      :session_configuration,
+      :session_id)
+      SENSITIVE = [:input]
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] session_id
+    #   The session ID.
+    #   @return [String]
+    #
+    # @!attribute [rw] stream
+    #   A stream of events from the model.
+    #   @return [Types::RetrieveAndGenerateStreamResponseOutput]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/RetrieveAndGenerateStreamResponse AWS API Documentation
+    #
+    class RetrieveAndGenerateStreamResponse < Struct.new(
+      :session_id,
+      :stream)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] guardrail_configuration
+    #   Guardrail settings.
+    #   @return [Types::GuardrailConfiguration]
+    #
     # @!attribute [rw] knowledge_base_id
     #   The unique identifier of the knowledge base to query.
     #   @return [String]
@@ -4474,6 +5168,7 @@ module Aws::BedrockAgentRuntime
     # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/RetrieveRequest AWS API Documentation
     #
     class RetrieveRequest < Struct.new(
+      :guardrail_configuration,
       :knowledge_base_id,
       :next_token,
       :retrieval_configuration,
@@ -4482,6 +5177,10 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
+    # @!attribute [rw] guardrail_action
+    #   Specifies if there is a guardrail intervention in the response.
+    #   @return [String]
+    #
     # @!attribute [rw] next_token
     #   If there are more results than can fit in the response, the response
     #   returns a `nextToken`. Use this token in the `nextToken` field of
@@ -4495,6 +5194,7 @@ module Aws::BedrockAgentRuntime
     # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/RetrieveResponse AWS API Documentation
     #
     class RetrieveResponse < Struct.new(
+      :guardrail_action,
       :next_token,
       :retrieval_results)
       SENSITIVE = [:retrieval_results]
@@ -4575,6 +5275,88 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
+    # An action invocation result.
+    #
+    # @!attribute [rw] invocation_id
+    #   The action's invocation ID.
+    #   @return [String]
+    #
+    # @!attribute [rw] return_control_invocation_results
+    #   The action invocation result.
+    #   @return [Array<Types::InvocationResultMember>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/ReturnControlResults AWS API Documentation
+    #
+    class ReturnControlResults < Struct.new(
+      :invocation_id,
+      :return_control_invocation_results)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Invocation output from a routing classifier model.
+    #
+    # @!attribute [rw] metadata
+    #   The invocation's metadata.
+    #   @return [Types::Metadata]
+    #
+    # @!attribute [rw] raw_response
+    #   The invocation's raw response.
+    #   @return [Types::RawResponse]
+    #
+    # @!attribute [rw] trace_id
+    #   The invocation's trace ID.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/RoutingClassifierModelInvocationOutput AWS API Documentation
+    #
+    class RoutingClassifierModelInvocationOutput < Struct.new(
+      :metadata,
+      :raw_response,
+      :trace_id)
+      SENSITIVE = [:metadata, :raw_response]
+      include Aws::Structure
+    end
+
+    # A trace for a routing classifier.
+    #
+    # @note RoutingClassifierTrace is a union - when returned from an API call exactly one value will be set and the returned type will be a subclass of RoutingClassifierTrace corresponding to the set member.
+    #
+    # @!attribute [rw] invocation_input
+    #   The classifier's invocation input.
+    #   @return [Types::InvocationInput]
+    #
+    # @!attribute [rw] model_invocation_input
+    #   The classifier's model invocation input.
+    #   @return [Types::ModelInvocationInput]
+    #
+    # @!attribute [rw] model_invocation_output
+    #   The classifier's model invocation output.
+    #   @return [Types::RoutingClassifierModelInvocationOutput]
+    #
+    # @!attribute [rw] observation
+    #   The classifier's observation.
+    #   @return [Types::Observation]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/RoutingClassifierTrace AWS API Documentation
+    #
+    class RoutingClassifierTrace < Struct.new(
+      :invocation_input,
+      :model_invocation_input,
+      :model_invocation_output,
+      :observation,
+      :unknown)
+      SENSITIVE = [:invocation_input, :model_invocation_input, :model_invocation_output, :observation]
+      include Aws::Structure
+      include Aws::Structure::Union
+
+      class InvocationInput < RoutingClassifierTrace; end
+      class ModelInvocationInput < RoutingClassifierTrace; end
+      class ModelInvocationOutput < RoutingClassifierTrace; end
+      class Observation < RoutingClassifierTrace; end
+      class Unknown < RoutingClassifierTrace; end
+    end
+
     # The identifier information for an Amazon S3 bucket.
     #
     # @!attribute [rw] s3_bucket_name
@@ -4651,6 +5433,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] conversation_history
+    #   The state's conversation history.
+    #   @return [Types::ConversationHistory]
+    #
     # @!attribute [rw] files
     #   Contains information about the files used by code interpreter.
     #   @return [Array<Types::InputFile>]
@@ -4709,6 +5495,7 @@ 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(
+      :conversation_history,
       :files,
       :invocation_id,
       :knowledge_base_configurations,
@@ -4926,6 +5713,10 @@ module Aws::BedrockAgentRuntime
     #   contextualizes and categorizes user inputs.
     #   @return [Types::PreProcessingTrace]
     #
+    # @!attribute [rw] routing_classifier_trace
+    #   A routing classifier's trace.
+    #   @return [Types::RoutingClassifierTrace]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/Trace AWS API Documentation
     #
     class Trace < Struct.new(
@@ -4935,8 +5726,9 @@ module Aws::BedrockAgentRuntime
       :orchestration_trace,
       :post_processing_trace,
       :pre_processing_trace,
+      :routing_classifier_trace,
       :unknown)
-      SENSITIVE = [:custom_orchestration_trace, :failure_trace, :guardrail_trace, :orchestration_trace, :post_processing_trace, :pre_processing_trace]
+      SENSITIVE = [:custom_orchestration_trace, :failure_trace, :guardrail_trace, :orchestration_trace, :post_processing_trace, :pre_processing_trace, :routing_classifier_trace]
       include Aws::Structure
       include Aws::Structure::Union
 
@@ -4946,6 +5738,7 @@ module Aws::BedrockAgentRuntime
       class OrchestrationTrace < Trace; end
       class PostProcessingTrace < Trace; end
       class PreProcessingTrace < Trace; end
+      class RoutingClassifierTrace < Trace; end
       class Unknown < Trace; end
     end
 
@@ -4971,6 +5764,14 @@ module Aws::BedrockAgentRuntime
     #   The version of the agent.
     #   @return [String]
     #
+    # @!attribute [rw] caller_chain
+    #   The part's caller chain.
+    #   @return [Array<Types::Caller>]
+    #
+    # @!attribute [rw] collaborator_name
+    #   The part's collaborator name.
+    #   @return [String]
+    #
     # @!attribute [rw] session_id
     #   The unique identifier of the session with the agent.
     #   @return [String]
@@ -4993,10 +5794,12 @@ module Aws::BedrockAgentRuntime
       :agent_alias_id,
       :agent_id,
       :agent_version,
+      :caller_chain,
+      :collaborator_name,
       :session_id,
       :trace,
       :event_type)
-      SENSITIVE = [:trace]
+      SENSITIVE = [:collaborator_name, :trace]
       include Aws::Structure
     end
 
@@ -5036,6 +5839,70 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
+    # Contains configurations for reranking with an Amazon Bedrock reranker
+    # model.
+    #
+    # @!attribute [rw] metadata_configuration
+    #   Contains configurations for the metadata to use in reranking.
+    #   @return [Types::MetadataConfigurationForReranking]
+    #
+    # @!attribute [rw] model_configuration
+    #   Contains configurations for the reranker model.
+    #   @return [Types::VectorSearchBedrockRerankingModelConfiguration]
+    #
+    # @!attribute [rw] number_of_reranked_results
+    #   The number of results to return after reranking.
+    #   @return [Integer]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/VectorSearchBedrockRerankingConfiguration AWS API Documentation
+    #
+    class VectorSearchBedrockRerankingConfiguration < Struct.new(
+      :metadata_configuration,
+      :model_configuration,
+      :number_of_reranked_results)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Contains configurations for an Amazon Bedrock reranker model.
+    #
+    # @!attribute [rw] additional_model_request_fields
+    #   A JSON object whose keys are request fields for the model and whose
+    #   values are values for those fields.
+    #   @return [Hash<String,Hash,Array,String,Numeric,Boolean>]
+    #
+    # @!attribute [rw] model_arn
+    #   The ARN of the reranker model to use.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/VectorSearchBedrockRerankingModelConfiguration AWS API Documentation
+    #
+    class VectorSearchBedrockRerankingModelConfiguration < Struct.new(
+      :additional_model_request_fields,
+      :model_arn)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Contains configurations for reranking the retrieved results.
+    #
+    # @!attribute [rw] bedrock_reranking_configuration
+    #   Contains configurations for an Amazon Bedrock reranker model.
+    #   @return [Types::VectorSearchBedrockRerankingConfiguration]
+    #
+    # @!attribute [rw] type
+    #   The type of reranker model.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/VectorSearchRerankingConfiguration AWS API Documentation
+    #
+    class VectorSearchRerankingConfiguration < Struct.new(
+      :bedrock_reranking_configuration,
+      :type)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # The output of the flow.
     #
     # EventStream is an Enumerator of Events.
@@ -5148,6 +6015,34 @@ module Aws::BedrockAgentRuntime
 
     end
 
+    # A retrieve and generate stream response output.
+    #
+    # 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/RetrieveAndGenerateStreamResponseOutput AWS API Documentation
+    #
+    class RetrieveAndGenerateStreamResponseOutput < Enumerator
+
+      def event_types
+        [
+          :access_denied_exception,
+          :bad_gateway_exception,
+          :citation,
+          :conflict_exception,
+          :dependency_failed_exception,
+          :guardrail,
+          :internal_server_exception,
+          :output,
+          :resource_not_found_exception,
+          :service_quota_exceeded_exception,
+          :throttling_exception,
+          :validation_exception
+        ]
+      end
+
+    end
+
   end
 end