aws-sdk-bedrockagentruntime 1.28.0 → 1.29.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4338575e5e4e1c40786134c71bd8381feff99a7dbf001e0b9c24bff649e80617
-  data.tar.gz: f6f86d4d746e7ddac1dcaab2c74c9d0abaf8a63f636cadf0290206666e5f1cab
+  metadata.gz: 874f1f6793b18532bc35891baa80c7ba272f425884814271fca7df08b2e5a5be
+  data.tar.gz: fd3769ce8236cff70661e686fb9d39bd37a43ffcb47ad06c6e229933e3f4e416
 SHA512:
-  metadata.gz: d3385e7ef442ceff7f263840efb5343a58d981b2b777851e27c0ff3cf662ee2d399abd15feb80e26e601d9684d0c61676d7713cef23fed87741e690d36fd056c
-  data.tar.gz: 5a9131c582f6526d88951088e05a86eb45b4d968e91ad17d4b207f8e2b1185638989552b3bb18a8323434d7f99bde1e756e39161848d484db4e720470e149100
+  metadata.gz: 3997c97c9d6484bdb249afc314d0c254db21297ca85b912b653801b84bb48b61ee224da8a2ae7e286774e90755ba1bfb4345fa22062f1d31545ea0db764a1ad7
+  data.tar.gz: fb395ef3e0549e25c726e6bbe7a168eb19ff580ee2c4c6b44d9ee55fb4e38026912c212c82c13b451545c37f3f7b71289709041c08080a217f7d7f362e1d4cf6

data/CHANGELOG.md

@@ -1,6 +1,11 @@
 Unreleased Changes
 ------------------
 
+1.29.0 (2024-10-21)
+------------------
+
+* Feature - Knowledge Bases for Amazon Bedrock now supports custom prompts and model parameters in the orchestrationConfiguration of the RetrieveAndGenerate API. The modelArn field accepts Custom Models and Imported Models ARNs.
+
 1.28.0 (2024-10-18)
 ------------------
 

data/VERSION

@@ -1 +1 @@
-1.28.0
+1.29.0

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

@@ -1773,7 +1773,22 @@ module Aws::BedrockAgentRuntime
     #         knowledge_base_id: "KnowledgeBaseId", # required
     #         model_arn: "BedrockModelArn", # required
     #         orchestration_configuration: {
-    #           query_transformation_configuration: { # required
+    #           additional_model_request_fields: {
+    #             "AdditionalModelRequestFieldsKey" => {
+    #             },
+    #           },
+    #           inference_config: {
+    #             text_inference_config: {
+    #               max_tokens: 1,
+    #               stop_sequences: ["RAGStopSequencesMemberString"],
+    #               temperature: 1.0,
+    #               top_p: 1.0,
+    #             },
+    #           },
+    #           prompt_template: {
+    #             text_prompt_template: "TextPromptTemplate",
+    #           },
+    #           query_transformation_configuration: {
     #             type: "QUERY_DECOMPOSITION", # required, accepts QUERY_DECOMPOSITION
     #           },
     #         },
@@ -1905,7 +1920,7 @@ module Aws::BedrockAgentRuntime
         tracer: tracer
       )
       context[:gem_name] = 'aws-sdk-bedrockagentruntime'
-      context[:gem_version] = '1.28.0'
+      context[:gem_version] = '1.29.0'
       Seahorse::Client::Request.new(handlers, context)
     end
 

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

@@ -713,7 +713,10 @@ module Aws::BedrockAgentRuntime
     Observation.add_member(:type, Shapes::ShapeRef.new(shape: Type, location_name: "type"))
     Observation.struct_class = Types::Observation
 
-    OrchestrationConfiguration.add_member(:query_transformation_configuration, Shapes::ShapeRef.new(shape: QueryTransformationConfiguration, required: true, location_name: "queryTransformationConfiguration"))
+    OrchestrationConfiguration.add_member(:additional_model_request_fields, Shapes::ShapeRef.new(shape: AdditionalModelRequestFields, location_name: "additionalModelRequestFields"))
+    OrchestrationConfiguration.add_member(:inference_config, Shapes::ShapeRef.new(shape: InferenceConfig, location_name: "inferenceConfig"))
+    OrchestrationConfiguration.add_member(:prompt_template, Shapes::ShapeRef.new(shape: PromptTemplate, location_name: "promptTemplate"))
+    OrchestrationConfiguration.add_member(:query_transformation_configuration, Shapes::ShapeRef.new(shape: QueryTransformationConfiguration, location_name: "queryTransformationConfiguration"))
     OrchestrationConfiguration.struct_class = Types::OrchestrationConfiguration
 
     OrchestrationModelInvocationOutput.add_member(:metadata, Shapes::ShapeRef.new(shape: Metadata, location_name: "metadata"))

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

@@ -2284,6 +2284,23 @@ module Aws::BedrockAgentRuntime
     # Settings for how the model processes the prompt prior to retrieval and
     # generation.
     #
+    # @!attribute [rw] additional_model_request_fields
+    #   Additional model parameters and corresponding values not included in
+    #   the textInferenceConfig structure for a knowledge base. This allows
+    #   users to provide custom model parameters specific to the language
+    #   model being used.
+    #   @return [Hash<String,Hash,Array,String,Numeric,Boolean>]
+    #
+    # @!attribute [rw] inference_config
+    #   Configuration settings for inference when using RetrieveAndGenerate
+    #   to generate responses while using a knowledge base as a source.
+    #   @return [Types::InferenceConfig]
+    #
+    # @!attribute [rw] prompt_template
+    #   Contains the template for the prompt that's sent to the model for
+    #   response generation.
+    #   @return [Types::PromptTemplate]
+    #
     # @!attribute [rw] query_transformation_configuration
     #   To split up the prompt and retrieve multiple sources, set the
     #   transformation type to `QUERY_DECOMPOSITION`.
@@ -2292,6 +2309,9 @@ module Aws::BedrockAgentRuntime
     # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/OrchestrationConfiguration AWS API Documentation
     #
     class OrchestrationConfiguration < Struct.new(
+      :additional_model_request_fields,
+      :inference_config,
+      :prompt_template,
       :query_transformation_configuration)
       SENSITIVE = []
       include Aws::Structure
@@ -2300,7 +2320,8 @@ module Aws::BedrockAgentRuntime
     # The foundation model output from the orchestration step.
     #
     # @!attribute [rw] metadata
-    #   Contains information about the foundation model output.
+    #   Contains information about the foundation model output from the
+    #   orchestration step.
     #   @return [Types::Metadata]
     #
     # @!attribute [rw] raw_response
@@ -2458,7 +2479,8 @@ module Aws::BedrockAgentRuntime
     # The foundation model output from the post-processing step.
     #
     # @!attribute [rw] metadata
-    #   Provides details of the foundation model.
+    #   Contains information about the foundation model output from the
+    #   post-processing step.
     #   @return [Types::Metadata]
     #
     # @!attribute [rw] parsed_response
@@ -2467,7 +2489,7 @@ module Aws::BedrockAgentRuntime
     #   @return [Types::PostProcessingParsedResponse]
     #
     # @!attribute [rw] raw_response
-    #   Contains the raw output from the foundation model.
+    #   Details of the raw response from the foundation model output.
     #   @return [Types::RawResponse]
     #
     # @!attribute [rw] trace_id
@@ -2543,7 +2565,8 @@ module Aws::BedrockAgentRuntime
     # The foundation model output from the pre-processing step.
     #
     # @!attribute [rw] metadata
-    #   Provides details of the foundation model.
+    #   Contains information about the foundation model output from the
+    #   pre-processing step.
     #   @return [Types::Metadata]
     #
     # @!attribute [rw] parsed_response
@@ -2552,7 +2575,7 @@ module Aws::BedrockAgentRuntime
     #   @return [Types::PreProcessingParsedResponse]
     #
     # @!attribute [rw] raw_response
-    #   Contains the raw output from the foundation model.
+    #   Details of the raw response from the foundation model output.
     #   @return [Types::RawResponse]
     #
     # @!attribute [rw] trace_id

data/lib/aws-sdk-bedrockagentruntime.rb

@@ -55,7 +55,7 @@ module Aws::BedrockAgentRuntime
   autoload :Endpoints, 'aws-sdk-bedrockagentruntime/endpoints'
   autoload :EventStreams, 'aws-sdk-bedrockagentruntime/event_streams'
 
-  GEM_VERSION = '1.28.0'
+  GEM_VERSION = '1.29.0'
 
 end
 

data/sig/client.rbs

@@ -419,9 +419,22 @@ module Aws
                                        knowledge_base_id: ::String,
                                        model_arn: ::String,
                                        orchestration_configuration: {
+                                         additional_model_request_fields: Hash[::String, {
+                                           }]?,
+                                         inference_config: {
+                                           text_inference_config: {
+                                             max_tokens: ::Integer?,
+                                             stop_sequences: Array[::String]?,
+                                             temperature: ::Float?,
+                                             top_p: ::Float?
+                                           }?
+                                         }?,
+                                         prompt_template: {
+                                           text_prompt_template: ::String?
+                                         }?,
                                          query_transformation_configuration: {
                                            type: ("QUERY_DECOMPOSITION")
-                                         }
+                                         }?
                                        }?,
                                        retrieval_configuration: {
                                          vector_search_configuration: {

data/sig/types.rbs

@@ -566,6 +566,9 @@ module Aws::BedrockAgentRuntime
     end
 
     class OrchestrationConfiguration
+      attr_accessor additional_model_request_fields: ::Hash[::String, untyped]
+      attr_accessor inference_config: Types::InferenceConfig
+      attr_accessor prompt_template: Types::PromptTemplate
       attr_accessor query_transformation_configuration: Types::QueryTransformationConfiguration
       SENSITIVE: []
     end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: aws-sdk-bedrockagentruntime
 version: !ruby/object:Gem::Version
-  version: 1.28.0
+  version: 1.29.0
 platform: ruby
 authors:
 - Amazon Web Services
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-10-18 00:00:00.000000000 Z
+date: 2024-10-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk-core