feat(client-bedrock-agent-runtime): Amazon Bedrock Knowledge Bases no…

…w supports using inference profiles to increase throughput and improve resilience.
aws · Sep 11, 2024 · a70cc8d · a70cc8d
1 parent 85d62db
commit a70cc8d
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 9 deletions.
diff --git a/clients/client-bedrock-agent-runtime/src/commands/RetrieveAndGenerateCommand.ts b/clients/client-bedrock-agent-runtime/src/commands/RetrieveAndGenerateCommand.ts
@@ -37,7 +37,7 @@ export interface RetrieveAndGenerateCommandInput extends RetrieveAndGenerateRequ
 export interface RetrieveAndGenerateCommandOutput extends RetrieveAndGenerateResponse, __MetadataBearer {}
 
 /**
- * <p>Queries a knowledge base and generates responses based on the retrieved results. The response only cites sources that are relevant to the query.</p>
+ * <p>Queries a knowledge base and generates responses based on the retrieved results and using the specified foundation model or <a href="https://docs.aws.amazon.com/bedrock/latest/userguide/cross-region-inference.html">inference profile</a>. The response only cites sources that are relevant to the query.</p>
  * @example
  * Use a bare-bones client and the command you need to make an API call.
  * ```javascript

diff --git a/clients/client-bedrock-agent-runtime/src/models/models_0.ts b/clients/client-bedrock-agent-runtime/src/models/models_0.ts
@@ -4840,13 +4840,13 @@ export interface KnowledgeBaseConfiguration {
  */
 export interface KnowledgeBaseRetrieveAndGenerateConfiguration {
   /**
-   * <p>The unique identifier of the knowledge base that is queried and the foundation model used for generation.</p>
+   * <p>The unique identifier of the knowledge base that is queried.</p>
    * @public
    */
   knowledgeBaseId: string | undefined;
 
   /**
-   * <p>The ARN of the foundation model used to generate a response.</p>
+   * <p>The ARN of the foundation model or <a href="https://docs.aws.amazon.com/bedrock/latest/userguide/cross-region-inference.html">inference profile</a> used to generate a response.</p>
    * @public
    */
   modelArn: string | undefined;

diff --git a/codegen/sdk-codegen/aws-models/bedrock-agent-runtime.json b/codegen/sdk-codegen/aws-models/bedrock-agent-runtime.json
@@ -1061,10 +1061,10 @@
       "type": "string",
       "traits": {
         "smithy.api#length": {
-          "min": 20,
-          "max": 1011
+          "min": 1,
+          "max": 2048
         },
-        "smithy.api#pattern": "^arn:aws(-[^:]+)?:bedrock:[a-z0-9-]{1,20}:(([0-9]{12}:custom-model/[a-z0-9-]{1,63}[.]{1}[a-z0-9-]{1,63}/[a-z0-9]{12})|(:foundation-model/[a-z0-9-]{1,63}[.]{1}[a-z0-9-]{1,63}))$"
+        "smithy.api#pattern": "^(arn:aws(-[^:]+)?:bedrock:[a-z0-9-]{1,20}:(([0-9]{12}:custom-model/[a-z0-9-]{1,63}[.]{1}[a-z0-9-]{1,63}/[a-z0-9]{12})|(:foundation-model/[a-z0-9-]{1,63}[.]{1}[a-z0-9-]{1,63}([.:]?[a-z0-9-]{1,63}))))|(arn:aws(|-us-gov|-cn|-iso|-iso-b):bedrock:(|[0-9a-z-]{1,20}):(|[0-9]{12}):inference-profile/[a-zA-Z0-9-:.]+)|([a-z0-9-]{1,63}[.]{1}[a-z0-9-]{1,63}([.:]?[a-z0-9-]{1,63}))|(([0-9a-zA-Z][_-]?)+)$"
       }
     },
     "com.amazonaws.bedrockagentruntime#ByteContentBlob": {
@@ -3627,14 +3627,14 @@
         "knowledgeBaseId": {
           "target": "com.amazonaws.bedrockagentruntime#KnowledgeBaseId",
           "traits": {
-            "smithy.api#documentation": "<p>The unique identifier of the knowledge base that is queried and the foundation model used for generation.</p>",
+            "smithy.api#documentation": "<p>The unique identifier of the knowledge base that is queried.</p>",
             "smithy.api#required": {}
           }
         },
         "modelArn": {
           "target": "com.amazonaws.bedrockagentruntime#BedrockModelArn",
           "traits": {
-            "smithy.api#documentation": "<p>The ARN of the foundation model used to generate a response.</p>",
+            "smithy.api#documentation": "<p>The ARN of the foundation model or <a href=\"https://docs.aws.amazon.com/bedrock/latest/userguide/cross-region-inference.html\">inference profile</a> used to generate a response.</p>",
             "smithy.api#required": {}
           }
         },
@@ -5024,7 +5024,7 @@
         }
       ],
       "traits": {
-        "smithy.api#documentation": "<p>Queries a knowledge base and generates responses based on the retrieved results. The response only cites sources that are relevant to the query.</p>",
+        "smithy.api#documentation": "<p>Queries a knowledge base and generates responses based on the retrieved results and using the specified foundation model or <a href=\"https://docs.aws.amazon.com/bedrock/latest/userguide/cross-region-inference.html\">inference profile</a>. The response only cites sources that are relevant to the query.</p>",
         "smithy.api#http": {
           "code": 200,
           "method": "POST",