aws-sdk-bedrockagentruntime 1.34.0 → 1.36.0

data/sig/types.rbs

@@ -69,6 +69,34 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: [:action_group_name, :description]
     end
 
+    class AgentCollaboratorInputPayload
+      attr_accessor return_control_results: Types::ReturnControlResults
+      attr_accessor text: ::String
+      attr_accessor type: ("TEXT" | "RETURN_CONTROL")
+      SENSITIVE: [:text]
+    end
+
+    class AgentCollaboratorInvocationInput
+      attr_accessor agent_collaborator_alias_arn: ::String
+      attr_accessor agent_collaborator_name: ::String
+      attr_accessor input: Types::AgentCollaboratorInputPayload
+      SENSITIVE: []
+    end
+
+    class AgentCollaboratorInvocationOutput
+      attr_accessor agent_collaborator_alias_arn: ::String
+      attr_accessor agent_collaborator_name: ::String
+      attr_accessor output: Types::AgentCollaboratorOutputPayload
+      SENSITIVE: []
+    end
+
+    class AgentCollaboratorOutputPayload
+      attr_accessor return_control_payload: Types::ReturnControlPayload
+      attr_accessor text: ::String
+      attr_accessor type: ("TEXT" | "RETURN_CONTROL")
+      SENSITIVE: [:return_control_payload, :text]
+    end
+
     class AnalyzePromptEvent
       attr_accessor message: ::String
       attr_accessor event_type: untyped
@@ -78,11 +106,13 @@ module Aws::BedrockAgentRuntime
     class ApiInvocationInput
       attr_accessor action_group: ::String
       attr_accessor action_invocation_type: ("RESULT" | "USER_CONFIRMATION" | "USER_CONFIRMATION_AND_RESULT")
+      attr_accessor agent_id: ::String
       attr_accessor api_path: ::String
+      attr_accessor collaborator_name: ::String
       attr_accessor http_method: ::String
       attr_accessor parameters: ::Array[Types::ApiParameter]
       attr_accessor request_body: Types::ApiRequestBody
-      SENSITIVE: [:api_path]
+      SENSITIVE: [:api_path, :collaborator_name]
     end
 
     class ApiParameter
@@ -99,6 +129,7 @@ module Aws::BedrockAgentRuntime
 
     class ApiResult
       attr_accessor action_group: ::String
+      attr_accessor agent_id: ::String
       attr_accessor api_path: ::String
       attr_accessor confirmation_state: ("CONFIRM" | "DENY")
       attr_accessor http_method: ::String
@@ -120,6 +151,18 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: []
     end
 
+    class BedrockRerankingConfiguration
+      attr_accessor model_configuration: Types::BedrockRerankingModelConfiguration
+      attr_accessor number_of_results: ::Integer
+      SENSITIVE: []
+    end
+
+    class BedrockRerankingModelConfiguration
+      attr_accessor additional_model_request_fields: ::Hash[::String, untyped]
+      attr_accessor model_arn: ::String
+      SENSITIVE: []
+    end
+
     class ByteContentDoc
       attr_accessor content_type: ::String
       attr_accessor data: ::String
@@ -133,12 +176,29 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: [:data]
     end
 
+    class Caller
+      attr_accessor agent_alias_arn: ::String
+      attr_accessor unknown: untyped
+      SENSITIVE: []
+
+      class AgentAliasArn < Caller
+      end
+      class Unknown < Caller
+      end
+    end
+
     class Citation
       attr_accessor generated_response_part: Types::GeneratedResponsePart
       attr_accessor retrieved_references: ::Array[Types::RetrievedReference]
       SENSITIVE: []
     end
 
+    class CitationEvent
+      attr_accessor citation: Types::Citation
+      attr_accessor event_type: untyped
+      SENSITIVE: []
+    end
+
     class CodeInterpreterInvocationInput
       attr_accessor code: ::String
       attr_accessor files: ::Array[::String]
@@ -159,11 +219,27 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: []
     end
 
+    class ContentBlock
+      attr_accessor text: ::String
+      attr_accessor unknown: untyped
+      SENSITIVE: []
+
+      class Text < ContentBlock
+      end
+      class Unknown < ContentBlock
+      end
+    end
+
     class ContentBody
       attr_accessor body: ::String
       SENSITIVE: []
     end
 
+    class ConversationHistory
+      attr_accessor messages: ::Array[Types::Message]
+      SENSITIVE: []
+    end
+
     class CustomOrchestrationTrace
       attr_accessor event: Types::CustomOrchestrationTraceEvent
       attr_accessor trace_id: ::String
@@ -220,6 +296,11 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: [:failure_reason]
     end
 
+    class FieldForReranking
+      attr_accessor field_name: ::String
+      SENSITIVE: []
+    end
+
     class FilePart
       attr_accessor files: ::Array[Types::OutputFile]
       attr_accessor event_type: untyped
@@ -381,9 +462,11 @@ module Aws::BedrockAgentRuntime
     class FunctionInvocationInput
       attr_accessor action_group: ::String
       attr_accessor action_invocation_type: ("RESULT" | "USER_CONFIRMATION" | "USER_CONFIRMATION_AND_RESULT")
+      attr_accessor agent_id: ::String
+      attr_accessor collaborator_name: ::String
       attr_accessor function: ::String
       attr_accessor parameters: ::Array[Types::FunctionParameter]
-      SENSITIVE: []
+      SENSITIVE: [:collaborator_name]
     end
 
     class FunctionParameter
@@ -395,6 +478,7 @@ module Aws::BedrockAgentRuntime
 
     class FunctionResult
       attr_accessor action_group: ::String
+      attr_accessor agent_id: ::String
       attr_accessor confirmation_state: ("CONFIRM" | "DENY")
       attr_accessor function: ::String
       attr_accessor response_body: ::Hash[::String, Types::ContentBody]
@@ -480,6 +564,12 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: []
     end
 
+    class GuardrailEvent
+      attr_accessor action: ("INTERVENED" | "NONE")
+      attr_accessor event_type: untyped
+      SENSITIVE: []
+    end
+
     class GuardrailManagedWord
       attr_accessor action: ("BLOCKED")
       attr_accessor match: ::String
@@ -534,6 +624,12 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: [:custom_words, :managed_word_lists]
     end
 
+    class ImplicitFilterConfiguration
+      attr_accessor metadata_attributes: ::Array[Types::MetadataAttributeSchema]
+      attr_accessor model_arn: ::String
+      SENSITIVE: []
+    end
+
     class InferenceConfig
       attr_accessor text_inference_config: Types::TextInferenceConfig
       SENSITIVE: []
@@ -610,8 +706,9 @@ module Aws::BedrockAgentRuntime
 
     class InvocationInput
       attr_accessor action_group_invocation_input: Types::ActionGroupInvocationInput
+      attr_accessor agent_collaborator_invocation_input: Types::AgentCollaboratorInvocationInput
       attr_accessor code_interpreter_invocation_input: Types::CodeInterpreterInvocationInput
-      attr_accessor invocation_type: ("ACTION_GROUP" | "KNOWLEDGE_BASE" | "FINISH" | "ACTION_GROUP_CODE_INTERPRETER")
+      attr_accessor invocation_type: ("ACTION_GROUP" | "KNOWLEDGE_BASE" | "FINISH" | "ACTION_GROUP_CODE_INTERPRETER" | "AGENT_COLLABORATOR")
       attr_accessor knowledge_base_lookup_input: Types::KnowledgeBaseLookupInput
       attr_accessor trace_id: ::String
       SENSITIVE: []
@@ -654,6 +751,7 @@ module Aws::BedrockAgentRuntime
       attr_accessor memory_id: ::String
       attr_accessor session_id: ::String
       attr_accessor session_state: Types::SessionState
+      attr_accessor source_arn: ::String
       attr_accessor streaming_configurations: Types::StreamingConfigurations
       SENSITIVE: [:input_text]
     end
@@ -756,8 +854,10 @@ module Aws::BedrockAgentRuntime
 
     class KnowledgeBaseVectorSearchConfiguration
       attr_accessor filter: Types::RetrievalFilter
+      attr_accessor implicit_filter_configuration: Types::ImplicitFilterConfiguration
       attr_accessor number_of_results: ::Integer
       attr_accessor override_search_type: ("HYBRID" | "SEMANTIC")
+      attr_accessor reranking_configuration: Types::VectorSearchRerankingConfiguration
       SENSITIVE: [:filter]
     end
 
@@ -781,12 +881,32 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: []
     end
 
+    class Message
+      attr_accessor content: ::Array[Types::ContentBlock]
+      attr_accessor role: ("user" | "assistant")
+      SENSITIVE: []
+    end
+
     class Metadata
       attr_accessor usage: Types::Usage
       SENSITIVE: [:usage]
     end
 
+    class MetadataAttributeSchema
+      attr_accessor description: ::String
+      attr_accessor key: ::String
+      attr_accessor type: ("STRING" | "NUMBER" | "BOOLEAN" | "STRING_LIST")
+      SENSITIVE: []
+    end
+
+    class MetadataConfigurationForReranking
+      attr_accessor selection_mode: ("SELECTIVE" | "ALL")
+      attr_accessor selective_mode_configuration: Types::RerankingMetadataSelectiveModeConfiguration
+      SENSITIVE: []
+    end
+
     class ModelInvocationInput
+      attr_accessor foundation_model: ::String
       attr_accessor inference_configuration: Types::InferenceConfiguration
       attr_accessor override_lambda: ::String
       attr_accessor parser_mode: ("DEFAULT" | "OVERRIDDEN")
@@ -799,12 +919,13 @@ module Aws::BedrockAgentRuntime
 
     class Observation
       attr_accessor action_group_invocation_output: Types::ActionGroupInvocationOutput
+      attr_accessor agent_collaborator_invocation_output: Types::AgentCollaboratorInvocationOutput
       attr_accessor code_interpreter_invocation_output: Types::CodeInterpreterInvocationOutput
       attr_accessor final_response: Types::FinalResponse
       attr_accessor knowledge_base_lookup_output: Types::KnowledgeBaseLookupOutput
       attr_accessor reprompt_response: Types::RepromptResponse
       attr_accessor trace_id: ::String
-      attr_accessor type: ("ACTION_GROUP" | "KNOWLEDGE_BASE" | "FINISH" | "ASK_USER" | "REPROMPT")
+      attr_accessor type: ("ACTION_GROUP" | "AGENT_COLLABORATOR" | "KNOWLEDGE_BASE" | "FINISH" | "ASK_USER" | "REPROMPT")
       SENSITIVE: [:reprompt_response]
     end
 
@@ -1010,6 +1131,71 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: []
     end
 
+    class RerankDocument
+      attr_accessor json_document: untyped
+      attr_accessor text_document: Types::RerankTextDocument
+      attr_accessor type: ("TEXT" | "JSON")
+      SENSITIVE: [:text_document]
+    end
+
+    class RerankQuery
+      attr_accessor text_query: Types::RerankTextDocument
+      attr_accessor type: ("TEXT")
+      SENSITIVE: [:text_query]
+    end
+
+    class RerankRequest
+      attr_accessor next_token: ::String
+      attr_accessor queries: ::Array[Types::RerankQuery]
+      attr_accessor reranking_configuration: Types::RerankingConfiguration
+      attr_accessor sources: ::Array[Types::RerankSource]
+      SENSITIVE: [:queries, :sources]
+    end
+
+    class RerankResponse
+      attr_accessor next_token: ::String
+      attr_accessor results: ::Array[Types::RerankResult]
+      SENSITIVE: []
+    end
+
+    class RerankResult
+      attr_accessor document: Types::RerankDocument
+      attr_accessor index: ::Integer
+      attr_accessor relevance_score: ::Float
+      SENSITIVE: [:document]
+    end
+
+    class RerankSource
+      attr_accessor inline_document_source: Types::RerankDocument
+      attr_accessor type: ("INLINE")
+      SENSITIVE: [:inline_document_source]
+    end
+
+    class RerankTextDocument
+      attr_accessor text: ::String
+      SENSITIVE: []
+    end
+
+    class RerankingConfiguration
+      attr_accessor bedrock_reranking_configuration: Types::BedrockRerankingConfiguration
+      attr_accessor type: ("BEDROCK_RERANKING_MODEL")
+      SENSITIVE: []
+    end
+
+    class RerankingMetadataSelectiveModeConfiguration
+      attr_accessor fields_to_exclude: ::Array[Types::FieldForReranking]
+      attr_accessor fields_to_include: ::Array[Types::FieldForReranking]
+      attr_accessor unknown: untyped
+      SENSITIVE: [:fields_to_exclude, :fields_to_include]
+
+      class FieldsToExclude < RerankingMetadataSelectiveModeConfiguration
+      end
+      class FieldsToInclude < RerankingMetadataSelectiveModeConfiguration
+      end
+      class Unknown < RerankingMetadataSelectiveModeConfiguration
+      end
+    end
+
     class ResourceNotFoundException
       attr_accessor message: ::String
       attr_accessor event_type: untyped
@@ -1073,12 +1259,18 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: []
     end
 
+    class RetrievalResultCustomDocumentLocation
+      attr_accessor id: ::String
+      SENSITIVE: []
+    end
+
     class RetrievalResultLocation
       attr_accessor confluence_location: Types::RetrievalResultConfluenceLocation
+      attr_accessor custom_document_location: Types::RetrievalResultCustomDocumentLocation
       attr_accessor s3_location: Types::RetrievalResultS3Location
       attr_accessor salesforce_location: Types::RetrievalResultSalesforceLocation
       attr_accessor share_point_location: Types::RetrievalResultSharePointLocation
-      attr_accessor type: ("S3" | "WEB" | "CONFLUENCE" | "SALESFORCE" | "SHAREPOINT")
+      attr_accessor type: ("S3" | "WEB" | "CONFLUENCE" | "SALESFORCE" | "SHAREPOINT" | "CUSTOM")
       attr_accessor web_location: Types::RetrievalResultWebLocation
       SENSITIVE: []
     end
@@ -1120,6 +1312,12 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: []
     end
 
+    class RetrieveAndGenerateOutputEvent
+      attr_accessor text: ::String
+      attr_accessor event_type: untyped
+      SENSITIVE: []
+    end
+
     class RetrieveAndGenerateRequest
       attr_accessor input: Types::RetrieveAndGenerateInput
       attr_accessor retrieve_and_generate_configuration: Types::RetrieveAndGenerateConfiguration
@@ -1141,7 +1339,22 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: []
     end
 
+    class RetrieveAndGenerateStreamRequest
+      attr_accessor input: Types::RetrieveAndGenerateInput
+      attr_accessor retrieve_and_generate_configuration: Types::RetrieveAndGenerateConfiguration
+      attr_accessor session_configuration: Types::RetrieveAndGenerateSessionConfiguration
+      attr_accessor session_id: ::String
+      SENSITIVE: [:input]
+    end
+
+    class RetrieveAndGenerateStreamResponse
+      attr_accessor session_id: ::String
+      attr_accessor stream: Types::RetrieveAndGenerateStreamResponseOutput
+      SENSITIVE: []
+    end
+
     class RetrieveRequest
+      attr_accessor guardrail_configuration: Types::GuardrailConfiguration
       attr_accessor knowledge_base_id: ::String
       attr_accessor next_token: ::String
       attr_accessor retrieval_configuration: Types::KnowledgeBaseRetrievalConfiguration
@@ -1150,6 +1363,7 @@ module Aws::BedrockAgentRuntime
     end
 
     class RetrieveResponse
+      attr_accessor guardrail_action: ("INTERVENED" | "NONE")
       attr_accessor next_token: ::String
       attr_accessor retrieval_results: ::Array[Types::KnowledgeBaseRetrievalResult]
       SENSITIVE: [:retrieval_results]
@@ -1169,6 +1383,39 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: []
     end
 
+    class ReturnControlResults
+      attr_accessor invocation_id: ::String
+      attr_accessor return_control_invocation_results: ::Array[Types::InvocationResultMember]
+      SENSITIVE: []
+    end
+
+    class RoutingClassifierModelInvocationOutput
+      attr_accessor metadata: Types::Metadata
+      attr_accessor raw_response: Types::RawResponse
+      attr_accessor trace_id: ::String
+      SENSITIVE: [:metadata, :raw_response]
+    end
+
+    class RoutingClassifierTrace
+      attr_accessor invocation_input: Types::InvocationInput
+      attr_accessor model_invocation_input: Types::ModelInvocationInput
+      attr_accessor model_invocation_output: Types::RoutingClassifierModelInvocationOutput
+      attr_accessor observation: Types::Observation
+      attr_accessor unknown: untyped
+      SENSITIVE: [:invocation_input, :model_invocation_input, :model_invocation_output, :observation]
+
+      class InvocationInput < RoutingClassifierTrace
+      end
+      class ModelInvocationInput < RoutingClassifierTrace
+      end
+      class ModelInvocationOutput < RoutingClassifierTrace
+      end
+      class Observation < RoutingClassifierTrace
+      end
+      class Unknown < RoutingClassifierTrace
+      end
+    end
+
     class S3Identifier
       attr_accessor s3_bucket_name: ::String
       attr_accessor s3_object_key: ::String
@@ -1192,6 +1439,7 @@ module Aws::BedrockAgentRuntime
     end
 
     class SessionState
+      attr_accessor conversation_history: Types::ConversationHistory
       attr_accessor files: ::Array[Types::InputFile]
       attr_accessor invocation_id: ::String
       attr_accessor knowledge_base_configurations: ::Array[Types::KnowledgeBaseConfiguration]
@@ -1245,8 +1493,9 @@ module Aws::BedrockAgentRuntime
       attr_accessor orchestration_trace: Types::OrchestrationTrace
       attr_accessor post_processing_trace: Types::PostProcessingTrace
       attr_accessor pre_processing_trace: Types::PreProcessingTrace
+      attr_accessor routing_classifier_trace: Types::RoutingClassifierTrace
       attr_accessor unknown: untyped
-      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]
 
       class CustomOrchestrationTrace < Trace
       end
@@ -1260,6 +1509,8 @@ module Aws::BedrockAgentRuntime
       end
       class PreProcessingTrace < Trace
       end
+      class RoutingClassifierTrace < Trace
+      end
       class Unknown < Trace
       end
     end
@@ -1268,10 +1519,12 @@ module Aws::BedrockAgentRuntime
       attr_accessor agent_alias_id: ::String
       attr_accessor agent_id: ::String
       attr_accessor agent_version: ::String
+      attr_accessor caller_chain: ::Array[Types::Caller]
+      attr_accessor collaborator_name: ::String
       attr_accessor session_id: ::String
       attr_accessor trace: Types::Trace
       attr_accessor event_type: untyped
-      SENSITIVE: [:trace]
+      SENSITIVE: [:collaborator_name, :trace]
     end
 
     class Usage
@@ -1286,6 +1539,25 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: []
     end
 
+    class VectorSearchBedrockRerankingConfiguration
+      attr_accessor metadata_configuration: Types::MetadataConfigurationForReranking
+      attr_accessor model_configuration: Types::VectorSearchBedrockRerankingModelConfiguration
+      attr_accessor number_of_reranked_results: ::Integer
+      SENSITIVE: []
+    end
+
+    class VectorSearchBedrockRerankingModelConfiguration
+      attr_accessor additional_model_request_fields: ::Hash[::String, untyped]
+      attr_accessor model_arn: ::String
+      SENSITIVE: []
+    end
+
+    class VectorSearchRerankingConfiguration
+      attr_accessor bedrock_reranking_configuration: Types::VectorSearchBedrockRerankingConfiguration
+      attr_accessor type: ("BEDROCK_RERANKING_MODEL")
+      SENSITIVE: []
+    end
+
     class FlowResponseStream < Enumerator[untyped, untyped]
       def event_types: () -> [:access_denied_exception, :bad_gateway_exception, :conflict_exception, :dependency_failed_exception, :flow_completion_event, :flow_output_event, :flow_trace_event, :internal_server_exception, :resource_not_found_exception, :service_quota_exceeded_exception, :throttling_exception, :validation_exception]
     end
@@ -1301,5 +1573,9 @@ module Aws::BedrockAgentRuntime
     class ResponseStream < Enumerator[untyped, untyped]
       def event_types: () -> [:access_denied_exception, :bad_gateway_exception, :chunk, :conflict_exception, :dependency_failed_exception, :files, :internal_server_exception, :resource_not_found_exception, :return_control, :service_quota_exceeded_exception, :throttling_exception, :trace, :validation_exception]
     end
+
+    class RetrieveAndGenerateStreamResponseOutput < Enumerator[untyped, untyped]
+      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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: aws-sdk-bedrockagentruntime
 version: !ruby/object:Gem::Version
-  version: 1.34.0
+  version: 1.36.0
 platform: ruby
 authors:
 - Amazon Web Services
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-11-26 00:00:00.000000000 Z
+date: 2024-12-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk-core