feat(client-sagemaker): This release 1/ enables optimization jobs tha…

…t allows customers to perform Ahead-of-time compilation and quantization. 2/ allows customers to control access to Amazon Q integration in SageMaker Studio. 3/ enables AdditionalModelDataSources for CreateModel action.
aws · Jul 9, 2024 · ec13654 · ec13654
1 parent cd5cd96
commit ec13654
Show file tree

Hide file tree

Showing 44 changed files with 14,460 additions and 11,083 deletions.
diff --git a/clients/client-sagemaker/README.md b/clients/client-sagemaker/README.md
@@ -601,6 +601,14 @@ CreateNotebookInstanceLifecycleConfig
 
 [Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/sagemaker/command/CreateNotebookInstanceLifecycleConfigCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/CreateNotebookInstanceLifecycleConfigCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/CreateNotebookInstanceLifecycleConfigCommandOutput/)
 
+</details>
+<details>
+<summary>
+CreateOptimizationJob
+</summary>
+
+[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/sagemaker/command/CreateOptimizationJobCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/CreateOptimizationJobCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/CreateOptimizationJobCommandOutput/)
+
 </details>
 <details>
 <summary>
@@ -1049,6 +1057,14 @@ DeleteNotebookInstanceLifecycleConfig
 
 [Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/sagemaker/command/DeleteNotebookInstanceLifecycleConfigCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/DeleteNotebookInstanceLifecycleConfigCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/DeleteNotebookInstanceLifecycleConfigCommandOutput/)
 
+</details>
+<details>
+<summary>
+DeleteOptimizationJob
+</summary>
+
+[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/sagemaker/command/DeleteOptimizationJobCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/DeleteOptimizationJobCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/DeleteOptimizationJobCommandOutput/)
+
 </details>
 <details>
 <summary>
@@ -1513,6 +1529,14 @@ DescribeNotebookInstanceLifecycleConfig
 
 [Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/sagemaker/command/DescribeNotebookInstanceLifecycleConfigCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/DescribeNotebookInstanceLifecycleConfigCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/DescribeNotebookInstanceLifecycleConfigCommandOutput/)
 
+</details>
+<details>
+<summary>
+DescribeOptimizationJob
+</summary>
+
+[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/sagemaker/command/DescribeOptimizationJobCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/DescribeOptimizationJobCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/DescribeOptimizationJobCommandOutput/)
+
 </details>
 <details>
 <summary>
@@ -2161,6 +2185,14 @@ ListNotebookInstances
 
 [Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/sagemaker/command/ListNotebookInstancesCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/ListNotebookInstancesCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/ListNotebookInstancesCommandOutput/)
 
+</details>
+<details>
+<summary>
+ListOptimizationJobs
+</summary>
+
+[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/sagemaker/command/ListOptimizationJobsCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/ListOptimizationJobsCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/ListOptimizationJobsCommandOutput/)
+
 </details>
 <details>
 <summary>
@@ -2521,6 +2553,14 @@ StopNotebookInstance
 
 [Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/sagemaker/command/StopNotebookInstanceCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/StopNotebookInstanceCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/StopNotebookInstanceCommandOutput/)
 
+</details>
+<details>
+<summary>
+StopOptimizationJob
+</summary>
+
+[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/sagemaker/command/StopOptimizationJobCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/StopOptimizationJobCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/StopOptimizationJobCommandOutput/)
+
 </details>
 <details>
 <summary>

diff --git a/clients/client-sagemaker/src/SageMaker.ts b/clients/client-sagemaker/src/SageMaker.ts
@@ -222,6 +222,11 @@ import {
   CreateNotebookInstanceLifecycleConfigCommandInput,
   CreateNotebookInstanceLifecycleConfigCommandOutput,
 } from "./commands/CreateNotebookInstanceLifecycleConfigCommand";
+import {
+  CreateOptimizationJobCommand,
+  CreateOptimizationJobCommandInput,
+  CreateOptimizationJobCommandOutput,
+} from "./commands/CreateOptimizationJobCommand";
 import {
   CreatePipelineCommand,
   CreatePipelineCommandInput,
@@ -478,6 +483,11 @@ import {
   DeleteNotebookInstanceLifecycleConfigCommandInput,
   DeleteNotebookInstanceLifecycleConfigCommandOutput,
 } from "./commands/DeleteNotebookInstanceLifecycleConfigCommand";
+import {
+  DeleteOptimizationJobCommand,
+  DeleteOptimizationJobCommandInput,
+  DeleteOptimizationJobCommandOutput,
+} from "./commands/DeleteOptimizationJobCommand";
 import {
   DeletePipelineCommand,
   DeletePipelineCommandInput,
@@ -748,6 +758,11 @@ import {
   DescribeNotebookInstanceLifecycleConfigCommandInput,
   DescribeNotebookInstanceLifecycleConfigCommandOutput,
 } from "./commands/DescribeNotebookInstanceLifecycleConfigCommand";
+import {
+  DescribeOptimizationJobCommand,
+  DescribeOptimizationJobCommandInput,
+  DescribeOptimizationJobCommandOutput,
+} from "./commands/DescribeOptimizationJobCommand";
 import {
   DescribePipelineCommand,
   DescribePipelineCommandInput,
@@ -1121,6 +1136,11 @@ import {
   ListNotebookInstancesCommandInput,
   ListNotebookInstancesCommandOutput,
 } from "./commands/ListNotebookInstancesCommand";
+import {
+  ListOptimizationJobsCommand,
+  ListOptimizationJobsCommandInput,
+  ListOptimizationJobsCommandOutput,
+} from "./commands/ListOptimizationJobsCommand";
 import {
   ListPipelineExecutionsCommand,
   ListPipelineExecutionsCommandInput,
@@ -1330,6 +1350,11 @@ import {
   StopNotebookInstanceCommandInput,
   StopNotebookInstanceCommandOutput,
 } from "./commands/StopNotebookInstanceCommand";
+import {
+  StopOptimizationJobCommand,
+  StopOptimizationJobCommandInput,
+  StopOptimizationJobCommandOutput,
+} from "./commands/StopOptimizationJobCommand";
 import {
   StopPipelineExecutionCommand,
   StopPipelineExecutionCommandInput,
@@ -1575,6 +1600,7 @@ const commands = {
   CreateMonitoringScheduleCommand,
   CreateNotebookInstanceCommand,
   CreateNotebookInstanceLifecycleConfigCommand,
+  CreateOptimizationJobCommand,
   CreatePipelineCommand,
   CreatePresignedDomainUrlCommand,
   CreatePresignedMlflowTrackingServerUrlCommand,
@@ -1631,6 +1657,7 @@ const commands = {
   DeleteMonitoringScheduleCommand,
   DeleteNotebookInstanceCommand,
   DeleteNotebookInstanceLifecycleConfigCommand,
+  DeleteOptimizationJobCommand,
   DeletePipelineCommand,
   DeleteProjectCommand,
   DeleteSpaceCommand,
@@ -1689,6 +1716,7 @@ const commands = {
   DescribeMonitoringScheduleCommand,
   DescribeNotebookInstanceCommand,
   DescribeNotebookInstanceLifecycleConfigCommand,
+  DescribeOptimizationJobCommand,
   DescribePipelineCommand,
   DescribePipelineDefinitionForExecutionCommand,
   DescribePipelineExecutionCommand,
@@ -1770,6 +1798,7 @@ const commands = {
   ListMonitoringSchedulesCommand,
   ListNotebookInstanceLifecycleConfigsCommand,
   ListNotebookInstancesCommand,
+  ListOptimizationJobsCommand,
   ListPipelineExecutionsCommand,
   ListPipelineExecutionStepsCommand,
   ListPipelineParametersForExecutionCommand,
@@ -1815,6 +1844,7 @@ const commands = {
   StopMlflowTrackingServerCommand,
   StopMonitoringScheduleCommand,
   StopNotebookInstanceCommand,
+  StopOptimizationJobCommand,
   StopPipelineExecutionCommand,
   StopProcessingJobCommand,
   StopTrainingJobCommand,
@@ -2605,6 +2635,23 @@ export interface SageMaker {
     cb: (err: any, data?: CreateNotebookInstanceLifecycleConfigCommandOutput) => void
   ): void;
 
+  /**
+   * @see {@link CreateOptimizationJobCommand}
+   */
+  createOptimizationJob(
+    args: CreateOptimizationJobCommandInput,
+    options?: __HttpHandlerOptions
+  ): Promise<CreateOptimizationJobCommandOutput>;
+  createOptimizationJob(
+    args: CreateOptimizationJobCommandInput,
+    cb: (err: any, data?: CreateOptimizationJobCommandOutput) => void
+  ): void;
+  createOptimizationJob(
+    args: CreateOptimizationJobCommandInput,
+    options: __HttpHandlerOptions,
+    cb: (err: any, data?: CreateOptimizationJobCommandOutput) => void
+  ): void;
+
   /**
    * @see {@link CreatePipelineCommand}
    */
@@ -3471,6 +3518,23 @@ export interface SageMaker {
     cb: (err: any, data?: DeleteNotebookInstanceLifecycleConfigCommandOutput) => void
   ): void;
 
+  /**
+   * @see {@link DeleteOptimizationJobCommand}
+   */
+  deleteOptimizationJob(
+    args: DeleteOptimizationJobCommandInput,
+    options?: __HttpHandlerOptions
+  ): Promise<DeleteOptimizationJobCommandOutput>;
+  deleteOptimizationJob(
+    args: DeleteOptimizationJobCommandInput,
+    cb: (err: any, data?: DeleteOptimizationJobCommandOutput) => void
+  ): void;
+  deleteOptimizationJob(
+    args: DeleteOptimizationJobCommandInput,
+    options: __HttpHandlerOptions,
+    cb: (err: any, data?: DeleteOptimizationJobCommandOutput) => void
+  ): void;
+
   /**
    * @see {@link DeletePipelineCommand}
    */
@@ -4385,6 +4449,23 @@ export interface SageMaker {
     cb: (err: any, data?: DescribeNotebookInstanceLifecycleConfigCommandOutput) => void
   ): void;
 
+  /**
+   * @see {@link DescribeOptimizationJobCommand}
+   */
+  describeOptimizationJob(
+    args: DescribeOptimizationJobCommandInput,
+    options?: __HttpHandlerOptions
+  ): Promise<DescribeOptimizationJobCommandOutput>;
+  describeOptimizationJob(
+    args: DescribeOptimizationJobCommandInput,
+    cb: (err: any, data?: DescribeOptimizationJobCommandOutput) => void
+  ): void;
+  describeOptimizationJob(
+    args: DescribeOptimizationJobCommandInput,
+    options: __HttpHandlerOptions,
+    cb: (err: any, data?: DescribeOptimizationJobCommandOutput) => void
+  ): void;
+
   /**
    * @see {@link DescribePipelineCommand}
    */
@@ -5708,6 +5789,24 @@ export interface SageMaker {
     cb: (err: any, data?: ListNotebookInstancesCommandOutput) => void
   ): void;
 
+  /**
+   * @see {@link ListOptimizationJobsCommand}
+   */
+  listOptimizationJobs(): Promise<ListOptimizationJobsCommandOutput>;
+  listOptimizationJobs(
+    args: ListOptimizationJobsCommandInput,
+    options?: __HttpHandlerOptions
+  ): Promise<ListOptimizationJobsCommandOutput>;
+  listOptimizationJobs(
+    args: ListOptimizationJobsCommandInput,
+    cb: (err: any, data?: ListOptimizationJobsCommandOutput) => void
+  ): void;
+  listOptimizationJobs(
+    args: ListOptimizationJobsCommandInput,
+    options: __HttpHandlerOptions,
+    cb: (err: any, data?: ListOptimizationJobsCommandOutput) => void
+  ): void;
+
   /**
    * @see {@link ListPipelineExecutionsCommand}
    */
@@ -6426,6 +6525,23 @@ export interface SageMaker {
     cb: (err: any, data?: StopNotebookInstanceCommandOutput) => void
   ): void;
 
+  /**
+   * @see {@link StopOptimizationJobCommand}
+   */
+  stopOptimizationJob(
+    args: StopOptimizationJobCommandInput,
+    options?: __HttpHandlerOptions
+  ): Promise<StopOptimizationJobCommandOutput>;
+  stopOptimizationJob(
+    args: StopOptimizationJobCommandInput,
+    cb: (err: any, data?: StopOptimizationJobCommandOutput) => void
+  ): void;
+  stopOptimizationJob(
+    args: StopOptimizationJobCommandInput,
+    options: __HttpHandlerOptions,
+    cb: (err: any, data?: StopOptimizationJobCommandOutput) => void
+  ): void;
+
   /**
    * @see {@link StopPipelineExecutionCommand}
    */