aws-sdk-bedrockagentruntime 1.75.0 → 1.77.0

data/lib/aws-sdk-bedrockagentruntime.rb

@@ -23,7 +23,7 @@ Aws::Plugins::GlobalConfiguration.add_identifier(:bedrockagentruntime)
 # structure.
 #
 #     bedrock_agent_runtime = Aws::BedrockAgentRuntime::Client.new
-#     resp = bedrock_agent_runtime.create_invocation(params)
+#     resp = bedrock_agent_runtime.agentic_retrieve_stream(params)
 #
 # See {Client} for more information.
 #
@@ -56,7 +56,7 @@ module Aws::BedrockAgentRuntime
   autoload :Endpoints, 'aws-sdk-bedrockagentruntime/endpoints'
   autoload :EventStreams, 'aws-sdk-bedrockagentruntime/event_streams'
 
-  GEM_VERSION = '1.75.0'
+  GEM_VERSION = '1.77.0'
 
 end
 

data/sig/client.rbs

@@ -81,6 +81,69 @@ module Aws
                   | (?Hash[Symbol, untyped]) -> instance
 
 
+      interface _AgenticRetrieveStreamResponseSuccess
+        include ::Seahorse::Client::_ResponseSuccess[Types::AgenticRetrieveStreamResponse]
+        def stream: () -> Types::AgenticRetrieveStreamResponseOutput
+      end
+      # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/BedrockAgentRuntime/Client.html#agentic_retrieve_stream-instance_method
+      def agentic_retrieve_stream: (
+                                     agentic_retrieve_configuration: {
+                                       foundation_model_configuration: {
+                                         bedrock_foundation_model_configuration: {
+                                           model_configuration: {
+                                             model_arn: ::String
+                                           }
+                                         }?,
+                                         type: ("BEDROCK_FOUNDATION_MODEL")
+                                       }?,
+                                       foundation_model_type: ("CUSTOM" | "MANAGED")?,
+                                       max_agent_iteration: ::Integer?,
+                                       reranking_configuration: {
+                                         bedrock_reranking_configuration: {
+                                           model_configuration: {
+                                             model_arn: ::String
+                                           }
+                                         }?,
+                                         type: ("BEDROCK_RERANKING_MODEL")
+                                       }?,
+                                       reranking_model_type: ("CUSTOM" | "MANAGED" | "NONE")?
+                                     },
+                                     ?generate_response: bool,
+                                     messages: Array[
+                                       {
+                                         content: {
+                                           text: ::String?
+                                         },
+                                         role: ("user" | "assistant")
+                                       }
+                                     ],
+                                     ?next_token: ::String,
+                                     ?policy_configuration: {
+                                       bedrock_guardrail_configuration: {
+                                         guardrail_id: ::String,
+                                         guardrail_version: ::String
+                                       }?
+                                     },
+                                     retrievers: Array[
+                                       {
+                                         configuration: {
+                                           knowledge_base: {
+                                             knowledge_base_id: ::String,
+                                             retrieval_overrides: {
+                                               filter: Params::retrieval_filter?,
+                                               max_number_of_results: ::Integer?
+                                             }?
+                                           }?
+                                         },
+                                         description: ::String?
+                                       }
+                                     ],
+                                     ?user_context: {
+                                       user_id: ::String
+                                     }
+                                   ) ?{ (*untyped) -> void } -> _AgenticRetrieveStreamResponseSuccess
+                                 | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) ?{ (*untyped) -> void } -> _AgenticRetrieveStreamResponseSuccess
+
       interface _CreateInvocationResponseSuccess
         include ::Seahorse::Client::_ResponseSuccess[Types::CreateInvocationResponse]
         def created_at: () -> ::Time
@@ -181,6 +244,24 @@ module Aws
                             ) -> _GetAgentMemoryResponseSuccess
                           | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _GetAgentMemoryResponseSuccess
 
+      interface _GetDocumentContentResponseSuccess
+        include ::Seahorse::Client::_ResponseSuccess[Types::GetDocumentContentResponse]
+        def document_content_length: () -> ::Integer
+        def mime_type: () -> ::String
+        def presigned_url: () -> ::String
+      end
+      # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/BedrockAgentRuntime/Client.html#get_document_content-instance_method
+      def get_document_content: (
+                                  data_source_id: ::String,
+                                  document_id: ::String,
+                                  knowledge_base_id: ::String,
+                                  ?output_format: ("RAW" | "EXTRACTED"),
+                                  ?user_context: {
+                                    user_id: ::String
+                                  }
+                                ) -> _GetDocumentContentResponseSuccess
+                              | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _GetDocumentContentResponseSuccess
+
       interface _GetExecutionFlowSnapshotResponseSuccess
         include ::Seahorse::Client::_ResponseSuccess[Types::GetExecutionFlowSnapshotResponse]
         def customer_encryption_key_arn: () -> ::String
@@ -606,6 +687,9 @@ module Aws
                         }?,
                         text: ::String?,
                         type: ("TEXT" | "IMAGE")?
+                      },
+                      ?user_context: {
+                        user_id: ::String
                       }
                     ) -> _RetrieveResponseSuccess
                   | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _RetrieveResponseSuccess
@@ -626,7 +710,10 @@ module Aws
                                    ?session_configuration: {
                                      kms_key_arn: ::String
                                    },
-                                   ?session_id: ::String
+                                   ?session_id: ::String,
+                                   ?user_context: {
+                                     user_id: ::String
+                                   }
                                  ) -> _RetrieveAndGenerateResponseSuccess
                                | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _RetrieveAndGenerateResponseSuccess
 
@@ -644,7 +731,10 @@ module Aws
                                           ?session_configuration: {
                                             kms_key_arn: ::String
                                           },
-                                          ?session_id: ::String
+                                          ?session_id: ::String,
+                                          ?user_context: {
+                                            user_id: ::String
+                                          }
                                         ) ?{ (*untyped) -> void } -> _RetrieveAndGenerateStreamResponseSuccess
                                       | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) ?{ (*untyped) -> void } -> _RetrieveAndGenerateStreamResponseSuccess
 

data/sig/params.rbs

@@ -8,36 +8,6 @@
 module Aws
   module BedrockAgentRuntime
     module Params
-      type conversation_history = {
-      messages: Array[
-        {
-          content: Array[
-            {
-              text: ::String?
-            }
-          ],
-          role: ("user" | "assistant")
-        }
-      ]?
-    }
-
-      type file_source = {
-      byte_content: {
-        data: ::String,
-        media_type: ::String
-      }?,
-      s3_location: {
-        uri: ::String
-      }?,
-      source_type: ("S3" | "BYTE_CONTENT")
-    }
-
-      type input_file = {
-      name: ::String,
-      source: Params::file_source,
-      use_case: ("CODE_INTERPRETER" | "CHAT")
-    }
-
       type retrieval_filter = {
       and_all: Array[
         {
@@ -218,15 +188,34 @@ module Aws
       }?
     }
 
-      type implicit_filter_configuration = {
-      metadata_attributes: Array[
+      type conversation_history = {
+      messages: Array[
         {
-          description: ::String,
-          key: ::String,
-          type: ("STRING" | "NUMBER" | "BOOLEAN" | "STRING_LIST")
+          content: Array[
+            {
+              text: ::String?
+            }
+          ],
+          role: ("user" | "assistant")
         }
-      ],
-      model_arn: ::String
+      ]?
+    }
+
+      type file_source = {
+      byte_content: {
+        data: ::String,
+        media_type: ::String
+      }?,
+      s3_location: {
+        uri: ::String
+      }?,
+      source_type: ("S3" | "BYTE_CONTENT")
+    }
+
+      type input_file = {
+      name: ::String,
+      source: Params::file_source,
+      use_case: ("CODE_INTERPRETER" | "CHAT")
     }
 
       type reranking_metadata_selective_mode_configuration = {
@@ -247,6 +236,39 @@ module Aws
       selective_mode_configuration: Params::reranking_metadata_selective_mode_configuration?
     }
 
+      type managed_search_bedrock_reranking_configuration = {
+      metadata_configuration: Params::metadata_configuration_for_reranking?,
+      model_configuration: {
+        additional_model_request_fields: Hash[::String, {
+          }]?,
+        model_arn: ::String
+      },
+      number_of_reranked_results: ::Integer?
+    }
+
+      type managed_search_reranking_configuration = {
+      bedrock_reranking_configuration: Params::managed_search_bedrock_reranking_configuration?,
+      type: ("BEDROCK_RERANKING_MODEL")
+    }
+
+      type managed_search_configuration = {
+      filter: Params::retrieval_filter?,
+      number_of_results: ::Integer?,
+      reranking_configuration: Params::managed_search_reranking_configuration?,
+      reranking_model_type: ("CUSTOM" | "MANAGED" | "NONE")?
+    }
+
+      type implicit_filter_configuration = {
+      metadata_attributes: Array[
+        {
+          description: ::String,
+          key: ::String,
+          type: ("STRING" | "NUMBER" | "BOOLEAN" | "STRING_LIST")
+        }
+      ],
+      model_arn: ::String
+    }
+
       type vector_search_bedrock_reranking_configuration = {
       metadata_configuration: Params::metadata_configuration_for_reranking?,
       model_configuration: {
@@ -271,7 +293,8 @@ module Aws
     }
 
       type knowledge_base_retrieval_configuration = {
-      vector_search_configuration: Params::knowledge_base_vector_search_configuration
+      managed_search_configuration: Params::managed_search_configuration?,
+      vector_search_configuration: Params::knowledge_base_vector_search_configuration?
     }
 
       type api_result = {