aws-sdk-bedrockagentruntime 1.0.0 → 1.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: fcb4655eafd88cab131cfc10bfd56c6ad2d437dec5b4ce5fbfa352b18fb87bd4
-  data.tar.gz: 17059c000a2db6036dbb6c28764d4284f448a6b2e3863414f1e0dc3865905486
+  metadata.gz: b1d4472fd40fedc45f494e9f5129562bdd972073c9b1bd7ef308f30b74935924
+  data.tar.gz: b3a5a22d247241afd94a114c0afb5cc80e3ef4e468f635f8c3e287e31bfd2677
 SHA512:
-  metadata.gz: '094c6ffc1ed6f89ceee403a8d4c4697c8fa5abb161e1c1778718847d78a576506c8a98e111ab697a8486c0b2eb456c11b9c274d4bc725acba1404ce20363f383'
-  data.tar.gz: 38272c59c4929245890f7f037cba4ae2fdee7cdb83ac16f00e36bfa988fca001666ba57cdbfd2b6aea54c57376d93e915e234280dec82c82df987c4771928937
+  metadata.gz: 292befa190a22a5ea57dd546882ba2fbf465553cbb8fd998a8ddd66779073a36c19297dac8d22482c90a0d207b702b9d662f8b1d4120281e2c42bbec2be67f6c
+  data.tar.gz: 1d57994919acfb2ee86fbfd68be0a7acf2b78511dc26bbc49aea544fdb8e3182751a6fba2a3209cae27a97bd90a333c8e7d7ec6eb0682d077f7d7e4ba6cc90fc

data/CHANGELOG.md

@@ -1,6 +1,16 @@
 Unreleased Changes
 ------------------
 
+1.2.0 (2024-02-28)
+------------------
+
+* Feature - This release adds support to override search strategy performed by the Retrieve and RetrieveAndGenerate APIs for Amazon Bedrock Agents
+
+1.1.0 (2024-01-26)
+------------------
+
+* Feature - Code Generated Changes, see `./build_tools` or `aws-sdk-core`'s CHANGELOG.md for details.
+
 1.0.0 (2023-11-28)
 ------------------
 

data/VERSION

@@ -1 +1 @@
-1.0.0
+1.2.0

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

@@ -805,7 +805,8 @@ module Aws::BedrockAgentRuntime
     #     },
     #     retrieval_configuration: {
     #       vector_search_configuration: { # required
-    #         number_of_results: 1, # required
+    #         number_of_results: 1,
+    #         override_search_type: "HYBRID", # accepts HYBRID, SEMANTIC
     #       },
     #     },
     #     next_token: "NextToken",
@@ -861,6 +862,12 @@ module Aws::BedrockAgentRuntime
     #       knowledge_base_configuration: {
     #         knowledge_base_id: "KnowledgeBaseId", # required
     #         model_arn: "BedrockModelArn", # required
+    #         retrieval_configuration: {
+    #           vector_search_configuration: { # required
+    #             number_of_results: 1,
+    #             override_search_type: "HYBRID", # accepts HYBRID, SEMANTIC
+    #           },
+    #         },
     #       },
     #     },
     #     session_configuration: {
@@ -903,7 +910,7 @@ module Aws::BedrockAgentRuntime
         params: params,
         config: config)
       context[:gem_name] = 'aws-sdk-bedrockagentruntime'
-      context[:gem_version] = '1.0.0'
+      context[:gem_version] = '1.2.0'
       Seahorse::Client::Request.new(handlers, context)
     end
 

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

@@ -101,6 +101,7 @@ module Aws::BedrockAgentRuntime
     RetrieveResponse = Shapes::StructureShape.new(name: 'RetrieveResponse')
     RetrievedReference = Shapes::StructureShape.new(name: 'RetrievedReference')
     RetrievedReferences = Shapes::ListShape.new(name: 'RetrievedReferences')
+    SearchType = Shapes::StringShape.new(name: 'SearchType')
     ServiceQuotaExceededException = Shapes::StructureShape.new(name: 'ServiceQuotaExceededException')
     SessionAttributesMap = Shapes::MapShape.new(name: 'SessionAttributesMap')
     SessionId = Shapes::StringShape.new(name: 'SessionId')
@@ -224,9 +225,11 @@ module Aws::BedrockAgentRuntime
 
     KnowledgeBaseRetrieveAndGenerateConfiguration.add_member(:knowledge_base_id, Shapes::ShapeRef.new(shape: KnowledgeBaseId, required: true, location_name: "knowledgeBaseId"))
     KnowledgeBaseRetrieveAndGenerateConfiguration.add_member(:model_arn, Shapes::ShapeRef.new(shape: BedrockModelArn, required: true, location_name: "modelArn"))
+    KnowledgeBaseRetrieveAndGenerateConfiguration.add_member(:retrieval_configuration, Shapes::ShapeRef.new(shape: KnowledgeBaseRetrievalConfiguration, location_name: "retrievalConfiguration"))
     KnowledgeBaseRetrieveAndGenerateConfiguration.struct_class = Types::KnowledgeBaseRetrieveAndGenerateConfiguration
 
-    KnowledgeBaseVectorSearchConfiguration.add_member(:number_of_results, Shapes::ShapeRef.new(shape: KnowledgeBaseVectorSearchConfigurationNumberOfResultsInteger, required: true, location_name: "numberOfResults"))
+    KnowledgeBaseVectorSearchConfiguration.add_member(:number_of_results, Shapes::ShapeRef.new(shape: KnowledgeBaseVectorSearchConfigurationNumberOfResultsInteger, location_name: "numberOfResults", metadata: {"box"=>true}))
+    KnowledgeBaseVectorSearchConfiguration.add_member(:override_search_type, Shapes::ShapeRef.new(shape: SearchType, location_name: "overrideSearchType"))
     KnowledgeBaseVectorSearchConfiguration.struct_class = Types::KnowledgeBaseVectorSearchConfiguration
 
     ModelInvocationInput.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))

data/lib/aws-sdk-bedrockagentruntime/plugins/endpoints.rb

@@ -14,6 +14,7 @@ module Aws::BedrockAgentRuntime
       option(
         :endpoint_provider,
         doc_type: 'Aws::BedrockAgentRuntime::EndpointProvider',
+        rbs_type: 'untyped',
         docstring: 'The endpoint provider used to resolve endpoints. Any '\
                    'object that responds to `#resolve_endpoint(parameters)` '\
                    'where `parameters` is a Struct similar to '\

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

@@ -209,7 +209,7 @@ module Aws::BedrockAgentRuntime
     #
     class GeneratedResponsePart < Struct.new(
       :text_response_part)
-      SENSITIVE = []
+      SENSITIVE = [:text_response_part]
       include Aws::Structure
     end
 
@@ -446,7 +446,7 @@ module Aws::BedrockAgentRuntime
       :content,
       :location,
       :score)
-      SENSITIVE = []
+      SENSITIVE = [:content, :location]
       include Aws::Structure
     end
 
@@ -460,11 +460,16 @@ module Aws::BedrockAgentRuntime
     #   Arn of a Bedrock model.
     #   @return [String]
     #
+    # @!attribute [rw] retrieval_configuration
+    #   Search parameters for retrieving from knowledge base.
+    #   @return [Types::KnowledgeBaseRetrievalConfiguration]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/KnowledgeBaseRetrieveAndGenerateConfiguration AWS API Documentation
     #
     class KnowledgeBaseRetrieveAndGenerateConfiguration < Struct.new(
       :knowledge_base_id,
-      :model_arn)
+      :model_arn,
+      :retrieval_configuration)
       SENSITIVE = []
       include Aws::Structure
     end
@@ -475,10 +480,15 @@ module Aws::BedrockAgentRuntime
     #   Top-K results to retrieve from knowledge base.
     #   @return [Integer]
     #
+    # @!attribute [rw] override_search_type
+    #   Override the type of query to be performed on data store
+    #   @return [String]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/KnowledgeBaseVectorSearchConfiguration AWS API Documentation
     #
     class KnowledgeBaseVectorSearchConfiguration < Struct.new(
-      :number_of_results)
+      :number_of_results,
+      :override_search_type)
       SENSITIVE = []
       include Aws::Structure
     end
@@ -1064,7 +1074,7 @@ module Aws::BedrockAgentRuntime
     class RetrievedReference < Struct.new(
       :content,
       :location)
-      SENSITIVE = []
+      SENSITIVE = [:content, :location]
       include Aws::Structure
     end
 

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.0.0'
+  GEM_VERSION = '1.2.0'
 
 end

data/sig/client.rbs

@@ -0,0 +1,153 @@
+# WARNING ABOUT GENERATED CODE
+#
+# This file is generated. See the contributing guide for more information:
+# https://github.com/aws/aws-sdk-ruby/blob/version-3/CONTRIBUTING.md
+#
+# WARNING ABOUT GENERATED CODE
+
+module Aws
+  module BedrockAgentRuntime
+    class Client < ::Seahorse::Client::Base
+      include ::Aws::ClientStubs
+
+      # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/BedrockAgentRuntime/Client.html#initialize-instance_method
+      def self.new: (
+                      ?credentials: untyped,
+                      ?region: String,
+                      ?access_key_id: String,
+                      ?active_endpoint_cache: bool,
+                      ?adaptive_retry_wait_to_fill: bool,
+                      ?client_side_monitoring: bool,
+                      ?client_side_monitoring_client_id: String,
+                      ?client_side_monitoring_host: String,
+                      ?client_side_monitoring_port: Integer,
+                      ?client_side_monitoring_publisher: untyped,
+                      ?convert_params: bool,
+                      ?correct_clock_skew: bool,
+                      ?defaults_mode: String,
+                      ?disable_host_prefix_injection: bool,
+                      ?disable_request_compression: bool,
+                      ?endpoint: String,
+                      ?endpoint_cache_max_entries: Integer,
+                      ?endpoint_cache_max_threads: Integer,
+                      ?endpoint_cache_poll_interval: Integer,
+                      ?endpoint_discovery: bool,
+                      ?event_stream_handler: Proc,
+                      ?ignore_configured_endpoint_urls: bool,
+                      ?input_event_stream_handler: Proc,
+                      ?log_formatter: untyped,
+                      ?log_level: Symbol,
+                      ?logger: untyped,
+                      ?max_attempts: Integer,
+                      ?output_event_stream_handler: Proc,
+                      ?profile: String,
+                      ?request_min_compression_size_bytes: Integer,
+                      ?retry_backoff: Proc,
+                      ?retry_base_delay: Float,
+                      ?retry_jitter: (:none | :equal | :full | ^(Integer) -> Integer),
+                      ?retry_limit: Integer,
+                      ?retry_max_delay: Integer,
+                      ?retry_mode: ("legacy" | "standard" | "adaptive"),
+                      ?sdk_ua_app_id: String,
+                      ?secret_access_key: String,
+                      ?session_token: String,
+                      ?stub_responses: untyped,
+                      ?token_provider: untyped,
+                      ?use_dualstack_endpoint: bool,
+                      ?use_fips_endpoint: bool,
+                      ?validate_params: bool,
+                      ?endpoint_provider: untyped,
+                      ?http_proxy: String,
+                      ?http_open_timeout: (Float | Integer),
+                      ?http_read_timeout: (Float | Integer),
+                      ?http_idle_timeout: (Float | Integer),
+                      ?http_continue_timeout: (Float | Integer),
+                      ?ssl_timeout: (Float | Integer | nil),
+                      ?http_wire_trace: bool,
+                      ?ssl_verify_peer: bool,
+                      ?ssl_ca_bundle: String,
+                      ?ssl_ca_directory: String,
+                      ?ssl_ca_store: String,
+                      ?on_chunk_received: Proc,
+                      ?on_chunk_sent: Proc,
+                      ?raise_response_errors: bool
+                    ) -> instance
+                  | (?Hash[Symbol, untyped]) -> instance
+
+
+      interface _InvokeAgentResponseSuccess
+        include ::Seahorse::Client::_ResponseSuccess[Types::InvokeAgentResponse]
+        def completion: () -> Types::ResponseStream
+        def content_type: () -> ::String
+        def session_id: () -> ::String
+      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,
+                          ?enable_trace: bool,
+                          input_text: ::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
+      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
+                      },
+                      ?retrieval_configuration: {
+                        vector_search_configuration: {
+                          number_of_results: ::Integer?,
+                          override_search_type: ("HYBRID" | "SEMANTIC")?
+                        }
+                      },
+                      ?next_token: ::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]
+      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: {
+                                       knowledge_base_id: ::String,
+                                       model_arn: ::String,
+                                       retrieval_configuration: {
+                                         vector_search_configuration: {
+                                           number_of_results: ::Integer?,
+                                           override_search_type: ("HYBRID" | "SEMANTIC")?
+                                         }
+                                       }?
+                                     }?
+                                   },
+                                   ?session_configuration: {
+                                     kms_key_arn: ::String
+                                   }
+                                 ) -> _RetrieveAndGenerateResponseSuccess
+                               | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _RetrieveAndGenerateResponseSuccess
+    end
+  end
+end
+

data/sig/errors.rbs

@@ -0,0 +1,45 @@
+# WARNING ABOUT GENERATED CODE
+#
+# This file is generated. See the contributing guide for more information:
+# https://github.com/aws/aws-sdk-ruby/blob/version-3/CONTRIBUTING.md
+#
+# WARNING ABOUT GENERATED CODE
+
+module Aws
+  module BedrockAgentRuntime
+    module Errors
+      class ServiceError < ::Aws::Errors::ServiceError
+      end
+
+      class AccessDeniedException < ::Aws::Errors::ServiceError
+        def message: () -> ::String
+      end
+      class BadGatewayException < ::Aws::Errors::ServiceError
+        def message: () -> ::String
+        def resource_name: () -> ::String
+      end
+      class ConflictException < ::Aws::Errors::ServiceError
+        def message: () -> ::String
+      end
+      class DependencyFailedException < ::Aws::Errors::ServiceError
+        def message: () -> ::String
+        def resource_name: () -> ::String
+      end
+      class InternalServerException < ::Aws::Errors::ServiceError
+        def message: () -> ::String
+      end
+      class ResourceNotFoundException < ::Aws::Errors::ServiceError
+        def message: () -> ::String
+      end
+      class ServiceQuotaExceededException < ::Aws::Errors::ServiceError
+        def message: () -> ::String
+      end
+      class ThrottlingException < ::Aws::Errors::ServiceError
+        def message: () -> ::String
+      end
+      class ValidationException < ::Aws::Errors::ServiceError
+        def message: () -> ::String
+      end
+    end
+  end
+end

data/sig/resource.rbs

@@ -0,0 +1,82 @@
+# WARNING ABOUT GENERATED CODE
+#
+# This file is generated. See the contributing guide for more information:
+# https://github.com/aws/aws-sdk-ruby/blob/version-3/CONTRIBUTING.md
+#
+# WARNING ABOUT GENERATED CODE
+
+module Aws
+  module BedrockAgentRuntime
+    # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/BedrockAgentRuntime/Resource.html
+    class Resource
+      # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/BedrockAgentRuntime/Resource.html#initialize-instance_method
+      def initialize: (
+                        ?client: Client,
+                        ?credentials: untyped,
+                        ?region: String,
+                        ?access_key_id: String,
+                        ?active_endpoint_cache: bool,
+                        ?adaptive_retry_wait_to_fill: bool,
+                        ?client_side_monitoring: bool,
+                        ?client_side_monitoring_client_id: String,
+                        ?client_side_monitoring_host: String,
+                        ?client_side_monitoring_port: Integer,
+                        ?client_side_monitoring_publisher: untyped,
+                        ?convert_params: bool,
+                        ?correct_clock_skew: bool,
+                        ?defaults_mode: String,
+                        ?disable_host_prefix_injection: bool,
+                        ?disable_request_compression: bool,
+                        ?endpoint: String,
+                        ?endpoint_cache_max_entries: Integer,
+                        ?endpoint_cache_max_threads: Integer,
+                        ?endpoint_cache_poll_interval: Integer,
+                        ?endpoint_discovery: bool,
+                        ?event_stream_handler: Proc,
+                        ?ignore_configured_endpoint_urls: bool,
+                        ?input_event_stream_handler: Proc,
+                        ?log_formatter: untyped,
+                        ?log_level: Symbol,
+                        ?logger: untyped,
+                        ?max_attempts: Integer,
+                        ?output_event_stream_handler: Proc,
+                        ?profile: String,
+                        ?request_min_compression_size_bytes: Integer,
+                        ?retry_backoff: Proc,
+                        ?retry_base_delay: Float,
+                        ?retry_jitter: (:none | :equal | :full | ^(Integer) -> Integer),
+                        ?retry_limit: Integer,
+                        ?retry_max_delay: Integer,
+                        ?retry_mode: ("legacy" | "standard" | "adaptive"),
+                        ?sdk_ua_app_id: String,
+                        ?secret_access_key: String,
+                        ?session_token: String,
+                        ?stub_responses: untyped,
+                        ?token_provider: untyped,
+                        ?use_dualstack_endpoint: bool,
+                        ?use_fips_endpoint: bool,
+                        ?validate_params: bool,
+                        ?endpoint_provider: untyped,
+                        ?http_proxy: String,
+                        ?http_open_timeout: (Float | Integer),
+                        ?http_read_timeout: (Float | Integer),
+                        ?http_idle_timeout: (Float | Integer),
+                        ?http_continue_timeout: (Float | Integer),
+                        ?ssl_timeout: (Float | Integer | nil),
+                        ?http_wire_trace: bool,
+                        ?ssl_verify_peer: bool,
+                        ?ssl_ca_bundle: String,
+                        ?ssl_ca_directory: String,
+                        ?ssl_ca_store: String,
+                        ?on_chunk_received: Proc,
+                        ?on_chunk_sent: Proc,
+                        ?raise_response_errors: bool
+                      ) -> void
+                    | (?Hash[Symbol, untyped]) -> void
+
+      def client: () -> Client
+
+
+    end
+  end
+end

data/sig/types.rbs

@@ -0,0 +1,430 @@
+# WARNING ABOUT GENERATED CODE
+#
+# This file is generated. See the contributing guide for more information:
+# https://github.com/aws/aws-sdk-ruby/blob/version-3/CONTRIBUTING.md
+#
+# WARNING ABOUT GENERATED CODE
+
+module Aws::BedrockAgentRuntime
+  module Types
+
+    class AccessDeniedException
+      attr_accessor message: ::String
+      attr_accessor event_type: untyped
+      SENSITIVE: []
+    end
+
+    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]
+    end
+
+    class ActionGroupInvocationOutput
+      attr_accessor text: ::String
+      SENSITIVE: [:text]
+    end
+
+    class Attribution
+      attr_accessor citations: ::Array[Types::Citation]
+      SENSITIVE: []
+    end
+
+    class BadGatewayException
+      attr_accessor message: ::String
+      attr_accessor resource_name: ::String
+      attr_accessor event_type: untyped
+      SENSITIVE: []
+    end
+
+    class Citation
+      attr_accessor generated_response_part: Types::GeneratedResponsePart
+      attr_accessor retrieved_references: ::Array[Types::RetrievedReference]
+      SENSITIVE: []
+    end
+
+    class ConflictException
+      attr_accessor message: ::String
+      attr_accessor event_type: untyped
+      SENSITIVE: []
+    end
+
+    class DependencyFailedException
+      attr_accessor message: ::String
+      attr_accessor resource_name: ::String
+      attr_accessor event_type: untyped
+      SENSITIVE: []
+    end
+
+    class FailureTrace
+      attr_accessor trace_id: ::String
+      attr_accessor failure_reason: ::String
+      SENSITIVE: [:failure_reason]
+    end
+
+    class FinalResponse
+      attr_accessor text: ::String
+      SENSITIVE: [:text]
+    end
+
+    class GeneratedResponsePart
+      attr_accessor text_response_part: Types::TextResponsePart
+      SENSITIVE: [:text_response_part]
+    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]
+      SENSITIVE: []
+    end
+
+    class InternalServerException
+      attr_accessor message: ::String
+      attr_accessor event_type: untyped
+      SENSITIVE: []
+    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 knowledge_base_lookup_input: Types::KnowledgeBaseLookupInput
+      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 enable_trace: bool
+      attr_accessor input_text: ::String
+      SENSITIVE: [:input_text]
+    end
+
+    class InvokeAgentResponse
+      attr_accessor completion: Types::ResponseStream
+      attr_accessor content_type: ::String
+      attr_accessor session_id: ::String
+      SENSITIVE: []
+    end
+
+    class KnowledgeBaseLookupInput
+      attr_accessor text: ::String
+      attr_accessor knowledge_base_id: ::String
+      SENSITIVE: [:text, :knowledge_base_id]
+    end
+
+    class KnowledgeBaseLookupOutput
+      attr_accessor retrieved_references: ::Array[Types::RetrievedReference]
+      SENSITIVE: []
+    end
+
+    class KnowledgeBaseQuery
+      attr_accessor text: ::String
+      SENSITIVE: []
+    end
+
+    class KnowledgeBaseRetrievalConfiguration
+      attr_accessor vector_search_configuration: Types::KnowledgeBaseVectorSearchConfiguration
+      SENSITIVE: []
+    end
+
+    class KnowledgeBaseRetrievalResult
+      attr_accessor content: Types::RetrievalResultContent
+      attr_accessor location: Types::RetrievalResultLocation
+      attr_accessor score: ::Float
+      SENSITIVE: [:content, :location]
+    end
+
+    class KnowledgeBaseRetrieveAndGenerateConfiguration
+      attr_accessor knowledge_base_id: ::String
+      attr_accessor model_arn: ::String
+      attr_accessor retrieval_configuration: Types::KnowledgeBaseRetrievalConfiguration
+      SENSITIVE: []
+    end
+
+    class KnowledgeBaseVectorSearchConfiguration
+      attr_accessor number_of_results: ::Integer
+      attr_accessor override_search_type: ("HYBRID" | "SEMANTIC")
+      SENSITIVE: []
+    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")
+      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 reprompt_response: Types::RepromptResponse
+      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 unknown: untyped
+      SENSITIVE: [:rationale, :invocation_input, :observation, :model_invocation_input]
+
+      class Rationale < OrchestrationTrace
+      end
+      class InvocationInput < OrchestrationTrace
+      end
+      class Observation < OrchestrationTrace
+      end
+      class ModelInvocationInput < OrchestrationTrace
+      end
+      class Unknown < OrchestrationTrace
+      end
+    end
+
+    class Parameter
+      attr_accessor name: ::String
+      attr_accessor type: ::String
+      attr_accessor value: ::String
+      SENSITIVE: []
+    end
+
+    class PayloadPart
+      attr_accessor bytes: ::String
+      attr_accessor attribution: Types::Attribution
+      attr_accessor event_type: untyped
+      SENSITIVE: [:bytes]
+    end
+
+    class PostProcessingModelInvocationOutput
+      attr_accessor trace_id: ::String
+      attr_accessor parsed_response: Types::PostProcessingParsedResponse
+      SENSITIVE: [:parsed_response]
+    end
+
+    class PostProcessingParsedResponse
+      attr_accessor text: ::String
+      SENSITIVE: [:text]
+    end
+
+    class PostProcessingTrace
+      attr_accessor model_invocation_input: Types::ModelInvocationInput
+      attr_accessor model_invocation_output: Types::PostProcessingModelInvocationOutput
+      attr_accessor unknown: untyped
+      SENSITIVE: [:model_invocation_input, :model_invocation_output]
+
+      class ModelInvocationInput < PostProcessingTrace
+      end
+      class ModelInvocationOutput < PostProcessingTrace
+      end
+      class Unknown < PostProcessingTrace
+      end
+    end
+
+    class PreProcessingModelInvocationOutput
+      attr_accessor trace_id: ::String
+      attr_accessor parsed_response: Types::PreProcessingParsedResponse
+      SENSITIVE: [:parsed_response]
+    end
+
+    class PreProcessingParsedResponse
+      attr_accessor rationale: ::String
+      attr_accessor is_valid: bool
+      SENSITIVE: [:rationale]
+    end
+
+    class PreProcessingTrace
+      attr_accessor model_invocation_input: Types::ModelInvocationInput
+      attr_accessor model_invocation_output: Types::PreProcessingModelInvocationOutput
+      attr_accessor unknown: untyped
+      SENSITIVE: [:model_invocation_input, :model_invocation_output]
+
+      class ModelInvocationInput < PreProcessingTrace
+      end
+      class ModelInvocationOutput < PreProcessingTrace
+      end
+      class Unknown < PreProcessingTrace
+      end
+    end
+
+    class Rationale
+      attr_accessor trace_id: ::String
+      attr_accessor text: ::String
+      SENSITIVE: [:text]
+    end
+
+    class RepromptResponse
+      attr_accessor text: ::String
+      attr_accessor source: ("ACTION_GROUP" | "KNOWLEDGE_BASE" | "PARSER")
+      SENSITIVE: [:source]
+    end
+
+    class RequestBody
+      attr_accessor content: ::Hash[::String, ::Array[Types::Parameter]]
+      SENSITIVE: []
+    end
+
+    class ResourceNotFoundException
+      attr_accessor message: ::String
+      attr_accessor event_type: untyped
+      SENSITIVE: []
+    end
+
+    class RetrievalResultContent
+      attr_accessor text: ::String
+      SENSITIVE: []
+    end
+
+    class RetrievalResultLocation
+      attr_accessor type: ("S3")
+      attr_accessor s3_location: Types::RetrievalResultS3Location
+      SENSITIVE: []
+    end
+
+    class RetrievalResultS3Location
+      attr_accessor uri: ::String
+      SENSITIVE: []
+    end
+
+    class RetrieveAndGenerateConfiguration
+      attr_accessor type: ("KNOWLEDGE_BASE")
+      attr_accessor knowledge_base_configuration: Types::KnowledgeBaseRetrieveAndGenerateConfiguration
+      SENSITIVE: []
+    end
+
+    class RetrieveAndGenerateInput
+      attr_accessor text: ::String
+      SENSITIVE: []
+    end
+
+    class RetrieveAndGenerateOutput
+      attr_accessor text: ::String
+      SENSITIVE: []
+    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
+      SENSITIVE: [:input]
+    end
+
+    class RetrieveAndGenerateResponse
+      attr_accessor session_id: ::String
+      attr_accessor output: Types::RetrieveAndGenerateOutput
+      attr_accessor citations: ::Array[Types::Citation]
+      SENSITIVE: [:output]
+    end
+
+    class RetrieveAndGenerateSessionConfiguration
+      attr_accessor kms_key_arn: ::String
+      SENSITIVE: []
+    end
+
+    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
+      SENSITIVE: [:retrieval_query]
+    end
+
+    class RetrieveResponse
+      attr_accessor retrieval_results: ::Array[Types::KnowledgeBaseRetrievalResult]
+      attr_accessor next_token: ::String
+      SENSITIVE: [:retrieval_results]
+    end
+
+    class RetrievedReference
+      attr_accessor content: Types::RetrievalResultContent
+      attr_accessor location: Types::RetrievalResultLocation
+      SENSITIVE: [:content, :location]
+    end
+
+    class ServiceQuotaExceededException
+      attr_accessor message: ::String
+      attr_accessor event_type: untyped
+      SENSITIVE: []
+    end
+
+    class SessionState
+      attr_accessor session_attributes: ::Hash[::String, ::String]
+      attr_accessor prompt_session_attributes: ::Hash[::String, ::String]
+      SENSITIVE: []
+    end
+
+    class Span
+      attr_accessor start: ::Integer
+      attr_accessor end: ::Integer
+      SENSITIVE: []
+    end
+
+    class TextResponsePart
+      attr_accessor text: ::String
+      attr_accessor span: Types::Span
+      SENSITIVE: []
+    end
+
+    class ThrottlingException
+      attr_accessor message: ::String
+      attr_accessor event_type: untyped
+      SENSITIVE: []
+    end
+
+    class Trace
+      attr_accessor pre_processing_trace: Types::PreProcessingTrace
+      attr_accessor orchestration_trace: Types::OrchestrationTrace
+      attr_accessor post_processing_trace: Types::PostProcessingTrace
+      attr_accessor failure_trace: Types::FailureTrace
+      attr_accessor unknown: untyped
+      SENSITIVE: [:pre_processing_trace, :orchestration_trace, :post_processing_trace, :failure_trace]
+
+      class PreProcessingTrace < Trace
+      end
+      class OrchestrationTrace < Trace
+      end
+      class PostProcessingTrace < Trace
+      end
+      class FailureTrace < Trace
+      end
+      class Unknown < Trace
+      end
+    end
+
+    class TracePart
+      attr_accessor agent_id: ::String
+      attr_accessor agent_alias_id: ::String
+      attr_accessor session_id: ::String
+      attr_accessor trace: Types::Trace
+      attr_accessor event_type: untyped
+      SENSITIVE: [:trace]
+    end
+
+    class ValidationException
+      attr_accessor message: ::String
+      attr_accessor event_type: untyped
+      SENSITIVE: []
+    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]
+    end
+  end
+end

data/sig/waiters.rbs

@@ -0,0 +1,13 @@
+# WARNING ABOUT GENERATED CODE
+#
+# This file is generated. See the contributing guide for more information:
+# https://github.com/aws/aws-sdk-ruby/blob/version-3/CONTRIBUTING.md
+#
+# WARNING ABOUT GENERATED CODE
+
+module Aws
+  module BedrockAgentRuntime
+    module Waiters
+    end
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: aws-sdk-bedrockagentruntime
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.2.0
 platform: ruby
 authors:
 - Amazon Web Services
-autorequire: 
+autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-11-28 00:00:00.000000000 Z
+date: 2024-02-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk-core
@@ -19,7 +19,7 @@ dependencies:
         version: '3'
     - - ">="
       - !ruby/object:Gem::Version
-        version: 3.188.0
+        version: 3.191.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
@@ -29,7 +29,7 @@ dependencies:
         version: '3'
     - - ">="
       - !ruby/object:Gem::Version
-        version: 3.188.0
+        version: 3.191.0
 - !ruby/object:Gem::Dependency
   name: aws-sigv4
   requirement: !ruby/object:Gem::Requirement
@@ -67,13 +67,18 @@ files:
 - lib/aws-sdk-bedrockagentruntime/plugins/endpoints.rb
 - lib/aws-sdk-bedrockagentruntime/resource.rb
 - lib/aws-sdk-bedrockagentruntime/types.rb
+- sig/client.rbs
+- sig/errors.rbs
+- sig/resource.rbs
+- sig/types.rbs
+- sig/waiters.rbs
 homepage: https://github.com/aws/aws-sdk-ruby
 licenses:
 - Apache-2.0
 metadata:
   source_code_uri: https://github.com/aws/aws-sdk-ruby/tree/version-3/gems/aws-sdk-bedrockagentruntime
   changelog_uri: https://github.com/aws/aws-sdk-ruby/tree/version-3/gems/aws-sdk-bedrockagentruntime/CHANGELOG.md
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -88,8 +93,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.1.6
-signing_key: 
+rubygems_version: 3.4.10
+signing_key:
 specification_version: 4
 summary: AWS SDK for Ruby - Agents for Amazon Bedrock Runtime
 test_files: []