feat(client-sagemaker): Amazon Sagemaker Autopilot now supports Text …

…Generation jobs.
aws · Oct 26, 2023 · fd8ca90 · fd8ca90
1 parent 0bc9a39
commit fd8ca90
Show file tree

Hide file tree

Showing 14 changed files with 851 additions and 551 deletions.
diff --git a/clients/client-sagemaker/src/commands/CreateAutoMLJobCommand.ts b/clients/client-sagemaker/src/commands/CreateAutoMLJobCommand.ts
@@ -43,7 +43,8 @@ export interface CreateAutoMLJobCommandOutput extends CreateAutoMLJobResponse, _
  *             <p>
  *                <code>CreateAutoMLJobV2</code> can manage tabular problem types identical to those of
  *             its previous version <code>CreateAutoMLJob</code>, as well as time-series forecasting,
- *             and non-tabular problem types such as image or text classification.</p>
+ *             non-tabular problem types such as image or text classification, and text generation
+ *             (LLMs fine-tuning).</p>
  *             <p>Find guidelines about how to migrate a <code>CreateAutoMLJob</code> to
  *                <code>CreateAutoMLJobV2</code> in <a href="https://docs.aws.amazon.com/sagemaker/latest/dg/autopilot-automate-model-development-create-experiment-api.html#autopilot-create-experiment-api-migrate-v1-v2">Migrate a CreateAutoMLJob to CreateAutoMLJobV2</a>.</p>
  *          </note>

diff --git a/clients/client-sagemaker/src/commands/CreateAutoMLJobV2Command.ts b/clients/client-sagemaker/src/commands/CreateAutoMLJobV2Command.ts
@@ -45,7 +45,8 @@ export interface CreateAutoMLJobV2CommandOutput extends CreateAutoMLJobV2Respons
  *             <p>
  *                <code>CreateAutoMLJobV2</code> can manage tabular problem types identical to those of
  *             its previous version <code>CreateAutoMLJob</code>, as well as time-series forecasting,
- *             and non-tabular problem types such as image or text classification.</p>
+ *             non-tabular problem types such as image or text classification, and text generation
+ *             (LLMs fine-tuning).</p>
  *             <p>Find guidelines about how to migrate a <code>CreateAutoMLJob</code> to
  *                <code>CreateAutoMLJobV2</code> in <a href="https://docs.aws.amazon.com/sagemaker/latest/dg/autopilot-automate-model-development-create-experiment-api.html#autopilot-create-experiment-api-migrate-v1-v2">Migrate a CreateAutoMLJob to CreateAutoMLJobV2</a>.</p>
  *          </note>
@@ -152,6 +153,14 @@ export interface CreateAutoMLJobV2CommandOutput extends CreateAutoMLJobV2Respons
  *         },
  *       ],
  *     },
+ *     TextGenerationJobConfig: { // TextGenerationJobConfig
+ *       CompletionCriteria: {
+ *         MaxCandidates: Number("int"),
+ *         MaxRuntimePerTrainingJobInSeconds: Number("int"),
+ *         MaxAutoMLJobRuntimeInSeconds: Number("int"),
+ *       },
+ *       BaseModelName: "STRING_VALUE",
+ *     },
  *   },
  *   RoleArn: "STRING_VALUE", // required
  *   Tags: [ // TagList

diff --git a/clients/client-sagemaker/src/commands/CreateDataQualityJobDefinitionCommand.ts b/clients/client-sagemaker/src/commands/CreateDataQualityJobDefinitionCommand.ts
@@ -14,8 +14,7 @@ import {
   SMITHY_CONTEXT_KEY,
 } from "@smithy/types";
 
-import { CreateDataQualityJobDefinitionRequest } from "../models/models_0";
-import { CreateDataQualityJobDefinitionResponse } from "../models/models_1";
+import { CreateDataQualityJobDefinitionRequest, CreateDataQualityJobDefinitionResponse } from "../models/models_1";
 import {
   de_CreateDataQualityJobDefinitionCommand,
   se_CreateDataQualityJobDefinitionCommand,

diff --git a/clients/client-sagemaker/src/commands/DeleteContextCommand.ts b/clients/client-sagemaker/src/commands/DeleteContextCommand.ts
@@ -14,7 +14,7 @@ import {
   SMITHY_CONTEXT_KEY,
 } from "@smithy/types";
 
-import { DeleteContextRequest, DeleteContextResponse } from "../models/models_1";
+import { DeleteContextRequest, DeleteContextResponse } from "../models/models_2";
 import { de_DeleteContextCommand, se_DeleteContextCommand } from "../protocols/Aws_json1_1";
 import { SageMakerClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../SageMakerClient";
 

diff --git a/clients/client-sagemaker/src/commands/DescribeAutoMLJobCommand.ts b/clients/client-sagemaker/src/commands/DescribeAutoMLJobCommand.ts
@@ -163,7 +163,7 @@ export interface DescribeAutoMLJobCommandOutput extends DescribeAutoMLJobRespons
  * //           MetricName: "Accuracy" || "MSE" || "F1" || "F1macro" || "AUC" || "RMSE" || "MAE" || "R2" || "BalancedAccuracy" || "Precision" || "PrecisionMacro" || "Recall" || "RecallMacro" || "MAPE" || "MASE" || "WAPE" || "AverageWeightedQuantileLoss",
  * //           Value: Number("float"),
  * //           Set: "Train" || "Validation" || "Test",
- * //           StandardMetricName: "Accuracy" || "MSE" || "F1" || "F1macro" || "AUC" || "RMSE" || "MAE" || "R2" || "BalancedAccuracy" || "Precision" || "PrecisionMacro" || "Recall" || "RecallMacro" || "LogLoss" || "InferenceLatency" || "MAPE" || "MASE" || "WAPE" || "AverageWeightedQuantileLoss",
+ * //           StandardMetricName: "Accuracy" || "MSE" || "F1" || "F1macro" || "AUC" || "RMSE" || "MAE" || "R2" || "BalancedAccuracy" || "Precision" || "PrecisionMacro" || "Recall" || "RecallMacro" || "LogLoss" || "InferenceLatency" || "MAPE" || "MASE" || "WAPE" || "AverageWeightedQuantileLoss" || "Rouge1" || "Rouge2" || "RougeL" || "RougeLSum" || "Perplexity" || "ValidationLoss" || "TrainingLoss",
  * //         },
  * //       ],
  * //     },

diff --git a/clients/client-sagemaker/src/commands/DescribeAutoMLJobV2Command.ts b/clients/client-sagemaker/src/commands/DescribeAutoMLJobV2Command.ts
@@ -149,6 +149,14 @@ export interface DescribeAutoMLJobV2CommandOutput extends DescribeAutoMLJobV2Res
  * //         },
  * //       ],
  * //     },
+ * //     TextGenerationJobConfig: { // TextGenerationJobConfig
+ * //       CompletionCriteria: {
+ * //         MaxCandidates: Number("int"),
+ * //         MaxRuntimePerTrainingJobInSeconds: Number("int"),
+ * //         MaxAutoMLJobRuntimeInSeconds: Number("int"),
+ * //       },
+ * //       BaseModelName: "STRING_VALUE",
+ * //     },
  * //   },
  * //   CreationTime: new Date("TIMESTAMP"), // required
  * //   EndTime: new Date("TIMESTAMP"),
@@ -200,7 +208,7 @@ export interface DescribeAutoMLJobV2CommandOutput extends DescribeAutoMLJobV2Res
  * //           MetricName: "Accuracy" || "MSE" || "F1" || "F1macro" || "AUC" || "RMSE" || "MAE" || "R2" || "BalancedAccuracy" || "Precision" || "PrecisionMacro" || "Recall" || "RecallMacro" || "MAPE" || "MASE" || "WAPE" || "AverageWeightedQuantileLoss",
  * //           Value: Number("float"),
  * //           Set: "Train" || "Validation" || "Test",
- * //           StandardMetricName: "Accuracy" || "MSE" || "F1" || "F1macro" || "AUC" || "RMSE" || "MAE" || "R2" || "BalancedAccuracy" || "Precision" || "PrecisionMacro" || "Recall" || "RecallMacro" || "LogLoss" || "InferenceLatency" || "MAPE" || "MASE" || "WAPE" || "AverageWeightedQuantileLoss",
+ * //           StandardMetricName: "Accuracy" || "MSE" || "F1" || "F1macro" || "AUC" || "RMSE" || "MAE" || "R2" || "BalancedAccuracy" || "Precision" || "PrecisionMacro" || "Recall" || "RecallMacro" || "LogLoss" || "InferenceLatency" || "MAPE" || "MASE" || "WAPE" || "AverageWeightedQuantileLoss" || "Rouge1" || "Rouge2" || "RougeL" || "RougeLSum" || "Perplexity" || "ValidationLoss" || "TrainingLoss",
  * //         },
  * //       ],
  * //     },
@@ -248,18 +256,17 @@ export interface DescribeAutoMLJobV2CommandOutput extends DescribeAutoMLJobV2Res
  * //     AutoMLJobObjective: {
  * //       MetricName: "Accuracy" || "MSE" || "F1" || "F1macro" || "AUC" || "RMSE" || "MAE" || "R2" || "BalancedAccuracy" || "Precision" || "PrecisionMacro" || "Recall" || "RecallMacro" || "MAPE" || "MASE" || "WAPE" || "AverageWeightedQuantileLoss", // required
  * //     },
- * //     CompletionCriteria: {
- * //       MaxCandidates: Number("int"),
- * //       MaxRuntimePerTrainingJobInSeconds: Number("int"),
- * //       MaxAutoMLJobRuntimeInSeconds: Number("int"),
- * //     },
+ * //     CompletionCriteria: "<AutoMLJobCompletionCriteria>",
  * //     AutoMLProblemTypeResolvedAttributes: { // AutoMLProblemTypeResolvedAttributes Union: only one key present
  * //       TabularResolvedAttributes: { // TabularResolvedAttributes
  * //         ProblemType: "BinaryClassification" || "MulticlassClassification" || "Regression",
  * //       },
+ * //       TextGenerationResolvedAttributes: { // TextGenerationResolvedAttributes
+ * //         BaseModelName: "STRING_VALUE",
+ * //       },
  * //     },
  * //   },
- * //   AutoMLProblemTypeConfigName: "ImageClassification" || "TextClassification" || "Tabular" || "TimeSeriesForecasting",
+ * //   AutoMLProblemTypeConfigName: "ImageClassification" || "TextClassification" || "Tabular" || "TimeSeriesForecasting" || "TextGeneration",
  * // };
  *
  * ```

diff --git a/clients/client-sagemaker/src/commands/ListCandidatesForAutoMLJobCommand.ts b/clients/client-sagemaker/src/commands/ListCandidatesForAutoMLJobCommand.ts
@@ -98,7 +98,7 @@ export interface ListCandidatesForAutoMLJobCommandOutput extends ListCandidatesF
  * //             MetricName: "Accuracy" || "MSE" || "F1" || "F1macro" || "AUC" || "RMSE" || "MAE" || "R2" || "BalancedAccuracy" || "Precision" || "PrecisionMacro" || "Recall" || "RecallMacro" || "MAPE" || "MASE" || "WAPE" || "AverageWeightedQuantileLoss",
  * //             Value: Number("float"),
  * //             Set: "Train" || "Validation" || "Test",
- * //             StandardMetricName: "Accuracy" || "MSE" || "F1" || "F1macro" || "AUC" || "RMSE" || "MAE" || "R2" || "BalancedAccuracy" || "Precision" || "PrecisionMacro" || "Recall" || "RecallMacro" || "LogLoss" || "InferenceLatency" || "MAPE" || "MASE" || "WAPE" || "AverageWeightedQuantileLoss",
+ * //             StandardMetricName: "Accuracy" || "MSE" || "F1" || "F1macro" || "AUC" || "RMSE" || "MAE" || "R2" || "BalancedAccuracy" || "Precision" || "PrecisionMacro" || "Recall" || "RecallMacro" || "LogLoss" || "InferenceLatency" || "MAPE" || "MASE" || "WAPE" || "AverageWeightedQuantileLoss" || "Rouge1" || "Rouge2" || "RougeL" || "RougeLSum" || "Perplexity" || "ValidationLoss" || "TrainingLoss",
  * //           },
  * //         ],
  * //       },