aws-sdk-bedrockagentruntime 1.2.0 → 1.4.0

data/lib/aws-sdk-bedrockagentruntime.rb

@@ -53,6 +53,6 @@ require_relative 'aws-sdk-bedrockagentruntime/event_streams'
 # @!group service
 module Aws::BedrockAgentRuntime
 
-  GEM_VERSION = '1.2.0'
+  GEM_VERSION = '1.4.0'
 
 end

data/sig/client.rbs

@@ -83,55 +83,58 @@ module Aws
       end
       # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/BedrockAgentRuntime/Client.html#invoke_agent-instance_method
       def invoke_agent: (
-                          ?session_state: {
-                            session_attributes: Hash[::String, ::String]?,
-                            prompt_session_attributes: Hash[::String, ::String]?
-                          },
-                          agent_id: ::String,
                           agent_alias_id: ::String,
-                          session_id: ::String,
-                          ?end_session: bool,
+                          agent_id: ::String,
                           ?enable_trace: bool,
-                          input_text: ::String
+                          ?end_session: bool,
+                          input_text: ::String,
+                          session_id: ::String,
+                          ?session_state: {
+                            prompt_session_attributes: Hash[::String, ::String]?,
+                            session_attributes: Hash[::String, ::String]?
+                          }
                         ) ?{ (*untyped) -> void } -> _InvokeAgentResponseSuccess
                       | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) ?{ (*untyped) -> void } -> _InvokeAgentResponseSuccess
 
       interface _RetrieveResponseSuccess
         include ::Seahorse::Client::_ResponseSuccess[Types::RetrieveResponse]
-        def retrieval_results: () -> ::Array[Types::KnowledgeBaseRetrievalResult]
         def next_token: () -> ::String
+        def retrieval_results: () -> ::Array[Types::KnowledgeBaseRetrievalResult]
       end
       # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/BedrockAgentRuntime/Client.html#retrieve-instance_method
       def retrieve: (
                       knowledge_base_id: ::String,
-                      retrieval_query: {
-                        text: ::String
-                      },
+                      ?next_token: ::String,
                       ?retrieval_configuration: {
                         vector_search_configuration: {
                           number_of_results: ::Integer?,
                           override_search_type: ("HYBRID" | "SEMANTIC")?
                         }
                       },
-                      ?next_token: ::String
+                      retrieval_query: {
+                        text: ::String
+                      }
                     ) -> _RetrieveResponseSuccess
                   | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _RetrieveResponseSuccess
 
       interface _RetrieveAndGenerateResponseSuccess
         include ::Seahorse::Client::_ResponseSuccess[Types::RetrieveAndGenerateResponse]
-        def session_id: () -> ::String
-        def output: () -> Types::RetrieveAndGenerateOutput
         def citations: () -> ::Array[Types::Citation]
+        def output: () -> Types::RetrieveAndGenerateOutput
+        def session_id: () -> ::String
       end
       # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/BedrockAgentRuntime/Client.html#retrieve_and_generate-instance_method
       def retrieve_and_generate: (
-                                   ?session_id: ::String,
                                    input: {
                                      text: ::String
                                    },
                                    ?retrieve_and_generate_configuration: {
-                                     type: ("KNOWLEDGE_BASE"),
                                      knowledge_base_configuration: {
+                                       generation_configuration: {
+                                         prompt_template: {
+                                           text_prompt_template: ::String?
+                                         }?
+                                       }?,
                                        knowledge_base_id: ::String,
                                        model_arn: ::String,
                                        retrieval_configuration: {
@@ -140,11 +143,13 @@ module Aws
                                            override_search_type: ("HYBRID" | "SEMANTIC")?
                                          }
                                        }?
-                                     }?
+                                     }?,
+                                     type: ("KNOWLEDGE_BASE")
                                    },
                                    ?session_configuration: {
                                      kms_key_arn: ::String
-                                   }
+                                   },
+                                   ?session_id: ::String
                                  ) -> _RetrieveAndGenerateResponseSuccess
                                | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _RetrieveAndGenerateResponseSuccess
     end

data/sig/types.rbs

@@ -16,11 +16,11 @@ module Aws::BedrockAgentRuntime
 
     class ActionGroupInvocationInput
       attr_accessor action_group_name: ::String
-      attr_accessor verb: ::String
       attr_accessor api_path: ::String
       attr_accessor parameters: ::Array[Types::Parameter]
       attr_accessor request_body: Types::RequestBody
-      SENSITIVE: [:action_group_name, :verb, :api_path]
+      attr_accessor verb: ::String
+      SENSITIVE: [:action_group_name, :api_path, :verb]
     end
 
     class ActionGroupInvocationOutput
@@ -60,8 +60,8 @@ module Aws::BedrockAgentRuntime
     end
 
     class FailureTrace
-      attr_accessor trace_id: ::String
       attr_accessor failure_reason: ::String
+      attr_accessor trace_id: ::String
       SENSITIVE: [:failure_reason]
     end
 
@@ -75,12 +75,17 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: [:text_response_part]
     end
 
+    class GenerationConfiguration
+      attr_accessor prompt_template: Types::PromptTemplate
+      SENSITIVE: []
+    end
+
     class InferenceConfiguration
-      attr_accessor temperature: ::Float
-      attr_accessor top_p: ::Float
-      attr_accessor top_k: ::Integer
       attr_accessor maximum_length: ::Integer
       attr_accessor stop_sequences: ::Array[::String]
+      attr_accessor temperature: ::Float
+      attr_accessor top_k: ::Integer
+      attr_accessor top_p: ::Float
       SENSITIVE: []
     end
 
@@ -91,21 +96,21 @@ module Aws::BedrockAgentRuntime
     end
 
     class InvocationInput
-      attr_accessor trace_id: ::String
-      attr_accessor invocation_type: ("ACTION_GROUP" | "KNOWLEDGE_BASE" | "FINISH")
       attr_accessor action_group_invocation_input: Types::ActionGroupInvocationInput
+      attr_accessor invocation_type: ("ACTION_GROUP" | "KNOWLEDGE_BASE" | "FINISH")
       attr_accessor knowledge_base_lookup_input: Types::KnowledgeBaseLookupInput
+      attr_accessor trace_id: ::String
       SENSITIVE: []
     end
 
     class InvokeAgentRequest
-      attr_accessor session_state: Types::SessionState
-      attr_accessor agent_id: ::String
       attr_accessor agent_alias_id: ::String
-      attr_accessor session_id: ::String
-      attr_accessor end_session: bool
+      attr_accessor agent_id: ::String
       attr_accessor enable_trace: bool
+      attr_accessor end_session: bool
       attr_accessor input_text: ::String
+      attr_accessor session_id: ::String
+      attr_accessor session_state: Types::SessionState
       SENSITIVE: [:input_text]
     end
 
@@ -117,9 +122,9 @@ module Aws::BedrockAgentRuntime
     end
 
     class KnowledgeBaseLookupInput
-      attr_accessor text: ::String
       attr_accessor knowledge_base_id: ::String
-      SENSITIVE: [:text, :knowledge_base_id]
+      attr_accessor text: ::String
+      SENSITIVE: [:knowledge_base_id, :text]
     end
 
     class KnowledgeBaseLookupOutput
@@ -145,6 +150,7 @@ module Aws::BedrockAgentRuntime
     end
 
     class KnowledgeBaseRetrieveAndGenerateConfiguration
+      attr_accessor generation_configuration: Types::GenerationConfiguration
       attr_accessor knowledge_base_id: ::String
       attr_accessor model_arn: ::String
       attr_accessor retrieval_configuration: Types::KnowledgeBaseRetrievalConfiguration
@@ -158,41 +164,41 @@ module Aws::BedrockAgentRuntime
     end
 
     class ModelInvocationInput
-      attr_accessor trace_id: ::String
-      attr_accessor text: ::String
-      attr_accessor type: ("PRE_PROCESSING" | "ORCHESTRATION" | "KNOWLEDGE_BASE_RESPONSE_GENERATION" | "POST_PROCESSING")
       attr_accessor inference_configuration: Types::InferenceConfiguration
       attr_accessor override_lambda: ::String
-      attr_accessor prompt_creation_mode: ("DEFAULT" | "OVERRIDDEN")
       attr_accessor parser_mode: ("DEFAULT" | "OVERRIDDEN")
+      attr_accessor prompt_creation_mode: ("DEFAULT" | "OVERRIDDEN")
+      attr_accessor text: ::String
+      attr_accessor trace_id: ::String
+      attr_accessor type: ("PRE_PROCESSING" | "ORCHESTRATION" | "KNOWLEDGE_BASE_RESPONSE_GENERATION" | "POST_PROCESSING")
       SENSITIVE: [:text]
     end
 
     class Observation
-      attr_accessor trace_id: ::String
-      attr_accessor type: ("ACTION_GROUP" | "KNOWLEDGE_BASE" | "FINISH" | "ASK_USER" | "REPROMPT")
       attr_accessor action_group_invocation_output: Types::ActionGroupInvocationOutput
-      attr_accessor knowledge_base_lookup_output: Types::KnowledgeBaseLookupOutput
       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")
       SENSITIVE: [:reprompt_response]
     end
 
     class OrchestrationTrace
-      attr_accessor rationale: Types::Rationale
       attr_accessor invocation_input: Types::InvocationInput
-      attr_accessor observation: Types::Observation
       attr_accessor model_invocation_input: Types::ModelInvocationInput
+      attr_accessor observation: Types::Observation
+      attr_accessor rationale: Types::Rationale
       attr_accessor unknown: untyped
-      SENSITIVE: [:rationale, :invocation_input, :observation, :model_invocation_input]
+      SENSITIVE: [:invocation_input, :model_invocation_input, :observation, :rationale]
 
-      class Rationale < OrchestrationTrace
-      end
       class InvocationInput < OrchestrationTrace
       end
+      class ModelInvocationInput < OrchestrationTrace
+      end
       class Observation < OrchestrationTrace
       end
-      class ModelInvocationInput < OrchestrationTrace
+      class Rationale < OrchestrationTrace
       end
       class Unknown < OrchestrationTrace
       end
@@ -206,15 +212,15 @@ module Aws::BedrockAgentRuntime
     end
 
     class PayloadPart
-      attr_accessor bytes: ::String
       attr_accessor attribution: Types::Attribution
+      attr_accessor bytes: ::String
       attr_accessor event_type: untyped
       SENSITIVE: [:bytes]
     end
 
     class PostProcessingModelInvocationOutput
-      attr_accessor trace_id: ::String
       attr_accessor parsed_response: Types::PostProcessingParsedResponse
+      attr_accessor trace_id: ::String
       SENSITIVE: [:parsed_response]
     end
 
@@ -238,14 +244,14 @@ module Aws::BedrockAgentRuntime
     end
 
     class PreProcessingModelInvocationOutput
-      attr_accessor trace_id: ::String
       attr_accessor parsed_response: Types::PreProcessingParsedResponse
+      attr_accessor trace_id: ::String
       SENSITIVE: [:parsed_response]
     end
 
     class PreProcessingParsedResponse
-      attr_accessor rationale: ::String
       attr_accessor is_valid: bool
+      attr_accessor rationale: ::String
       SENSITIVE: [:rationale]
     end
 
@@ -263,15 +269,20 @@ module Aws::BedrockAgentRuntime
       end
     end
 
+    class PromptTemplate
+      attr_accessor text_prompt_template: ::String
+      SENSITIVE: [:text_prompt_template]
+    end
+
     class Rationale
-      attr_accessor trace_id: ::String
       attr_accessor text: ::String
+      attr_accessor trace_id: ::String
       SENSITIVE: [:text]
     end
 
     class RepromptResponse
-      attr_accessor text: ::String
       attr_accessor source: ("ACTION_GROUP" | "KNOWLEDGE_BASE" | "PARSER")
+      attr_accessor text: ::String
       SENSITIVE: [:source]
     end
 
@@ -292,8 +303,8 @@ module Aws::BedrockAgentRuntime
     end
 
     class RetrievalResultLocation
-      attr_accessor type: ("S3")
       attr_accessor s3_location: Types::RetrievalResultS3Location
+      attr_accessor type: ("S3")
       SENSITIVE: []
     end
 
@@ -303,8 +314,8 @@ module Aws::BedrockAgentRuntime
     end
 
     class RetrieveAndGenerateConfiguration
-      attr_accessor type: ("KNOWLEDGE_BASE")
       attr_accessor knowledge_base_configuration: Types::KnowledgeBaseRetrieveAndGenerateConfiguration
+      attr_accessor type: ("KNOWLEDGE_BASE")
       SENSITIVE: []
     end
 
@@ -319,17 +330,17 @@ module Aws::BedrockAgentRuntime
     end
 
     class RetrieveAndGenerateRequest
-      attr_accessor session_id: ::String
       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 RetrieveAndGenerateResponse
-      attr_accessor session_id: ::String
-      attr_accessor output: Types::RetrieveAndGenerateOutput
       attr_accessor citations: ::Array[Types::Citation]
+      attr_accessor output: Types::RetrieveAndGenerateOutput
+      attr_accessor session_id: ::String
       SENSITIVE: [:output]
     end
 
@@ -340,15 +351,15 @@ module Aws::BedrockAgentRuntime
 
     class RetrieveRequest
       attr_accessor knowledge_base_id: ::String
-      attr_accessor retrieval_query: Types::KnowledgeBaseQuery
-      attr_accessor retrieval_configuration: Types::KnowledgeBaseRetrievalConfiguration
       attr_accessor next_token: ::String
+      attr_accessor retrieval_configuration: Types::KnowledgeBaseRetrievalConfiguration
+      attr_accessor retrieval_query: Types::KnowledgeBaseQuery
       SENSITIVE: [:retrieval_query]
     end
 
     class RetrieveResponse
-      attr_accessor retrieval_results: ::Array[Types::KnowledgeBaseRetrievalResult]
       attr_accessor next_token: ::String
+      attr_accessor retrieval_results: ::Array[Types::KnowledgeBaseRetrievalResult]
       SENSITIVE: [:retrieval_results]
     end
 
@@ -365,20 +376,20 @@ module Aws::BedrockAgentRuntime
     end
 
     class SessionState
-      attr_accessor session_attributes: ::Hash[::String, ::String]
       attr_accessor prompt_session_attributes: ::Hash[::String, ::String]
+      attr_accessor session_attributes: ::Hash[::String, ::String]
       SENSITIVE: []
     end
 
     class Span
-      attr_accessor start: ::Integer
       attr_accessor end: ::Integer
+      attr_accessor start: ::Integer
       SENSITIVE: []
     end
 
     class TextResponsePart
-      attr_accessor text: ::String
       attr_accessor span: Types::Span
+      attr_accessor text: ::String
       SENSITIVE: []
     end
 
@@ -389,28 +400,28 @@ module Aws::BedrockAgentRuntime
     end
 
     class Trace
-      attr_accessor pre_processing_trace: Types::PreProcessingTrace
+      attr_accessor failure_trace: Types::FailureTrace
       attr_accessor orchestration_trace: Types::OrchestrationTrace
       attr_accessor post_processing_trace: Types::PostProcessingTrace
-      attr_accessor failure_trace: Types::FailureTrace
+      attr_accessor pre_processing_trace: Types::PreProcessingTrace
       attr_accessor unknown: untyped
-      SENSITIVE: [:pre_processing_trace, :orchestration_trace, :post_processing_trace, :failure_trace]
+      SENSITIVE: [:failure_trace, :orchestration_trace, :post_processing_trace, :pre_processing_trace]
 
-      class PreProcessingTrace < Trace
+      class FailureTrace < Trace
       end
       class OrchestrationTrace < Trace
       end
       class PostProcessingTrace < Trace
       end
-      class FailureTrace < Trace
+      class PreProcessingTrace < Trace
       end
       class Unknown < Trace
       end
     end
 
     class TracePart
-      attr_accessor agent_id: ::String
       attr_accessor agent_alias_id: ::String
+      attr_accessor agent_id: ::String
       attr_accessor session_id: ::String
       attr_accessor trace: Types::Trace
       attr_accessor event_type: untyped
@@ -424,7 +435,7 @@ module Aws::BedrockAgentRuntime
     end
 
     class ResponseStream < Enumerator[untyped, untyped]
-      def event_types: () -> [:chunk, :trace, :internal_server_exception, :validation_exception, :resource_not_found_exception, :service_quota_exceeded_exception, :throttling_exception, :access_denied_exception, :conflict_exception, :dependency_failed_exception, :bad_gateway_exception]
+      def event_types: () -> [:access_denied_exception, :bad_gateway_exception, :chunk, :conflict_exception, :dependency_failed_exception, :internal_server_exception, :resource_not_found_exception, :service_quota_exceeded_exception, :throttling_exception, :trace, :validation_exception]
     end
   end
 end

metadata

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