Skip to content

Commit 44b1172

Browse files
author
awstools
committed
feat(client-bedrock): Tagging support for Async Invoke resources. Added support for Distillation in CreateModelCustomizationJob API. Support for videoDataDeliveryEnabled flag in invocation logging.
1 parent c1cceb3 commit 44b1172

16 files changed

+938
-46
lines changed

‎clients/client-bedrock/src/commands/CreateGuardrailCommand.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ export interface CreateGuardrailCommandOutput extends CreateGuardrailResponse, _
6262
* </ul>
6363
* <p>In addition to the above policies, you can also configure the messages to be returned to
6464
* the user if a user input or model response is in violation of the policies defined in the guardrail.</p>
65-
* <p>For more information, see <a href="https://docs.aws.amazon.com/bedrock/latest/userguide/guardrails.html">Guardrails for Amazon Bedrock</a> in
65+
* <p>For more information, see <a href="https://docs.aws.amazon.com/bedrock/latest/userguide/guardrails.html">Amazon Bedrock Guardrails</a> in
6666
* the <i>Amazon Bedrock User Guide</i>.</p>
6767
* @example
6868
* Use a bare-bones client and the command you need to make an API call.

‎clients/client-bedrock/src/commands/CreateModelCustomizationJobCommand.ts

Lines changed: 49 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,11 @@ import { MetadataBearer as __MetadataBearer } from "@smithy/types";
66

77
import { BedrockClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../BedrockClient";
88
import { commonParams } from "../endpoint/EndpointParameters";
9-
import { CreateModelCustomizationJobRequest, CreateModelCustomizationJobResponse } from "../models/models_0";
9+
import {
10+
CreateModelCustomizationJobRequest,
11+
CreateModelCustomizationJobRequestFilterSensitiveLog,
12+
CreateModelCustomizationJobResponse,
13+
} from "../models/models_0";
1014
import {
1115
de_CreateModelCustomizationJobCommand,
1216
se_CreateModelCustomizationJobCommand,
@@ -54,7 +58,7 @@ export interface CreateModelCustomizationJobCommandOutput
5458
* roleArn: "STRING_VALUE", // required
5559
* clientRequestToken: "STRING_VALUE",
5660
* baseModelIdentifier: "STRING_VALUE", // required
57-
* customizationType: "FINE_TUNING" || "CONTINUED_PRE_TRAINING",
61+
* customizationType: "FINE_TUNING" || "CONTINUED_PRE_TRAINING" || "DISTILLATION",
5862
* customModelKmsKeyId: "STRING_VALUE",
5963
* jobTags: [ // TagList
6064
* { // Tag
@@ -69,7 +73,39 @@ export interface CreateModelCustomizationJobCommandOutput
6973
* },
7074
* ],
7175
* trainingDataConfig: { // TrainingDataConfig
72-
* s3Uri: "STRING_VALUE", // required
76+
* s3Uri: "STRING_VALUE",
77+
* invocationLogsConfig: { // InvocationLogsConfig
78+
* usePromptResponse: true || false,
79+
* invocationLogSource: { // InvocationLogSource Union: only one key present
80+
* s3Uri: "STRING_VALUE",
81+
* },
82+
* requestMetadataFilters: { // RequestMetadataFilters Union: only one key present
83+
* equals: { // RequestMetadataMap
84+
* "<keys>": "STRING_VALUE",
85+
* },
86+
* notEquals: {
87+
* "<keys>": "STRING_VALUE",
88+
* },
89+
* andAll: [ // RequestMetadataFiltersList
90+
* { // RequestMetadataBaseFilters
91+
* equals: {
92+
* "<keys>": "STRING_VALUE",
93+
* },
94+
* notEquals: {
95+
* "<keys>": "STRING_VALUE",
96+
* },
97+
* },
98+
* ],
99+
* orAll: [
100+
* {
101+
* equals: {
102+
* "<keys>": "STRING_VALUE",
103+
* },
104+
* notEquals: "<RequestMetadataMap>",
105+
* },
106+
* ],
107+
* },
108+
* },
73109
* },
74110
* validationDataConfig: { // ValidationDataConfig
75111
* validators: [ // Validators // required
@@ -81,7 +117,7 @@ export interface CreateModelCustomizationJobCommandOutput
81117
* outputDataConfig: { // OutputDataConfig
82118
* s3Uri: "STRING_VALUE", // required
83119
* },
84-
* hyperParameters: { // ModelCustomizationHyperParameters // required
120+
* hyperParameters: { // ModelCustomizationHyperParameters
85121
* "<keys>": "STRING_VALUE",
86122
* },
87123
* vpcConfig: { // VpcConfig
@@ -92,6 +128,14 @@ export interface CreateModelCustomizationJobCommandOutput
92128
* "STRING_VALUE",
93129
* ],
94130
* },
131+
* customizationConfig: { // CustomizationConfig Union: only one key present
132+
* distillationConfig: { // DistillationConfig
133+
* teacherModelConfig: { // TeacherModelConfig
134+
* teacherModelIdentifier: "STRING_VALUE", // required
135+
* maxResponseLengthForInference: Number("int"),
136+
* },
137+
* },
138+
* },
95139
* };
96140
* const command = new CreateModelCustomizationJobCommand(input);
97141
* const response = await client.send(command);
@@ -154,7 +198,7 @@ export class CreateModelCustomizationJobCommand extends $Command
154198
})
155199
.s("AmazonBedrockControlPlaneService", "CreateModelCustomizationJob", {})
156200
.n("BedrockClient", "CreateModelCustomizationJobCommand")
157-
.f(void 0, void 0)
201+
.f(CreateModelCustomizationJobRequestFilterSensitiveLog, void 0)
158202
.ser(se_CreateModelCustomizationJobCommand)
159203
.de(de_CreateModelCustomizationJobCommand)
160204
.build() {

‎clients/client-bedrock/src/commands/GetCustomModelCommand.ts

Lines changed: 48 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,11 @@ import { MetadataBearer as __MetadataBearer } from "@smithy/types";
66

77
import { BedrockClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../BedrockClient";
88
import { commonParams } from "../endpoint/EndpointParameters";
9-
import { GetCustomModelRequest, GetCustomModelResponse } from "../models/models_0";
9+
import {
10+
GetCustomModelRequest,
11+
GetCustomModelResponse,
12+
GetCustomModelResponseFilterSensitiveLog,
13+
} from "../models/models_0";
1014
import { de_GetCustomModelCommand, se_GetCustomModelCommand } from "../protocols/Aws_restJson1";
1115

1216
/**
@@ -46,13 +50,45 @@ export interface GetCustomModelCommandOutput extends GetCustomModelResponse, __M
4650
* // jobName: "STRING_VALUE",
4751
* // jobArn: "STRING_VALUE", // required
4852
* // baseModelArn: "STRING_VALUE", // required
49-
* // customizationType: "FINE_TUNING" || "CONTINUED_PRE_TRAINING",
53+
* // customizationType: "FINE_TUNING" || "CONTINUED_PRE_TRAINING" || "DISTILLATION",
5054
* // modelKmsKeyArn: "STRING_VALUE",
5155
* // hyperParameters: { // ModelCustomizationHyperParameters
5256
* // "<keys>": "STRING_VALUE",
5357
* // },
5458
* // trainingDataConfig: { // TrainingDataConfig
55-
* // s3Uri: "STRING_VALUE", // required
59+
* // s3Uri: "STRING_VALUE",
60+
* // invocationLogsConfig: { // InvocationLogsConfig
61+
* // usePromptResponse: true || false,
62+
* // invocationLogSource: { // InvocationLogSource Union: only one key present
63+
* // s3Uri: "STRING_VALUE",
64+
* // },
65+
* // requestMetadataFilters: { // RequestMetadataFilters Union: only one key present
66+
* // equals: { // RequestMetadataMap
67+
* // "<keys>": "STRING_VALUE",
68+
* // },
69+
* // notEquals: {
70+
* // "<keys>": "STRING_VALUE",
71+
* // },
72+
* // andAll: [ // RequestMetadataFiltersList
73+
* // { // RequestMetadataBaseFilters
74+
* // equals: {
75+
* // "<keys>": "STRING_VALUE",
76+
* // },
77+
* // notEquals: {
78+
* // "<keys>": "STRING_VALUE",
79+
* // },
80+
* // },
81+
* // ],
82+
* // orAll: [
83+
* // {
84+
* // equals: {
85+
* // "<keys>": "STRING_VALUE",
86+
* // },
87+
* // notEquals: "<RequestMetadataMap>",
88+
* // },
89+
* // ],
90+
* // },
91+
* // },
5692
* // },
5793
* // validationDataConfig: { // ValidationDataConfig
5894
* // validators: [ // Validators // required
@@ -73,6 +109,14 @@ export interface GetCustomModelCommandOutput extends GetCustomModelResponse, __M
73109
* // },
74110
* // ],
75111
* // creationTime: new Date("TIMESTAMP"), // required
112+
* // customizationConfig: { // CustomizationConfig Union: only one key present
113+
* // distillationConfig: { // DistillationConfig
114+
* // teacherModelConfig: { // TeacherModelConfig
115+
* // teacherModelIdentifier: "STRING_VALUE", // required
116+
* // maxResponseLengthForInference: Number("int"),
117+
* // },
118+
* // },
119+
* // },
76120
* // };
77121
*
78122
* ```
@@ -120,7 +164,7 @@ export class GetCustomModelCommand extends $Command
120164
})
121165
.s("AmazonBedrockControlPlaneService", "GetCustomModel", {})
122166
.n("BedrockClient", "GetCustomModelCommand")
123-
.f(void 0, void 0)
167+
.f(void 0, GetCustomModelResponseFilterSensitiveLog)
124168
.ser(se_GetCustomModelCommand)
125169
.de(de_GetCustomModelCommand)
126170
.build() {

‎clients/client-bedrock/src/commands/GetFoundationModelCommand.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ export interface GetFoundationModelCommandOutput extends GetFoundationModelRespo
5454
* // ],
5555
* // responseStreamingSupported: true || false,
5656
* // customizationsSupported: [ // ModelCustomizationList
57-
* // "FINE_TUNING" || "CONTINUED_PRE_TRAINING",
57+
* // "FINE_TUNING" || "CONTINUED_PRE_TRAINING" || "DISTILLATION",
5858
* // ],
5959
* // inferenceTypesSupported: [ // InferenceTypeList
6060
* // "ON_DEMAND" || "PROVISIONED",

‎clients/client-bedrock/src/commands/GetModelCustomizationJobCommand.ts

Lines changed: 49 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,11 @@ import { MetadataBearer as __MetadataBearer } from "@smithy/types";
66

77
import { BedrockClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../BedrockClient";
88
import { commonParams } from "../endpoint/EndpointParameters";
9-
import { GetModelCustomizationJobRequest, GetModelCustomizationJobResponse } from "../models/models_0";
9+
import {
10+
GetModelCustomizationJobRequest,
11+
GetModelCustomizationJobResponse,
12+
GetModelCustomizationJobResponseFilterSensitiveLog,
13+
} from "../models/models_0";
1014
import { de_GetModelCustomizationJobCommand, se_GetModelCustomizationJobCommand } from "../protocols/Aws_restJson1";
1115

1216
/**
@@ -54,11 +58,43 @@ export interface GetModelCustomizationJobCommandOutput extends GetModelCustomiza
5458
* // lastModifiedTime: new Date("TIMESTAMP"),
5559
* // endTime: new Date("TIMESTAMP"),
5660
* // baseModelArn: "STRING_VALUE", // required
57-
* // hyperParameters: { // ModelCustomizationHyperParameters // required
61+
* // hyperParameters: { // ModelCustomizationHyperParameters
5862
* // "<keys>": "STRING_VALUE",
5963
* // },
6064
* // trainingDataConfig: { // TrainingDataConfig
61-
* // s3Uri: "STRING_VALUE", // required
65+
* // s3Uri: "STRING_VALUE",
66+
* // invocationLogsConfig: { // InvocationLogsConfig
67+
* // usePromptResponse: true || false,
68+
* // invocationLogSource: { // InvocationLogSource Union: only one key present
69+
* // s3Uri: "STRING_VALUE",
70+
* // },
71+
* // requestMetadataFilters: { // RequestMetadataFilters Union: only one key present
72+
* // equals: { // RequestMetadataMap
73+
* // "<keys>": "STRING_VALUE",
74+
* // },
75+
* // notEquals: {
76+
* // "<keys>": "STRING_VALUE",
77+
* // },
78+
* // andAll: [ // RequestMetadataFiltersList
79+
* // { // RequestMetadataBaseFilters
80+
* // equals: {
81+
* // "<keys>": "STRING_VALUE",
82+
* // },
83+
* // notEquals: {
84+
* // "<keys>": "STRING_VALUE",
85+
* // },
86+
* // },
87+
* // ],
88+
* // orAll: [
89+
* // {
90+
* // equals: {
91+
* // "<keys>": "STRING_VALUE",
92+
* // },
93+
* // notEquals: "<RequestMetadataMap>",
94+
* // },
95+
* // ],
96+
* // },
97+
* // },
6298
* // },
6399
* // validationDataConfig: { // ValidationDataConfig
64100
* // validators: [ // Validators // required
@@ -70,7 +106,7 @@ export interface GetModelCustomizationJobCommandOutput extends GetModelCustomiza
70106
* // outputDataConfig: { // OutputDataConfig
71107
* // s3Uri: "STRING_VALUE", // required
72108
* // },
73-
* // customizationType: "FINE_TUNING" || "CONTINUED_PRE_TRAINING",
109+
* // customizationType: "FINE_TUNING" || "CONTINUED_PRE_TRAINING" || "DISTILLATION",
74110
* // outputModelKmsKeyArn: "STRING_VALUE",
75111
* // trainingMetrics: { // TrainingMetrics
76112
* // trainingLoss: Number("float"),
@@ -88,6 +124,14 @@ export interface GetModelCustomizationJobCommandOutput extends GetModelCustomiza
88124
* // "STRING_VALUE",
89125
* // ],
90126
* // },
127+
* // customizationConfig: { // CustomizationConfig Union: only one key present
128+
* // distillationConfig: { // DistillationConfig
129+
* // teacherModelConfig: { // TeacherModelConfig
130+
* // teacherModelIdentifier: "STRING_VALUE", // required
131+
* // maxResponseLengthForInference: Number("int"),
132+
* // },
133+
* // },
134+
* // },
91135
* // };
92136
*
93137
* ```
@@ -135,7 +179,7 @@ export class GetModelCustomizationJobCommand extends $Command
135179
})
136180
.s("AmazonBedrockControlPlaneService", "GetModelCustomizationJob", {})
137181
.n("BedrockClient", "GetModelCustomizationJobCommand")
138-
.f(void 0, void 0)
182+
.f(void 0, GetModelCustomizationJobResponseFilterSensitiveLog)
139183
.ser(se_GetModelCustomizationJobCommand)
140184
.de(de_GetModelCustomizationJobCommand)
141185
.build() {

‎clients/client-bedrock/src/commands/GetModelInvocationJobCommand.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ export interface GetModelInvocationJobCommandInput extends GetModelInvocationJob
3232
export interface GetModelInvocationJobCommandOutput extends GetModelInvocationJobResponse, __MetadataBearer {}
3333

3434
/**
35-
* <p>Gets details about a batch inference job. For more information, see <a href="https://docs.aws.amazon.com/bedrock/latest/userguide/batch-inference-manage.html#batch-inference-view">View details about a batch inference job</a>
35+
* <p>Gets details about a batch inference job. For more information, see <a href="https://docs.aws.amazon.com/bedrock/latest/userguide/batch-inference-monitor">Monitor batch inference jobs</a>
3636
* </p>
3737
* @example
3838
* Use a bare-bones client and the command you need to make an API call.

‎clients/client-bedrock/src/commands/GetModelInvocationLoggingConfigurationCommand.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -64,6 +64,7 @@ export interface GetModelInvocationLoggingConfigurationCommandOutput
6464
* // textDataDeliveryEnabled: true || false,
6565
* // imageDataDeliveryEnabled: true || false,
6666
* // embeddingDataDeliveryEnabled: true || false,
67+
* // videoDataDeliveryEnabled: true || false,
6768
* // },
6869
* // };
6970
*

‎clients/client-bedrock/src/commands/ListCustomModelsCommand.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ export interface ListCustomModelsCommandOutput extends ListCustomModelsResponse,
5959
* // creationTime: new Date("TIMESTAMP"), // required
6060
* // baseModelArn: "STRING_VALUE", // required
6161
* // baseModelName: "STRING_VALUE", // required
62-
* // customizationType: "FINE_TUNING" || "CONTINUED_PRE_TRAINING",
62+
* // customizationType: "FINE_TUNING" || "CONTINUED_PRE_TRAINING" || "DISTILLATION",
6363
* // ownerAccountId: "STRING_VALUE",
6464
* // },
6565
* // ],

‎clients/client-bedrock/src/commands/ListFoundationModelsCommand.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ export interface ListFoundationModelsCommandOutput extends ListFoundationModelsR
3737
* const client = new BedrockClient(config);
3838
* const input = { // ListFoundationModelsRequest
3939
* byProvider: "STRING_VALUE",
40-
* byCustomizationType: "FINE_TUNING" || "CONTINUED_PRE_TRAINING",
40+
* byCustomizationType: "FINE_TUNING" || "CONTINUED_PRE_TRAINING" || "DISTILLATION",
4141
* byOutputModality: "TEXT" || "IMAGE" || "EMBEDDING",
4242
* byInferenceType: "ON_DEMAND" || "PROVISIONED",
4343
* };
@@ -58,7 +58,7 @@ export interface ListFoundationModelsCommandOutput extends ListFoundationModelsR
5858
* // ],
5959
* // responseStreamingSupported: true || false,
6060
* // customizationsSupported: [ // ModelCustomizationList
61-
* // "FINE_TUNING" || "CONTINUED_PRE_TRAINING",
61+
* // "FINE_TUNING" || "CONTINUED_PRE_TRAINING" || "DISTILLATION",
6262
* // ],
6363
* // inferenceTypesSupported: [ // InferenceTypeList
6464
* // "ON_DEMAND" || "PROVISIONED",

‎clients/client-bedrock/src/commands/ListModelCustomizationJobsCommand.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ export interface ListModelCustomizationJobsCommandOutput extends ListModelCustom
6262
* // endTime: new Date("TIMESTAMP"),
6363
* // customModelArn: "STRING_VALUE",
6464
* // customModelName: "STRING_VALUE",
65-
* // customizationType: "FINE_TUNING" || "CONTINUED_PRE_TRAINING",
65+
* // customizationType: "FINE_TUNING" || "CONTINUED_PRE_TRAINING" || "DISTILLATION",
6666
* // },
6767
* // ],
6868
* // };

‎clients/client-bedrock/src/commands/ListModelInvocationJobsCommand.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ export interface ListModelInvocationJobsCommandInput extends ListModelInvocation
3232
export interface ListModelInvocationJobsCommandOutput extends ListModelInvocationJobsResponse, __MetadataBearer {}
3333

3434
/**
35-
* <p>Lists all batch inference jobs in the account. For more information, see <a href="https://docs.aws.amazon.com/bedrock/latest/userguide/batch-inference-manage.html#batch-inference-view">View details about a batch inference job</a>.</p>
35+
* <p>Lists all batch inference jobs in the account. For more information, see <a href="https://docs.aws.amazon.com/bedrock/latest/userguide/batch-inference-view.html">View details about a batch inference job</a>.</p>
3636
* @example
3737
* Use a bare-bones client and the command you need to make an API call.
3838
* ```javascript

‎clients/client-bedrock/src/commands/PutModelInvocationLoggingConfigurationCommand.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,7 @@ export interface PutModelInvocationLoggingConfigurationCommandOutput
6161
* textDataDeliveryEnabled: true || false,
6262
* imageDataDeliveryEnabled: true || false,
6363
* embeddingDataDeliveryEnabled: true || false,
64+
* videoDataDeliveryEnabled: true || false,
6465
* },
6566
* };
6667
* const command = new PutModelInvocationLoggingConfigurationCommand(input);

‎clients/client-bedrock/src/commands/StopModelInvocationJobCommand.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ export interface StopModelInvocationJobCommandInput extends StopModelInvocationJ
2828
export interface StopModelInvocationJobCommandOutput extends StopModelInvocationJobResponse, __MetadataBearer {}
2929

3030
/**
31-
* <p>Stops a batch inference job. You're only charged for tokens that were already processed. For more information, see <a href="https://docs.aws.amazon.com/bedrock/latest/userguide/batch-inference-manage.html#batch-inference-stop">Stop a batch inference job</a>.</p>
31+
* <p>Stops a batch inference job. You're only charged for tokens that were already processed. For more information, see <a href="https://docs.aws.amazon.com/bedrock/latest/userguide/batch-inference-stop.html">Stop a batch inference job</a>.</p>
3232
* @example
3333
* Use a bare-bones client and the command you need to make an API call.
3434
* ```javascript

0 commit comments

Comments
 (0)