Skip to content

Commit 0a6ad34

Browse files
cmgrotecmgrote
authored
Merge pull request #2347 from atlanhq/sync-models-20260323-091238
Sync Java SDK models from atlanhq/models
2 parents b7c5298 + f5f8b73 commit 0a6ad34

13 files changed

Lines changed: 155 additions & 0 deletions

‎sdk/src/main/java/com/atlan/model/assets/AIModelVersion.java‎

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
import com.atlan.exception.ErrorCode;
88
import com.atlan.exception.InvalidRequestException;
99
import com.atlan.exception.NotFoundException;
10+
import com.atlan.model.enums.AIModelVersionStage;
1011
import com.atlan.model.enums.AtlanAnnouncementType;
1112
import com.atlan.model.enums.CertificateStatus;
1213
import com.atlan.model.enums.EthicalAIAccountabilityConfig;
@@ -59,6 +60,19 @@ public class AIModelVersion extends Asset implements IAIModelVersion, IAI, ICata
5960
@Attribute
6061
IAIModel aiModel;
6162

63+
/** Unique name of the AI model to which this version belongs, used to navigate from a version back to its parent model. */
64+
@Attribute
65+
String aiModelQualifiedName;
66+
67+
/** Evaluation and performance metrics recorded for this AI model version, stored as key-value pairs (e.g. accuracy, F1 score, precision, recall). */
68+
@Attribute
69+
@Singular
70+
Map<String, String> aiModelVersionMetrics;
71+
72+
/** Lifecycle deployment stage of this AI model version, indicating its readiness for production use. */
73+
@Attribute
74+
AIModelVersionStage aiModelVersionStage;
75+
6276
/** Accountability configuration for ensuring the ethical use of an AI asset */
6377
@Attribute
6478
EthicalAIAccountabilityConfig ethicalAIAccountabilityConfig;

‎sdk/src/main/java/com/atlan/model/assets/DatabricksAIModelVersion.java‎

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
import com.atlan.exception.ErrorCode;
88
import com.atlan.exception.InvalidRequestException;
99
import com.atlan.exception.NotFoundException;
10+
import com.atlan.model.enums.AIModelVersionStage;
1011
import com.atlan.model.enums.AtlanAnnouncementType;
1112
import com.atlan.model.enums.CertificateStatus;
1213
import com.atlan.model.enums.EthicalAIAccountabilityConfig;
@@ -68,6 +69,19 @@ public class DatabricksAIModelVersion extends Asset
6869
@Attribute
6970
IAIModel aiModel;
7071

72+
/** Unique name of the AI model to which this version belongs, used to navigate from a version back to its parent model. */
73+
@Attribute
74+
String aiModelQualifiedName;
75+
76+
/** Evaluation and performance metrics recorded for this AI model version, stored as key-value pairs (e.g. accuracy, F1 score, precision, recall). */
77+
@Attribute
78+
@Singular
79+
Map<String, String> aiModelVersionMetrics;
80+
81+
/** Lifecycle deployment stage of this AI model version, indicating its readiness for production use. */
82+
@Attribute
83+
AIModelVersionStage aiModelVersionStage;
84+
7185
/** Simple name of the calculation view in which this SQL asset exists, or empty if it does not exist within a calculation view. */
7286
@Attribute
7387
String calculationViewName;

‎sdk/src/main/java/com/atlan/model/assets/IAIModelVersion.java‎

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
Copyright 2023 Atlan Pte. Ltd. */
33
package com.atlan.model.assets;
44

5+
import com.atlan.model.enums.AIModelVersionStage;
56
import com.atlan.model.enums.AssetDQRunStatus;
67
import com.atlan.model.enums.AtlanAnnouncementType;
78
import com.atlan.model.enums.AtlanConnectorType;
@@ -20,6 +21,7 @@
2021
import com.atlan.model.enums.EthicalAIReliabilityAndSafetyConfig;
2122
import com.atlan.model.enums.EthicalAITransparencyConfig;
2223
import com.atlan.model.enums.SourceCostUnitType;
24+
import com.atlan.model.fields.KeywordField;
2325
import com.atlan.model.fields.RelationField;
2426
import com.atlan.model.relations.RelationshipAttributes;
2527
import com.atlan.model.relations.UniqueAttributes;
@@ -50,6 +52,15 @@ public interface IAIModelVersion {
5052
/** Model containing the versions. */
5153
RelationField AI_MODEL = new RelationField("aiModel");
5254

55+
/** Unique name of the AI model to which this version belongs, used to navigate from a version back to its parent model. */
56+
KeywordField AI_MODEL_QUALIFIED_NAME = new KeywordField("aiModelQualifiedName", "aiModelQualifiedName");
57+
58+
/** Evaluation and performance metrics recorded for this AI model version, stored as key-value pairs (e.g. accuracy, F1 score, precision, recall). */
59+
KeywordField AI_MODEL_VERSION_METRICS = new KeywordField("aiModelVersionMetrics", "aiModelVersionMetrics");
60+
61+
/** Lifecycle deployment stage of this AI model version, indicating its readiness for production use. */
62+
KeywordField AI_MODEL_VERSION_STAGE = new KeywordField("aiModelVersionStage", "aiModelVersionStage");
63+
5364
/** List of groups who administer this asset. (This is only used for certain asset types.) */
5465
SortedSet<String> getAdminGroups();
5566

@@ -64,6 +75,15 @@ default IAIModel getAiModel() {
6475
return null;
6576
}
6677

78+
/** Unique name of the AI model to which this version belongs, used to navigate from a version back to its parent model. */
79+
String getAiModelQualifiedName();
80+
81+
/** Evaluation and performance metrics recorded for this AI model version, stored as key-value pairs (e.g. accuracy, F1 score, precision, recall). */
82+
Map<String, String> getAiModelVersionMetrics();
83+
84+
/** Lifecycle deployment stage of this AI model version, indicating its readiness for production use. */
85+
AIModelVersionStage getAiModelVersionStage();
86+
6787
/** Detailed message to include in the announcement on this asset. */
6888
String getAnnouncementMessage();
6989

‎sdk/src/main/java/com/atlan/model/assets/IDatabricksAIModelVersion.java‎

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
Copyright 2023 Atlan Pte. Ltd. */
33
package com.atlan.model.assets;
44

5+
import com.atlan.model.enums.AIModelVersionStage;
56
import com.atlan.model.enums.AssetDQRunStatus;
67
import com.atlan.model.enums.AtlanAnnouncementType;
78
import com.atlan.model.enums.AtlanConnectorType;
@@ -115,6 +116,15 @@ default IAIModel getAiModel() {
115116
return null;
116117
}
117118

119+
/** Unique name of the AI model to which this version belongs, used to navigate from a version back to its parent model. */
120+
String getAiModelQualifiedName();
121+
122+
/** Evaluation and performance metrics recorded for this AI model version, stored as key-value pairs (e.g. accuracy, F1 score, precision, recall). */
123+
Map<String, String> getAiModelVersionMetrics();
124+
125+
/** Lifecycle deployment stage of this AI model version, indicating its readiness for production use. */
126+
AIModelVersionStage getAiModelVersionStage();
127+
118128
/** Detailed message to include in the announcement on this asset. */
119129
String getAnnouncementMessage();
120130

‎sdk/src/main/java/com/atlan/model/assets/ISageMakerModel.java‎

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
Copyright 2023 Atlan Pte. Ltd. */
33
package com.atlan.model.assets;
44

5+
import com.atlan.model.enums.AIModelVersionStage;
56
import com.atlan.model.enums.AssetDQRunStatus;
67
import com.atlan.model.enums.AtlanAnnouncementType;
78
import com.atlan.model.enums.AtlanConnectorType;
@@ -91,6 +92,15 @@ default IAIModel getAiModel() {
9192
return null;
9293
}
9394

95+
/** Unique name of the AI model to which this version belongs, used to navigate from a version back to its parent model. */
96+
String getAiModelQualifiedName();
97+
98+
/** Evaluation and performance metrics recorded for this AI model version, stored as key-value pairs (e.g. accuracy, F1 score, precision, recall). */
99+
Map<String, String> getAiModelVersionMetrics();
100+
101+
/** Lifecycle deployment stage of this AI model version, indicating its readiness for production use. */
102+
AIModelVersionStage getAiModelVersionStage();
103+
94104
/** Detailed message to include in the announcement on this asset. */
95105
String getAnnouncementMessage();
96106

‎sdk/src/main/java/com/atlan/model/assets/ISnowflakeAIModelVersion.java‎

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
Copyright 2023 Atlan Pte. Ltd. */
33
package com.atlan.model.assets;
44

5+
import com.atlan.model.enums.AIModelVersionStage;
56
import com.atlan.model.enums.AssetDQRunStatus;
67
import com.atlan.model.enums.AtlanAnnouncementType;
78
import com.atlan.model.enums.AtlanConnectorType;
@@ -85,6 +86,15 @@ default IAIModel getAiModel() {
8586
return null;
8687
}
8788

89+
/** Unique name of the AI model to which this version belongs, used to navigate from a version back to its parent model. */
90+
String getAiModelQualifiedName();
91+
92+
/** Evaluation and performance metrics recorded for this AI model version, stored as key-value pairs (e.g. accuracy, F1 score, precision, recall). */
93+
Map<String, String> getAiModelVersionMetrics();
94+
95+
/** Lifecycle deployment stage of this AI model version, indicating its readiness for production use. */
96+
AIModelVersionStage getAiModelVersionStage();
97+
8898
/** Detailed message to include in the announcement on this asset. */
8999
String getAnnouncementMessage();
90100

‎sdk/src/main/java/com/atlan/model/assets/SageMakerModel.java‎

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
import com.atlan.exception.ErrorCode;
88
import com.atlan.exception.InvalidRequestException;
99
import com.atlan.exception.NotFoundException;
10+
import com.atlan.model.enums.AIModelVersionStage;
1011
import com.atlan.model.enums.AtlanAnnouncementType;
1112
import com.atlan.model.enums.CertificateStatus;
1213
import com.atlan.model.enums.EthicalAIAccountabilityConfig;
@@ -61,6 +62,19 @@ public class SageMakerModel extends Asset
6162
@Attribute
6263
IAIModel aiModel;
6364

65+
/** Unique name of the AI model to which this version belongs, used to navigate from a version back to its parent model. */
66+
@Attribute
67+
String aiModelQualifiedName;
68+
69+
/** Evaluation and performance metrics recorded for this AI model version, stored as key-value pairs (e.g. accuracy, F1 score, precision, recall). */
70+
@Attribute
71+
@Singular
72+
Map<String, String> aiModelVersionMetrics;
73+
74+
/** Lifecycle deployment stage of this AI model version, indicating its readiness for production use. */
75+
@Attribute
76+
AIModelVersionStage aiModelVersionStage;
77+
6478
/** 12-digit number that uniquely identifies an AWS account. */
6579
@Attribute
6680
String awsAccountId;

‎sdk/src/main/java/com/atlan/model/assets/SnowflakeAIModelVersion.java‎

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
import com.atlan.exception.ErrorCode;
88
import com.atlan.exception.InvalidRequestException;
99
import com.atlan.exception.NotFoundException;
10+
import com.atlan.model.enums.AIModelVersionStage;
1011
import com.atlan.model.enums.AtlanAnnouncementType;
1112
import com.atlan.model.enums.CertificateStatus;
1213
import com.atlan.model.enums.EthicalAIAccountabilityConfig;
@@ -60,6 +61,19 @@ public class SnowflakeAIModelVersion extends Asset
6061
@Attribute
6162
IAIModel aiModel;
6263

64+
/** Unique name of the AI model to which this version belongs, used to navigate from a version back to its parent model. */
65+
@Attribute
66+
String aiModelQualifiedName;
67+
68+
/** Evaluation and performance metrics recorded for this AI model version, stored as key-value pairs (e.g. accuracy, F1 score, precision, recall). */
69+
@Attribute
70+
@Singular
71+
Map<String, String> aiModelVersionMetrics;
72+
73+
/** Lifecycle deployment stage of this AI model version, indicating its readiness for production use. */
74+
@Attribute
75+
AIModelVersionStage aiModelVersionStage;
76+
6377
/** Simple name of the calculation view in which this SQL asset exists, or empty if it does not exist within a calculation view. */
6478
@Attribute
6579
String calculationViewName;

‎sdk/src/main/java/com/atlan/model/enums/AIModelVersionStage.java‎

Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
1+
/* SPDX-License-Identifier: Apache-2.0
2+
Copyright 2022 Atlan Pte. Ltd. */
3+
package com.atlan.model.enums;
4+
5+
import com.fasterxml.jackson.annotation.JsonValue;
6+
import javax.annotation.processing.Generated;
7+
import lombok.Getter;
8+
9+
@Generated(value = "com.atlan.generators.ModelGeneratorV2")
10+
public enum AIModelVersionStage implements AtlanEnum {
11+
NONE("NONE"),
12+
STAGING("STAGING"),
13+
PRODUCTION("PRODUCTION"),
14+
ARCHIVED("ARCHIVED"),
15+
;
16+
17+
@JsonValue
18+
@Getter(onMethod_ = {@Override})
19+
private final String value;
20+
21+
AIModelVersionStage(String value) {
22+
this.value = value;
23+
}
24+
25+
public static AIModelVersionStage fromValue(String value) {
26+
for (AIModelVersionStage b : AIModelVersionStage.values()) {
27+
if (b.value.equals(value)) {
28+
return b;
29+
}
30+
}
31+
return null;
32+
}
33+
}

‎sdk/src/test/java/com/atlan/model/assets/AIModelVersionTest.java‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -677,6 +677,10 @@ public class AIModelVersionTest {
677677
.assetMetadataFormRevision("String1")
678678
.assetMetadataFormFields(List.of(Map.of("key1", "value1"), Map.of("key2", "value2")))
679679
.build())
680+
.aiModelQualifiedName("String0")
681+
.aiModelVersionStage(AIModelVersionStage.NONE)
682+
.aiModelVersionMetric("String0", "String0")
683+
.aiModelVersionMetric("String1", "String1")
680684
.aiModel(AIModel.refByGuid("705d96f4-bdb6-4792-8dfe-8dc4ca3d2c23"))
681685
.assignedTerm(GlossaryTerm.refByGuid("705d96f4-bdb6-4792-8dfe-8dc4ca3d2c23"))
682686
.assignedTerm(GlossaryTerm.refByQualifiedName("default/snowflake/1234567890/test/qualifiedName"))

0 commit comments

Comments
 (0)