aws-sdk-bedrockagentruntime 1.1.0 → 1.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a51b3e0aefdc1ee5b97c4d76b8ff3a9839cd14ebc2dedcd05be55b76606562f4
-  data.tar.gz: a6eff7b8ef45b8fe4b3d11ee271c75d48aa01c82bd9b209cecdd10b117b8ba92
+  metadata.gz: b1d4472fd40fedc45f494e9f5129562bdd972073c9b1bd7ef308f30b74935924
+  data.tar.gz: b3a5a22d247241afd94a114c0afb5cc80e3ef4e468f635f8c3e287e31bfd2677
 SHA512:
-  metadata.gz: d3f60cd08b92bca7724ffe64eaee07400caa94f03ab9749f7bbc4dbbf80e602f1b963e75201b8c94921e4cf83d2f546c00a68f77b497f5dd014dc3feb231f43a
-  data.tar.gz: 9ce3396628d4abc23982fe8dbf4bcb51260fd37b329077803eb66f13c2656f45138a07142d609cb1b37fa892881e5d8ebc88afdd4961a0b704fcfe8d19acf43f
+  metadata.gz: 292befa190a22a5ea57dd546882ba2fbf465553cbb8fd998a8ddd66779073a36c19297dac8d22482c90a0d207b702b9d662f8b1d4120281e2c42bbec2be67f6c
+  data.tar.gz: 1d57994919acfb2ee86fbfd68be0a7acf2b78511dc26bbc49aea544fdb8e3182751a6fba2a3209cae27a97bd90a333c8e7d7ec6eb0682d077f7d7e4ba6cc90fc

data/CHANGELOG.md

@@ -1,6 +1,11 @@
 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)
 ------------------
 

data/VERSION

@@ -1 +1 @@
-1.1.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.1.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/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.1.0'
+  GEM_VERSION = '1.2.0'
 
 end

data/sig/client.rbs

@@ -109,7 +109,8 @@ module Aws
                       },
                       ?retrieval_configuration: {
                         vector_search_configuration: {
-                          number_of_results: ::Integer
+                          number_of_results: ::Integer?,
+                          override_search_type: ("HYBRID" | "SEMANTIC")?
                         }
                       },
                       ?next_token: ::String
@@ -132,7 +133,13 @@ module Aws
                                      type: ("KNOWLEDGE_BASE"),
                                      knowledge_base_configuration: {
                                        knowledge_base_id: ::String,
-                                       model_arn: ::String
+                                       model_arn: ::String,
+                                       retrieval_configuration: {
+                                         vector_search_configuration: {
+                                           number_of_results: ::Integer?,
+                                           override_search_type: ("HYBRID" | "SEMANTIC")?
+                                         }
+                                       }?
                                      }?
                                    },
                                    ?session_configuration: {

data/sig/types.rbs

@@ -72,7 +72,7 @@ module Aws::BedrockAgentRuntime
 
     class GeneratedResponsePart
       attr_accessor text_response_part: Types::TextResponsePart
-      SENSITIVE: []
+      SENSITIVE: [:text_response_part]
     end
 
     class InferenceConfiguration
@@ -141,17 +141,19 @@ module Aws::BedrockAgentRuntime
       attr_accessor content: Types::RetrievalResultContent
       attr_accessor location: Types::RetrievalResultLocation
       attr_accessor score: ::Float
-      SENSITIVE: []
+      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
 
@@ -353,7 +355,7 @@ module Aws::BedrockAgentRuntime
     class RetrievedReference
       attr_accessor content: Types::RetrievalResultContent
       attr_accessor location: Types::RetrievalResultLocation
-      SENSITIVE: []
+      SENSITIVE: [:content, :location]
     end
 
     class ServiceQuotaExceededException

metadata

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