Skip to content

Commit 99cda6c

Browse files
author
awstools
committed
feat(client-sagemaker): Added support for serverless MLflow Apps.
Added support for new HubContentTypes (DataSet and JsonDoc) in Private Hub for AI model customization assets, enabling tracking and management of training datasets and evaluators (reward functions/prompts) throughout the ML lifecycle.
1 parent 45c4bd1 commit 99cda6c

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

52 files changed

+7206
-4872
lines changed

clients/client-sagemaker/README.md

Lines changed: 48 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -554,6 +554,14 @@ CreateLabelingJob
554554

555555
[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/sagemaker/command/CreateLabelingJobCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/CreateLabelingJobCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/CreateLabelingJobCommandOutput/)
556556

557+
</details>
558+
<details>
559+
<summary>
560+
CreateMlflowApp
561+
</summary>
562+
563+
[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/sagemaker/command/CreateMlflowAppCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/CreateMlflowAppCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/CreateMlflowAppCommandOutput/)
564+
557565
</details>
558566
<details>
559567
<summary>
@@ -690,6 +698,14 @@ CreatePresignedDomainUrl
690698

691699
[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/sagemaker/command/CreatePresignedDomainUrlCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/CreatePresignedDomainUrlCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/CreatePresignedDomainUrlCommandOutput/)
692700

701+
</details>
702+
<details>
703+
<summary>
704+
CreatePresignedMlflowAppUrl
705+
</summary>
706+
707+
[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/sagemaker/command/CreatePresignedMlflowAppUrlCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/CreatePresignedMlflowAppUrlCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/CreatePresignedMlflowAppUrlCommandOutput/)
708+
693709
</details>
694710
<details>
695711
<summary>
@@ -1050,6 +1066,14 @@ DeleteInferenceExperiment
10501066

10511067
[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/sagemaker/command/DeleteInferenceExperimentCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/DeleteInferenceExperimentCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/DeleteInferenceExperimentCommandOutput/)
10521068

1069+
</details>
1070+
<details>
1071+
<summary>
1072+
DeleteMlflowApp
1073+
</summary>
1074+
1075+
[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/sagemaker/command/DeleteMlflowAppCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/DeleteMlflowAppCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/DeleteMlflowAppCommandOutput/)
1076+
10531077
</details>
10541078
<details>
10551079
<summary>
@@ -1570,6 +1594,14 @@ DescribeLineageGroup
15701594

15711595
[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/sagemaker/command/DescribeLineageGroupCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/DescribeLineageGroupCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/DescribeLineageGroupCommandOutput/)
15721596

1597+
</details>
1598+
<details>
1599+
<summary>
1600+
DescribeMlflowApp
1601+
</summary>
1602+
1603+
[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/sagemaker/command/DescribeMlflowAppCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/DescribeMlflowAppCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/DescribeMlflowAppCommandOutput/)
1604+
15731605
</details>
15741606
<details>
15751607
<summary>
@@ -2242,6 +2274,14 @@ ListLineageGroups
22422274

22432275
[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/sagemaker/command/ListLineageGroupsCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/ListLineageGroupsCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/ListLineageGroupsCommandOutput/)
22442276

2277+
</details>
2278+
<details>
2279+
<summary>
2280+
ListMlflowApps
2281+
</summary>
2282+
2283+
[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/sagemaker/command/ListMlflowAppsCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/ListMlflowAppsCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/ListMlflowAppsCommandOutput/)
2284+
22452285
</details>
22462286
<details>
22472287
<summary>
@@ -3034,6 +3074,14 @@ UpdateInferenceExperiment
30343074

30353075
[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/sagemaker/command/UpdateInferenceExperimentCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/UpdateInferenceExperimentCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/UpdateInferenceExperimentCommandOutput/)
30363076

3077+
</details>
3078+
<details>
3079+
<summary>
3080+
UpdateMlflowApp
3081+
</summary>
3082+
3083+
[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/sagemaker/command/UpdateMlflowAppCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/UpdateMlflowAppCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/UpdateMlflowAppCommandOutput/)
3084+
30373085
</details>
30383086
<details>
30393087
<summary>

clients/client-sagemaker/src/SageMaker.ts

Lines changed: 127 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -206,6 +206,11 @@ import {
206206
CreateLabelingJobCommandInput,
207207
CreateLabelingJobCommandOutput,
208208
} from "./commands/CreateLabelingJobCommand";
209+
import {
210+
CreateMlflowAppCommand,
211+
CreateMlflowAppCommandInput,
212+
CreateMlflowAppCommandOutput,
213+
} from "./commands/CreateMlflowAppCommand";
209214
import {
210215
CreateMlflowTrackingServerCommand,
211216
CreateMlflowTrackingServerCommandInput,
@@ -287,6 +292,11 @@ import {
287292
CreatePresignedDomainUrlCommandInput,
288293
CreatePresignedDomainUrlCommandOutput,
289294
} from "./commands/CreatePresignedDomainUrlCommand";
295+
import {
296+
CreatePresignedMlflowAppUrlCommand,
297+
CreatePresignedMlflowAppUrlCommandInput,
298+
CreatePresignedMlflowAppUrlCommandOutput,
299+
} from "./commands/CreatePresignedMlflowAppUrlCommand";
290300
import {
291301
CreatePresignedMlflowTrackingServerUrlCommand,
292302
CreatePresignedMlflowTrackingServerUrlCommandInput,
@@ -492,6 +502,11 @@ import {
492502
DeleteInferenceExperimentCommandInput,
493503
DeleteInferenceExperimentCommandOutput,
494504
} from "./commands/DeleteInferenceExperimentCommand";
505+
import {
506+
DeleteMlflowAppCommand,
507+
DeleteMlflowAppCommandInput,
508+
DeleteMlflowAppCommandOutput,
509+
} from "./commands/DeleteMlflowAppCommand";
495510
import {
496511
DeleteMlflowTrackingServerCommand,
497512
DeleteMlflowTrackingServerCommandInput,
@@ -793,6 +808,11 @@ import {
793808
DescribeLineageGroupCommandInput,
794809
DescribeLineageGroupCommandOutput,
795810
} from "./commands/DescribeLineageGroupCommand";
811+
import {
812+
DescribeMlflowAppCommand,
813+
DescribeMlflowAppCommandInput,
814+
DescribeMlflowAppCommandOutput,
815+
} from "./commands/DescribeMlflowAppCommand";
796816
import {
797817
DescribeMlflowTrackingServerCommand,
798818
DescribeMlflowTrackingServerCommandInput,
@@ -1185,6 +1205,11 @@ import {
11851205
ListLineageGroupsCommandInput,
11861206
ListLineageGroupsCommandOutput,
11871207
} from "./commands/ListLineageGroupsCommand";
1208+
import {
1209+
ListMlflowAppsCommand,
1210+
ListMlflowAppsCommandInput,
1211+
ListMlflowAppsCommandOutput,
1212+
} from "./commands/ListMlflowAppsCommand";
11881213
import {
11891214
ListMlflowTrackingServersCommand,
11901215
ListMlflowTrackingServersCommandInput,
@@ -1652,6 +1677,11 @@ import {
16521677
UpdateInferenceExperimentCommandInput,
16531678
UpdateInferenceExperimentCommandOutput,
16541679
} from "./commands/UpdateInferenceExperimentCommand";
1680+
import {
1681+
UpdateMlflowAppCommand,
1682+
UpdateMlflowAppCommandInput,
1683+
UpdateMlflowAppCommandOutput,
1684+
} from "./commands/UpdateMlflowAppCommand";
16551685
import {
16561686
UpdateMlflowTrackingServerCommand,
16571687
UpdateMlflowTrackingServerCommandInput,
@@ -1786,6 +1816,7 @@ const commands = {
17861816
CreateInferenceExperimentCommand,
17871817
CreateInferenceRecommendationsJobCommand,
17881818
CreateLabelingJobCommand,
1819+
CreateMlflowAppCommand,
17891820
CreateMlflowTrackingServerCommand,
17901821
CreateModelCommand,
17911822
CreateModelBiasJobDefinitionCommand,
@@ -1803,6 +1834,7 @@ const commands = {
18031834
CreatePartnerAppPresignedUrlCommand,
18041835
CreatePipelineCommand,
18051836
CreatePresignedDomainUrlCommand,
1837+
CreatePresignedMlflowAppUrlCommand,
18061838
CreatePresignedMlflowTrackingServerUrlCommand,
18071839
CreatePresignedNotebookInstanceUrlCommand,
18081840
CreateProcessingJobCommand,
@@ -1848,6 +1880,7 @@ const commands = {
18481880
DeleteImageVersionCommand,
18491881
DeleteInferenceComponentCommand,
18501882
DeleteInferenceExperimentCommand,
1883+
DeleteMlflowAppCommand,
18511884
DeleteMlflowTrackingServerCommand,
18521885
DeleteModelCommand,
18531886
DeleteModelBiasJobDefinitionCommand,
@@ -1913,6 +1946,7 @@ const commands = {
19131946
DescribeInferenceRecommendationsJobCommand,
19141947
DescribeLabelingJobCommand,
19151948
DescribeLineageGroupCommand,
1949+
DescribeMlflowAppCommand,
19161950
DescribeMlflowTrackingServerCommand,
19171951
DescribeModelCommand,
19181952
DescribeModelBiasJobDefinitionCommand,
@@ -1997,6 +2031,7 @@ const commands = {
19972031
ListLabelingJobsCommand,
19982032
ListLabelingJobsForWorkteamCommand,
19992033
ListLineageGroupsCommand,
2034+
ListMlflowAppsCommand,
20002035
ListMlflowTrackingServersCommand,
20012036
ListModelBiasJobDefinitionsCommand,
20022037
ListModelCardExportJobsCommand,
@@ -2096,6 +2131,7 @@ const commands = {
20962131
UpdateInferenceComponentCommand,
20972132
UpdateInferenceComponentRuntimeConfigCommand,
20982133
UpdateInferenceExperimentCommand,
2134+
UpdateMlflowAppCommand,
20992135
UpdateMlflowTrackingServerCommand,
21002136
UpdateModelCardCommand,
21012137
UpdateModelPackageCommand,
@@ -2803,6 +2839,20 @@ export interface SageMaker {
28032839
cb: (err: any, data?: CreateLabelingJobCommandOutput) => void
28042840
): void;
28052841

2842+
/**
2843+
* @see {@link CreateMlflowAppCommand}
2844+
*/
2845+
createMlflowApp(
2846+
args: CreateMlflowAppCommandInput,
2847+
options?: __HttpHandlerOptions
2848+
): Promise<CreateMlflowAppCommandOutput>;
2849+
createMlflowApp(args: CreateMlflowAppCommandInput, cb: (err: any, data?: CreateMlflowAppCommandOutput) => void): void;
2850+
createMlflowApp(
2851+
args: CreateMlflowAppCommandInput,
2852+
options: __HttpHandlerOptions,
2853+
cb: (err: any, data?: CreateMlflowAppCommandOutput) => void
2854+
): void;
2855+
28062856
/**
28072857
* @see {@link CreateMlflowTrackingServerCommand}
28082858
*/
@@ -3081,6 +3131,23 @@ export interface SageMaker {
30813131
cb: (err: any, data?: CreatePresignedDomainUrlCommandOutput) => void
30823132
): void;
30833133

3134+
/**
3135+
* @see {@link CreatePresignedMlflowAppUrlCommand}
3136+
*/
3137+
createPresignedMlflowAppUrl(
3138+
args: CreatePresignedMlflowAppUrlCommandInput,
3139+
options?: __HttpHandlerOptions
3140+
): Promise<CreatePresignedMlflowAppUrlCommandOutput>;
3141+
createPresignedMlflowAppUrl(
3142+
args: CreatePresignedMlflowAppUrlCommandInput,
3143+
cb: (err: any, data?: CreatePresignedMlflowAppUrlCommandOutput) => void
3144+
): void;
3145+
createPresignedMlflowAppUrl(
3146+
args: CreatePresignedMlflowAppUrlCommandInput,
3147+
options: __HttpHandlerOptions,
3148+
cb: (err: any, data?: CreatePresignedMlflowAppUrlCommandOutput) => void
3149+
): void;
3150+
30843151
/**
30853152
* @see {@link CreatePresignedMlflowTrackingServerUrlCommand}
30863153
*/
@@ -3772,6 +3839,20 @@ export interface SageMaker {
37723839
cb: (err: any, data?: DeleteInferenceExperimentCommandOutput) => void
37733840
): void;
37743841

3842+
/**
3843+
* @see {@link DeleteMlflowAppCommand}
3844+
*/
3845+
deleteMlflowApp(
3846+
args: DeleteMlflowAppCommandInput,
3847+
options?: __HttpHandlerOptions
3848+
): Promise<DeleteMlflowAppCommandOutput>;
3849+
deleteMlflowApp(args: DeleteMlflowAppCommandInput, cb: (err: any, data?: DeleteMlflowAppCommandOutput) => void): void;
3850+
deleteMlflowApp(
3851+
args: DeleteMlflowAppCommandInput,
3852+
options: __HttpHandlerOptions,
3853+
cb: (err: any, data?: DeleteMlflowAppCommandOutput) => void
3854+
): void;
3855+
37753856
/**
37763857
* @see {@link DeleteMlflowTrackingServerCommand}
37773858
*/
@@ -4802,6 +4883,23 @@ export interface SageMaker {
48024883
cb: (err: any, data?: DescribeLineageGroupCommandOutput) => void
48034884
): void;
48044885

4886+
/**
4887+
* @see {@link DescribeMlflowAppCommand}
4888+
*/
4889+
describeMlflowApp(
4890+
args: DescribeMlflowAppCommandInput,
4891+
options?: __HttpHandlerOptions
4892+
): Promise<DescribeMlflowAppCommandOutput>;
4893+
describeMlflowApp(
4894+
args: DescribeMlflowAppCommandInput,
4895+
cb: (err: any, data?: DescribeMlflowAppCommandOutput) => void
4896+
): void;
4897+
describeMlflowApp(
4898+
args: DescribeMlflowAppCommandInput,
4899+
options: __HttpHandlerOptions,
4900+
cb: (err: any, data?: DescribeMlflowAppCommandOutput) => void
4901+
): void;
4902+
48054903
/**
48064904
* @see {@link DescribeMlflowTrackingServerCommand}
48074905
*/
@@ -6167,6 +6265,21 @@ export interface SageMaker {
61676265
cb: (err: any, data?: ListLineageGroupsCommandOutput) => void
61686266
): void;
61696267

6268+
/**
6269+
* @see {@link ListMlflowAppsCommand}
6270+
*/
6271+
listMlflowApps(): Promise<ListMlflowAppsCommandOutput>;
6272+
listMlflowApps(
6273+
args: ListMlflowAppsCommandInput,
6274+
options?: __HttpHandlerOptions
6275+
): Promise<ListMlflowAppsCommandOutput>;
6276+
listMlflowApps(args: ListMlflowAppsCommandInput, cb: (err: any, data?: ListMlflowAppsCommandOutput) => void): void;
6277+
listMlflowApps(
6278+
args: ListMlflowAppsCommandInput,
6279+
options: __HttpHandlerOptions,
6280+
cb: (err: any, data?: ListMlflowAppsCommandOutput) => void
6281+
): void;
6282+
61706283
/**
61716284
* @see {@link ListMlflowTrackingServersCommand}
61726285
*/
@@ -7751,6 +7864,20 @@ export interface SageMaker {
77517864
cb: (err: any, data?: UpdateInferenceExperimentCommandOutput) => void
77527865
): void;
77537866

7867+
/**
7868+
* @see {@link UpdateMlflowAppCommand}
7869+
*/
7870+
updateMlflowApp(
7871+
args: UpdateMlflowAppCommandInput,
7872+
options?: __HttpHandlerOptions
7873+
): Promise<UpdateMlflowAppCommandOutput>;
7874+
updateMlflowApp(args: UpdateMlflowAppCommandInput, cb: (err: any, data?: UpdateMlflowAppCommandOutput) => void): void;
7875+
updateMlflowApp(
7876+
args: UpdateMlflowAppCommandInput,
7877+
options: __HttpHandlerOptions,
7878+
cb: (err: any, data?: UpdateMlflowAppCommandOutput) => void
7879+
): void;
7880+
77547881
/**
77557882
* @see {@link UpdateMlflowTrackingServerCommand}
77567883
*/

0 commit comments

Comments
 (0)