diff --git a/clients/client-cleanroomsml/src/commands/GetAudienceGenerationJobCommand.ts b/clients/client-cleanroomsml/src/commands/GetAudienceGenerationJobCommand.ts index b53da5e25aa08..54d3e713808c7 100644 --- a/clients/client-cleanroomsml/src/commands/GetAudienceGenerationJobCommand.ts +++ b/clients/client-cleanroomsml/src/commands/GetAudienceGenerationJobCommand.ts @@ -68,6 +68,12 @@ export interface GetAudienceGenerationJobCommandOutput extends GetAudienceGenera * // "": "STRING_VALUE", * // }, * // }, + * // sqlComputeConfiguration: { // ComputeConfiguration Union: only one key present + * // worker: { // WorkerComputeConfiguration + * // type: "CR.1X" || "CR.4X", + * // number: Number("int"), + * // }, + * // }, * // }, * // includeSeedInOutput: true || false, * // collaborationId: "STRING_VALUE", diff --git a/clients/client-cleanroomsml/src/commands/StartAudienceGenerationJobCommand.ts b/clients/client-cleanroomsml/src/commands/StartAudienceGenerationJobCommand.ts index 7d7adb13c1fdb..c9d4d0ed45f67 100644 --- a/clients/client-cleanroomsml/src/commands/StartAudienceGenerationJobCommand.ts +++ b/clients/client-cleanroomsml/src/commands/StartAudienceGenerationJobCommand.ts @@ -54,6 +54,12 @@ export interface StartAudienceGenerationJobCommandOutput extends StartAudienceGe * "": "STRING_VALUE", * }, * }, + * sqlComputeConfiguration: { // ComputeConfiguration Union: only one key present + * worker: { // WorkerComputeConfiguration + * type: "CR.1X" || "CR.4X", + * number: Number("int"), + * }, + * }, * }, * includeSeedInOutput: true || false, * collaborationId: "STRING_VALUE", diff --git a/clients/client-cleanroomsml/src/models/models_0.ts b/clients/client-cleanroomsml/src/models/models_0.ts index 48df6fc9a4672..5ea8ca79a9c21 100644 --- a/clients/client-cleanroomsml/src/models/models_0.ts +++ b/clients/client-cleanroomsml/src/models/models_0.ts @@ -383,6 +383,76 @@ export interface AudienceQualityMetrics { recallMetric?: number | undefined; } +/** + * @public + * @enum + */ +export const WorkerComputeType = { + CR1X: "CR.1X", + CR4X: "CR.4X", +} as const; + +/** + * @public + */ +export type WorkerComputeType = (typeof WorkerComputeType)[keyof typeof WorkerComputeType]; + +/** + *

Configuration information about the compute workers that perform the transform job.

+ * @public + */ +export interface WorkerComputeConfiguration { + /** + *

The instance type of the compute workers that are used.

+ * @public + */ + type?: WorkerComputeType | undefined; + + /** + *

The number of compute workers that are used.

+ * @public + */ + number?: number | undefined; +} + +/** + *

Provides configuration information for the instances that will perform the compute work.

+ * @public + */ +export type ComputeConfiguration = ComputeConfiguration.WorkerMember | ComputeConfiguration.$UnknownMember; + +/** + * @public + */ +export namespace ComputeConfiguration { + /** + *

The worker instances that will perform the compute work.

+ * @public + */ + export interface WorkerMember { + worker: WorkerComputeConfiguration; + $unknown?: never; + } + + /** + * @public + */ + export interface $UnknownMember { + worker?: never; + $unknown: [string, any]; + } + + export interface Visitor { + worker: (value: WorkerComputeConfiguration) => T; + _: (name: string, value: any) => T; + } + + export const visit = (value: ComputeConfiguration, visitor: Visitor): T => { + if (value.worker !== undefined) return visitor.worker(value.worker); + return visitor._(value.$unknown[0], value.$unknown[1]); + }; +} + /** *

The parameters for the SQL type Protected Query.

* @public @@ -439,6 +509,12 @@ export interface AudienceGenerationJobDataSource { * @public */ sqlParameters?: ProtectedQuerySQLParameters | undefined; + + /** + *

Provides configuration information for the instances that will perform the compute work.

+ * @public + */ + sqlComputeConfiguration?: ComputeConfiguration | undefined; } /** @@ -3337,76 +3413,6 @@ export interface PutMLConfigurationRequest { defaultOutputLocation: MLOutputConfiguration | undefined; } -/** - * @public - * @enum - */ -export const WorkerComputeType = { - CR1X: "CR.1X", - CR4X: "CR.4X", -} as const; - -/** - * @public - */ -export type WorkerComputeType = (typeof WorkerComputeType)[keyof typeof WorkerComputeType]; - -/** - *

Configuration information about the compute workers that perform the transform job.

- * @public - */ -export interface WorkerComputeConfiguration { - /** - *

The instance type of the compute workers that are used.

- * @public - */ - type?: WorkerComputeType | undefined; - - /** - *

The number of compute workers that are used.

- * @public - */ - number?: number | undefined; -} - -/** - *

Provides configuration information for the instances that will perform the compute work.

- * @public - */ -export type ComputeConfiguration = ComputeConfiguration.WorkerMember | ComputeConfiguration.$UnknownMember; - -/** - * @public - */ -export namespace ComputeConfiguration { - /** - *

The worker instances that will perform the compute work.

- * @public - */ - export interface WorkerMember { - worker: WorkerComputeConfiguration; - $unknown?: never; - } - - /** - * @public - */ - export interface $UnknownMember { - worker?: never; - $unknown: [string, any]; - } - - export interface Visitor { - worker: (value: WorkerComputeConfiguration) => T; - _: (name: string, value: any) => T; - } - - export const visit = (value: ComputeConfiguration, visitor: Visitor): T => { - if (value.worker !== undefined) return visitor.worker(value.worker); - return visitor._(value.$unknown[0], value.$unknown[1]); - }; -} - /** *

Provides information necessary to perform the protected query.

* @public @@ -5260,7 +5266,7 @@ export interface StartTrainedModelInferenceJobRequest { outputConfiguration: InferenceOutputConfiguration | undefined; /** - *

Defines he data source that is used for the trained model inference job.

+ *

Defines the data source that is used for the trained model inference job.

* @public */ dataSource: ModelInferenceDataSource | undefined; @@ -5727,6 +5733,7 @@ export const ProtectedQuerySQLParametersFilterSensitiveLog = (obj: ProtectedQuer export const AudienceGenerationJobDataSourceFilterSensitiveLog = (obj: AudienceGenerationJobDataSource): any => ({ ...obj, ...(obj.sqlParameters && { sqlParameters: SENSITIVE_STRING }), + ...(obj.sqlComputeConfiguration && { sqlComputeConfiguration: obj.sqlComputeConfiguration }), }); /** diff --git a/codegen/sdk-codegen/aws-models/cleanroomsml.json b/codegen/sdk-codegen/aws-models/cleanroomsml.json index e3594fa63570a..82c84b001ddfd 100644 --- a/codegen/sdk-codegen/aws-models/cleanroomsml.json +++ b/codegen/sdk-codegen/aws-models/cleanroomsml.json @@ -1030,6 +1030,9 @@ "traits": { "smithy.api#documentation": "

The protected SQL query parameters.

" } + }, + "sqlComputeConfiguration": { + "target": "com.amazonaws.cleanroomsml#ComputeConfiguration" } }, "traits": { @@ -9921,7 +9924,7 @@ "dataSource": { "target": "com.amazonaws.cleanroomsml#ModelInferenceDataSource", "traits": { - "smithy.api#documentation": "

Defines he data source that is used for the trained model inference job.

", + "smithy.api#documentation": "

Defines the data source that is used for the trained model inference job.

", "smithy.api#required": {} } },