aws-sdk-bedrockagentruntime 1.27.0 → 1.29.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4e7448cbe295f831a06c04f2bdb7bd006b193c4317d4961d8a3b1543bc008b46
-  data.tar.gz: 9362b237ea89785d4e402e2a27ac8fcf5e566a341c22e70da5da92fc1e1a8c67
+  metadata.gz: 874f1f6793b18532bc35891baa80c7ba272f425884814271fca7df08b2e5a5be
+  data.tar.gz: fd3769ce8236cff70661e686fb9d39bd37a43ffcb47ad06c6e229933e3f4e416
 SHA512:
-  metadata.gz: '059bcc004b19a3ac1457b915bf272703043bc086e068a125d5dc7d20efcacbb17edcb2ba55091c17bbb844a79342ad856e202de9290f4368b16b32d05e1c5333'
-  data.tar.gz: d531fcd88a45991466c06d6fee6227dc252fe7a8b0519a133b529470dc8224a71d67221fa6c6ca36f40180a92217e2cf8d4c00552e2b171b3354216921edd008
+  metadata.gz: 3997c97c9d6484bdb249afc314d0c254db21297ca85b912b653801b84bb48b61ee224da8a2ae7e286774e90755ba1bfb4345fa22062f1d31545ea0db764a1ad7
+  data.tar.gz: fb395ef3e0549e25c726e6bbe7a168eb19ff580ee2c4c6b44d9ee55fb4e38026912c212c82c13b451545c37f3f7b71289709041c08080a217f7d7f362e1d4cf6

data/CHANGELOG.md

@@ -1,6 +1,16 @@
 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)
+------------------
+
+* Feature - Code Generated Changes, see `./build_tools` or `aws-sdk-core`'s CHANGELOG.md for details.
+
 1.27.0 (2024-10-02)
 ------------------
 

data/VERSION

@@ -1 +1 @@
-1.27.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.27.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/endpoint_parameters.rb

@@ -52,15 +52,18 @@ module Aws::BedrockAgentRuntime
       self[:region] = options[:region]
       self[:use_dual_stack] = options[:use_dual_stack]
       self[:use_dual_stack] = false if self[:use_dual_stack].nil?
-      if self[:use_dual_stack].nil?
-        raise ArgumentError, "Missing required EndpointParameter: :use_dual_stack"
-      end
       self[:use_fips] = options[:use_fips]
       self[:use_fips] = false if self[:use_fips].nil?
-      if self[:use_fips].nil?
-        raise ArgumentError, "Missing required EndpointParameter: :use_fips"
-      end
       self[:endpoint] = options[:endpoint]
     end
+
+    def self.create(config, options={})
+      new({
+        region: config.region,
+        use_dual_stack: config.use_dualstack_endpoint,
+        use_fips: config.use_fips_endpoint,
+        endpoint: (config.endpoint.to_s unless config.regional_endpoint),
+      }.merge(options))
+    end
   end
 end

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

@@ -12,71 +12,9 @@ module Aws::BedrockAgentRuntime
   # @api private
   module Endpoints
 
-    class DeleteAgentMemory
-      def self.build(context)
-        Aws::BedrockAgentRuntime::EndpointParameters.new(
-          region: context.config.region,
-          use_dual_stack: context.config.use_dualstack_endpoint,
-          use_fips: context.config.use_fips_endpoint,
-          endpoint: context.config.regional_endpoint ? nil : context.config.endpoint.to_s,
-        )
-      end
-    end
-
-    class GetAgentMemory
-      def self.build(context)
-        Aws::BedrockAgentRuntime::EndpointParameters.new(
-          region: context.config.region,
-          use_dual_stack: context.config.use_dualstack_endpoint,
-          use_fips: context.config.use_fips_endpoint,
-          endpoint: context.config.regional_endpoint ? nil : context.config.endpoint.to_s,
-        )
-      end
-    end
-
-    class InvokeAgent
-      def self.build(context)
-        Aws::BedrockAgentRuntime::EndpointParameters.new(
-          region: context.config.region,
-          use_dual_stack: context.config.use_dualstack_endpoint,
-          use_fips: context.config.use_fips_endpoint,
-          endpoint: context.config.regional_endpoint ? nil : context.config.endpoint.to_s,
-        )
-      end
-    end
 
-    class InvokeFlow
-      def self.build(context)
-        Aws::BedrockAgentRuntime::EndpointParameters.new(
-          region: context.config.region,
-          use_dual_stack: context.config.use_dualstack_endpoint,
-          use_fips: context.config.use_fips_endpoint,
-          endpoint: context.config.regional_endpoint ? nil : context.config.endpoint.to_s,
-        )
-      end
+    def self.parameters_for_operation(context)
+      Aws::BedrockAgentRuntime::EndpointParameters.create(context.config)
     end
-
-    class Retrieve
-      def self.build(context)
-        Aws::BedrockAgentRuntime::EndpointParameters.new(
-          region: context.config.region,
-          use_dual_stack: context.config.use_dualstack_endpoint,
-          use_fips: context.config.use_fips_endpoint,
-          endpoint: context.config.regional_endpoint ? nil : context.config.endpoint.to_s,
-        )
-      end
-    end
-
-    class RetrieveAndGenerate
-      def self.build(context)
-        Aws::BedrockAgentRuntime::EndpointParameters.new(
-          region: context.config.region,
-          use_dual_stack: context.config.use_dualstack_endpoint,
-          use_fips: context.config.use_fips_endpoint,
-          endpoint: context.config.regional_endpoint ? nil : context.config.endpoint.to_s,
-        )
-      end
-    end
-
   end
 end

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

@@ -27,7 +27,7 @@ The endpoint provider used to resolve endpoints. Any object that responds to
       class Handler < Seahorse::Client::Handler
         def call(context)
           unless context[:discovered_endpoint]
-            params = parameters_for_operation(context)
+            params = Aws::BedrockAgentRuntime::Endpoints.parameters_for_operation(context)
             endpoint = context.config.endpoint_provider.resolve_endpoint(params)
 
             context.http_request.endpoint = endpoint.url
@@ -67,23 +67,6 @@ The endpoint provider used to resolve endpoints. Any object that responds to
             context.http_request.headers[key] = value
           end
         end
-
-        def parameters_for_operation(context)
-          case context.operation_name
-          when :delete_agent_memory
-            Aws::BedrockAgentRuntime::Endpoints::DeleteAgentMemory.build(context)
-          when :get_agent_memory
-            Aws::BedrockAgentRuntime::Endpoints::GetAgentMemory.build(context)
-          when :invoke_agent
-            Aws::BedrockAgentRuntime::Endpoints::InvokeAgent.build(context)
-          when :invoke_flow
-            Aws::BedrockAgentRuntime::Endpoints::InvokeFlow.build(context)
-          when :retrieve
-            Aws::BedrockAgentRuntime::Endpoints::Retrieve.build(context)
-          when :retrieve_and_generate
-            Aws::BedrockAgentRuntime::Endpoints::RetrieveAndGenerate.build(context)
-          end
-        end
       end
 
       def add_handlers(handlers, _config)

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.27.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.27.0
+  version: 1.29.0
 platform: ruby
 authors:
 - Amazon Web Services
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-10-02 00:00:00.000000000 Z
+date: 2024-10-21 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.207.0
+        version: 3.210.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
@@ -29,7 +29,7 @@ dependencies:
         version: '3'
     - - ">="
       - !ruby/object:Gem::Version
-        version: 3.207.0
+        version: 3.210.0
 - !ruby/object:Gem::Dependency
   name: aws-sigv4
   requirement: !ruby/object:Gem::Requirement