Skip to content

Commit 7c1d339

Browse files
author
awstools
committed
feat(client-bedrock-agent): This release adds support for using Provisioned Throughput with Bedrock Agents.
1 parent 79fb03b commit 7c1d339

File tree

7 files changed

+72
-20
lines changed

7 files changed

+72
-20
lines changed

‎clients/client-bedrock-agent/src/commands/CreateAgentAliasCommand.ts

+9-3
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,8 @@ export interface CreateAgentAliasCommandOutput extends CreateAgentAliasResponse,
4141
* description: "STRING_VALUE",
4242
* routingConfiguration: [ // AgentAliasRoutingConfiguration
4343
* { // AgentAliasRoutingConfigurationListItem
44-
* agentVersion: "STRING_VALUE", // required
44+
* agentVersion: "STRING_VALUE",
45+
* provisionedThroughput: "STRING_VALUE",
4546
* },
4647
* ],
4748
* tags: { // TagsMap
@@ -60,7 +61,8 @@ export interface CreateAgentAliasCommandOutput extends CreateAgentAliasResponse,
6061
* // description: "STRING_VALUE",
6162
* // routingConfiguration: [ // AgentAliasRoutingConfiguration // required
6263
* // { // AgentAliasRoutingConfigurationListItem
63-
* // agentVersion: "STRING_VALUE", // required
64+
* // agentVersion: "STRING_VALUE",
65+
* // provisionedThroughput: "STRING_VALUE",
6466
* // },
6567
* // ],
6668
* // createdAt: new Date("TIMESTAMP"), // required
@@ -69,14 +71,18 @@ export interface CreateAgentAliasCommandOutput extends CreateAgentAliasResponse,
6971
* // { // AgentAliasHistoryEvent
7072
* // routingConfiguration: [
7173
* // {
72-
* // agentVersion: "STRING_VALUE", // required
74+
* // agentVersion: "STRING_VALUE",
75+
* // provisionedThroughput: "STRING_VALUE",
7376
* // },
7477
* // ],
7578
* // endDate: new Date("TIMESTAMP"),
7679
* // startDate: new Date("TIMESTAMP"),
7780
* // },
7881
* // ],
7982
* // agentAliasStatus: "CREATING" || "PREPARED" || "FAILED" || "UPDATING" || "DELETING", // required
83+
* // failureReasons: [ // FailureReasons
84+
* // "STRING_VALUE",
85+
* // ],
8086
* // },
8187
* // };
8288
*

‎clients/client-bedrock-agent/src/commands/GetAgentAliasCommand.ts

+7-2
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,8 @@ export interface GetAgentAliasCommandOutput extends GetAgentAliasResponse, __Met
5050
* // description: "STRING_VALUE",
5151
* // routingConfiguration: [ // AgentAliasRoutingConfiguration // required
5252
* // { // AgentAliasRoutingConfigurationListItem
53-
* // agentVersion: "STRING_VALUE", // required
53+
* // agentVersion: "STRING_VALUE",
54+
* // provisionedThroughput: "STRING_VALUE",
5455
* // },
5556
* // ],
5657
* // createdAt: new Date("TIMESTAMP"), // required
@@ -59,14 +60,18 @@ export interface GetAgentAliasCommandOutput extends GetAgentAliasResponse, __Met
5960
* // { // AgentAliasHistoryEvent
6061
* // routingConfiguration: [
6162
* // {
62-
* // agentVersion: "STRING_VALUE", // required
63+
* // agentVersion: "STRING_VALUE",
64+
* // provisionedThroughput: "STRING_VALUE",
6365
* // },
6466
* // ],
6567
* // endDate: new Date("TIMESTAMP"),
6668
* // startDate: new Date("TIMESTAMP"),
6769
* // },
6870
* // ],
6971
* // agentAliasStatus: "CREATING" || "PREPARED" || "FAILED" || "UPDATING" || "DELETING", // required
72+
* // failureReasons: [ // FailureReasons
73+
* // "STRING_VALUE",
74+
* // ],
7075
* // },
7176
* // };
7277
*

‎clients/client-bedrock-agent/src/commands/ListAgentAliasesCommand.ts

+2-1
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,8 @@ export interface ListAgentAliasesCommandOutput extends ListAgentAliasesResponse,
4949
* // description: "STRING_VALUE",
5050
* // routingConfiguration: [ // AgentAliasRoutingConfiguration
5151
* // { // AgentAliasRoutingConfigurationListItem
52-
* // agentVersion: "STRING_VALUE", // required
52+
* // agentVersion: "STRING_VALUE",
53+
* // provisionedThroughput: "STRING_VALUE",
5354
* // },
5455
* // ],
5556
* // agentAliasStatus: "CREATING" || "PREPARED" || "FAILED" || "UPDATING" || "DELETING", // required

‎clients/client-bedrock-agent/src/commands/UpdateAgentAliasCommand.ts

+9-3
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,8 @@ export interface UpdateAgentAliasCommandOutput extends UpdateAgentAliasResponse,
4141
* description: "STRING_VALUE",
4242
* routingConfiguration: [ // AgentAliasRoutingConfiguration
4343
* { // AgentAliasRoutingConfigurationListItem
44-
* agentVersion: "STRING_VALUE", // required
44+
* agentVersion: "STRING_VALUE",
45+
* provisionedThroughput: "STRING_VALUE",
4546
* },
4647
* ],
4748
* };
@@ -57,7 +58,8 @@ export interface UpdateAgentAliasCommandOutput extends UpdateAgentAliasResponse,
5758
* // description: "STRING_VALUE",
5859
* // routingConfiguration: [ // AgentAliasRoutingConfiguration // required
5960
* // { // AgentAliasRoutingConfigurationListItem
60-
* // agentVersion: "STRING_VALUE", // required
61+
* // agentVersion: "STRING_VALUE",
62+
* // provisionedThroughput: "STRING_VALUE",
6163
* // },
6264
* // ],
6365
* // createdAt: new Date("TIMESTAMP"), // required
@@ -66,14 +68,18 @@ export interface UpdateAgentAliasCommandOutput extends UpdateAgentAliasResponse,
6668
* // { // AgentAliasHistoryEvent
6769
* // routingConfiguration: [
6870
* // {
69-
* // agentVersion: "STRING_VALUE", // required
71+
* // agentVersion: "STRING_VALUE",
72+
* // provisionedThroughput: "STRING_VALUE",
7073
* // },
7174
* // ],
7275
* // endDate: new Date("TIMESTAMP"),
7376
* // startDate: new Date("TIMESTAMP"),
7477
* // },
7578
* // ],
7679
* // agentAliasStatus: "CREATING" || "PREPARED" || "FAILED" || "UPDATING" || "DELETING", // required
80+
* // failureReasons: [ // FailureReasons
81+
* // "STRING_VALUE",
82+
* // ],
7783
* // },
7884
* // };
7985
*

‎clients/client-bedrock-agent/src/models/models_0.ts

+17-5
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ export const CustomControlMethod = {
3737
export type CustomControlMethod = (typeof CustomControlMethod)[keyof typeof CustomControlMethod];
3838

3939
/**
40-
* <p>Contains details about the Lambda function containing the business logic that is carried out upon invoking the action.</p>
40+
* <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>
4141
* @public
4242
*/
4343
export type ActionGroupExecutor =
@@ -438,7 +438,7 @@ export interface CreateAgentActionGroupRequest {
438438
parentActionGroupSignature?: ActionGroupSignature;
439439

440440
/**
441-
* <p>The Amazon Resource Name (ARN) of the Lambda function containing the business logic that is carried out upon invoking the action.</p>
441+
* <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>
442442
* @public
443443
*/
444444
actionGroupExecutor?: ActionGroupExecutor;
@@ -524,7 +524,7 @@ export interface AgentActionGroup {
524524
parentActionSignature?: ActionGroupSignature;
525525

526526
/**
527-
* <p>The Amazon Resource Name (ARN) of the Lambda function containing the business logic that is carried out upon invoking the action.</p>
527+
* <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>
528528
* @public
529529
*/
530530
actionGroupExecutor?: ActionGroupExecutor;
@@ -1243,7 +1243,13 @@ export interface AgentAliasRoutingConfigurationListItem {
12431243
* <p>The version of the agent with which the alias is associated.</p>
12441244
* @public
12451245
*/
1246-
agentVersion: string | undefined;
1246+
agentVersion?: string;
1247+
1248+
/**
1249+
* <p>Information on the Provisioned Throughput assigned to an agent alias.</p>
1250+
* @public
1251+
*/
1252+
provisionedThroughput?: string;
12471253
}
12481254

12491255
/**
@@ -1375,6 +1381,12 @@ export interface AgentAlias {
13751381
* @public
13761382
*/
13771383
agentAliasStatus: AgentAliasStatus | undefined;
1384+
1385+
/**
1386+
* <p>Information on the failure of Provisioned Throughput assigned to an agent alias.</p>
1387+
* @public
1388+
*/
1389+
failureReasons?: string[];
13781390
}
13791391

13801392
/**
@@ -2694,7 +2706,7 @@ export interface UpdateDataSourceRequest {
26942706
dataSourceConfiguration: DataSourceConfiguration | undefined;
26952707

26962708
/**
2697-
* <p>The data deletion policy assigned to the data source.</p>
2709+
* <p>The data deletion policy of the updated data source.</p>
26982710
* @public
26992711
*/
27002712
dataDeletionPolicy?: DataDeletionPolicy;

‎clients/client-bedrock-agent/src/protocols/Aws_restJson1.ts

+1
Original file line numberDiff line numberDiff line change
@@ -2300,6 +2300,7 @@ const de_AgentAlias = (output: any, context: __SerdeContext): AgentAlias => {
23002300
clientToken: __expectString,
23012301
createdAt: (_: any) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
23022302
description: __expectString,
2303+
failureReasons: _json,
23032304
routingConfiguration: _json,
23042305
updatedAt: (_: any) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
23052306
}) as any;

‎codegen/sdk-codegen/aws-models/bedrock-agent.json

+27-6
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@
5151
}
5252
},
5353
"traits": {
54-
"smithy.api#documentation": "<p>Contains details about the Lambda function containing the business logic that is carried out upon invoking the action.</p>"
54+
"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>"
5555
}
5656
},
5757
"com.amazonaws.bedrockagent#ActionGroupResource": {
@@ -346,7 +346,7 @@
346346
"actionGroupExecutor": {
347347
"target": "com.amazonaws.bedrockagent#ActionGroupExecutor",
348348
"traits": {
349-
"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>"
349+
"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>"
350350
}
351351
},
352352
"apiSchema": {
@@ -449,6 +449,12 @@
449449
"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>",
450450
"smithy.api#required": {}
451451
}
452+
},
453+
"failureReasons": {
454+
"target": "com.amazonaws.bedrockagent#FailureReasons",
455+
"traits": {
456+
"smithy.api#documentation": "<p>Information on the failure of Provisioned Throughput assigned to an agent alias.</p>"
457+
}
452458
}
453459
},
454460
"traits": {
@@ -528,8 +534,13 @@
528534
"agentVersion": {
529535
"target": "com.amazonaws.bedrockagent#Version",
530536
"traits": {
531-
"smithy.api#documentation": "<p>The version of the agent with which the alias is associated.</p>",
532-
"smithy.api#required": {}
537+
"smithy.api#documentation": "<p>The version of the agent with which the alias is associated.</p>"
538+
}
539+
},
540+
"provisionedThroughput": {
541+
"target": "com.amazonaws.bedrockagent#ProvisionedModelIdentifier",
542+
"traits": {
543+
"smithy.api#documentation": "<p>Information on the Provisioned Throughput assigned to an agent alias.</p>"
533544
}
534545
}
535546
},
@@ -2136,7 +2147,7 @@
21362147
"actionGroupExecutor": {
21372148
"target": "com.amazonaws.bedrockagent#ActionGroupExecutor",
21382149
"traits": {
2139-
"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>"
2150+
"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>"
21402151
}
21412152
},
21422153
"apiSchema": {
@@ -6376,6 +6387,16 @@
63766387
}
63776388
}
63786389
},
6390+
"com.amazonaws.bedrockagent#ProvisionedModelIdentifier": {
6391+
"type": "string",
6392+
"traits": {
6393+
"smithy.api#length": {
6394+
"min": 1,
6395+
"max": 2048
6396+
},
6397+
"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}))$"
6398+
}
6399+
},
63796400
"com.amazonaws.bedrockagent#RdsArn": {
63806401
"type": "string",
63816402
"traits": {
@@ -7711,7 +7732,7 @@
77117732
"dataDeletionPolicy": {
77127733
"target": "com.amazonaws.bedrockagent#DataDeletionPolicy",
77137734
"traits": {
7714-
"smithy.api#documentation": "<p>The data deletion policy assigned to the data source.</p>"
7735+
"smithy.api#documentation": "<p>The data deletion policy of the updated data source.</p>"
77157736
}
77167737
},
77177738
"serverSideEncryptionConfiguration": {

0 commit comments

Comments
 (0)