feat(client-cleanrooms): This release introduces support for Custom M…

…odels in AWS Clean Rooms ML.
aws · Nov 7, 2024 · ef8b258 · ef8b258
1 parent 7f72a17
commit ef8b258
Show file tree

Hide file tree

Showing 12 changed files with 753 additions and 200 deletions.
diff --git a/clients/client-cleanrooms/src/commands/CreateCollaborationCommand.ts b/clients/client-cleanrooms/src/commands/CreateCollaborationCommand.ts
@@ -42,11 +42,24 @@ export interface CreateCollaborationCommandOutput extends CreateCollaborationOut
  *       memberAbilities: [ // MemberAbilities // required
  *         "CAN_QUERY" || "CAN_RECEIVE_RESULTS",
  *       ],
+ *       mlMemberAbilities: { // MLMemberAbilities
+ *         customMLMemberAbilities: [ // CustomMLMemberAbilities // required
+ *           "CAN_RECEIVE_MODEL_OUTPUT" || "CAN_RECEIVE_INFERENCE_OUTPUT",
+ *         ],
+ *       },
  *       displayName: "STRING_VALUE", // required
  *       paymentConfiguration: { // PaymentConfiguration
  *         queryCompute: { // QueryComputePaymentConfig
  *           isResponsible: true || false, // required
  *         },
+ *         machineLearning: { // MLPaymentConfig
+ *           modelTraining: { // ModelTrainingPaymentConfig
+ *             isResponsible: true || false, // required
+ *           },
+ *           modelInference: { // ModelInferencePaymentConfig
+ *             isResponsible: true || false, // required
+ *           },
+ *         },
  *       },
  *     },
  *   ],
@@ -55,6 +68,11 @@ export interface CreateCollaborationCommandOutput extends CreateCollaborationOut
  *   creatorMemberAbilities: [ // required
  *     "CAN_QUERY" || "CAN_RECEIVE_RESULTS",
  *   ],
+ *   creatorMLMemberAbilities: {
+ *     customMLMemberAbilities: [ // required
+ *       "CAN_RECEIVE_MODEL_OUTPUT" || "CAN_RECEIVE_INFERENCE_OUTPUT",
+ *     ],
+ *   },
  *   creatorDisplayName: "STRING_VALUE", // required
  *   dataEncryptionMetadata: { // DataEncryptionMetadata
  *     allowCleartext: true || false, // required
@@ -70,6 +88,14 @@ export interface CreateCollaborationCommandOutput extends CreateCollaborationOut
  *     queryCompute: {
  *       isResponsible: true || false, // required
  *     },
+ *     machineLearning: {
+ *       modelTraining: {
+ *         isResponsible: true || false, // required
+ *       },
+ *       modelInference: {
+ *         isResponsible: true || false, // required
+ *       },
+ *     },
  *   },
  *   analyticsEngine: "SPARK" || "CLEAN_ROOMS_SQL",
  * };

diff --git a/clients/client-cleanrooms/src/commands/CreateMembershipCommand.ts b/clients/client-cleanrooms/src/commands/CreateMembershipCommand.ts
@@ -57,6 +57,14 @@ export interface CreateMembershipCommandOutput extends CreateMembershipOutput, _
  *     queryCompute: { // MembershipQueryComputePaymentConfig
  *       isResponsible: true || false, // required
  *     },
+ *     machineLearning: { // MembershipMLPaymentConfig
+ *       modelTraining: { // MembershipModelTrainingPaymentConfig
+ *         isResponsible: true || false, // required
+ *       },
+ *       modelInference: { // MembershipModelInferencePaymentConfig
+ *         isResponsible: true || false, // required
+ *       },
+ *     },
  *   },
  * };
  * const command = new CreateMembershipCommand(input);
@@ -76,6 +84,11 @@ export interface CreateMembershipCommandOutput extends CreateMembershipOutput, _
  * //     memberAbilities: [ // MemberAbilities // required
  * //       "CAN_QUERY" || "CAN_RECEIVE_RESULTS",
  * //     ],
+ * //     mlMemberAbilities: { // MLMemberAbilities
+ * //       customMLMemberAbilities: [ // CustomMLMemberAbilities // required
+ * //         "CAN_RECEIVE_MODEL_OUTPUT" || "CAN_RECEIVE_INFERENCE_OUTPUT",
+ * //       ],
+ * //     },
  * //     queryLogStatus: "STRING_VALUE", // required
  * //     defaultResultConfiguration: { // MembershipProtectedQueryResultConfiguration
  * //       outputConfiguration: { // MembershipProtectedQueryOutputConfiguration Union: only one key present
@@ -92,6 +105,14 @@ export interface CreateMembershipCommandOutput extends CreateMembershipOutput, _
  * //       queryCompute: { // MembershipQueryComputePaymentConfig
  * //         isResponsible: true || false, // required
  * //       },
+ * //       machineLearning: { // MembershipMLPaymentConfig
+ * //         modelTraining: { // MembershipModelTrainingPaymentConfig
+ * //           isResponsible: true || false, // required
+ * //         },
+ * //         modelInference: { // MembershipModelInferencePaymentConfig
+ * //           isResponsible: true || false, // required
+ * //         },
+ * //       },
  * //     },
  * //   },
  * // };

diff --git a/clients/client-cleanrooms/src/commands/GetMembershipCommand.ts b/clients/client-cleanrooms/src/commands/GetMembershipCommand.ts
@@ -55,6 +55,11 @@ export interface GetMembershipCommandOutput extends GetMembershipOutput, __Metad
  * //     memberAbilities: [ // MemberAbilities // required
  * //       "CAN_QUERY" || "CAN_RECEIVE_RESULTS",
  * //     ],
+ * //     mlMemberAbilities: { // MLMemberAbilities
+ * //       customMLMemberAbilities: [ // CustomMLMemberAbilities // required
+ * //         "CAN_RECEIVE_MODEL_OUTPUT" || "CAN_RECEIVE_INFERENCE_OUTPUT",
+ * //       ],
+ * //     },
  * //     queryLogStatus: "STRING_VALUE", // required
  * //     defaultResultConfiguration: { // MembershipProtectedQueryResultConfiguration
  * //       outputConfiguration: { // MembershipProtectedQueryOutputConfiguration Union: only one key present
@@ -71,6 +76,14 @@ export interface GetMembershipCommandOutput extends GetMembershipOutput, __Metad
  * //       queryCompute: { // MembershipQueryComputePaymentConfig
  * //         isResponsible: true || false, // required
  * //       },
+ * //       machineLearning: { // MembershipMLPaymentConfig
+ * //         modelTraining: { // MembershipModelTrainingPaymentConfig
+ * //           isResponsible: true || false, // required
+ * //         },
+ * //         modelInference: { // MembershipModelInferencePaymentConfig
+ * //           isResponsible: true || false, // required
+ * //         },
+ * //       },
  * //     },
  * //   },
  * // };

diff --git a/clients/client-cleanrooms/src/commands/ListMembersCommand.ts b/clients/client-cleanrooms/src/commands/ListMembersCommand.ts
@@ -52,6 +52,11 @@ export interface ListMembersCommandOutput extends ListMembersOutput, __MetadataB
  * //       abilities: [ // MemberAbilities // required
  * //         "CAN_QUERY" || "CAN_RECEIVE_RESULTS",
  * //       ],
+ * //       mlAbilities: { // MLMemberAbilities
+ * //         customMLMemberAbilities: [ // CustomMLMemberAbilities // required
+ * //           "CAN_RECEIVE_MODEL_OUTPUT" || "CAN_RECEIVE_INFERENCE_OUTPUT",
+ * //         ],
+ * //       },
  * //       createTime: new Date("TIMESTAMP"), // required
  * //       updateTime: new Date("TIMESTAMP"), // required
  * //       membershipId: "STRING_VALUE",
@@ -60,6 +65,14 @@ export interface ListMembersCommandOutput extends ListMembersOutput, __MetadataB
  * //         queryCompute: { // QueryComputePaymentConfig
  * //           isResponsible: true || false, // required
  * //         },
+ * //         machineLearning: { // MLPaymentConfig
+ * //           modelTraining: { // ModelTrainingPaymentConfig
+ * //             isResponsible: true || false, // required
+ * //           },
+ * //           modelInference: { // ModelInferencePaymentConfig
+ * //             isResponsible: true || false, // required
+ * //           },
+ * //         },
  * //       },
  * //     },
  * //   ],

diff --git a/clients/client-cleanrooms/src/commands/ListMembershipsCommand.ts b/clients/client-cleanrooms/src/commands/ListMembershipsCommand.ts
@@ -59,10 +59,23 @@ export interface ListMembershipsCommandOutput extends ListMembershipsOutput, __M
  * //       memberAbilities: [ // MemberAbilities // required
  * //         "CAN_QUERY" || "CAN_RECEIVE_RESULTS",
  * //       ],
+ * //       mlMemberAbilities: { // MLMemberAbilities
+ * //         customMLMemberAbilities: [ // CustomMLMemberAbilities // required
+ * //           "CAN_RECEIVE_MODEL_OUTPUT" || "CAN_RECEIVE_INFERENCE_OUTPUT",
+ * //         ],
+ * //       },
  * //       paymentConfiguration: { // MembershipPaymentConfiguration
  * //         queryCompute: { // MembershipQueryComputePaymentConfig
  * //           isResponsible: true || false, // required
  * //         },
+ * //         machineLearning: { // MembershipMLPaymentConfig
+ * //           modelTraining: { // MembershipModelTrainingPaymentConfig
+ * //             isResponsible: true || false, // required
+ * //           },
+ * //           modelInference: { // MembershipModelInferencePaymentConfig
+ * //             isResponsible: true || false, // required
+ * //           },
+ * //         },
  * //       },
  * //     },
  * //   ],

diff --git a/clients/client-cleanrooms/src/commands/PreviewPrivacyImpactCommand.ts b/clients/client-cleanrooms/src/commands/PreviewPrivacyImpactCommand.ts
@@ -6,7 +6,7 @@ import { MetadataBearer as __MetadataBearer } from "@smithy/types";
 
 import { CleanRoomsClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../CleanRoomsClient";
 import { commonParams } from "../endpoint/EndpointParameters";
-import { PreviewPrivacyImpactInput, PreviewPrivacyImpactOutput } from "../models/models_0";
+import { PreviewPrivacyImpactInput, PreviewPrivacyImpactOutput } from "../models/models_1";
 import { de_PreviewPrivacyImpactCommand, se_PreviewPrivacyImpactCommand } from "../protocols/Aws_restJson1";
 
 /**

diff --git a/clients/client-cleanrooms/src/commands/StartProtectedQueryCommand.ts b/clients/client-cleanrooms/src/commands/StartProtectedQueryCommand.ts
@@ -6,8 +6,12 @@ import { MetadataBearer as __MetadataBearer } from "@smithy/types";
 
 import { CleanRoomsClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../CleanRoomsClient";
 import { commonParams } from "../endpoint/EndpointParameters";
-import { StartProtectedQueryInput, StartProtectedQueryInputFilterSensitiveLog } from "../models/models_0";
-import { StartProtectedQueryOutput, StartProtectedQueryOutputFilterSensitiveLog } from "../models/models_1";
+import {
+  StartProtectedQueryInput,
+  StartProtectedQueryInputFilterSensitiveLog,
+  StartProtectedQueryOutput,
+  StartProtectedQueryOutputFilterSensitiveLog,
+} from "../models/models_1";
 import { de_StartProtectedQueryCommand, se_StartProtectedQueryCommand } from "../protocols/Aws_restJson1";
 
 /**

diff --git a/clients/client-cleanrooms/src/commands/UpdateMembershipCommand.ts b/clients/client-cleanrooms/src/commands/UpdateMembershipCommand.ts
@@ -67,6 +67,11 @@ export interface UpdateMembershipCommandOutput extends UpdateMembershipOutput, _
  * //     memberAbilities: [ // MemberAbilities // required
  * //       "CAN_QUERY" || "CAN_RECEIVE_RESULTS",
  * //     ],
+ * //     mlMemberAbilities: { // MLMemberAbilities
+ * //       customMLMemberAbilities: [ // CustomMLMemberAbilities // required
+ * //         "CAN_RECEIVE_MODEL_OUTPUT" || "CAN_RECEIVE_INFERENCE_OUTPUT",
+ * //       ],
+ * //     },
  * //     queryLogStatus: "STRING_VALUE", // required
  * //     defaultResultConfiguration: { // MembershipProtectedQueryResultConfiguration
  * //       outputConfiguration: { // MembershipProtectedQueryOutputConfiguration Union: only one key present
@@ -83,6 +88,14 @@ export interface UpdateMembershipCommandOutput extends UpdateMembershipOutput, _
  * //       queryCompute: { // MembershipQueryComputePaymentConfig
  * //         isResponsible: true || false, // required
  * //       },
+ * //       machineLearning: { // MembershipMLPaymentConfig
+ * //         modelTraining: { // MembershipModelTrainingPaymentConfig
+ * //           isResponsible: true || false, // required
+ * //         },
+ * //         modelInference: { // MembershipModelInferencePaymentConfig
+ * //           isResponsible: true || false, // required
+ * //         },
+ * //       },
  * //     },
  * //   },
  * // };