Skip to content

Commit 3f89139

Browse files
author
awstools
committed
feat(client-sagemaker): Amazon SageMaker Edge Manager provides lightweight model deployment feature to deploy machine learning models on requested devices.
1 parent e65c9b2 commit 3f89139

30 files changed

+11019
-6881
lines changed

clients/client-sagemaker/src/SageMaker.ts

Lines changed: 333 additions & 0 deletions
Large diffs are not rendered by default.

clients/client-sagemaker/src/SageMakerClient.ts

Lines changed: 51 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -87,6 +87,14 @@ import {
8787
} from "./commands/CreateDataQualityJobDefinitionCommand";
8888
import { CreateDeviceFleetCommandInput, CreateDeviceFleetCommandOutput } from "./commands/CreateDeviceFleetCommand";
8989
import { CreateDomainCommandInput, CreateDomainCommandOutput } from "./commands/CreateDomainCommand";
90+
import {
91+
CreateEdgeDeploymentPlanCommandInput,
92+
CreateEdgeDeploymentPlanCommandOutput,
93+
} from "./commands/CreateEdgeDeploymentPlanCommand";
94+
import {
95+
CreateEdgeDeploymentStageCommandInput,
96+
CreateEdgeDeploymentStageCommandOutput,
97+
} from "./commands/CreateEdgeDeploymentStageCommand";
9098
import {
9199
CreateEdgePackagingJobCommandInput,
92100
CreateEdgePackagingJobCommandOutput,
@@ -192,6 +200,14 @@ import {
192200
} from "./commands/DeleteDataQualityJobDefinitionCommand";
193201
import { DeleteDeviceFleetCommandInput, DeleteDeviceFleetCommandOutput } from "./commands/DeleteDeviceFleetCommand";
194202
import { DeleteDomainCommandInput, DeleteDomainCommandOutput } from "./commands/DeleteDomainCommand";
203+
import {
204+
DeleteEdgeDeploymentPlanCommandInput,
205+
DeleteEdgeDeploymentPlanCommandOutput,
206+
} from "./commands/DeleteEdgeDeploymentPlanCommand";
207+
import {
208+
DeleteEdgeDeploymentStageCommandInput,
209+
DeleteEdgeDeploymentStageCommandOutput,
210+
} from "./commands/DeleteEdgeDeploymentStageCommand";
195211
import { DeleteEndpointCommandInput, DeleteEndpointCommandOutput } from "./commands/DeleteEndpointCommand";
196212
import {
197213
DeleteEndpointConfigCommandInput,
@@ -284,6 +300,10 @@ import {
284300
DescribeDeviceFleetCommandOutput,
285301
} from "./commands/DescribeDeviceFleetCommand";
286302
import { DescribeDomainCommandInput, DescribeDomainCommandOutput } from "./commands/DescribeDomainCommand";
303+
import {
304+
DescribeEdgeDeploymentPlanCommandInput,
305+
DescribeEdgeDeploymentPlanCommandOutput,
306+
} from "./commands/DescribeEdgeDeploymentPlanCommand";
287307
import {
288308
DescribeEdgePackagingJobCommandInput,
289309
DescribeEdgePackagingJobCommandOutput,
@@ -467,6 +487,10 @@ import {
467487
import { ListDeviceFleetsCommandInput, ListDeviceFleetsCommandOutput } from "./commands/ListDeviceFleetsCommand";
468488
import { ListDevicesCommandInput, ListDevicesCommandOutput } from "./commands/ListDevicesCommand";
469489
import { ListDomainsCommandInput, ListDomainsCommandOutput } from "./commands/ListDomainsCommand";
490+
import {
491+
ListEdgeDeploymentPlansCommandInput,
492+
ListEdgeDeploymentPlansCommandOutput,
493+
} from "./commands/ListEdgeDeploymentPlansCommand";
470494
import {
471495
ListEdgePackagingJobsCommandInput,
472496
ListEdgePackagingJobsCommandOutput,
@@ -549,6 +573,7 @@ import {
549573
import { ListPipelinesCommandInput, ListPipelinesCommandOutput } from "./commands/ListPipelinesCommand";
550574
import { ListProcessingJobsCommandInput, ListProcessingJobsCommandOutput } from "./commands/ListProcessingJobsCommand";
551575
import { ListProjectsCommandInput, ListProjectsCommandOutput } from "./commands/ListProjectsCommand";
576+
import { ListStageDevicesCommandInput, ListStageDevicesCommandOutput } from "./commands/ListStageDevicesCommand";
552577
import {
553578
ListStudioLifecycleConfigsCommandInput,
554579
ListStudioLifecycleConfigsCommandOutput,
@@ -592,6 +617,10 @@ import {
592617
SendPipelineExecutionStepSuccessCommandInput,
593618
SendPipelineExecutionStepSuccessCommandOutput,
594619
} from "./commands/SendPipelineExecutionStepSuccessCommand";
620+
import {
621+
StartEdgeDeploymentStageCommandInput,
622+
StartEdgeDeploymentStageCommandOutput,
623+
} from "./commands/StartEdgeDeploymentStageCommand";
595624
import {
596625
StartMonitoringScheduleCommandInput,
597626
StartMonitoringScheduleCommandOutput,
@@ -606,6 +635,10 @@ import {
606635
} from "./commands/StartPipelineExecutionCommand";
607636
import { StopAutoMLJobCommandInput, StopAutoMLJobCommandOutput } from "./commands/StopAutoMLJobCommand";
608637
import { StopCompilationJobCommandInput, StopCompilationJobCommandOutput } from "./commands/StopCompilationJobCommand";
638+
import {
639+
StopEdgeDeploymentStageCommandInput,
640+
StopEdgeDeploymentStageCommandOutput,
641+
} from "./commands/StopEdgeDeploymentStageCommand";
609642
import {
610643
StopEdgePackagingJobCommandInput,
611644
StopEdgePackagingJobCommandOutput,
@@ -707,6 +740,8 @@ export type ServiceInputTypes =
707740
| CreateDataQualityJobDefinitionCommandInput
708741
| CreateDeviceFleetCommandInput
709742
| CreateDomainCommandInput
743+
| CreateEdgeDeploymentPlanCommandInput
744+
| CreateEdgeDeploymentStageCommandInput
710745
| CreateEdgePackagingJobCommandInput
711746
| CreateEndpointCommandInput
712747
| CreateEndpointConfigCommandInput
@@ -752,6 +787,8 @@ export type ServiceInputTypes =
752787
| DeleteDataQualityJobDefinitionCommandInput
753788
| DeleteDeviceFleetCommandInput
754789
| DeleteDomainCommandInput
790+
| DeleteEdgeDeploymentPlanCommandInput
791+
| DeleteEdgeDeploymentStageCommandInput
755792
| DeleteEndpointCommandInput
756793
| DeleteEndpointConfigCommandInput
757794
| DeleteExperimentCommandInput
@@ -793,6 +830,7 @@ export type ServiceInputTypes =
793830
| DescribeDeviceCommandInput
794831
| DescribeDeviceFleetCommandInput
795832
| DescribeDomainCommandInput
833+
| DescribeEdgeDeploymentPlanCommandInput
796834
| DescribeEdgePackagingJobCommandInput
797835
| DescribeEndpointCommandInput
798836
| DescribeEndpointConfigCommandInput
@@ -853,6 +891,7 @@ export type ServiceInputTypes =
853891
| ListDeviceFleetsCommandInput
854892
| ListDevicesCommandInput
855893
| ListDomainsCommandInput
894+
| ListEdgeDeploymentPlansCommandInput
856895
| ListEdgePackagingJobsCommandInput
857896
| ListEndpointConfigsCommandInput
858897
| ListEndpointsCommandInput
@@ -884,6 +923,7 @@ export type ServiceInputTypes =
884923
| ListPipelinesCommandInput
885924
| ListProcessingJobsCommandInput
886925
| ListProjectsCommandInput
926+
| ListStageDevicesCommandInput
887927
| ListStudioLifecycleConfigsCommandInput
888928
| ListSubscribedWorkteamsCommandInput
889929
| ListTagsCommandInput
@@ -903,11 +943,13 @@ export type ServiceInputTypes =
903943
| SearchCommandInput
904944
| SendPipelineExecutionStepFailureCommandInput
905945
| SendPipelineExecutionStepSuccessCommandInput
946+
| StartEdgeDeploymentStageCommandInput
906947
| StartMonitoringScheduleCommandInput
907948
| StartNotebookInstanceCommandInput
908949
| StartPipelineExecutionCommandInput
909950
| StopAutoMLJobCommandInput
910951
| StopCompilationJobCommandInput
952+
| StopEdgeDeploymentStageCommandInput
911953
| StopEdgePackagingJobCommandInput
912954
| StopHyperParameterTuningJobCommandInput
913955
| StopInferenceRecommendationsJobCommandInput
@@ -963,6 +1005,8 @@ export type ServiceOutputTypes =
9631005
| CreateDataQualityJobDefinitionCommandOutput
9641006
| CreateDeviceFleetCommandOutput
9651007
| CreateDomainCommandOutput
1008+
| CreateEdgeDeploymentPlanCommandOutput
1009+
| CreateEdgeDeploymentStageCommandOutput
9661010
| CreateEdgePackagingJobCommandOutput
9671011
| CreateEndpointCommandOutput
9681012
| CreateEndpointConfigCommandOutput
@@ -1008,6 +1052,8 @@ export type ServiceOutputTypes =
10081052
| DeleteDataQualityJobDefinitionCommandOutput
10091053
| DeleteDeviceFleetCommandOutput
10101054
| DeleteDomainCommandOutput
1055+
| DeleteEdgeDeploymentPlanCommandOutput
1056+
| DeleteEdgeDeploymentStageCommandOutput
10111057
| DeleteEndpointCommandOutput
10121058
| DeleteEndpointConfigCommandOutput
10131059
| DeleteExperimentCommandOutput
@@ -1049,6 +1095,7 @@ export type ServiceOutputTypes =
10491095
| DescribeDeviceCommandOutput
10501096
| DescribeDeviceFleetCommandOutput
10511097
| DescribeDomainCommandOutput
1098+
| DescribeEdgeDeploymentPlanCommandOutput
10521099
| DescribeEdgePackagingJobCommandOutput
10531100
| DescribeEndpointCommandOutput
10541101
| DescribeEndpointConfigCommandOutput
@@ -1109,6 +1156,7 @@ export type ServiceOutputTypes =
11091156
| ListDeviceFleetsCommandOutput
11101157
| ListDevicesCommandOutput
11111158
| ListDomainsCommandOutput
1159+
| ListEdgeDeploymentPlansCommandOutput
11121160
| ListEdgePackagingJobsCommandOutput
11131161
| ListEndpointConfigsCommandOutput
11141162
| ListEndpointsCommandOutput
@@ -1140,6 +1188,7 @@ export type ServiceOutputTypes =
11401188
| ListPipelinesCommandOutput
11411189
| ListProcessingJobsCommandOutput
11421190
| ListProjectsCommandOutput
1191+
| ListStageDevicesCommandOutput
11431192
| ListStudioLifecycleConfigsCommandOutput
11441193
| ListSubscribedWorkteamsCommandOutput
11451194
| ListTagsCommandOutput
@@ -1159,11 +1208,13 @@ export type ServiceOutputTypes =
11591208
| SearchCommandOutput
11601209
| SendPipelineExecutionStepFailureCommandOutput
11611210
| SendPipelineExecutionStepSuccessCommandOutput
1211+
| StartEdgeDeploymentStageCommandOutput
11621212
| StartMonitoringScheduleCommandOutput
11631213
| StartNotebookInstanceCommandOutput
11641214
| StartPipelineExecutionCommandOutput
11651215
| StopAutoMLJobCommandOutput
11661216
| StopCompilationJobCommandOutput
1217+
| StopEdgeDeploymentStageCommandOutput
11671218
| StopEdgePackagingJobCommandOutput
11681219
| StopHyperParameterTuningJobCommandOutput
11691220
| StopInferenceRecommendationsJobCommandOutput
Lines changed: 96 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,96 @@
1+
// smithy-typescript generated code
2+
import { getSerdePlugin } from "@aws-sdk/middleware-serde";
3+
import { HttpRequest as __HttpRequest, HttpResponse as __HttpResponse } from "@aws-sdk/protocol-http";
4+
import { Command as $Command } from "@aws-sdk/smithy-client";
5+
import {
6+
FinalizeHandlerArguments,
7+
Handler,
8+
HandlerExecutionContext,
9+
HttpHandlerOptions as __HttpHandlerOptions,
10+
MetadataBearer as __MetadataBearer,
11+
MiddlewareStack,
12+
SerdeContext as __SerdeContext,
13+
} from "@aws-sdk/types";
14+
15+
import { CreateEdgeDeploymentPlanRequest, CreateEdgeDeploymentPlanResponse } from "../models/models_0";
16+
import {
17+
deserializeAws_json1_1CreateEdgeDeploymentPlanCommand,
18+
serializeAws_json1_1CreateEdgeDeploymentPlanCommand,
19+
} from "../protocols/Aws_json1_1";
20+
import { SageMakerClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../SageMakerClient";
21+
22+
export interface CreateEdgeDeploymentPlanCommandInput extends CreateEdgeDeploymentPlanRequest {}
23+
export interface CreateEdgeDeploymentPlanCommandOutput extends CreateEdgeDeploymentPlanResponse, __MetadataBearer {}
24+
25+
/**
26+
* <p>Creates an edge deployment plan, consisting of multiple stages. Each stage may have a different deployment configuration and devices.</p>
27+
* @example
28+
* Use a bare-bones client and the command you need to make an API call.
29+
* ```javascript
30+
* import { SageMakerClient, CreateEdgeDeploymentPlanCommand } from "@aws-sdk/client-sagemaker"; // ES Modules import
31+
* // const { SageMakerClient, CreateEdgeDeploymentPlanCommand } = require("@aws-sdk/client-sagemaker"); // CommonJS import
32+
* const client = new SageMakerClient(config);
33+
* const command = new CreateEdgeDeploymentPlanCommand(input);
34+
* const response = await client.send(command);
35+
* ```
36+
*
37+
* @see {@link CreateEdgeDeploymentPlanCommandInput} for command's `input` shape.
38+
* @see {@link CreateEdgeDeploymentPlanCommandOutput} for command's `response` shape.
39+
* @see {@link SageMakerClientResolvedConfig | config} for SageMakerClient's `config` shape.
40+
*
41+
*/
42+
export class CreateEdgeDeploymentPlanCommand extends $Command<
43+
CreateEdgeDeploymentPlanCommandInput,
44+
CreateEdgeDeploymentPlanCommandOutput,
45+
SageMakerClientResolvedConfig
46+
> {
47+
// Start section: command_properties
48+
// End section: command_properties
49+
50+
constructor(readonly input: CreateEdgeDeploymentPlanCommandInput) {
51+
// Start section: command_constructor
52+
super();
53+
// End section: command_constructor
54+
}
55+
56+
/**
57+
* @internal
58+
*/
59+
resolveMiddleware(
60+
clientStack: MiddlewareStack<ServiceInputTypes, ServiceOutputTypes>,
61+
configuration: SageMakerClientResolvedConfig,
62+
options?: __HttpHandlerOptions
63+
): Handler<CreateEdgeDeploymentPlanCommandInput, CreateEdgeDeploymentPlanCommandOutput> {
64+
this.middlewareStack.use(getSerdePlugin(configuration, this.serialize, this.deserialize));
65+
66+
const stack = clientStack.concat(this.middlewareStack);
67+
68+
const { logger } = configuration;
69+
const clientName = "SageMakerClient";
70+
const commandName = "CreateEdgeDeploymentPlanCommand";
71+
const handlerExecutionContext: HandlerExecutionContext = {
72+
logger,
73+
clientName,
74+
commandName,
75+
inputFilterSensitiveLog: CreateEdgeDeploymentPlanRequest.filterSensitiveLog,
76+
outputFilterSensitiveLog: CreateEdgeDeploymentPlanResponse.filterSensitiveLog,
77+
};
78+
const { requestHandler } = configuration;
79+
return stack.resolve(
80+
(request: FinalizeHandlerArguments<any>) =>
81+
requestHandler.handle(request.request as __HttpRequest, options || {}),
82+
handlerExecutionContext
83+
);
84+
}
85+
86+
private serialize(input: CreateEdgeDeploymentPlanCommandInput, context: __SerdeContext): Promise<__HttpRequest> {
87+
return serializeAws_json1_1CreateEdgeDeploymentPlanCommand(input, context);
88+
}
89+
90+
private deserialize(output: __HttpResponse, context: __SerdeContext): Promise<CreateEdgeDeploymentPlanCommandOutput> {
91+
return deserializeAws_json1_1CreateEdgeDeploymentPlanCommand(output, context);
92+
}
93+
94+
// Start section: command_body_extra
95+
// End section: command_body_extra
96+
}
Lines changed: 99 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,99 @@
1+
// smithy-typescript generated code
2+
import { getSerdePlugin } from "@aws-sdk/middleware-serde";
3+
import { HttpRequest as __HttpRequest, HttpResponse as __HttpResponse } from "@aws-sdk/protocol-http";
4+
import { Command as $Command } from "@aws-sdk/smithy-client";
5+
import {
6+
FinalizeHandlerArguments,
7+
Handler,
8+
HandlerExecutionContext,
9+
HttpHandlerOptions as __HttpHandlerOptions,
10+
MetadataBearer as __MetadataBearer,
11+
MiddlewareStack,
12+
SerdeContext as __SerdeContext,
13+
} from "@aws-sdk/types";
14+
15+
import { CreateEdgeDeploymentStageRequest } from "../models/models_0";
16+
import {
17+
deserializeAws_json1_1CreateEdgeDeploymentStageCommand,
18+
serializeAws_json1_1CreateEdgeDeploymentStageCommand,
19+
} from "../protocols/Aws_json1_1";
20+
import { SageMakerClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../SageMakerClient";
21+
22+
export interface CreateEdgeDeploymentStageCommandInput extends CreateEdgeDeploymentStageRequest {}
23+
export interface CreateEdgeDeploymentStageCommandOutput extends __MetadataBearer {}
24+
25+
/**
26+
* <p>Creates a new stage in an existing edge deployment plan.</p>
27+
* @example
28+
* Use a bare-bones client and the command you need to make an API call.
29+
* ```javascript
30+
* import { SageMakerClient, CreateEdgeDeploymentStageCommand } from "@aws-sdk/client-sagemaker"; // ES Modules import
31+
* // const { SageMakerClient, CreateEdgeDeploymentStageCommand } = require("@aws-sdk/client-sagemaker"); // CommonJS import
32+
* const client = new SageMakerClient(config);
33+
* const command = new CreateEdgeDeploymentStageCommand(input);
34+
* const response = await client.send(command);
35+
* ```
36+
*
37+
* @see {@link CreateEdgeDeploymentStageCommandInput} for command's `input` shape.
38+
* @see {@link CreateEdgeDeploymentStageCommandOutput} for command's `response` shape.
39+
* @see {@link SageMakerClientResolvedConfig | config} for SageMakerClient's `config` shape.
40+
*
41+
*/
42+
export class CreateEdgeDeploymentStageCommand extends $Command<
43+
CreateEdgeDeploymentStageCommandInput,
44+
CreateEdgeDeploymentStageCommandOutput,
45+
SageMakerClientResolvedConfig
46+
> {
47+
// Start section: command_properties
48+
// End section: command_properties
49+
50+
constructor(readonly input: CreateEdgeDeploymentStageCommandInput) {
51+
// Start section: command_constructor
52+
super();
53+
// End section: command_constructor
54+
}
55+
56+
/**
57+
* @internal
58+
*/
59+
resolveMiddleware(
60+
clientStack: MiddlewareStack<ServiceInputTypes, ServiceOutputTypes>,
61+
configuration: SageMakerClientResolvedConfig,
62+
options?: __HttpHandlerOptions
63+
): Handler<CreateEdgeDeploymentStageCommandInput, CreateEdgeDeploymentStageCommandOutput> {
64+
this.middlewareStack.use(getSerdePlugin(configuration, this.serialize, this.deserialize));
65+
66+
const stack = clientStack.concat(this.middlewareStack);
67+
68+
const { logger } = configuration;
69+
const clientName = "SageMakerClient";
70+
const commandName = "CreateEdgeDeploymentStageCommand";
71+
const handlerExecutionContext: HandlerExecutionContext = {
72+
logger,
73+
clientName,
74+
commandName,
75+
inputFilterSensitiveLog: CreateEdgeDeploymentStageRequest.filterSensitiveLog,
76+
outputFilterSensitiveLog: (output: any) => output,
77+
};
78+
const { requestHandler } = configuration;
79+
return stack.resolve(
80+
(request: FinalizeHandlerArguments<any>) =>
81+
requestHandler.handle(request.request as __HttpRequest, options || {}),
82+
handlerExecutionContext
83+
);
84+
}
85+
86+
private serialize(input: CreateEdgeDeploymentStageCommandInput, context: __SerdeContext): Promise<__HttpRequest> {
87+
return serializeAws_json1_1CreateEdgeDeploymentStageCommand(input, context);
88+
}
89+
90+
private deserialize(
91+
output: __HttpResponse,
92+
context: __SerdeContext
93+
): Promise<CreateEdgeDeploymentStageCommandOutput> {
94+
return deserializeAws_json1_1CreateEdgeDeploymentStageCommand(output, context);
95+
}
96+
97+
// Start section: command_body_extra
98+
// End section: command_body_extra
99+
}

clients/client-sagemaker/src/commands/CreateHyperParameterTuningJobCommand.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ import {
1212
SerdeContext as __SerdeContext,
1313
} from "@aws-sdk/types";
1414

15-
import { CreateHyperParameterTuningJobRequest, CreateHyperParameterTuningJobResponse } from "../models/models_0";
15+
import { CreateHyperParameterTuningJobRequest, CreateHyperParameterTuningJobResponse } from "../models/models_1";
1616
import {
1717
deserializeAws_json1_1CreateHyperParameterTuningJobCommand,
1818
serializeAws_json1_1CreateHyperParameterTuningJobCommand,

0 commit comments

Comments
 (0)