feat(client-bedrock-agent): This release adds support for using Provi…

…sioned Throughput with Bedrock Agents.
aws · May 3, 2024 · 7c1d339 · 7c1d339
1 parent 79fb03b
commit 7c1d339
Show file tree

Hide file tree

Showing 7 changed files with 72 additions and 20 deletions.
diff --git a/clients/client-bedrock-agent/src/commands/CreateAgentAliasCommand.ts b/clients/client-bedrock-agent/src/commands/CreateAgentAliasCommand.ts
@@ -41,7 +41,8 @@ export interface CreateAgentAliasCommandOutput extends CreateAgentAliasResponse,
  *   description: "STRING_VALUE",
  *   routingConfiguration: [ // AgentAliasRoutingConfiguration
  *     { // AgentAliasRoutingConfigurationListItem
- *       agentVersion: "STRING_VALUE", // required
+ *       agentVersion: "STRING_VALUE",
+ *       provisionedThroughput: "STRING_VALUE",
  *     },
  *   ],
  *   tags: { // TagsMap
@@ -60,7 +61,8 @@ export interface CreateAgentAliasCommandOutput extends CreateAgentAliasResponse,
  * //     description: "STRING_VALUE",
  * //     routingConfiguration: [ // AgentAliasRoutingConfiguration // required
  * //       { // AgentAliasRoutingConfigurationListItem
- * //         agentVersion: "STRING_VALUE", // required
+ * //         agentVersion: "STRING_VALUE",
+ * //         provisionedThroughput: "STRING_VALUE",
  * //       },
  * //     ],
  * //     createdAt: new Date("TIMESTAMP"), // required
@@ -69,14 +71,18 @@ export interface CreateAgentAliasCommandOutput extends CreateAgentAliasResponse,
  * //       { // AgentAliasHistoryEvent
  * //         routingConfiguration: [
  * //           {
- * //             agentVersion: "STRING_VALUE", // required
+ * //             agentVersion: "STRING_VALUE",
+ * //             provisionedThroughput: "STRING_VALUE",
  * //           },
  * //         ],
  * //         endDate: new Date("TIMESTAMP"),
  * //         startDate: new Date("TIMESTAMP"),
  * //       },
  * //     ],
  * //     agentAliasStatus: "CREATING" || "PREPARED" || "FAILED" || "UPDATING" || "DELETING", // required
+ * //     failureReasons: [ // FailureReasons
+ * //       "STRING_VALUE",
+ * //     ],
  * //   },
  * // };
  *

diff --git a/clients/client-bedrock-agent/src/commands/GetAgentAliasCommand.ts b/clients/client-bedrock-agent/src/commands/GetAgentAliasCommand.ts
@@ -50,7 +50,8 @@ export interface GetAgentAliasCommandOutput extends GetAgentAliasResponse, __Met
  * //     description: "STRING_VALUE",
  * //     routingConfiguration: [ // AgentAliasRoutingConfiguration // required
  * //       { // AgentAliasRoutingConfigurationListItem
- * //         agentVersion: "STRING_VALUE", // required
+ * //         agentVersion: "STRING_VALUE",
+ * //         provisionedThroughput: "STRING_VALUE",
  * //       },
  * //     ],
  * //     createdAt: new Date("TIMESTAMP"), // required
@@ -59,14 +60,18 @@ export interface GetAgentAliasCommandOutput extends GetAgentAliasResponse, __Met
  * //       { // AgentAliasHistoryEvent
  * //         routingConfiguration: [
  * //           {
- * //             agentVersion: "STRING_VALUE", // required
+ * //             agentVersion: "STRING_VALUE",
+ * //             provisionedThroughput: "STRING_VALUE",
  * //           },
  * //         ],
  * //         endDate: new Date("TIMESTAMP"),
  * //         startDate: new Date("TIMESTAMP"),
  * //       },
  * //     ],
  * //     agentAliasStatus: "CREATING" || "PREPARED" || "FAILED" || "UPDATING" || "DELETING", // required
+ * //     failureReasons: [ // FailureReasons
+ * //       "STRING_VALUE",
+ * //     ],
  * //   },
  * // };
  *

diff --git a/clients/client-bedrock-agent/src/commands/ListAgentAliasesCommand.ts b/clients/client-bedrock-agent/src/commands/ListAgentAliasesCommand.ts
@@ -49,7 +49,8 @@ export interface ListAgentAliasesCommandOutput extends ListAgentAliasesResponse,
  * //       description: "STRING_VALUE",
  * //       routingConfiguration: [ // AgentAliasRoutingConfiguration
  * //         { // AgentAliasRoutingConfigurationListItem
- * //           agentVersion: "STRING_VALUE", // required
+ * //           agentVersion: "STRING_VALUE",
+ * //           provisionedThroughput: "STRING_VALUE",
  * //         },
  * //       ],
  * //       agentAliasStatus: "CREATING" || "PREPARED" || "FAILED" || "UPDATING" || "DELETING", // required

diff --git a/clients/client-bedrock-agent/src/commands/UpdateAgentAliasCommand.ts b/clients/client-bedrock-agent/src/commands/UpdateAgentAliasCommand.ts
@@ -41,7 +41,8 @@ export interface UpdateAgentAliasCommandOutput extends UpdateAgentAliasResponse,
  *   description: "STRING_VALUE",
  *   routingConfiguration: [ // AgentAliasRoutingConfiguration
  *     { // AgentAliasRoutingConfigurationListItem
- *       agentVersion: "STRING_VALUE", // required
+ *       agentVersion: "STRING_VALUE",
+ *       provisionedThroughput: "STRING_VALUE",
  *     },
  *   ],
  * };
@@ -57,7 +58,8 @@ export interface UpdateAgentAliasCommandOutput extends UpdateAgentAliasResponse,
  * //     description: "STRING_VALUE",
  * //     routingConfiguration: [ // AgentAliasRoutingConfiguration // required
  * //       { // AgentAliasRoutingConfigurationListItem
- * //         agentVersion: "STRING_VALUE", // required
+ * //         agentVersion: "STRING_VALUE",
+ * //         provisionedThroughput: "STRING_VALUE",
  * //       },
  * //     ],
  * //     createdAt: new Date("TIMESTAMP"), // required
@@ -66,14 +68,18 @@ export interface UpdateAgentAliasCommandOutput extends UpdateAgentAliasResponse,
  * //       { // AgentAliasHistoryEvent
  * //         routingConfiguration: [
  * //           {
- * //             agentVersion: "STRING_VALUE", // required
+ * //             agentVersion: "STRING_VALUE",
+ * //             provisionedThroughput: "STRING_VALUE",
  * //           },
  * //         ],
  * //         endDate: new Date("TIMESTAMP"),
  * //         startDate: new Date("TIMESTAMP"),
  * //       },
  * //     ],
  * //     agentAliasStatus: "CREATING" || "PREPARED" || "FAILED" || "UPDATING" || "DELETING", // required
+ * //     failureReasons: [ // FailureReasons
+ * //       "STRING_VALUE",
+ * //     ],
  * //   },
  * // };
  *

diff --git a/clients/client-bedrock-agent/src/models/models_0.ts b/clients/client-bedrock-agent/src/models/models_0.ts
@@ -37,7 +37,7 @@ export const CustomControlMethod = {
 export type CustomControlMethod = (typeof CustomControlMethod)[keyof typeof CustomControlMethod];
 
 /**
- * <p>Contains details about the Lambda function containing the business logic that is carried out upon invoking the action.</p>
+ * <p>Contains details about the Lambda function containing the business logic that is carried out upon invoking the action or the custom control method for handling the information elicited from the user.</p>
  * @public
  */
 export type ActionGroupExecutor =
@@ -438,7 +438,7 @@ export interface CreateAgentActionGroupRequest {
   parentActionGroupSignature?: ActionGroupSignature;
 
   /**
-   * <p>The Amazon Resource Name (ARN) of the Lambda function containing the business logic that is carried out upon invoking the action.</p>
+   * <p>The Amazon Resource Name (ARN) of the Lambda function containing the business logic that is carried out upon invoking the action or the custom control method for handling the information elicited from the user.</p>
    * @public
    */
   actionGroupExecutor?: ActionGroupExecutor;
@@ -524,7 +524,7 @@ export interface AgentActionGroup {
   parentActionSignature?: ActionGroupSignature;
 
   /**
-   * <p>The Amazon Resource Name (ARN) of the Lambda function containing the business logic that is carried out upon invoking the action.</p>
+   * <p>The Amazon Resource Name (ARN) of the Lambda function containing the business logic that is carried out upon invoking the action or the custom control method for handling the information elicited from the user.</p>
    * @public
    */
   actionGroupExecutor?: ActionGroupExecutor;
@@ -1243,7 +1243,13 @@ export interface AgentAliasRoutingConfigurationListItem {
    * <p>The version of the agent with which the alias is associated.</p>
    * @public
    */
-  agentVersion: string | undefined;
+  agentVersion?: string;
+
+  /**
+   * <p>Information on the Provisioned Throughput assigned to an agent alias.</p>
+   * @public
+   */
+  provisionedThroughput?: string;
 }
 
 /**
@@ -1375,6 +1381,12 @@ export interface AgentAlias {
    * @public
    */
   agentAliasStatus: AgentAliasStatus | undefined;
+
+  /**
+   * <p>Information on the failure of Provisioned Throughput assigned to an agent alias.</p>
+   * @public
+   */
+  failureReasons?: string[];
 }
 
 /**
@@ -2694,7 +2706,7 @@ export interface UpdateDataSourceRequest {
   dataSourceConfiguration: DataSourceConfiguration | undefined;
 
   /**
-   * <p>The data deletion policy assigned to the data source.</p>
+   * <p>The data deletion policy of the updated data source.</p>
    * @public
    */
   dataDeletionPolicy?: DataDeletionPolicy;

diff --git a/clients/client-bedrock-agent/src/protocols/Aws_restJson1.ts b/clients/client-bedrock-agent/src/protocols/Aws_restJson1.ts
@@ -2300,6 +2300,7 @@ const de_AgentAlias = (output: any, context: __SerdeContext): AgentAlias => {
     clientToken: __expectString,
     createdAt: (_: any) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
     description: __expectString,
+    failureReasons: _json,
     routingConfiguration: _json,
     updatedAt: (_: any) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
   }) as any;

diff --git a/codegen/sdk-codegen/aws-models/bedrock-agent.json b/codegen/sdk-codegen/aws-models/bedrock-agent.json
@@ -51,7 +51,7 @@
         }
       },
       "traits": {
-        "smithy.api#documentation": "<p>Contains details about the Lambda function containing the business logic that is carried out upon invoking the action.</p>"
+        "smithy.api#documentation": "<p>Contains details about the Lambda function containing the business logic that is carried out upon invoking the action or the custom control method for handling the information elicited from the user.</p>"
       }
     },
     "com.amazonaws.bedrockagent#ActionGroupResource": {
@@ -346,7 +346,7 @@
         "actionGroupExecutor": {
           "target": "com.amazonaws.bedrockagent#ActionGroupExecutor",
           "traits": {
-            "smithy.api#documentation": "<p>The Amazon Resource Name (ARN) of the Lambda function containing the business logic that is carried out upon invoking the action.</p>"
+            "smithy.api#documentation": "<p>The Amazon Resource Name (ARN) of the Lambda function containing the business logic that is carried out upon invoking the action or the custom control method for handling the information elicited from the user.</p>"
           }
         },
         "apiSchema": {
@@ -449,6 +449,12 @@
             "smithy.api#documentation": "<p>The status of the alias of the agent and whether it is ready for use. The following statuses are possible:</p>\n         <ul>\n            <li>\n               <p>CREATING – The agent alias is being created.</p>\n            </li>\n            <li>\n               <p>PREPARED – The agent alias is finished being created or updated and is ready to be invoked.</p>\n            </li>\n            <li>\n               <p>FAILED – The agent alias API operation failed.</p>\n            </li>\n            <li>\n               <p>UPDATING – The agent alias is being updated.</p>\n            </li>\n            <li>\n               <p>DELETING – The agent alias is being deleted.</p>\n            </li>\n         </ul>",
             "smithy.api#required": {}
           }
+        },
+        "failureReasons": {
+          "target": "com.amazonaws.bedrockagent#FailureReasons",
+          "traits": {
+            "smithy.api#documentation": "<p>Information on the failure of Provisioned Throughput assigned to an agent alias.</p>"
+          }
         }
       },
       "traits": {
@@ -528,8 +534,13 @@
         "agentVersion": {
           "target": "com.amazonaws.bedrockagent#Version",
           "traits": {
-            "smithy.api#documentation": "<p>The version of the agent with which the alias is associated.</p>",
-            "smithy.api#required": {}
+            "smithy.api#documentation": "<p>The version of the agent with which the alias is associated.</p>"
+          }
+        },
+        "provisionedThroughput": {
+          "target": "com.amazonaws.bedrockagent#ProvisionedModelIdentifier",
+          "traits": {
+            "smithy.api#documentation": "<p>Information on the Provisioned Throughput assigned to an agent alias.</p>"
           }
         }
       },
@@ -2136,7 +2147,7 @@
         "actionGroupExecutor": {
           "target": "com.amazonaws.bedrockagent#ActionGroupExecutor",
           "traits": {
-            "smithy.api#documentation": "<p>The Amazon Resource Name (ARN) of the Lambda function containing the business logic that is carried out upon invoking the action.</p>"
+            "smithy.api#documentation": "<p>The Amazon Resource Name (ARN) of the Lambda function containing the business logic that is carried out upon invoking the action or the custom control method for handling the information elicited from the user.</p>"
           }
         },
         "apiSchema": {
@@ -6376,6 +6387,16 @@
         }
       }
     },
+    "com.amazonaws.bedrockagent#ProvisionedModelIdentifier": {
+      "type": "string",
+      "traits": {
+        "smithy.api#length": {
+          "min": 1,
+          "max": 2048
+        },
+        "smithy.api#pattern": "^((([0-9a-zA-Z][_-]?){1,63})|(arn:aws(-[^:]+)?:bedrock:[a-z0-9-]{1,20}:[0-9]{12}:provisioned-model/[a-z0-9]{12}))$"
+      }
+    },
     "com.amazonaws.bedrockagent#RdsArn": {
       "type": "string",
       "traits": {
@@ -7711,7 +7732,7 @@
         "dataDeletionPolicy": {
           "target": "com.amazonaws.bedrockagent#DataDeletionPolicy",
           "traits": {
-            "smithy.api#documentation": "<p>The data deletion policy assigned to the data source.</p>"
+            "smithy.api#documentation": "<p>The data deletion policy of the updated data source.</p>"
           }
         },
         "serverSideEncryptionConfiguration": {