Skip to content

Commit 1bd75f0

Browse files
author
awstools
committed
feat(client-datazone): This release adds support for MLflow connections Creation in DataZone
1 parent 610cb93 commit 1bd75f0

File tree

11 files changed

+432
-167
lines changed

11 files changed

+432
-167
lines changed

clients/client-datazone/src/commands/CreateConnectionCommand.ts

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -184,6 +184,10 @@ export interface CreateConnectionCommandOutput extends CreateConnectionOutput, _
184184
* profileArn: "STRING_VALUE",
185185
* authMode: "STRING_VALUE",
186186
* },
187+
* mlflowProperties: { // MlflowPropertiesInput
188+
* trackingServerName: "STRING_VALUE",
189+
* trackingServerArn: "STRING_VALUE",
190+
* },
187191
* },
188192
* enableTrustedIdentityPropagation: true || false,
189193
* scope: "DOMAIN" || "PROJECT",
@@ -209,7 +213,7 @@ export interface CreateConnectionCommandOutput extends CreateConnectionOutput, _
209213
* // glueConnection: { // GlueConnection
210214
* // name: "STRING_VALUE",
211215
* // description: "STRING_VALUE",
212-
* // connectionType: "ATHENA" || "BIGQUERY" || "DATABRICKS" || "DOCUMENTDB" || "DYNAMODB" || "HYPERPOD" || "IAM" || "MYSQL" || "OPENSEARCH" || "ORACLE" || "POSTGRESQL" || "REDSHIFT" || "S3" || "SAPHANA" || "SNOWFLAKE" || "SPARK" || "SQLSERVER" || "TERADATA" || "VERTICA" || "WORKFLOWS_MWAA" || "AMAZON_Q",
216+
* // connectionType: "ATHENA" || "BIGQUERY" || "DATABRICKS" || "DOCUMENTDB" || "DYNAMODB" || "HYPERPOD" || "IAM" || "MYSQL" || "OPENSEARCH" || "ORACLE" || "POSTGRESQL" || "REDSHIFT" || "S3" || "SAPHANA" || "SNOWFLAKE" || "SPARK" || "SQLSERVER" || "TERADATA" || "VERTICA" || "WORKFLOWS_MWAA" || "AMAZON_Q" || "MLFLOW",
213217
* // matchCriteria: [ // MatchCriteria
214218
* // "STRING_VALUE",
215219
* // ],
@@ -367,8 +371,12 @@ export interface CreateConnectionCommandOutput extends CreateConnectionOutput, _
367371
* // profileArn: "STRING_VALUE",
368372
* // authMode: "STRING_VALUE",
369373
* // },
374+
* // mlflowProperties: { // MlflowPropertiesOutput
375+
* // trackingServerName: "STRING_VALUE",
376+
* // trackingServerArn: "STRING_VALUE",
377+
* // },
370378
* // },
371-
* // type: "ATHENA" || "BIGQUERY" || "DATABRICKS" || "DOCUMENTDB" || "DYNAMODB" || "HYPERPOD" || "IAM" || "MYSQL" || "OPENSEARCH" || "ORACLE" || "POSTGRESQL" || "REDSHIFT" || "S3" || "SAPHANA" || "SNOWFLAKE" || "SPARK" || "SQLSERVER" || "TERADATA" || "VERTICA" || "WORKFLOWS_MWAA" || "AMAZON_Q", // required
379+
* // type: "ATHENA" || "BIGQUERY" || "DATABRICKS" || "DOCUMENTDB" || "DYNAMODB" || "HYPERPOD" || "IAM" || "MYSQL" || "OPENSEARCH" || "ORACLE" || "POSTGRESQL" || "REDSHIFT" || "S3" || "SAPHANA" || "SNOWFLAKE" || "SPARK" || "SQLSERVER" || "TERADATA" || "VERTICA" || "WORKFLOWS_MWAA" || "AMAZON_Q" || "MLFLOW", // required
372380
* // scope: "DOMAIN" || "PROJECT",
373381
* // };
374382
*

clients/client-datazone/src/commands/GetConnectionCommand.ts

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ export interface GetConnectionCommandOutput extends GetConnectionOutput, __Metad
7070
* // glueConnection: { // GlueConnection
7171
* // name: "STRING_VALUE",
7272
* // description: "STRING_VALUE",
73-
* // connectionType: "ATHENA" || "BIGQUERY" || "DATABRICKS" || "DOCUMENTDB" || "DYNAMODB" || "HYPERPOD" || "IAM" || "MYSQL" || "OPENSEARCH" || "ORACLE" || "POSTGRESQL" || "REDSHIFT" || "S3" || "SAPHANA" || "SNOWFLAKE" || "SPARK" || "SQLSERVER" || "TERADATA" || "VERTICA" || "WORKFLOWS_MWAA" || "AMAZON_Q",
73+
* // connectionType: "ATHENA" || "BIGQUERY" || "DATABRICKS" || "DOCUMENTDB" || "DYNAMODB" || "HYPERPOD" || "IAM" || "MYSQL" || "OPENSEARCH" || "ORACLE" || "POSTGRESQL" || "REDSHIFT" || "S3" || "SAPHANA" || "SNOWFLAKE" || "SPARK" || "SQLSERVER" || "TERADATA" || "VERTICA" || "WORKFLOWS_MWAA" || "AMAZON_Q" || "MLFLOW",
7474
* // matchCriteria: [ // MatchCriteria
7575
* // "STRING_VALUE",
7676
* // ],
@@ -228,8 +228,12 @@ export interface GetConnectionCommandOutput extends GetConnectionOutput, __Metad
228228
* // profileArn: "STRING_VALUE",
229229
* // authMode: "STRING_VALUE",
230230
* // },
231+
* // mlflowProperties: { // MlflowPropertiesOutput
232+
* // trackingServerName: "STRING_VALUE",
233+
* // trackingServerArn: "STRING_VALUE",
234+
* // },
231235
* // },
232-
* // type: "ATHENA" || "BIGQUERY" || "DATABRICKS" || "DOCUMENTDB" || "DYNAMODB" || "HYPERPOD" || "IAM" || "MYSQL" || "OPENSEARCH" || "ORACLE" || "POSTGRESQL" || "REDSHIFT" || "S3" || "SAPHANA" || "SNOWFLAKE" || "SPARK" || "SQLSERVER" || "TERADATA" || "VERTICA" || "WORKFLOWS_MWAA" || "AMAZON_Q", // required
236+
* // type: "ATHENA" || "BIGQUERY" || "DATABRICKS" || "DOCUMENTDB" || "DYNAMODB" || "HYPERPOD" || "IAM" || "MYSQL" || "OPENSEARCH" || "ORACLE" || "POSTGRESQL" || "REDSHIFT" || "S3" || "SAPHANA" || "SNOWFLAKE" || "SPARK" || "SQLSERVER" || "TERADATA" || "VERTICA" || "WORKFLOWS_MWAA" || "AMAZON_Q" || "MLFLOW", // required
233237
* // scope: "DOMAIN" || "PROJECT",
234238
* // };
235239
*

clients/client-datazone/src/commands/ListConnectionsCommand.ts

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ export interface ListConnectionsCommandOutput extends ListConnectionsOutput, __M
5050
* name: "STRING_VALUE",
5151
* environmentIdentifier: "STRING_VALUE",
5252
* projectIdentifier: "STRING_VALUE",
53-
* type: "ATHENA" || "BIGQUERY" || "DATABRICKS" || "DOCUMENTDB" || "DYNAMODB" || "HYPERPOD" || "IAM" || "MYSQL" || "OPENSEARCH" || "ORACLE" || "POSTGRESQL" || "REDSHIFT" || "S3" || "SAPHANA" || "SNOWFLAKE" || "SPARK" || "SQLSERVER" || "TERADATA" || "VERTICA" || "WORKFLOWS_MWAA" || "AMAZON_Q",
53+
* type: "ATHENA" || "BIGQUERY" || "DATABRICKS" || "DOCUMENTDB" || "DYNAMODB" || "HYPERPOD" || "IAM" || "MYSQL" || "OPENSEARCH" || "ORACLE" || "POSTGRESQL" || "REDSHIFT" || "S3" || "SAPHANA" || "SNOWFLAKE" || "SPARK" || "SQLSERVER" || "TERADATA" || "VERTICA" || "WORKFLOWS_MWAA" || "AMAZON_Q" || "MLFLOW",
5454
* scope: "DOMAIN" || "PROJECT",
5555
* };
5656
* const command = new ListConnectionsCommand(input);
@@ -75,7 +75,7 @@ export interface ListConnectionsCommandOutput extends ListConnectionsOutput, __M
7575
* // glueConnection: { // GlueConnection
7676
* // name: "STRING_VALUE",
7777
* // description: "STRING_VALUE",
78-
* // connectionType: "ATHENA" || "BIGQUERY" || "DATABRICKS" || "DOCUMENTDB" || "DYNAMODB" || "HYPERPOD" || "IAM" || "MYSQL" || "OPENSEARCH" || "ORACLE" || "POSTGRESQL" || "REDSHIFT" || "S3" || "SAPHANA" || "SNOWFLAKE" || "SPARK" || "SQLSERVER" || "TERADATA" || "VERTICA" || "WORKFLOWS_MWAA" || "AMAZON_Q",
78+
* // connectionType: "ATHENA" || "BIGQUERY" || "DATABRICKS" || "DOCUMENTDB" || "DYNAMODB" || "HYPERPOD" || "IAM" || "MYSQL" || "OPENSEARCH" || "ORACLE" || "POSTGRESQL" || "REDSHIFT" || "S3" || "SAPHANA" || "SNOWFLAKE" || "SPARK" || "SQLSERVER" || "TERADATA" || "VERTICA" || "WORKFLOWS_MWAA" || "AMAZON_Q" || "MLFLOW",
7979
* // matchCriteria: [ // MatchCriteria
8080
* // "STRING_VALUE",
8181
* // ],
@@ -233,8 +233,12 @@ export interface ListConnectionsCommandOutput extends ListConnectionsOutput, __M
233233
* // profileArn: "STRING_VALUE",
234234
* // authMode: "STRING_VALUE",
235235
* // },
236+
* // mlflowProperties: { // MlflowPropertiesOutput
237+
* // trackingServerName: "STRING_VALUE",
238+
* // trackingServerArn: "STRING_VALUE",
239+
* // },
236240
* // },
237-
* // type: "ATHENA" || "BIGQUERY" || "DATABRICKS" || "DOCUMENTDB" || "DYNAMODB" || "HYPERPOD" || "IAM" || "MYSQL" || "OPENSEARCH" || "ORACLE" || "POSTGRESQL" || "REDSHIFT" || "S3" || "SAPHANA" || "SNOWFLAKE" || "SPARK" || "SQLSERVER" || "TERADATA" || "VERTICA" || "WORKFLOWS_MWAA" || "AMAZON_Q", // required
241+
* // type: "ATHENA" || "BIGQUERY" || "DATABRICKS" || "DOCUMENTDB" || "DYNAMODB" || "HYPERPOD" || "IAM" || "MYSQL" || "OPENSEARCH" || "ORACLE" || "POSTGRESQL" || "REDSHIFT" || "S3" || "SAPHANA" || "SNOWFLAKE" || "SPARK" || "SQLSERVER" || "TERADATA" || "VERTICA" || "WORKFLOWS_MWAA" || "AMAZON_Q" || "MLFLOW", // required
238242
* // scope: "DOMAIN" || "PROJECT",
239243
* // },
240244
* // ],

clients/client-datazone/src/commands/ListSubscriptionTargetsCommand.ts

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -6,11 +6,8 @@ import { MetadataBearer as __MetadataBearer } from "@smithy/types";
66

77
import { DataZoneClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../DataZoneClient";
88
import { commonParams } from "../endpoint/EndpointParameters";
9-
import {
10-
ListSubscriptionTargetsInput,
11-
ListSubscriptionTargetsOutput,
12-
ListSubscriptionTargetsOutputFilterSensitiveLog,
13-
} from "../models/models_1";
9+
import { ListSubscriptionTargetsInput } from "../models/models_1";
10+
import { ListSubscriptionTargetsOutput, ListSubscriptionTargetsOutputFilterSensitiveLog } from "../models/models_2";
1411
import { de_ListSubscriptionTargetsCommand, se_ListSubscriptionTargetsCommand } from "../protocols/Aws_restJson1";
1512

1613
/**

clients/client-datazone/src/commands/ListTagsForResourceCommand.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ import { MetadataBearer as __MetadataBearer } from "@smithy/types";
66

77
import { DataZoneClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../DataZoneClient";
88
import { commonParams } from "../endpoint/EndpointParameters";
9-
import { ListTagsForResourceRequest, ListTagsForResourceResponse } from "../models/models_1";
9+
import { ListTagsForResourceRequest, ListTagsForResourceResponse } from "../models/models_2";
1010
import { de_ListTagsForResourceCommand, se_ListTagsForResourceCommand } from "../protocols/Aws_restJson1";
1111

1212
/**

clients/client-datazone/src/commands/UpdateConnectionCommand.ts

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -115,6 +115,10 @@ export interface UpdateConnectionCommandOutput extends UpdateConnectionOutput, _
115115
* profileArn: "STRING_VALUE",
116116
* authMode: "STRING_VALUE",
117117
* },
118+
* mlflowProperties: { // MlflowPropertiesPatch
119+
* trackingServerName: "STRING_VALUE",
120+
* trackingServerArn: "STRING_VALUE",
121+
* },
118122
* },
119123
* };
120124
* const command = new UpdateConnectionCommand(input);
@@ -138,7 +142,7 @@ export interface UpdateConnectionCommandOutput extends UpdateConnectionOutput, _
138142
* // glueConnection: { // GlueConnection
139143
* // name: "STRING_VALUE",
140144
* // description: "STRING_VALUE",
141-
* // connectionType: "ATHENA" || "BIGQUERY" || "DATABRICKS" || "DOCUMENTDB" || "DYNAMODB" || "HYPERPOD" || "IAM" || "MYSQL" || "OPENSEARCH" || "ORACLE" || "POSTGRESQL" || "REDSHIFT" || "S3" || "SAPHANA" || "SNOWFLAKE" || "SPARK" || "SQLSERVER" || "TERADATA" || "VERTICA" || "WORKFLOWS_MWAA" || "AMAZON_Q",
145+
* // connectionType: "ATHENA" || "BIGQUERY" || "DATABRICKS" || "DOCUMENTDB" || "DYNAMODB" || "HYPERPOD" || "IAM" || "MYSQL" || "OPENSEARCH" || "ORACLE" || "POSTGRESQL" || "REDSHIFT" || "S3" || "SAPHANA" || "SNOWFLAKE" || "SPARK" || "SQLSERVER" || "TERADATA" || "VERTICA" || "WORKFLOWS_MWAA" || "AMAZON_Q" || "MLFLOW",
142146
* // matchCriteria: [ // MatchCriteria
143147
* // "STRING_VALUE",
144148
* // ],
@@ -296,8 +300,12 @@ export interface UpdateConnectionCommandOutput extends UpdateConnectionOutput, _
296300
* // profileArn: "STRING_VALUE",
297301
* // authMode: "STRING_VALUE",
298302
* // },
303+
* // mlflowProperties: { // MlflowPropertiesOutput
304+
* // trackingServerName: "STRING_VALUE",
305+
* // trackingServerArn: "STRING_VALUE",
306+
* // },
299307
* // },
300-
* // type: "ATHENA" || "BIGQUERY" || "DATABRICKS" || "DOCUMENTDB" || "DYNAMODB" || "HYPERPOD" || "IAM" || "MYSQL" || "OPENSEARCH" || "ORACLE" || "POSTGRESQL" || "REDSHIFT" || "S3" || "SAPHANA" || "SNOWFLAKE" || "SPARK" || "SQLSERVER" || "TERADATA" || "VERTICA" || "WORKFLOWS_MWAA" || "AMAZON_Q", // required
308+
* // type: "ATHENA" || "BIGQUERY" || "DATABRICKS" || "DOCUMENTDB" || "DYNAMODB" || "HYPERPOD" || "IAM" || "MYSQL" || "OPENSEARCH" || "ORACLE" || "POSTGRESQL" || "REDSHIFT" || "S3" || "SAPHANA" || "SNOWFLAKE" || "SPARK" || "SQLSERVER" || "TERADATA" || "VERTICA" || "WORKFLOWS_MWAA" || "AMAZON_Q" || "MLFLOW", // required
301309
* // scope: "DOMAIN" || "PROJECT",
302310
* // };
303311
*

0 commit comments

Comments
 (0)