aws-sdk-bedrockagentruntime 1.26.0 → 1.27.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 125500bac53e227983f1a95df4208a701f877f6579424dd6985e23e580755c88
-  data.tar.gz: fc1b91f68acc8862dedd8024cfa970947f2a43d6112d8d8f6e877ba1bc5fdb06
+  metadata.gz: 4e7448cbe295f831a06c04f2bdb7bd006b193c4317d4961d8a3b1543bc008b46
+  data.tar.gz: 9362b237ea89785d4e402e2a27ac8fcf5e566a341c22e70da5da92fc1e1a8c67
 SHA512:
-  metadata.gz: ab5a18eceb7a9375574fabff7a665d908f0fbe4e03b539c2d421aff8cd07ffa0e5fa913e0c93f361075faacc93565c490001281056b43d4aa5421170f7cb0329
-  data.tar.gz: 7d246d3302df4d49a7106821513cfda5644fdae59a07741a9045b4e45e3697eee0b597bb0bf42cbc7df78482ef65637e790574de1a00cdcdf1ea31b9f4d78c9f
+  metadata.gz: '059bcc004b19a3ac1457b915bf272703043bc086e068a125d5dc7d20efcacbb17edcb2ba55091c17bbb844a79342ad856e202de9290f4368b16b32d05e1c5333'
+  data.tar.gz: d531fcd88a45991466c06d6fee6227dc252fe7a8b0519a133b529470dc8224a71d67221fa6c6ca36f40180a92217e2cf8d4c00552e2b171b3354216921edd008

data/CHANGELOG.md

@@ -1,6 +1,11 @@
 Unreleased Changes
 ------------------
 
+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.27.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.27.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/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.27.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.27.0
 platform: ruby
 authors:
 - Amazon Web Services
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-09-24 00:00:00.000000000 Z
+date: 2024-10-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk-core