aws-sdk-bedrockagentruntime 1.3.0 → 1.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 12cf1b84a20e0969d3e184e5b93f9455474446b7e54b938b72fb071fb458f26b
-  data.tar.gz: 8aff379d2148dcf2554d477750d148e0dfdb42f42aa85e6e50cce3347f76bc4b
+  metadata.gz: 24e10fc42b8ea0b4e35d44ca0c691949feaab6fe28ebe2a12c290ea005730c27
+  data.tar.gz: 7239ae58b7fc4a5343de154490f9c81eb7011fdd9a1be9472fc64575f6b4f803
 SHA512:
-  metadata.gz: 94b70ee38eeab8f9df78cfcef7f8ec0de55e22da72660b37846584294b27b68412491d79464a0ec1c69b6fa0332c110efc4b32315ecba769a47c14d00bd396cd
-  data.tar.gz: d6ba32ed95ac5aaeaac0a6f87d64bd27a5719e905101cf4741c570d8c175396b2102b58325901241a08918c37a197beef563e8fd8cd0f0907717de48cd348fbb
+  metadata.gz: 3f4d8ef61cdf0882cea1ab50df77212c91dda6f4a9457ad6632702c328f21fa20c1ff2dcff52544e935f747bb7b973c65f68ee40d77c63342c14f079599be7d7
+  data.tar.gz: 7b77085b8fe83e0f72e6276a82403b2a891515a51c3f2957860d516cb4b9657ea8be881250cda8abe01cd5417ef783b2ee499b8a84484bf72ca3f82f2e483ae6

data/CHANGELOG.md

@@ -1,6 +1,16 @@
 Unreleased Changes
 ------------------
 
+1.5.0 (2024-03-27)
+------------------
+
+* Feature - This release introduces filtering support on Retrieve and RetrieveAndGenerate APIs.
+
+1.4.0 (2024-03-26)
+------------------
+
+* Feature - This release adds support to customize prompts sent through the RetrieveAndGenerate API in Agents for Amazon Bedrock.
+
 1.3.0 (2024-03-08)
 ------------------
 

data/VERSION

@@ -1 +1 @@
-1.3.0
+1.5.0

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

@@ -459,6 +459,11 @@ module Aws::BedrockAgentRuntime
     #
     # @option params [Types::SessionState] :session_state
     #   Contains parameters that specify various attributes of the session.
+    #   For more information, see [Control session context][1].
+    #
+    #
+    #
+    #   [1]: https://docs.aws.amazon.com/bedrock/latest/userguide/agents-session-state.html
     #
     # @return [Types::InvokeAgentResponse] Returns a {Seahorse::Client::Response response} object which responds to the following methods:
     #
@@ -681,6 +686,7 @@ module Aws::BedrockAgentRuntime
     #   event.attribution.citations[0].retrieved_references[0].content.text #=> String
     #   event.attribution.citations[0].retrieved_references[0].location.s3_location.uri #=> String
     #   event.attribution.citations[0].retrieved_references[0].location.type #=> String, one of "S3"
+    #   event.attribution.citations[0].retrieved_references[0].metadata #=> Hash
     #   event.bytes #=> String
     #
     #   For :conflict_exception event available at #on_conflict_exception_event callback and response eventstream enumerator:
@@ -742,6 +748,7 @@ module Aws::BedrockAgentRuntime
     #   event.trace.orchestration_trace.observation.knowledge_base_lookup_output.retrieved_references[0].content.text #=> String
     #   event.trace.orchestration_trace.observation.knowledge_base_lookup_output.retrieved_references[0].location.s3_location.uri #=> String
     #   event.trace.orchestration_trace.observation.knowledge_base_lookup_output.retrieved_references[0].location.type #=> String, one of "S3"
+    #   event.trace.orchestration_trace.observation.knowledge_base_lookup_output.retrieved_references[0].metadata #=> Hash
     #   event.trace.orchestration_trace.observation.reprompt_response.source #=> String, one of "ACTION_GROUP", "KNOWLEDGE_BASE", "PARSER"
     #   event.trace.orchestration_trace.observation.reprompt_response.text #=> String
     #   event.trace.orchestration_trace.observation.trace_id #=> String
@@ -822,10 +829,15 @@ module Aws::BedrockAgentRuntime
     #   another request to retrieve the next batch of results.
     #
     # @option params [Types::KnowledgeBaseRetrievalConfiguration] :retrieval_configuration
-    #   Contains details about how the results should be returned.
+    #   Contains configurations for the knowledge base query and retrieval
+    #   process. For more information, see [Query configurations][1].
+    #
+    #
+    #
+    #   [1]: https://docs.aws.amazon.com/bedrock/latest/userguide/kb-test-config.html
     #
     # @option params [required, Types::KnowledgeBaseQuery] :retrieval_query
-    #   The query to send the knowledge base.
+    #   Contains the query to send the knowledge base.
     #
     # @return [Types::RetrieveResponse] Returns a {Seahorse::Client::Response response} object which responds to the following methods:
     #
@@ -841,6 +853,63 @@ module Aws::BedrockAgentRuntime
     #     next_token: "NextToken",
     #     retrieval_configuration: {
     #       vector_search_configuration: { # required
+    #         filter: {
+    #           and_all: [
+    #             {
+    #               # recursive RetrievalFilter
+    #             },
+    #           ],
+    #           equals: {
+    #             key: "FilterKey", # required
+    #             value: { # required
+    #             },
+    #           },
+    #           greater_than: {
+    #             key: "FilterKey", # required
+    #             value: { # required
+    #             },
+    #           },
+    #           greater_than_or_equals: {
+    #             key: "FilterKey", # required
+    #             value: { # required
+    #             },
+    #           },
+    #           in: {
+    #             key: "FilterKey", # required
+    #             value: { # required
+    #             },
+    #           },
+    #           less_than: {
+    #             key: "FilterKey", # required
+    #             value: { # required
+    #             },
+    #           },
+    #           less_than_or_equals: {
+    #             key: "FilterKey", # required
+    #             value: { # required
+    #             },
+    #           },
+    #           not_equals: {
+    #             key: "FilterKey", # required
+    #             value: { # required
+    #             },
+    #           },
+    #           not_in: {
+    #             key: "FilterKey", # required
+    #             value: { # required
+    #             },
+    #           },
+    #           or_all: [
+    #             {
+    #               # recursive RetrievalFilter
+    #             },
+    #           ],
+    #           starts_with: {
+    #             key: "FilterKey", # required
+    #             value: { # required
+    #             },
+    #           },
+    #         },
     #         number_of_results: 1,
     #         override_search_type: "HYBRID", # accepts HYBRID, SEMANTIC
     #       },
@@ -857,6 +926,7 @@ module Aws::BedrockAgentRuntime
     #   resp.retrieval_results[0].content.text #=> String
     #   resp.retrieval_results[0].location.s3_location.uri #=> String
     #   resp.retrieval_results[0].location.type #=> String, one of "S3"
+    #   resp.retrieval_results[0].metadata #=> Hash
     #   resp.retrieval_results[0].score #=> Float
     #
     # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/Retrieve AWS API Documentation
@@ -869,25 +939,19 @@ module Aws::BedrockAgentRuntime
     end
 
     # Queries a knowledge base and generates responses based on the
-    # retrieved results. The response cites up to five sources but only
-    # selects the ones that are relevant to the query.
-    #
-    # <note markdown="1"> The `numberOfResults` field is currently unsupported for
-    # `RetrieveAndGenerate`. Don't include it in the
-    # [vectorSearchConfiguration][1] object.
-    #
-    #  </note>
+    # retrieved results. The response only cites sources that are relevant
+    # to the query.
     #
+    # @option params [required, Types::RetrieveAndGenerateInput] :input
+    #   Contains the query to be made to the knowledge base.
     #
+    # @option params [Types::RetrieveAndGenerateConfiguration] :retrieve_and_generate_configuration
+    #   Contains configurations for the knowledge base query and retrieval
+    #   process. For more information, see [Query configurations][1].
     #
-    # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_KnowledgeBaseVectorSearchConfiguration.html
     #
-    # @option params [required, Types::RetrieveAndGenerateInput] :input
-    #   Contains the query made to the knowledge base.
     #
-    # @option params [Types::RetrieveAndGenerateConfiguration] :retrieve_and_generate_configuration
-    #   Contains details about the resource being queried and the foundation
-    #   model used for generation.
+    #   [1]: https://docs.aws.amazon.com/bedrock/latest/userguide/kb-test-config.html
     #
     # @option params [Types::RetrieveAndGenerateSessionConfiguration] :session_configuration
     #   Contains details about the session with the knowledge base.
@@ -910,10 +974,72 @@ module Aws::BedrockAgentRuntime
     #     },
     #     retrieve_and_generate_configuration: {
     #       knowledge_base_configuration: {
+    #         generation_configuration: {
+    #           prompt_template: {
+    #             text_prompt_template: "TextPromptTemplate",
+    #           },
+    #         },
     #         knowledge_base_id: "KnowledgeBaseId", # required
     #         model_arn: "BedrockModelArn", # required
     #         retrieval_configuration: {
     #           vector_search_configuration: { # required
+    #             filter: {
+    #               and_all: [
+    #                 {
+    #                   # recursive RetrievalFilter
+    #                 },
+    #               ],
+    #               equals: {
+    #                 key: "FilterKey", # required
+    #                 value: { # required
+    #                 },
+    #               },
+    #               greater_than: {
+    #                 key: "FilterKey", # required
+    #                 value: { # required
+    #                 },
+    #               },
+    #               greater_than_or_equals: {
+    #                 key: "FilterKey", # required
+    #                 value: { # required
+    #                 },
+    #               },
+    #               in: {
+    #                 key: "FilterKey", # required
+    #                 value: { # required
+    #                 },
+    #               },
+    #               less_than: {
+    #                 key: "FilterKey", # required
+    #                 value: { # required
+    #                 },
+    #               },
+    #               less_than_or_equals: {
+    #                 key: "FilterKey", # required
+    #                 value: { # required
+    #                 },
+    #               },
+    #               not_equals: {
+    #                 key: "FilterKey", # required
+    #                 value: { # required
+    #                 },
+    #               },
+    #               not_in: {
+    #                 key: "FilterKey", # required
+    #                 value: { # required
+    #                 },
+    #               },
+    #               or_all: [
+    #                 {
+    #                   # recursive RetrievalFilter
+    #                 },
+    #               ],
+    #               starts_with: {
+    #                 key: "FilterKey", # required
+    #                 value: { # required
+    #                 },
+    #               },
+    #             },
     #             number_of_results: 1,
     #             override_search_type: "HYBRID", # accepts HYBRID, SEMANTIC
     #           },
@@ -937,6 +1063,7 @@ module Aws::BedrockAgentRuntime
     #   resp.citations[0].retrieved_references[0].content.text #=> String
     #   resp.citations[0].retrieved_references[0].location.s3_location.uri #=> String
     #   resp.citations[0].retrieved_references[0].location.type #=> String, one of "S3"
+    #   resp.citations[0].retrieved_references[0].metadata #=> Hash
     #   resp.output.text #=> String
     #   resp.session_id #=> String
     #
@@ -962,7 +1089,7 @@ module Aws::BedrockAgentRuntime
         params: params,
         config: config)
       context[:gem_name] = 'aws-sdk-bedrockagentruntime'
-      context[:gem_version] = '1.3.0'
+      context[:gem_version] = '1.5.0'
       Seahorse::Client::Request.new(handlers, context)
     end
 

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

@@ -34,9 +34,13 @@ module Aws::BedrockAgentRuntime
     Double = Shapes::FloatShape.new(name: 'Double')
     FailureReasonString = Shapes::StringShape.new(name: 'FailureReasonString')
     FailureTrace = Shapes::StructureShape.new(name: 'FailureTrace')
+    FilterAttribute = Shapes::StructureShape.new(name: 'FilterAttribute')
+    FilterKey = Shapes::StringShape.new(name: 'FilterKey')
+    FilterValue = Shapes::DocumentShape.new(name: 'FilterValue', document: true)
     FinalResponse = Shapes::StructureShape.new(name: 'FinalResponse')
     FinalResponseString = Shapes::StringShape.new(name: 'FinalResponseString')
     GeneratedResponsePart = Shapes::StructureShape.new(name: 'GeneratedResponsePart')
+    GenerationConfiguration = Shapes::StructureShape.new(name: 'GenerationConfiguration')
     InferenceConfiguration = Shapes::StructureShape.new(name: 'InferenceConfiguration')
     InputText = Shapes::StringShape.new(name: 'InputText')
     InternalServerException = Shapes::StructureShape.new(name: 'InternalServerException')
@@ -77,6 +81,7 @@ module Aws::BedrockAgentRuntime
     PreProcessingParsedResponse = Shapes::StructureShape.new(name: 'PreProcessingParsedResponse')
     PreProcessingTrace = Shapes::UnionShape.new(name: 'PreProcessingTrace')
     PromptSessionAttributesMap = Shapes::MapShape.new(name: 'PromptSessionAttributesMap')
+    PromptTemplate = Shapes::StructureShape.new(name: 'PromptTemplate')
     PromptText = Shapes::StringShape.new(name: 'PromptText')
     PromptType = Shapes::StringShape.new(name: 'PromptType')
     Rationale = Shapes::StructureShape.new(name: 'Rationale')
@@ -85,9 +90,14 @@ module Aws::BedrockAgentRuntime
     RequestBody = Shapes::StructureShape.new(name: 'RequestBody')
     ResourceNotFoundException = Shapes::StructureShape.new(name: 'ResourceNotFoundException')
     ResponseStream = Shapes::StructureShape.new(name: 'ResponseStream')
+    RetrievalFilter = Shapes::UnionShape.new(name: 'RetrievalFilter')
+    RetrievalFilterList = Shapes::ListShape.new(name: 'RetrievalFilterList')
     RetrievalResultContent = Shapes::StructureShape.new(name: 'RetrievalResultContent')
     RetrievalResultLocation = Shapes::StructureShape.new(name: 'RetrievalResultLocation')
     RetrievalResultLocationType = Shapes::StringShape.new(name: 'RetrievalResultLocationType')
+    RetrievalResultMetadata = Shapes::MapShape.new(name: 'RetrievalResultMetadata')
+    RetrievalResultMetadataKey = Shapes::StringShape.new(name: 'RetrievalResultMetadataKey')
+    RetrievalResultMetadataValue = Shapes::DocumentShape.new(name: 'RetrievalResultMetadataValue', document: true)
     RetrievalResultS3Location = Shapes::StructureShape.new(name: 'RetrievalResultS3Location')
     RetrieveAndGenerateConfiguration = Shapes::StructureShape.new(name: 'RetrieveAndGenerateConfiguration')
     RetrieveAndGenerateInput = Shapes::StructureShape.new(name: 'RetrieveAndGenerateInput')
@@ -113,6 +123,7 @@ module Aws::BedrockAgentRuntime
     StopSequences = Shapes::ListShape.new(name: 'StopSequences')
     String = Shapes::StringShape.new(name: 'String')
     Temperature = Shapes::FloatShape.new(name: 'Temperature')
+    TextPromptTemplate = Shapes::StringShape.new(name: 'TextPromptTemplate')
     TextResponsePart = Shapes::StructureShape.new(name: 'TextResponsePart')
     ThrottlingException = Shapes::StructureShape.new(name: 'ThrottlingException')
     TopK = Shapes::IntegerShape.new(name: 'TopK')
@@ -165,12 +176,19 @@ module Aws::BedrockAgentRuntime
     FailureTrace.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))
     FailureTrace.struct_class = Types::FailureTrace
 
+    FilterAttribute.add_member(:key, Shapes::ShapeRef.new(shape: FilterKey, required: true, location_name: "key"))
+    FilterAttribute.add_member(:value, Shapes::ShapeRef.new(shape: FilterValue, required: true, location_name: "value"))
+    FilterAttribute.struct_class = Types::FilterAttribute
+
     FinalResponse.add_member(:text, Shapes::ShapeRef.new(shape: FinalResponseString, location_name: "text"))
     FinalResponse.struct_class = Types::FinalResponse
 
     GeneratedResponsePart.add_member(:text_response_part, Shapes::ShapeRef.new(shape: TextResponsePart, location_name: "textResponsePart"))
     GeneratedResponsePart.struct_class = Types::GeneratedResponsePart
 
+    GenerationConfiguration.add_member(:prompt_template, Shapes::ShapeRef.new(shape: PromptTemplate, location_name: "promptTemplate"))
+    GenerationConfiguration.struct_class = Types::GenerationConfiguration
+
     InferenceConfiguration.add_member(:maximum_length, Shapes::ShapeRef.new(shape: MaximumLength, location_name: "maximumLength"))
     InferenceConfiguration.add_member(:stop_sequences, Shapes::ShapeRef.new(shape: StopSequences, location_name: "stopSequences"))
     InferenceConfiguration.add_member(:temperature, Shapes::ShapeRef.new(shape: Temperature, location_name: "temperature"))
@@ -218,16 +236,19 @@ module Aws::BedrockAgentRuntime
 
     KnowledgeBaseRetrievalResult.add_member(:content, Shapes::ShapeRef.new(shape: RetrievalResultContent, required: true, location_name: "content"))
     KnowledgeBaseRetrievalResult.add_member(:location, Shapes::ShapeRef.new(shape: RetrievalResultLocation, location_name: "location"))
+    KnowledgeBaseRetrievalResult.add_member(:metadata, Shapes::ShapeRef.new(shape: RetrievalResultMetadata, location_name: "metadata"))
     KnowledgeBaseRetrievalResult.add_member(:score, Shapes::ShapeRef.new(shape: Double, location_name: "score"))
     KnowledgeBaseRetrievalResult.struct_class = Types::KnowledgeBaseRetrievalResult
 
     KnowledgeBaseRetrievalResults.member = Shapes::ShapeRef.new(shape: KnowledgeBaseRetrievalResult)
 
+    KnowledgeBaseRetrieveAndGenerateConfiguration.add_member(:generation_configuration, Shapes::ShapeRef.new(shape: GenerationConfiguration, location_name: "generationConfiguration"))
     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(:filter, Shapes::ShapeRef.new(shape: RetrievalFilter, location_name: "filter"))
     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
@@ -306,6 +327,9 @@ module Aws::BedrockAgentRuntime
     PromptSessionAttributesMap.key = Shapes::ShapeRef.new(shape: String)
     PromptSessionAttributesMap.value = Shapes::ShapeRef.new(shape: String)
 
+    PromptTemplate.add_member(:text_prompt_template, Shapes::ShapeRef.new(shape: TextPromptTemplate, location_name: "textPromptTemplate"))
+    PromptTemplate.struct_class = Types::PromptTemplate
+
     Rationale.add_member(:text, Shapes::ShapeRef.new(shape: RationaleString, location_name: "text"))
     Rationale.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))
     Rationale.struct_class = Types::Rationale
@@ -333,6 +357,34 @@ module Aws::BedrockAgentRuntime
     ResponseStream.add_member(:validation_exception, Shapes::ShapeRef.new(shape: ValidationException, location_name: "validationException"))
     ResponseStream.struct_class = Types::ResponseStream
 
+    RetrievalFilter.add_member(:and_all, Shapes::ShapeRef.new(shape: RetrievalFilterList, location_name: "andAll"))
+    RetrievalFilter.add_member(:equals, Shapes::ShapeRef.new(shape: FilterAttribute, location_name: "equals"))
+    RetrievalFilter.add_member(:greater_than, Shapes::ShapeRef.new(shape: FilterAttribute, location_name: "greaterThan"))
+    RetrievalFilter.add_member(:greater_than_or_equals, Shapes::ShapeRef.new(shape: FilterAttribute, location_name: "greaterThanOrEquals"))
+    RetrievalFilter.add_member(:in, Shapes::ShapeRef.new(shape: FilterAttribute, location_name: "in"))
+    RetrievalFilter.add_member(:less_than, Shapes::ShapeRef.new(shape: FilterAttribute, location_name: "lessThan"))
+    RetrievalFilter.add_member(:less_than_or_equals, Shapes::ShapeRef.new(shape: FilterAttribute, location_name: "lessThanOrEquals"))
+    RetrievalFilter.add_member(:not_equals, Shapes::ShapeRef.new(shape: FilterAttribute, location_name: "notEquals"))
+    RetrievalFilter.add_member(:not_in, Shapes::ShapeRef.new(shape: FilterAttribute, location_name: "notIn"))
+    RetrievalFilter.add_member(:or_all, Shapes::ShapeRef.new(shape: RetrievalFilterList, location_name: "orAll"))
+    RetrievalFilter.add_member(:starts_with, Shapes::ShapeRef.new(shape: FilterAttribute, location_name: "startsWith"))
+    RetrievalFilter.add_member(:unknown, Shapes::ShapeRef.new(shape: nil, location_name: 'unknown'))
+    RetrievalFilter.add_member_subclass(:and_all, Types::RetrievalFilter::AndAll)
+    RetrievalFilter.add_member_subclass(:equals, Types::RetrievalFilter::Equals)
+    RetrievalFilter.add_member_subclass(:greater_than, Types::RetrievalFilter::GreaterThan)
+    RetrievalFilter.add_member_subclass(:greater_than_or_equals, Types::RetrievalFilter::GreaterThanOrEquals)
+    RetrievalFilter.add_member_subclass(:in, Types::RetrievalFilter::In)
+    RetrievalFilter.add_member_subclass(:less_than, Types::RetrievalFilter::LessThan)
+    RetrievalFilter.add_member_subclass(:less_than_or_equals, Types::RetrievalFilter::LessThanOrEquals)
+    RetrievalFilter.add_member_subclass(:not_equals, Types::RetrievalFilter::NotEquals)
+    RetrievalFilter.add_member_subclass(:not_in, Types::RetrievalFilter::NotIn)
+    RetrievalFilter.add_member_subclass(:or_all, Types::RetrievalFilter::OrAll)
+    RetrievalFilter.add_member_subclass(:starts_with, Types::RetrievalFilter::StartsWith)
+    RetrievalFilter.add_member_subclass(:unknown, Types::RetrievalFilter::Unknown)
+    RetrievalFilter.struct_class = Types::RetrievalFilter
+
+    RetrievalFilterList.member = Shapes::ShapeRef.new(shape: RetrievalFilter)
+
     RetrievalResultContent.add_member(:text, Shapes::ShapeRef.new(shape: String, required: true, location_name: "text"))
     RetrievalResultContent.struct_class = Types::RetrievalResultContent
 
@@ -340,6 +392,9 @@ module Aws::BedrockAgentRuntime
     RetrievalResultLocation.add_member(:type, Shapes::ShapeRef.new(shape: RetrievalResultLocationType, required: true, location_name: "type"))
     RetrievalResultLocation.struct_class = Types::RetrievalResultLocation
 
+    RetrievalResultMetadata.key = Shapes::ShapeRef.new(shape: RetrievalResultMetadataKey)
+    RetrievalResultMetadata.value = Shapes::ShapeRef.new(shape: RetrievalResultMetadataValue)
+
     RetrievalResultS3Location.add_member(:uri, Shapes::ShapeRef.new(shape: String, location_name: "uri"))
     RetrievalResultS3Location.struct_class = Types::RetrievalResultS3Location
 
@@ -379,6 +434,7 @@ module Aws::BedrockAgentRuntime
 
     RetrievedReference.add_member(:content, Shapes::ShapeRef.new(shape: RetrievalResultContent, location_name: "content"))
     RetrievedReference.add_member(:location, Shapes::ShapeRef.new(shape: RetrievalResultLocation, location_name: "location"))
+    RetrievedReference.add_member(:metadata, Shapes::ShapeRef.new(shape: RetrievalResultMetadata, location_name: "metadata"))
     RetrievedReference.struct_class = Types::RetrievedReference
 
     RetrievedReferences.member = Shapes::ShapeRef.new(shape: RetrievedReference)