aws-sdk-bedrockagentruntime 1.26.0 → 1.28.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 125500bac53e227983f1a95df4208a701f877f6579424dd6985e23e580755c88
-  data.tar.gz: fc1b91f68acc8862dedd8024cfa970947f2a43d6112d8d8f6e877ba1bc5fdb06
+  metadata.gz: 4338575e5e4e1c40786134c71bd8381feff99a7dbf001e0b9c24bff649e80617
+  data.tar.gz: f6f86d4d746e7ddac1dcaab2c74c9d0abaf8a63f636cadf0290206666e5f1cab
 SHA512:
-  metadata.gz: ab5a18eceb7a9375574fabff7a665d908f0fbe4e03b539c2d421aff8cd07ffa0e5fa913e0c93f361075faacc93565c490001281056b43d4aa5421170f7cb0329
-  data.tar.gz: 7d246d3302df4d49a7106821513cfda5644fdae59a07741a9045b4e45e3697eee0b597bb0bf42cbc7df78482ef65637e790574de1a00cdcdf1ea31b9f4d78c9f
+  metadata.gz: d3385e7ef442ceff7f263840efb5343a58d981b2b777851e27c0ff3cf662ee2d399abd15feb80e26e601d9684d0c61676d7713cef23fed87741e690d36fd056c
+  data.tar.gz: 5a9131c582f6526d88951088e05a86eb45b4d968e91ad17d4b207f8e2b1185638989552b3bb18a8323434d7f99bde1e756e39161848d484db4e720470e149100

data/CHANGELOG.md

@@ -1,6 +1,16 @@
 Unreleased Changes
 ------------------
 
+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)
+------------------
+
+* Feature - Added raw model response and usage metrics to PreProcessing and PostProcessing Trace
+
 1.26.0 (2024-09-24)
 ------------------
 

data/VERSION

@@ -1 +1 @@
-1.26.0
+1.28.0

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

@@ -1187,7 +1187,10 @@ module Aws::BedrockAgentRuntime
     #   event.trace.post_processing_trace.model_invocation_input.text #=> String
     #   event.trace.post_processing_trace.model_invocation_input.trace_id #=> String
     #   event.trace.post_processing_trace.model_invocation_input.type #=> String, one of "PRE_PROCESSING", "ORCHESTRATION", "KNOWLEDGE_BASE_RESPONSE_GENERATION", "POST_PROCESSING"
+    #   event.trace.post_processing_trace.model_invocation_output.metadata.usage.input_tokens #=> Integer
+    #   event.trace.post_processing_trace.model_invocation_output.metadata.usage.output_tokens #=> Integer
     #   event.trace.post_processing_trace.model_invocation_output.parsed_response.text #=> String
+    #   event.trace.post_processing_trace.model_invocation_output.raw_response.content #=> String
     #   event.trace.post_processing_trace.model_invocation_output.trace_id #=> String
     #   event.trace.pre_processing_trace.model_invocation_input.inference_configuration.maximum_length #=> Integer
     #   event.trace.pre_processing_trace.model_invocation_input.inference_configuration.stop_sequences #=> Array
@@ -1201,8 +1204,11 @@ module Aws::BedrockAgentRuntime
     #   event.trace.pre_processing_trace.model_invocation_input.text #=> String
     #   event.trace.pre_processing_trace.model_invocation_input.trace_id #=> String
     #   event.trace.pre_processing_trace.model_invocation_input.type #=> String, one of "PRE_PROCESSING", "ORCHESTRATION", "KNOWLEDGE_BASE_RESPONSE_GENERATION", "POST_PROCESSING"
+    #   event.trace.pre_processing_trace.model_invocation_output.metadata.usage.input_tokens #=> Integer
+    #   event.trace.pre_processing_trace.model_invocation_output.metadata.usage.output_tokens #=> Integer
     #   event.trace.pre_processing_trace.model_invocation_output.parsed_response.is_valid #=> Boolean
     #   event.trace.pre_processing_trace.model_invocation_output.parsed_response.rationale #=> String
+    #   event.trace.pre_processing_trace.model_invocation_output.raw_response.content #=> String
     #   event.trace.pre_processing_trace.model_invocation_output.trace_id #=> String
     #
     #   For :validation_exception event available at #on_validation_exception_event callback and response eventstream enumerator:
@@ -1899,7 +1905,7 @@ module Aws::BedrockAgentRuntime
         tracer: tracer
       )
       context[:gem_name] = 'aws-sdk-bedrockagentruntime'
-      context[:gem_version] = '1.26.0'
+      context[:gem_version] = '1.28.0'
       Seahorse::Client::Request.new(handlers, context)
     end
 

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

@@ -755,7 +755,9 @@ module Aws::BedrockAgentRuntime
     PayloadPart.add_member(:bytes, Shapes::ShapeRef.new(shape: PartBody, location_name: "bytes"))
     PayloadPart.struct_class = Types::PayloadPart
 
+    PostProcessingModelInvocationOutput.add_member(:metadata, Shapes::ShapeRef.new(shape: Metadata, location_name: "metadata"))
     PostProcessingModelInvocationOutput.add_member(:parsed_response, Shapes::ShapeRef.new(shape: PostProcessingParsedResponse, location_name: "parsedResponse"))
+    PostProcessingModelInvocationOutput.add_member(:raw_response, Shapes::ShapeRef.new(shape: RawResponse, location_name: "rawResponse"))
     PostProcessingModelInvocationOutput.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))
     PostProcessingModelInvocationOutput.struct_class = Types::PostProcessingModelInvocationOutput
 
@@ -770,7 +772,9 @@ module Aws::BedrockAgentRuntime
     PostProcessingTrace.add_member_subclass(:unknown, Types::PostProcessingTrace::Unknown)
     PostProcessingTrace.struct_class = Types::PostProcessingTrace
 
+    PreProcessingModelInvocationOutput.add_member(:metadata, Shapes::ShapeRef.new(shape: Metadata, location_name: "metadata"))
     PreProcessingModelInvocationOutput.add_member(:parsed_response, Shapes::ShapeRef.new(shape: PreProcessingParsedResponse, location_name: "parsedResponse"))
+    PreProcessingModelInvocationOutput.add_member(:raw_response, Shapes::ShapeRef.new(shape: RawResponse, location_name: "rawResponse"))
     PreProcessingModelInvocationOutput.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))
     PreProcessingModelInvocationOutput.struct_class = Types::PreProcessingModelInvocationOutput
 

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

@@ -591,21 +591,21 @@ module Aws::BedrockAgentRuntime
     end
 
     # The configurations of the external source wrapper object in the
-    # retrieveAndGenerate function.
+    # `retrieveAndGenerate` function.
     #
     # @!attribute [rw] generation_configuration
     #   The prompt used with the external source wrapper object with the
-    #   retrieveAndGenerate function.
+    #   `retrieveAndGenerate` function.
     #   @return [Types::ExternalSourcesGenerationConfiguration]
     #
     # @!attribute [rw] model_arn
-    #   The modelArn used with the external source wrapper object in the
-    #   retrieveAndGenerate function.
+    #   The model Amazon Resource Name (ARN) for the external source wrapper
+    #   object in the `retrieveAndGenerate` function.
     #   @return [String]
     #
     # @!attribute [rw] sources
-    #   The document used with the external source wrapper object in the
-    #   retrieveAndGenerate function.
+    #   The document for the external source wrapper object in the
+    #   `retrieveAndGenerate` function.
     #   @return [Array<Types::ExternalSource>]
     #
     # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/ExternalSourcesRetrieveAndGenerateConfiguration AWS API Documentation
@@ -2457,11 +2457,19 @@ module Aws::BedrockAgentRuntime
 
     # The foundation model output from the post-processing step.
     #
+    # @!attribute [rw] metadata
+    #   Provides details of the foundation model.
+    #   @return [Types::Metadata]
+    #
     # @!attribute [rw] parsed_response
     #   Details about the response from the Lambda parsing of the output of
     #   the post-processing step.
     #   @return [Types::PostProcessingParsedResponse]
     #
+    # @!attribute [rw] raw_response
+    #   Contains the raw output from the foundation model.
+    #   @return [Types::RawResponse]
+    #
     # @!attribute [rw] trace_id
     #   The unique identifier of the trace.
     #   @return [String]
@@ -2469,9 +2477,11 @@ module Aws::BedrockAgentRuntime
     # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/PostProcessingModelInvocationOutput AWS API Documentation
     #
     class PostProcessingModelInvocationOutput < Struct.new(
+      :metadata,
       :parsed_response,
+      :raw_response,
       :trace_id)
-      SENSITIVE = [:parsed_response]
+      SENSITIVE = [:metadata, :parsed_response, :raw_response]
       include Aws::Structure
     end
 
@@ -2532,11 +2542,19 @@ module Aws::BedrockAgentRuntime
 
     # The foundation model output from the pre-processing step.
     #
+    # @!attribute [rw] metadata
+    #   Provides details of the foundation model.
+    #   @return [Types::Metadata]
+    #
     # @!attribute [rw] parsed_response
     #   Details about the response from the Lambda parsing of the output of
     #   the pre-processing step.
     #   @return [Types::PreProcessingParsedResponse]
     #
+    # @!attribute [rw] raw_response
+    #   Contains the raw output from the foundation model.
+    #   @return [Types::RawResponse]
+    #
     # @!attribute [rw] trace_id
     #   The unique identifier of the trace.
     #   @return [String]
@@ -2544,9 +2562,11 @@ module Aws::BedrockAgentRuntime
     # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/PreProcessingModelInvocationOutput AWS API Documentation
     #
     class PreProcessingModelInvocationOutput < Struct.new(
+      :metadata,
       :parsed_response,
+      :raw_response,
       :trace_id)
-      SENSITIVE = [:parsed_response]
+      SENSITIVE = [:metadata, :parsed_response, :raw_response]
       include Aws::Structure
     end
 
@@ -3144,16 +3164,21 @@ module Aws::BedrockAgentRuntime
     # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_RetrieveAndGenerate.html#API_agent-runtime_RetrieveAndGenerate_RequestSyntax
     #
     # @!attribute [rw] external_sources_configuration
-    #   The configuration used with the external source wrapper object in
-    #   the retrieveAndGenerate function.
+    #   The configuration for the external source wrapper object in the
+    #   `retrieveAndGenerate` function.
     #   @return [Types::ExternalSourcesRetrieveAndGenerateConfiguration]
     #
     # @!attribute [rw] knowledge_base_configuration
-    #   Contains details about the resource being queried.
+    #   Contains details about the knowledge base for retrieving information
+    #   and generating responses.
     #   @return [Types::KnowledgeBaseRetrieveAndGenerateConfiguration]
     #
     # @!attribute [rw] type
-    #   The type of resource that is queried by the request.
+    #   The type of resource that contains your data for retrieving
+    #   information and generating responses.
+    #
+    #   If you choose ot use `EXTERNAL_SOURCES`, then currently only Claude
+    #   3 Sonnet models for knowledge bases are supported.
     #   @return [String]
     #
     # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/RetrieveAndGenerateConfiguration AWS API Documentation

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.26.0'
+  GEM_VERSION = '1.28.0'
 
 end
 

data/sig/types.rbs

@@ -622,9 +622,11 @@ module Aws::BedrockAgentRuntime
     end
 
     class PostProcessingModelInvocationOutput
+      attr_accessor metadata: Types::Metadata
       attr_accessor parsed_response: Types::PostProcessingParsedResponse
+      attr_accessor raw_response: Types::RawResponse
       attr_accessor trace_id: ::String
-      SENSITIVE: [:parsed_response]
+      SENSITIVE: [:metadata, :parsed_response, :raw_response]
     end
 
     class PostProcessingParsedResponse
@@ -647,9 +649,11 @@ module Aws::BedrockAgentRuntime
     end
 
     class PreProcessingModelInvocationOutput
+      attr_accessor metadata: Types::Metadata
       attr_accessor parsed_response: Types::PreProcessingParsedResponse
+      attr_accessor raw_response: Types::RawResponse
       attr_accessor trace_id: ::String
-      SENSITIVE: [:parsed_response]
+      SENSITIVE: [:metadata, :parsed_response, :raw_response]
     end
 
     class PreProcessingParsedResponse

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: aws-sdk-bedrockagentruntime
 version: !ruby/object:Gem::Version
-  version: 1.26.0
+  version: 1.28.0
 platform: ruby
 authors:
 - Amazon Web Services
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-09-24 00:00:00.000000000 Z
+date: 2024-10-18 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