acryl-datahub-cloud 0.3.10.4rc2__py3-none-any.whl → 0.3.11__py3-none-any.whl

acryl_datahub_cloud/metadata/schema_classes.py

@@ -1756,37 +1756,24 @@ class AnomalySourcePropertiesClass(DictWrapper):
     
     RECORD_SCHEMA = get_schema_type("com.linkedin.pegasus2avro.anomaly.AnomalySourceProperties")
     def __init__(self,
-        assertionRunEventTime: Union[None, int]=None,
-        metricCubeTimestamp: Union[None, int]=None,
+        assertionMetric: Union[None, "AssertionMetricClass"]=None,
     ):
         super().__init__()
         
-        self.assertionRunEventTime = assertionRunEventTime
-        self.metricCubeTimestamp = metricCubeTimestamp
+        self.assertionMetric = assertionMetric
     
     def _restore_defaults(self) -> None:
-        self.assertionRunEventTime = self.RECORD_SCHEMA.fields_dict["assertionRunEventTime"].default
-        self.metricCubeTimestamp = self.RECORD_SCHEMA.fields_dict["metricCubeTimestamp"].default
+        self.assertionMetric = self.RECORD_SCHEMA.fields_dict["assertionMetric"].default
     
     
     @property
-    def assertionRunEventTime(self) -> Union[None, int]:
-        """The timestampMillis field of the AssertionRunEvent which altered the anomaly status the anomaly (if applicable)."""
-        return self._inner_dict.get('assertionRunEventTime')  # type: ignore
+    def assertionMetric(self) -> Union[None, "AssertionMetricClass"]:
+        """The monitor metric associated with the anomaly, if generated from an assertion monitor (the norm)."""
+        return self._inner_dict.get('assertionMetric')  # type: ignore
     
-    @assertionRunEventTime.setter
-    def assertionRunEventTime(self, value: Union[None, int]) -> None:
-        self._inner_dict['assertionRunEventTime'] = value
-    
-    
-    @property
-    def metricCubeTimestamp(self) -> Union[None, int]:
-        """The timestamp associated with the metric cube value that triggered the anomaly."""
-        return self._inner_dict.get('metricCubeTimestamp')  # type: ignore
-    
-    @metricCubeTimestamp.setter
-    def metricCubeTimestamp(self, value: Union[None, int]) -> None:
-        self._inner_dict['metricCubeTimestamp'] = value
+    @assertionMetric.setter
+    def assertionMetric(self, value: Union[None, "AssertionMetricClass"]) -> None:
+        self._inner_dict['assertionMetric'] = value
     
     
 class AnomalySourceTypeClass(object):
@@ -1808,13 +1795,13 @@ class MonitorAnomalyEventClass(_Aspect):
 
     def __init__(self,
         timestampMillis: int,
-        state: Union[str, "AnomalyReviewStateClass"],
         source: "AnomalySourceClass",
         created: "TimeStampClass",
         lastUpdated: "TimeStampClass",
         eventGranularity: Union[None, "TimeWindowSizeClass"]=None,
         partitionSpec: Optional[Union["PartitionSpecClass", None]]=None,
         messageId: Union[None, str]=None,
+        state: Union[None, Union[str, "AnomalyReviewStateClass"]]=None,
     ):
         super().__init__()
         
@@ -1836,7 +1823,7 @@ class MonitorAnomalyEventClass(_Aspect):
         self.eventGranularity = self.RECORD_SCHEMA.fields_dict["eventGranularity"].default
         self.partitionSpec = _json_converter.from_json_object(self.RECORD_SCHEMA.fields_dict["partitionSpec"].default, writers_schema=self.RECORD_SCHEMA.fields_dict["partitionSpec"].type)
         self.messageId = self.RECORD_SCHEMA.fields_dict["messageId"].default
-        self.state = AnomalyReviewStateClass.CONFIRMED
+        self.state = self.RECORD_SCHEMA.fields_dict["state"].default
         self.source = AnomalySourceClass._construct_with_defaults()
         self.created = TimeStampClass._construct_with_defaults()
         self.lastUpdated = TimeStampClass._construct_with_defaults()
@@ -1883,13 +1870,13 @@ class MonitorAnomalyEventClass(_Aspect):
     
     
     @property
-    def state(self) -> Union[str, "AnomalyReviewStateClass"]:
+    def state(self) -> Union[None, Union[str, "AnomalyReviewStateClass"]]:
         """The review of the anomaly, based on human-provided feedback.
     If this is not present, then the Anomaly has not yet been reviewed."""
         return self._inner_dict.get('state')  # type: ignore
     
     @state.setter
-    def state(self, value: Union[str, "AnomalyReviewStateClass"]) -> None:
+    def state(self, value: Union[None, Union[str, "AnomalyReviewStateClass"]]) -> None:
         self._inner_dict['state'] = value
     
     
@@ -2995,6 +2982,44 @@ class AssertionInfoClass(_Aspect):
         self._inner_dict['description'] = value
     
     
+class AssertionMetricClass(DictWrapper):
+    # No docs available.
+    
+    RECORD_SCHEMA = get_schema_type("com.linkedin.pegasus2avro.assertion.AssertionMetric")
+    def __init__(self,
+        timestampMs: int,
+        value: float,
+    ):
+        super().__init__()
+        
+        self.timestampMs = timestampMs
+        self.value = value
+    
+    def _restore_defaults(self) -> None:
+        self.timestampMs = int()
+        self.value = float()
+    
+    
+    @property
+    def timestampMs(self) -> int:
+        """The timestamp associated with the metric sampling time in milliseconds since epoch"""
+        return self._inner_dict.get('timestampMs')  # type: ignore
+    
+    @timestampMs.setter
+    def timestampMs(self, value: int) -> None:
+        self._inner_dict['timestampMs'] = value
+    
+    
+    @property
+    def value(self) -> float:
+        """The value of the metric that was sampled"""
+        return self._inner_dict.get('value')  # type: ignore
+    
+    @value.setter
+    def value(self, value: float) -> None:
+        self._inner_dict['value'] = value
+    
+    
 class AssertionMonitorSensitivityClass(DictWrapper):
     """Assertion monitor sensitivity is a measure of how sensitive the assertion monitor is to the assertion condition."""
     
@@ -3037,6 +3062,7 @@ class AssertionResultClass(DictWrapper):
         error: Union[None, "AssertionResultErrorClass"]=None,
         parameters: Union[None, "AssertionEvaluationParametersClass"]=None,
         assertionInferenceDetails: Union[None, "AssertionInferenceDetailsClass"]=None,
+        metric: Union[None, "AssertionMetricClass"]=None,
     ):
         super().__init__()
         
@@ -3052,6 +3078,7 @@ class AssertionResultClass(DictWrapper):
         self.error = error
         self.parameters = parameters
         self.assertionInferenceDetails = assertionInferenceDetails
+        self.metric = metric
     
     def _restore_defaults(self) -> None:
         self.assertion = self.RECORD_SCHEMA.fields_dict["assertion"].default
@@ -3066,6 +3093,7 @@ class AssertionResultClass(DictWrapper):
         self.error = self.RECORD_SCHEMA.fields_dict["error"].default
         self.parameters = self.RECORD_SCHEMA.fields_dict["parameters"].default
         self.assertionInferenceDetails = self.RECORD_SCHEMA.fields_dict["assertionInferenceDetails"].default
+        self.metric = self.RECORD_SCHEMA.fields_dict["metric"].default
     
     
     @property
@@ -3196,6 +3224,19 @@ class AssertionResultClass(DictWrapper):
         self._inner_dict['assertionInferenceDetails'] = value
     
     
+    @property
+    def metric(self) -> Union[None, "AssertionMetricClass"]:
+        """Information about the metric that was sampled & used when evaluating the assertion.
+    
+    Currently, this is only populated for Volume & Field Metric Assertions and used for anomaly
+    logging for Smart Assertions."""
+        return self._inner_dict.get('metric')  # type: ignore
+    
+    @metric.setter
+    def metric(self, value: Union[None, "AssertionMetricClass"]) -> None:
+        self._inner_dict['metric'] = value
+    
+    
 class AssertionResultErrorClass(DictWrapper):
     """ An error encountered when evaluating an AssertionResult"""
     
@@ -21870,7 +21911,7 @@ class DataContractKeyClass(_Aspect):
 
 
     ASPECT_NAME = 'dataContractKey'
-    ASPECT_INFO = {'keyForEntity': 'dataContract', 'entityCategory': 'core', 'entityAspects': ['dataContractProperties', 'dataContractStatus', 'status']}
+    ASPECT_INFO = {'keyForEntity': 'dataContract', 'entityCategory': 'core', 'entityAspects': ['dataContractProperties', 'dataContractStatus', 'status', 'structuredProperties']}
     RECORD_SCHEMA = get_schema_type("com.linkedin.pegasus2avro.metadata.key.DataContractKey")
 
     def __init__(self,
@@ -22094,6 +22135,35 @@ class DataHubMetricCubeKeyClass(_Aspect):
         self._inner_dict['id'] = value
     
     
+class DataHubOpenAPISchemaKeyClass(_Aspect):
+    """Key for a Query"""
+
+
+    ASPECT_NAME = 'dataHubOpenAPISchemaKey'
+    ASPECT_INFO = {'keyForEntity': 'dataHubOpenAPISchema', 'entityCategory': 'internal', 'entityAspects': ['systemMetadata'], 'entityDoc': 'Contains aspects which are used in OpenAPI requests/responses which are not otherwise present in the data model.'}
+    RECORD_SCHEMA = get_schema_type("com.linkedin.pegasus2avro.metadata.key.DataHubOpenAPISchemaKey")
+
+    def __init__(self,
+        id: str,
+    ):
+        super().__init__()
+        
+        self.id = id
+    
+    def _restore_defaults(self) -> None:
+        self.id = str()
+    
+    
+    @property
+    def id(self) -> str:
+        """A unique id for the DataHub OpenAPI schema."""
+        return self._inner_dict.get('id')  # type: ignore
+    
+    @id.setter
+    def id(self, value: str) -> None:
+        self._inner_dict['id'] = value
+    
+    
 class DataHubPersonaKeyClass(_Aspect):
     """Key for a persona type"""
 
@@ -28041,15 +28111,18 @@ class AssertionMonitorClass(DictWrapper):
     def __init__(self,
         assertions: List["AssertionEvaluationSpecClass"],
         settings: Union[None, "AssertionMonitorSettingsClass"]=None,
+        bootstrapStatus: Union[None, "AssertionMonitorBootstrapStatusClass"]=None,
     ):
         super().__init__()
         
         self.assertions = assertions
         self.settings = settings
+        self.bootstrapStatus = bootstrapStatus
     
     def _restore_defaults(self) -> None:
         self.assertions = list()
         self.settings = self.RECORD_SCHEMA.fields_dict["settings"].default
+        self.bootstrapStatus = self.RECORD_SCHEMA.fields_dict["bootstrapStatus"].default
     
     
     @property
@@ -28072,6 +28145,41 @@ class AssertionMonitorClass(DictWrapper):
         self._inner_dict['settings'] = value
     
     
+    @property
+    def bootstrapStatus(self) -> Union[None, "AssertionMonitorBootstrapStatusClass"]:
+        """The status of the bootstrap actions performed on the assertion."""
+        return self._inner_dict.get('bootstrapStatus')  # type: ignore
+    
+    @bootstrapStatus.setter
+    def bootstrapStatus(self, value: Union[None, "AssertionMonitorBootstrapStatusClass"]) -> None:
+        self._inner_dict['bootstrapStatus'] = value
+    
+    
+class AssertionMonitorBootstrapStatusClass(DictWrapper):
+    # No docs available.
+    
+    RECORD_SCHEMA = get_schema_type("com.linkedin.pegasus2avro.monitor.AssertionMonitorBootstrapStatus")
+    def __init__(self,
+        metricsCubeBootstrapStatus: Union[None, "AssertionMonitorMetricsCubeBootstrapStatusClass"]=None,
+    ):
+        super().__init__()
+        
+        self.metricsCubeBootstrapStatus = metricsCubeBootstrapStatus
+    
+    def _restore_defaults(self) -> None:
+        self.metricsCubeBootstrapStatus = self.RECORD_SCHEMA.fields_dict["metricsCubeBootstrapStatus"].default
+    
+    
+    @property
+    def metricsCubeBootstrapStatus(self) -> Union[None, "AssertionMonitorMetricsCubeBootstrapStatusClass"]:
+        """Whether the metrics cube for this monitor has been bootstrapped."""
+        return self._inner_dict.get('metricsCubeBootstrapStatus')  # type: ignore
+    
+    @metricsCubeBootstrapStatus.setter
+    def metricsCubeBootstrapStatus(self, value: Union[None, "AssertionMonitorMetricsCubeBootstrapStatusClass"]) -> None:
+        self._inner_dict['metricsCubeBootstrapStatus'] = value
+    
+    
 class AssertionMonitorCapabilityClass(object):
     """Individual toggle-able capability for an assertion monitor."""
     
@@ -28090,6 +28198,59 @@ class AssertionMonitorCapabilityClass(object):
     
     
     
+class AssertionMonitorMetricsCubeBootstrapStateClass(object):
+    # No docs available.
+    
+    PENDING = "PENDING"
+    """The metrics cube for this monitor has not been bootstrapped."""
+    
+    FAILED = "FAILED"
+    """The metrics cube for this monitor has failed to bootstrap."""
+    
+    COMPLETED = "COMPLETED"
+    """The metrics cube for this monitor has been bootstrapped."""
+    
+    
+    
+class AssertionMonitorMetricsCubeBootstrapStatusClass(DictWrapper):
+    # No docs available.
+    
+    RECORD_SCHEMA = get_schema_type("com.linkedin.pegasus2avro.monitor.AssertionMonitorMetricsCubeBootstrapStatus")
+    def __init__(self,
+        state: Union[str, "AssertionMonitorMetricsCubeBootstrapStateClass"],
+        message: Union[None, str]=None,
+    ):
+        super().__init__()
+        
+        self.state = state
+        self.message = message
+    
+    def _restore_defaults(self) -> None:
+        self.state = AssertionMonitorMetricsCubeBootstrapStateClass.PENDING
+        self.message = self.RECORD_SCHEMA.fields_dict["message"].default
+    
+    
+    @property
+    def state(self) -> Union[str, "AssertionMonitorMetricsCubeBootstrapStateClass"]:
+        """Whether the metrics cube for this monitor has been bootstrapped."""
+        return self._inner_dict.get('state')  # type: ignore
+    
+    @state.setter
+    def state(self, value: Union[str, "AssertionMonitorMetricsCubeBootstrapStateClass"]) -> None:
+        self._inner_dict['state'] = value
+    
+    
+    @property
+    def message(self) -> Union[None, str]:
+        """The message associated with the bootstrap status.
+    I.e., an error message if the bootstrap failed."""
+        return self._inner_dict.get('message')  # type: ignore
+    
+    @message.setter
+    def message(self, value: Union[None, str]) -> None:
+        self._inner_dict['message'] = value
+    
+    
 class AssertionMonitorSettingsClass(DictWrapper):
     """General purpose settings for the assertion monitor."""
     
@@ -29606,10 +29767,14 @@ class PlatformEventHeaderClass(DictWrapper):
         self._inner_dict['timestampMillis'] = value
     
     
-class SystemMetadataClass(DictWrapper):
+class SystemMetadataClass(_Aspect):
     """Metadata associated with each metadata change that is processed by the system"""
-    
+
+
+    ASPECT_NAME = 'systemMetadata'
+    ASPECT_INFO = {}
     RECORD_SCHEMA = get_schema_type("com.linkedin.pegasus2avro.mxe.SystemMetadata")
+
     def __init__(self,
         lastObserved: Optional[Union[int, None]]=None,
         runId: Optional[Union[str, None]]=None,
@@ -31228,6 +31393,9 @@ class QueryLanguageClass(object):
     SQL = "SQL"
     """A SQL Query"""
     
+    UNKNOWN = "UNKNOWN"
+    """Unknown query language"""
+    
     
     
 class QueryPropertiesClass(_Aspect):
@@ -37007,6 +37175,7 @@ __SCHEMA_TYPES = {
     'com.linkedin.pegasus2avro.assertion.AssertionExclusionWindowType': AssertionExclusionWindowTypeClass,
     'com.linkedin.pegasus2avro.assertion.AssertionInferenceDetails': AssertionInferenceDetailsClass,
     'com.linkedin.pegasus2avro.assertion.AssertionInfo': AssertionInfoClass,
+    'com.linkedin.pegasus2avro.assertion.AssertionMetric': AssertionMetricClass,
     'com.linkedin.pegasus2avro.assertion.AssertionMonitorSensitivity': AssertionMonitorSensitivityClass,
     'com.linkedin.pegasus2avro.assertion.AssertionResult': AssertionResultClass,
     'com.linkedin.pegasus2avro.assertion.AssertionResultError': AssertionResultErrorClass,
@@ -37350,6 +37519,7 @@ __SCHEMA_TYPES = {
     'com.linkedin.pegasus2avro.metadata.key.DataHubConnectionKey': DataHubConnectionKeyClass,
     'com.linkedin.pegasus2avro.metadata.key.DataHubIngestionSourceKey': DataHubIngestionSourceKeyClass,
     'com.linkedin.pegasus2avro.metadata.key.DataHubMetricCubeKey': DataHubMetricCubeKeyClass,
+    'com.linkedin.pegasus2avro.metadata.key.DataHubOpenAPISchemaKey': DataHubOpenAPISchemaKeyClass,
     'com.linkedin.pegasus2avro.metadata.key.DataHubPersonaKey': DataHubPersonaKeyClass,
     'com.linkedin.pegasus2avro.metadata.key.DataHubPolicyKey': DataHubPolicyKeyClass,
     'com.linkedin.pegasus2avro.metadata.key.DataHubRetentionKey': DataHubRetentionKeyClass,
@@ -37479,7 +37649,10 @@ __SCHEMA_TYPES = {
     'com.linkedin.pegasus2avro.monitor.AssertionEvaluationParametersType': AssertionEvaluationParametersTypeClass,
     'com.linkedin.pegasus2avro.monitor.AssertionEvaluationSpec': AssertionEvaluationSpecClass,
     'com.linkedin.pegasus2avro.monitor.AssertionMonitor': AssertionMonitorClass,
+    'com.linkedin.pegasus2avro.monitor.AssertionMonitorBootstrapStatus': AssertionMonitorBootstrapStatusClass,
     'com.linkedin.pegasus2avro.monitor.AssertionMonitorCapability': AssertionMonitorCapabilityClass,
+    'com.linkedin.pegasus2avro.monitor.AssertionMonitorMetricsCubeBootstrapState': AssertionMonitorMetricsCubeBootstrapStateClass,
+    'com.linkedin.pegasus2avro.monitor.AssertionMonitorMetricsCubeBootstrapStatus': AssertionMonitorMetricsCubeBootstrapStatusClass,
     'com.linkedin.pegasus2avro.monitor.AssertionMonitorSettings': AssertionMonitorSettingsClass,
     'com.linkedin.pegasus2avro.monitor.AuditLogSpec': AuditLogSpecClass,
     'com.linkedin.pegasus2avro.monitor.DataHubOperationSpec': DataHubOperationSpecClass,
@@ -37708,6 +37881,7 @@ __SCHEMA_TYPES = {
     'AssertionExclusionWindowType': AssertionExclusionWindowTypeClass,
     'AssertionInferenceDetails': AssertionInferenceDetailsClass,
     'AssertionInfo': AssertionInfoClass,
+    'AssertionMetric': AssertionMetricClass,
     'AssertionMonitorSensitivity': AssertionMonitorSensitivityClass,
     'AssertionResult': AssertionResultClass,
     'AssertionResultError': AssertionResultErrorClass,
@@ -38051,6 +38225,7 @@ __SCHEMA_TYPES = {
     'DataHubConnectionKey': DataHubConnectionKeyClass,
     'DataHubIngestionSourceKey': DataHubIngestionSourceKeyClass,
     'DataHubMetricCubeKey': DataHubMetricCubeKeyClass,
+    'DataHubOpenAPISchemaKey': DataHubOpenAPISchemaKeyClass,
     'DataHubPersonaKey': DataHubPersonaKeyClass,
     'DataHubPolicyKey': DataHubPolicyKeyClass,
     'DataHubRetentionKey': DataHubRetentionKeyClass,
@@ -38180,7 +38355,10 @@ __SCHEMA_TYPES = {
     'AssertionEvaluationParametersType': AssertionEvaluationParametersTypeClass,
     'AssertionEvaluationSpec': AssertionEvaluationSpecClass,
     'AssertionMonitor': AssertionMonitorClass,
+    'AssertionMonitorBootstrapStatus': AssertionMonitorBootstrapStatusClass,
     'AssertionMonitorCapability': AssertionMonitorCapabilityClass,
+    'AssertionMonitorMetricsCubeBootstrapState': AssertionMonitorMetricsCubeBootstrapStateClass,
+    'AssertionMonitorMetricsCubeBootstrapStatus': AssertionMonitorMetricsCubeBootstrapStatusClass,
     'AssertionMonitorSettings': AssertionMonitorSettingsClass,
     'AuditLogSpec': AuditLogSpecClass,
     'DataHubOperationSpec': DataHubOperationSpecClass,
@@ -38371,765 +38549,772 @@ avrojson.set_global_json_converter(_json_converter)
     
 
 ASPECT_CLASSES: List[Type[_Aspect]] = [
-    DataHubRetentionConfigClass,
-    TagPropertiesClass,
-    DataProductKeyClass,
-    DataProductPropertiesClass,
-    ActionRequestStatusClass,
-    ActionRequestArchivedClass,
-    ActionRequestInfoClass,
-    MonitorAnomalyEventClass,
-    SchemaFieldsInferredNeighborsClass,
-    SchemaFieldsInferredMetadataClass,
-    InferredMetadataClass,
-    InferredNeighborsClass,
-    ExecutionRequestResultClass,
-    ExecutionRequestInputClass,
-    ExecutionRequestSignalClass,
-    DataPlatformInfoClass,
-    SlackUserInfoClass,
-    VersionSetPropertiesClass,
+    BatchTestRunEventClass,
+    TestResultsClass,
+    TestInfoClass,
+    SchemaFieldAliasesClass,
+    SchemaFieldInfoClass,
     GlossaryNodeInfoClass,
     GlossaryTermInfoClass,
     GlossaryRelatedTermsClass,
-    ActorsClass,
-    RolePropertiesClass,
-    DataProcessInstanceInputClass,
-    DataProcessInstanceRunEventClass,
-    DataProcessInstanceRelationshipsClass,
-    DataProcessInstancePropertiesClass,
-    DataProcessInstanceOutputClass,
-    DataProcessInfoClass,
-    DataContractPropertiesClass,
-    DataContractStatusClass,
-    QuerySubjectsClass,
-    QueryUsageFeaturesClass,
-    QueryPropertiesClass,
-    QueryUsageStatisticsClass,
-    DataHubMetricCubeDefinitionClass,
-    DataHubMetricCubeEventClass,
-    NotebookInfoClass,
-    NotebookContentClass,
-    EditableNotebookPropertiesClass,
-    ChartInfoClass,
-    EditableChartPropertiesClass,
-    ChartQueryClass,
-    ChartUsageStatisticsClass,
-    RemoteExecutorPoolInfoClass,
-    DataHubAccessTokenInfoClass,
-    DataHubStepStatePropertiesClass,
+    DomainsClass,
+    DomainPropertiesClass,
     DataHubUpgradeResultClass,
     DataHubUpgradeRequestClass,
-    CorpGroupEditableInfoClass,
-    CorpGroupInfoClass,
-    CorpUserInfoClass,
-    NativeGroupMembershipClass,
-    GroupMembershipClass,
-    CorpGroupSettingsClass,
-    CorpUserCredentialsClass,
-    CorpUserEditableInfoClass,
-    RoleMembershipClass,
-    CorpUserStatusClass,
-    InviteTokenClass,
-    CorpUserSettingsClass,
-    EditableMLFeaturePropertiesClass,
+    SchemaProposalsClass,
+    SchemaMetadataClass,
+    EditableSchemaMetadataClass,
+    MLHyperParamClass,
+    TrainingDataClass,
+    EditableMLModelGroupPropertiesClass,
+    EthicalConsiderationsClass,
     MLPrimaryKeyPropertiesClass,
-    MLFeaturePropertiesClass,
+    CaveatsAndRecommendationsClass,
     EditableMLFeatureTablePropertiesClass,
-    SourceCodeClass,
-    MetricsClass,
-    MLFeatureTablePropertiesClass,
-    MLHyperParamClass,
+    QuantitativeAnalysesClass,
+    EvaluationDataClass,
     MLModelFactorPromptsClass,
-    CaveatsAndRecommendationsClass,
-    EthicalConsiderationsClass,
     MLTrainingRunPropertiesClass,
-    EditableMLPrimaryKeyPropertiesClass,
-    EditableMLModelPropertiesClass,
+    MetricsClass,
+    EditableMLFeaturePropertiesClass,
+    MLModelGroupPropertiesClass,
     MLModelDeploymentPropertiesClass,
-    MLMetricClass,
-    TrainingDataClass,
+    EditableMLModelPropertiesClass,
+    SourceCodeClass,
+    EditableMLPrimaryKeyPropertiesClass,
+    MLFeaturePropertiesClass,
     IntendedUseClass,
-    MLModelGroupPropertiesClass,
     MLModelPropertiesClass,
-    EditableMLModelGroupPropertiesClass,
-    QuantitativeAnalysesClass,
-    EvaluationDataClass,
-    GlobalSettingsInfoClass,
-    DataHubConnectionDetailsClass,
+    MLMetricClass,
+    MLFeatureTablePropertiesClass,
+    DataHubViewInfoClass,
+    EditableDataFlowPropertiesClass,
+    DataFlowInfoClass,
     VersionInfoClass,
-    DataJobInputOutputClass,
     EditableDataJobPropertiesClass,
+    DataJobInputOutputClass,
     DataJobInfoClass,
-    DataFlowInfoClass,
-    EditableDataFlowPropertiesClass,
-    DatahubIngestionCheckpointClass,
     DatahubIngestionRunSummaryClass,
-    DataHubRoleInfoClass,
-    DataHubPolicyInfoClass,
-    DataHubIngestionSourceInfoClass,
-    DomainPropertiesClass,
-    DomainsClass,
-    DataTypeInfoClass,
-    DataTypeKeyClass,
-    DatasetPropertiesClass,
-    UpstreamLineageClass,
-    DatasetProfileClass,
-    PartitionsSummaryClass,
-    ViewPropertiesClass,
-    IcebergCatalogInfoClass,
-    DatasetDeprecationClass,
-    SchemaFieldProfileClass,
-    EditableDatasetPropertiesClass,
-    DatasetUsageStatisticsClass,
-    DatasetUpstreamLineageClass,
-    BusinessAttributeInfoClass,
-    BusinessAttributeKeyClass,
-    BusinessAttributesClass,
-    StructuredPropertiesClass,
-    StructuredPropertySettingsClass,
-    StructuredPropertyKeyClass,
-    StructuredPropertyDefinitionClass,
-    EntityInferenceMetadataClass,
-    AiInferenceMetadataClass,
-    IcebergWarehouseInfoClass,
-    DataPlatformInstancePropertiesClass,
-    MonitorSuiteInfoClass,
-    MonitorInfoClass,
-    MonitorTimeseriesStateClass,
+    DatahubIngestionCheckpointClass,
+    ActionRequestArchivedClass,
+    ActionRequestInfoClass,
+    ActionRequestStatusClass,
     OwnershipTypeInfoClass,
-    PlatformResourceInfoClass,
-    PlatformResourceKeyClass,
-    ConstraintInfoClass,
-    RemoteExecutorStatusClass,
-    AssertionActionsClass,
-    AssertionSummaryClass,
-    AssertionRunEventClass,
-    AssertionInferenceDetailsClass,
-    AssertionAnalyticsRunEventClass,
-    AssertionDryRunEventClass,
-    AssertionInfoClass,
-    EditableContainerPropertiesClass,
-    ContainerPropertiesClass,
-    ContainerClass,
-    LinkPreviewInfoClass,
-    DataHubViewInfoClass,
-    DataHubActionInfoClass,
-    DataHubActionStatusClass,
     EntityTypeKeyClass,
     EntityTypeInfoClass,
-    InstitutionalMemoryClass,
-    AccessClass,
-    StatusClass,
-    EmbedClass,
-    DocumentationClass,
-    SiblingsClass,
-    CostClass,
-    IncidentsSummaryClass,
-    AssertionsSummaryClass,
-    SubTypesClass,
-    DisplayPropertiesClass,
-    OperationClass,
-    AnomaliesSummaryClass,
-    GlossaryTermsClass,
-    InputFieldsClass,
-    DataPlatformInstanceClass,
-    ShareClass,
-    OwnershipClass,
-    DataTransformLogicClass,
-    GlobalTagsClass,
-    VersionPropertiesClass,
-    DeprecationClass,
-    ProposalsClass,
-    BrowsePathsClass,
-    FormsClass,
-    BrowsePathsV2Class,
-    OriginClass,
-    TestResultsClass,
-    BatchTestRunEventClass,
-    TestInfoClass,
-    DataHubPersonaInfoClass,
-    EditableSchemaMetadataClass,
-    SchemaProposalsClass,
-    SchemaMetadataClass,
-    PostInfoClass,
-    SubscriptionInfoClass,
-    TelemetryClientIdClass,
-    DynamicFormAssignmentClass,
-    FormInfoClass,
-    DataHubSecretValueClass,
-    DashboardInfoClass,
-    EditableDashboardPropertiesClass,
-    DashboardUsageStatisticsClass,
-    SchemaFieldInfoClass,
-    SchemaFieldAliasesClass,
-    IncidentInfoClass,
-    IncidentNotificationDetailsClass,
-    IncidentActivityEventClass,
-    IncidentSourceClass,
-    RemoteExecutorPoolGlobalConfigClass,
-    CorpGroupKeyClass,
-    OwnershipTypeKeyClass,
-    GenericEntityKeyClass,
-    RecommendationModuleKeyClass,
-    DataProcessKeyClass,
-    MLFeatureKeyClass,
-    RemoteExecutorGlobalConfigKeyClass,
-    DataHubPersonaKeyClass,
-    DataPlatformInstanceKeyClass,
-    ERModelRelationshipKeyClass,
-    ContainerKeyClass,
+    TagPropertiesClass,
+    ERModelRelationshipPropertiesClass,
+    EditableERModelRelationshipPropertiesClass,
+    DataHubIngestionSourceInfoClass,
+    SystemMetadataClass,
+    ExecutionRequestInputClass,
+    ExecutionRequestSignalClass,
+    ExecutionRequestResultClass,
+    EditableChartPropertiesClass,
+    ChartQueryClass,
+    ChartUsageStatisticsClass,
+    ChartInfoClass,
+    GlobalSettingsInfoClass,
+    DataHubOpenAPISchemaKeyClass,
+    QueryKeyClass,
+    DataJobKeyClass,
+    DataHubAccessTokenKeyClass,
     TelemetryKeyClass,
-    DatasetKeyClass,
-    DataHubIngestionSourceKeyClass,
-    DataFlowKeyClass,
-    DataHubSecretKeyClass,
-    DataHubUpgradeKeyClass,
-    DataContractKeyClass,
-    InviteTokenKeyClass,
-    ActionRequestKeyClass,
     ChartKeyClass,
-    PostKeyClass,
+    DataHubViewKeyClass,
+    DatasetKeyClass,
+    DataPlatformKeyClass,
+    DataPlatformInstanceKeyClass,
+    DataHubPolicyKeyClass,
+    ExecutionRequestKeyClass,
+    DataHubRetentionKeyClass,
+    DataHubStepStateKeyClass,
+    RecommendationModuleKeyClass,
+    ContainerKeyClass,
+    RoleKeyClass,
+    DataProcessInstanceKeyClass,
+    DomainKeyClass,
+    ERModelRelationshipKeyClass,
+    RemoteExecutorPoolKeyClass,
+    DataHubMetricCubeKeyClass,
     GlobalSettingsKeyClass,
-    GlossaryTermKeyClass,
+    ActionRequestKeyClass,
+    CorpUserKeyClass,
+    GlossaryNodeKeyClass,
     SubscriptionKeyClass,
-    NotebookKeyClass,
-    MLFeatureTableKeyClass,
+    MLPrimaryKeyKeyClass,
+    TestKeyClass,
     MLModelGroupKeyClass,
-    ExecutionRequestKeyClass,
-    DataProcessInstanceKeyClass,
+    DataFlowKeyClass,
+    DataHubPersonaKeyClass,
     MonitorKeyClass,
-    IncidentKeyClass,
-    ConstraintKeyClass,
-    RoleKeyClass,
-    DashboardKeyClass,
-    GlossaryNodeKeyClass,
-    DataHubViewKeyClass,
-    DataHubRetentionKeyClass,
-    RemoteExecutorPoolKeyClass,
-    AnomalyKeyClass,
-    SchemaFieldKeyClass,
     VersionSetKeyClass,
-    DataJobKeyClass,
-    MLPrimaryKeyKeyClass,
-    DataHubAccessTokenKeyClass,
-    LinkPreviewKeyClass,
-    DataHubRoleKeyClass,
-    DataHubMetricCubeKeyClass,
-    DataHubActionKeyClass,
-    DataHubStepStateKeyClass,
+    DataHubUpgradeKeyClass,
     AssertionKeyClass,
-    QueryKeyClass,
-    MLModelDeploymentKeyClass,
-    MLModelKeyClass,
+    ConstraintKeyClass,
+    CorpGroupKeyClass,
+    GenericEntityKeyClass,
+    DataHubActionKeyClass,
+    DataContractKeyClass,
+    RemoteExecutorGlobalConfigKeyClass,
+    DataHubSecretKeyClass,
+    FormKeyClass,
+    MLFeatureTableKeyClass,
+    SchemaFieldKeyClass,
     TagKeyClass,
+    DataProcessKeyClass,
+    GlossaryTermKeyClass,
+    PostKeyClass,
+    DashboardKeyClass,
     DataHubConnectionKeyClass,
+    AnomalyKeyClass,
+    MLModelKeyClass,
+    DataHubIngestionSourceKeyClass,
     RemoteExecutorKeyClass,
-    DomainKeyClass,
+    MLFeatureKeyClass,
     MonitorSuiteKeyClass,
-    DataHubPolicyKeyClass,
-    TestKeyClass,
-    FormKeyClass,
-    CorpUserKeyClass,
-    DataPlatformKeyClass,
-    LineageFeaturesClass,
+    MLModelDeploymentKeyClass,
+    OwnershipTypeKeyClass,
+    IncidentKeyClass,
+    InviteTokenKeyClass,
+    LinkPreviewKeyClass,
+    NotebookKeyClass,
+    DataHubRoleKeyClass,
+    CostFeaturesClass,
     StorageFeaturesClass,
     UsageFeaturesClass,
-    CostFeaturesClass,
+    LineageFeaturesClass,
     RecommendationModuleClass,
-    EditableERModelRelationshipPropertiesClass,
-    ERModelRelationshipPropertiesClass
-]
-
-ASPECT_NAME_MAP: Dict[str, Type[_Aspect]] = {
-    aspect.get_aspect_name(): aspect
-    for aspect in ASPECT_CLASSES
-}
-
-from typing import Literal
-from typing_extensions import TypedDict
-
-class AspectBag(TypedDict, total=False):
-    dataHubRetentionConfig: DataHubRetentionConfigClass
-    tagProperties: TagPropertiesClass
-    dataProductKey: DataProductKeyClass
-    dataProductProperties: DataProductPropertiesClass
-    actionRequestStatus: ActionRequestStatusClass
-    actionRequestArchived: ActionRequestArchivedClass
-    actionRequestInfo: ActionRequestInfoClass
-    monitorAnomalyEvent: MonitorAnomalyEventClass
-    schemaFieldsInferredNeighbors: SchemaFieldsInferredNeighborsClass
-    schemaFieldsInferredMetadata: SchemaFieldsInferredMetadataClass
-    inferredMetadata: InferredMetadataClass
-    inferredNeighbors: InferredNeighborsClass
-    dataHubExecutionRequestResult: ExecutionRequestResultClass
-    dataHubExecutionRequestInput: ExecutionRequestInputClass
-    dataHubExecutionRequestSignal: ExecutionRequestSignalClass
-    dataPlatformInfo: DataPlatformInfoClass
-    slackUserInfo: SlackUserInfoClass
-    versionSetProperties: VersionSetPropertiesClass
-    glossaryNodeInfo: GlossaryNodeInfoClass
-    glossaryTermInfo: GlossaryTermInfoClass
-    glossaryRelatedTerms: GlossaryRelatedTermsClass
-    actors: ActorsClass
-    roleProperties: RolePropertiesClass
-    dataProcessInstanceInput: DataProcessInstanceInputClass
-    dataProcessInstanceRunEvent: DataProcessInstanceRunEventClass
-    dataProcessInstanceRelationships: DataProcessInstanceRelationshipsClass
-    dataProcessInstanceProperties: DataProcessInstancePropertiesClass
-    dataProcessInstanceOutput: DataProcessInstanceOutputClass
-    dataProcessInfo: DataProcessInfoClass
-    dataContractProperties: DataContractPropertiesClass
-    dataContractStatus: DataContractStatusClass
-    querySubjects: QuerySubjectsClass
-    queryUsageFeatures: QueryUsageFeaturesClass
-    queryProperties: QueryPropertiesClass
-    queryUsageStatistics: QueryUsageStatisticsClass
-    dataHubMetricCubeDefinition: DataHubMetricCubeDefinitionClass
-    dataHubMetricCubeEvent: DataHubMetricCubeEventClass
-    notebookInfo: NotebookInfoClass
-    notebookContent: NotebookContentClass
-    editableNotebookProperties: EditableNotebookPropertiesClass
-    chartInfo: ChartInfoClass
-    editableChartProperties: EditableChartPropertiesClass
-    chartQuery: ChartQueryClass
-    chartUsageStatistics: ChartUsageStatisticsClass
-    dataHubRemoteExecutorPoolInfo: RemoteExecutorPoolInfoClass
-    dataHubAccessTokenInfo: DataHubAccessTokenInfoClass
-    dataHubStepStateProperties: DataHubStepStatePropertiesClass
+    LinkPreviewInfoClass,
+    DataHubActionInfoClass,
+    DataHubActionStatusClass,
+    DataHubSecretValueClass,
+    EntityInferenceMetadataClass,
+    AiInferenceMetadataClass,
+    RemoteExecutorStatusClass,
+    DataPlatformInstancePropertiesClass,
+    IcebergWarehouseInfoClass,
+    DataHubAccessTokenInfoClass,
+    BusinessAttributeKeyClass,
+    BusinessAttributesClass,
+    BusinessAttributeInfoClass,
+    ConstraintInfoClass,
+    TelemetryClientIdClass,
+    StructuredPropertySettingsClass,
+    StructuredPropertiesClass,
+    StructuredPropertyKeyClass,
+    StructuredPropertyDefinitionClass,
+    DataTypeKeyClass,
+    DataTypeInfoClass,
+    ContainerPropertiesClass,
+    ContainerClass,
+    EditableContainerPropertiesClass,
+    DynamicFormAssignmentClass,
+    FormInfoClass,
+    QueryUsageFeaturesClass,
+    QuerySubjectsClass,
+    QueryPropertiesClass,
+    QueryUsageStatisticsClass,
+    EditableNotebookPropertiesClass,
+    NotebookInfoClass,
+    NotebookContentClass,
+    DataPlatformInfoClass,
+    SlackUserInfoClass,
+    EditableDashboardPropertiesClass,
+    DashboardUsageStatisticsClass,
+    DashboardInfoClass,
+    DataProcessInstanceOutputClass,
+    DataProcessInstanceInputClass,
+    DataProcessInstanceRunEventClass,
+    DataProcessInstanceRelationshipsClass,
+    DataProcessInfoClass,
+    DataProcessInstancePropertiesClass,
+    SubscriptionInfoClass,
+    DataHubRetentionConfigClass,
+    RemoteExecutorPoolInfoClass,
+    OperationClass,
+    ShareClass,
+    IncidentsSummaryClass,
+    BrowsePathsClass,
+    DataTransformLogicClass,
+    SubTypesClass,
+    DisplayPropertiesClass,
+    OriginClass,
+    InstitutionalMemoryClass,
+    DeprecationClass,
+    AssertionsSummaryClass,
+    VersionPropertiesClass,
+    OwnershipClass,
+    SiblingsClass,
+    DataPlatformInstanceClass,
+    FormsClass,
+    BrowsePathsV2Class,
+    GlobalTagsClass,
+    GlossaryTermsClass,
+    AccessClass,
+    EmbedClass,
+    StatusClass,
+    AnomaliesSummaryClass,
+    InputFieldsClass,
+    CostClass,
+    DocumentationClass,
+    ProposalsClass,
+    DataProductKeyClass,
+    DataProductPropertiesClass,
+    DataHubConnectionDetailsClass,
+    PlatformResourceInfoClass,
+    PlatformResourceKeyClass,
+    ActorsClass,
+    RolePropertiesClass,
+    RemoteExecutorPoolGlobalConfigClass,
+    AssertionRunEventClass,
+    AssertionAnalyticsRunEventClass,
+    AssertionActionsClass,
+    AssertionInferenceDetailsClass,
+    AssertionInfoClass,
+    AssertionDryRunEventClass,
+    AssertionSummaryClass,
+    MonitorInfoClass,
+    MonitorSuiteInfoClass,
+    MonitorTimeseriesStateClass,
+    DataHubStepStatePropertiesClass,
+    VersionSetPropertiesClass,
+    SchemaFieldsInferredNeighborsClass,
+    InferredMetadataClass,
+    InferredNeighborsClass,
+    SchemaFieldsInferredMetadataClass,
+    IncidentNotificationDetailsClass,
+    IncidentInfoClass,
+    IncidentActivityEventClass,
+    IncidentSourceClass,
+    DataHubPolicyInfoClass,
+    DataHubRoleInfoClass,
+    CorpGroupSettingsClass,
+    CorpUserInfoClass,
+    CorpGroupInfoClass,
+    CorpGroupEditableInfoClass,
+    InviteTokenClass,
+    NativeGroupMembershipClass,
+    RoleMembershipClass,
+    GroupMembershipClass,
+    CorpUserStatusClass,
+    CorpUserCredentialsClass,
+    CorpUserEditableInfoClass,
+    CorpUserSettingsClass,
+    DataHubMetricCubeDefinitionClass,
+    DataHubMetricCubeEventClass,
+    PostInfoClass,
+    DataContractStatusClass,
+    DataContractPropertiesClass,
+    MonitorAnomalyEventClass,
+    DatasetUsageStatisticsClass,
+    EditableDatasetPropertiesClass,
+    UpstreamLineageClass,
+    ViewPropertiesClass,
+    PartitionsSummaryClass,
+    SchemaFieldProfileClass,
+    DatasetProfileClass,
+    DatasetDeprecationClass,
+    DatasetUpstreamLineageClass,
+    DatasetPropertiesClass,
+    IcebergCatalogInfoClass,
+    DataHubPersonaInfoClass
+]
+
+ASPECT_NAME_MAP: Dict[str, Type[_Aspect]] = {
+    aspect.get_aspect_name(): aspect
+    for aspect in ASPECT_CLASSES
+}
+
+from typing import Literal
+from typing_extensions import TypedDict
+
+class AspectBag(TypedDict, total=False):
+    batchTestRunEvent: BatchTestRunEventClass
+    testResults: TestResultsClass
+    testInfo: TestInfoClass
+    schemaFieldAliases: SchemaFieldAliasesClass
+    schemafieldInfo: SchemaFieldInfoClass
+    glossaryNodeInfo: GlossaryNodeInfoClass
+    glossaryTermInfo: GlossaryTermInfoClass
+    glossaryRelatedTerms: GlossaryRelatedTermsClass
+    domains: DomainsClass
+    domainProperties: DomainPropertiesClass
     dataHubUpgradeResult: DataHubUpgradeResultClass
     dataHubUpgradeRequest: DataHubUpgradeRequestClass
-    corpGroupEditableInfo: CorpGroupEditableInfoClass
-    corpGroupInfo: CorpGroupInfoClass
-    corpUserInfo: CorpUserInfoClass
-    nativeGroupMembership: NativeGroupMembershipClass
-    groupMembership: GroupMembershipClass
-    corpGroupSettings: CorpGroupSettingsClass
-    corpUserCredentials: CorpUserCredentialsClass
-    corpUserEditableInfo: CorpUserEditableInfoClass
-    roleMembership: RoleMembershipClass
-    corpUserStatus: CorpUserStatusClass
-    inviteToken: InviteTokenClass
-    corpUserSettings: CorpUserSettingsClass
-    editableMlFeatureProperties: EditableMLFeaturePropertiesClass
+    schemaProposals: SchemaProposalsClass
+    schemaMetadata: SchemaMetadataClass
+    editableSchemaMetadata: EditableSchemaMetadataClass
+    mlHyperParam: MLHyperParamClass
+    mlModelTrainingData: TrainingDataClass
+    editableMlModelGroupProperties: EditableMLModelGroupPropertiesClass
+    mlModelEthicalConsiderations: EthicalConsiderationsClass
     mlPrimaryKeyProperties: MLPrimaryKeyPropertiesClass
-    mlFeatureProperties: MLFeaturePropertiesClass
+    mlModelCaveatsAndRecommendations: CaveatsAndRecommendationsClass
     editableMlFeatureTableProperties: EditableMLFeatureTablePropertiesClass
-    sourceCode: SourceCodeClass
-    mlModelMetrics: MetricsClass
-    mlFeatureTableProperties: MLFeatureTablePropertiesClass
-    mlHyperParam: MLHyperParamClass
+    mlModelQuantitativeAnalyses: QuantitativeAnalysesClass
+    mlModelEvaluationData: EvaluationDataClass
     mlModelFactorPrompts: MLModelFactorPromptsClass
-    mlModelCaveatsAndRecommendations: CaveatsAndRecommendationsClass
-    mlModelEthicalConsiderations: EthicalConsiderationsClass
     mlTrainingRunProperties: MLTrainingRunPropertiesClass
-    editableMlPrimaryKeyProperties: EditableMLPrimaryKeyPropertiesClass
-    editableMlModelProperties: EditableMLModelPropertiesClass
+    mlModelMetrics: MetricsClass
+    editableMlFeatureProperties: EditableMLFeaturePropertiesClass
+    mlModelGroupProperties: MLModelGroupPropertiesClass
     mlModelDeploymentProperties: MLModelDeploymentPropertiesClass
-    mlMetric: MLMetricClass
-    mlModelTrainingData: TrainingDataClass
+    editableMlModelProperties: EditableMLModelPropertiesClass
+    sourceCode: SourceCodeClass
+    editableMlPrimaryKeyProperties: EditableMLPrimaryKeyPropertiesClass
+    mlFeatureProperties: MLFeaturePropertiesClass
     intendedUse: IntendedUseClass
-    mlModelGroupProperties: MLModelGroupPropertiesClass
     mlModelProperties: MLModelPropertiesClass
-    editableMlModelGroupProperties: EditableMLModelGroupPropertiesClass
-    mlModelQuantitativeAnalyses: QuantitativeAnalysesClass
-    mlModelEvaluationData: EvaluationDataClass
-    globalSettingsInfo: GlobalSettingsInfoClass
-    dataHubConnectionDetails: DataHubConnectionDetailsClass
+    mlMetric: MLMetricClass
+    mlFeatureTableProperties: MLFeatureTablePropertiesClass
+    dataHubViewInfo: DataHubViewInfoClass
+    editableDataFlowProperties: EditableDataFlowPropertiesClass
+    dataFlowInfo: DataFlowInfoClass
     versionInfo: VersionInfoClass
-    dataJobInputOutput: DataJobInputOutputClass
     editableDataJobProperties: EditableDataJobPropertiesClass
+    dataJobInputOutput: DataJobInputOutputClass
     dataJobInfo: DataJobInfoClass
-    dataFlowInfo: DataFlowInfoClass
-    editableDataFlowProperties: EditableDataFlowPropertiesClass
-    datahubIngestionCheckpoint: DatahubIngestionCheckpointClass
     datahubIngestionRunSummary: DatahubIngestionRunSummaryClass
-    dataHubRoleInfo: DataHubRoleInfoClass
-    dataHubPolicyInfo: DataHubPolicyInfoClass
-    dataHubIngestionSourceInfo: DataHubIngestionSourceInfoClass
-    domainProperties: DomainPropertiesClass
-    domains: DomainsClass
-    dataTypeInfo: DataTypeInfoClass
-    dataTypeKey: DataTypeKeyClass
-    datasetProperties: DatasetPropertiesClass
-    upstreamLineage: UpstreamLineageClass
-    datasetProfile: DatasetProfileClass
-    partitionsSummary: PartitionsSummaryClass
-    viewProperties: ViewPropertiesClass
-    icebergCatalogInfo: IcebergCatalogInfoClass
-    datasetDeprecation: DatasetDeprecationClass
-    schemaFieldProfile: SchemaFieldProfileClass
-    editableDatasetProperties: EditableDatasetPropertiesClass
-    datasetUsageStatistics: DatasetUsageStatisticsClass
-    datasetUpstreamLineage: DatasetUpstreamLineageClass
-    businessAttributeInfo: BusinessAttributeInfoClass
-    businessAttributeKey: BusinessAttributeKeyClass
-    businessAttributes: BusinessAttributesClass
-    structuredProperties: StructuredPropertiesClass
-    structuredPropertySettings: StructuredPropertySettingsClass
-    structuredPropertyKey: StructuredPropertyKeyClass
-    propertyDefinition: StructuredPropertyDefinitionClass
-    entityInferenceMetadata: EntityInferenceMetadataClass
-    aiInferenceMetadata: AiInferenceMetadataClass
-    icebergWarehouseInfo: IcebergWarehouseInfoClass
-    dataPlatformInstanceProperties: DataPlatformInstancePropertiesClass
-    monitorSuiteInfo: MonitorSuiteInfoClass
-    monitorInfo: MonitorInfoClass
-    monitorTimeseriesState: MonitorTimeseriesStateClass
+    datahubIngestionCheckpoint: DatahubIngestionCheckpointClass
+    actionRequestArchived: ActionRequestArchivedClass
+    actionRequestInfo: ActionRequestInfoClass
+    actionRequestStatus: ActionRequestStatusClass
     ownershipTypeInfo: OwnershipTypeInfoClass
-    platformResourceInfo: PlatformResourceInfoClass
-    platformResourceKey: PlatformResourceKeyClass
-    constraintInfo: ConstraintInfoClass
-    dataHubRemoteExecutorStatus: RemoteExecutorStatusClass
-    assertionActions: AssertionActionsClass
-    assertionSummary: AssertionSummaryClass
-    assertionRunEvent: AssertionRunEventClass
-    assertionInferenceDetails: AssertionInferenceDetailsClass
-    assertionAnalyticsRunEvent: AssertionAnalyticsRunEventClass
-    assertionDryRunEvent: AssertionDryRunEventClass
-    assertionInfo: AssertionInfoClass
-    editableContainerProperties: EditableContainerPropertiesClass
-    containerProperties: ContainerPropertiesClass
-    container: ContainerClass
-    linkPreviewInfo: LinkPreviewInfoClass
-    dataHubViewInfo: DataHubViewInfoClass
-    dataHubActionInfo: DataHubActionInfoClass
-    dataHubActionStatus: DataHubActionStatusClass
     entityTypeKey: EntityTypeKeyClass
     entityTypeInfo: EntityTypeInfoClass
-    institutionalMemory: InstitutionalMemoryClass
-    access: AccessClass
-    status: StatusClass
-    embed: EmbedClass
-    documentation: DocumentationClass
-    siblings: SiblingsClass
-    cost: CostClass
-    incidentsSummary: IncidentsSummaryClass
-    assertionsSummary: AssertionsSummaryClass
-    subTypes: SubTypesClass
-    displayProperties: DisplayPropertiesClass
-    operation: OperationClass
-    anomaliesSummary: AnomaliesSummaryClass
-    glossaryTerms: GlossaryTermsClass
-    inputFields: InputFieldsClass
-    dataPlatformInstance: DataPlatformInstanceClass
-    share: ShareClass
-    ownership: OwnershipClass
-    dataTransformLogic: DataTransformLogicClass
-    globalTags: GlobalTagsClass
-    versionProperties: VersionPropertiesClass
-    deprecation: DeprecationClass
-    proposals: ProposalsClass
-    browsePaths: BrowsePathsClass
-    forms: FormsClass
-    browsePathsV2: BrowsePathsV2Class
-    origin: OriginClass
-    testResults: TestResultsClass
-    batchTestRunEvent: BatchTestRunEventClass
-    testInfo: TestInfoClass
-    dataHubPersonaInfo: DataHubPersonaInfoClass
-    editableSchemaMetadata: EditableSchemaMetadataClass
-    schemaProposals: SchemaProposalsClass
-    schemaMetadata: SchemaMetadataClass
-    postInfo: PostInfoClass
-    subscriptionInfo: SubscriptionInfoClass
-    telemetryClientId: TelemetryClientIdClass
-    dynamicFormAssignment: DynamicFormAssignmentClass
-    formInfo: FormInfoClass
-    dataHubSecretValue: DataHubSecretValueClass
-    dashboardInfo: DashboardInfoClass
-    editableDashboardProperties: EditableDashboardPropertiesClass
-    dashboardUsageStatistics: DashboardUsageStatisticsClass
-    schemafieldInfo: SchemaFieldInfoClass
-    schemaFieldAliases: SchemaFieldAliasesClass
-    incidentInfo: IncidentInfoClass
-    incidentNotificationDetails: IncidentNotificationDetailsClass
-    incidentActivityEvent: IncidentActivityEventClass
-    incidentSource: IncidentSourceClass
-    dataHubRemoteExecutorPoolGlobalConfig: RemoteExecutorPoolGlobalConfigClass
-    corpGroupKey: CorpGroupKeyClass
-    ownershipTypeKey: OwnershipTypeKeyClass
-    genericEntityKey: GenericEntityKeyClass
-    recommendationModuleKey: RecommendationModuleKeyClass
-    dataProcessKey: DataProcessKeyClass
-    mlFeatureKey: MLFeatureKeyClass
-    dataHubRemoteExecutorGlobalConfigKey: RemoteExecutorGlobalConfigKeyClass
-    dataHubPersonaKey: DataHubPersonaKeyClass
-    dataPlatformInstanceKey: DataPlatformInstanceKeyClass
-    erModelRelationshipKey: ERModelRelationshipKeyClass
-    containerKey: ContainerKeyClass
+    tagProperties: TagPropertiesClass
+    erModelRelationshipProperties: ERModelRelationshipPropertiesClass
+    editableERModelRelationshipProperties: EditableERModelRelationshipPropertiesClass
+    dataHubIngestionSourceInfo: DataHubIngestionSourceInfoClass
+    systemMetadata: SystemMetadataClass
+    dataHubExecutionRequestInput: ExecutionRequestInputClass
+    dataHubExecutionRequestSignal: ExecutionRequestSignalClass
+    dataHubExecutionRequestResult: ExecutionRequestResultClass
+    editableChartProperties: EditableChartPropertiesClass
+    chartQuery: ChartQueryClass
+    chartUsageStatistics: ChartUsageStatisticsClass
+    chartInfo: ChartInfoClass
+    globalSettingsInfo: GlobalSettingsInfoClass
+    dataHubOpenAPISchemaKey: DataHubOpenAPISchemaKeyClass
+    queryKey: QueryKeyClass
+    dataJobKey: DataJobKeyClass
+    dataHubAccessTokenKey: DataHubAccessTokenKeyClass
     telemetryKey: TelemetryKeyClass
-    datasetKey: DatasetKeyClass
-    dataHubIngestionSourceKey: DataHubIngestionSourceKeyClass
-    dataFlowKey: DataFlowKeyClass
-    dataHubSecretKey: DataHubSecretKeyClass
-    dataHubUpgradeKey: DataHubUpgradeKeyClass
-    dataContractKey: DataContractKeyClass
-    inviteTokenKey: InviteTokenKeyClass
-    actionRequestKey: ActionRequestKeyClass
     chartKey: ChartKeyClass
-    postKey: PostKeyClass
+    dataHubViewKey: DataHubViewKeyClass
+    datasetKey: DatasetKeyClass
+    dataPlatformKey: DataPlatformKeyClass
+    dataPlatformInstanceKey: DataPlatformInstanceKeyClass
+    dataHubPolicyKey: DataHubPolicyKeyClass
+    dataHubExecutionRequestKey: ExecutionRequestKeyClass
+    dataHubRetentionKey: DataHubRetentionKeyClass
+    dataHubStepStateKey: DataHubStepStateKeyClass
+    recommendationModuleKey: RecommendationModuleKeyClass
+    containerKey: ContainerKeyClass
+    roleKey: RoleKeyClass
+    dataProcessInstanceKey: DataProcessInstanceKeyClass
+    domainKey: DomainKeyClass
+    erModelRelationshipKey: ERModelRelationshipKeyClass
+    dataHubRemoteExecutorPoolKey: RemoteExecutorPoolKeyClass
+    dataHubMetricCubeKey: DataHubMetricCubeKeyClass
     globalSettingsKey: GlobalSettingsKeyClass
-    glossaryTermKey: GlossaryTermKeyClass
+    actionRequestKey: ActionRequestKeyClass
+    corpUserKey: CorpUserKeyClass
+    glossaryNodeKey: GlossaryNodeKeyClass
     subscriptionKey: SubscriptionKeyClass
-    notebookKey: NotebookKeyClass
-    mlFeatureTableKey: MLFeatureTableKeyClass
+    mlPrimaryKeyKey: MLPrimaryKeyKeyClass
+    testKey: TestKeyClass
     mlModelGroupKey: MLModelGroupKeyClass
-    dataHubExecutionRequestKey: ExecutionRequestKeyClass
-    dataProcessInstanceKey: DataProcessInstanceKeyClass
+    dataFlowKey: DataFlowKeyClass
+    dataHubPersonaKey: DataHubPersonaKeyClass
     monitorKey: MonitorKeyClass
-    incidentKey: IncidentKeyClass
-    constraintKey: ConstraintKeyClass
-    roleKey: RoleKeyClass
-    dashboardKey: DashboardKeyClass
-    glossaryNodeKey: GlossaryNodeKeyClass
-    dataHubViewKey: DataHubViewKeyClass
-    dataHubRetentionKey: DataHubRetentionKeyClass
-    dataHubRemoteExecutorPoolKey: RemoteExecutorPoolKeyClass
-    anomalyKey: AnomalyKeyClass
-    schemaFieldKey: SchemaFieldKeyClass
     versionSetKey: VersionSetKeyClass
-    dataJobKey: DataJobKeyClass
-    mlPrimaryKeyKey: MLPrimaryKeyKeyClass
-    dataHubAccessTokenKey: DataHubAccessTokenKeyClass
-    linkPreviewKey: LinkPreviewKeyClass
-    dataHubRoleKey: DataHubRoleKeyClass
-    dataHubMetricCubeKey: DataHubMetricCubeKeyClass
-    dataHubActionKey: DataHubActionKeyClass
-    dataHubStepStateKey: DataHubStepStateKeyClass
+    dataHubUpgradeKey: DataHubUpgradeKeyClass
     assertionKey: AssertionKeyClass
-    queryKey: QueryKeyClass
-    mlModelDeploymentKey: MLModelDeploymentKeyClass
-    mlModelKey: MLModelKeyClass
+    constraintKey: ConstraintKeyClass
+    corpGroupKey: CorpGroupKeyClass
+    genericEntityKey: GenericEntityKeyClass
+    dataHubActionKey: DataHubActionKeyClass
+    dataContractKey: DataContractKeyClass
+    dataHubRemoteExecutorGlobalConfigKey: RemoteExecutorGlobalConfigKeyClass
+    dataHubSecretKey: DataHubSecretKeyClass
+    formKey: FormKeyClass
+    mlFeatureTableKey: MLFeatureTableKeyClass
+    schemaFieldKey: SchemaFieldKeyClass
     tagKey: TagKeyClass
+    dataProcessKey: DataProcessKeyClass
+    glossaryTermKey: GlossaryTermKeyClass
+    postKey: PostKeyClass
+    dashboardKey: DashboardKeyClass
     dataHubConnectionKey: DataHubConnectionKeyClass
+    anomalyKey: AnomalyKeyClass
+    mlModelKey: MLModelKeyClass
+    dataHubIngestionSourceKey: DataHubIngestionSourceKeyClass
     dataHubRemoteExecutorKey: RemoteExecutorKeyClass
-    domainKey: DomainKeyClass
+    mlFeatureKey: MLFeatureKeyClass
     monitorSuiteKey: MonitorSuiteKeyClass
-    dataHubPolicyKey: DataHubPolicyKeyClass
-    testKey: TestKeyClass
-    formKey: FormKeyClass
-    corpUserKey: CorpUserKeyClass
-    dataPlatformKey: DataPlatformKeyClass
-    lineageFeatures: LineageFeaturesClass
+    mlModelDeploymentKey: MLModelDeploymentKeyClass
+    ownershipTypeKey: OwnershipTypeKeyClass
+    incidentKey: IncidentKeyClass
+    inviteTokenKey: InviteTokenKeyClass
+    linkPreviewKey: LinkPreviewKeyClass
+    notebookKey: NotebookKeyClass
+    dataHubRoleKey: DataHubRoleKeyClass
+    costFeatures: CostFeaturesClass
     storageFeatures: StorageFeaturesClass
     usageFeatures: UsageFeaturesClass
-    costFeatures: CostFeaturesClass
+    lineageFeatures: LineageFeaturesClass
     recommendationModule: RecommendationModuleClass
-    editableERModelRelationshipProperties: EditableERModelRelationshipPropertiesClass
-    erModelRelationshipProperties: ERModelRelationshipPropertiesClass
+    linkPreviewInfo: LinkPreviewInfoClass
+    dataHubActionInfo: DataHubActionInfoClass
+    dataHubActionStatus: DataHubActionStatusClass
+    dataHubSecretValue: DataHubSecretValueClass
+    entityInferenceMetadata: EntityInferenceMetadataClass
+    aiInferenceMetadata: AiInferenceMetadataClass
+    dataHubRemoteExecutorStatus: RemoteExecutorStatusClass
+    dataPlatformInstanceProperties: DataPlatformInstancePropertiesClass
+    icebergWarehouseInfo: IcebergWarehouseInfoClass
+    dataHubAccessTokenInfo: DataHubAccessTokenInfoClass
+    businessAttributeKey: BusinessAttributeKeyClass
+    businessAttributes: BusinessAttributesClass
+    businessAttributeInfo: BusinessAttributeInfoClass
+    constraintInfo: ConstraintInfoClass
+    telemetryClientId: TelemetryClientIdClass
+    structuredPropertySettings: StructuredPropertySettingsClass
+    structuredProperties: StructuredPropertiesClass
+    structuredPropertyKey: StructuredPropertyKeyClass
+    propertyDefinition: StructuredPropertyDefinitionClass
+    dataTypeKey: DataTypeKeyClass
+    dataTypeInfo: DataTypeInfoClass
+    containerProperties: ContainerPropertiesClass
+    container: ContainerClass
+    editableContainerProperties: EditableContainerPropertiesClass
+    dynamicFormAssignment: DynamicFormAssignmentClass
+    formInfo: FormInfoClass
+    queryUsageFeatures: QueryUsageFeaturesClass
+    querySubjects: QuerySubjectsClass
+    queryProperties: QueryPropertiesClass
+    queryUsageStatistics: QueryUsageStatisticsClass
+    editableNotebookProperties: EditableNotebookPropertiesClass
+    notebookInfo: NotebookInfoClass
+    notebookContent: NotebookContentClass
+    dataPlatformInfo: DataPlatformInfoClass
+    slackUserInfo: SlackUserInfoClass
+    editableDashboardProperties: EditableDashboardPropertiesClass
+    dashboardUsageStatistics: DashboardUsageStatisticsClass
+    dashboardInfo: DashboardInfoClass
+    dataProcessInstanceOutput: DataProcessInstanceOutputClass
+    dataProcessInstanceInput: DataProcessInstanceInputClass
+    dataProcessInstanceRunEvent: DataProcessInstanceRunEventClass
+    dataProcessInstanceRelationships: DataProcessInstanceRelationshipsClass
+    dataProcessInfo: DataProcessInfoClass
+    dataProcessInstanceProperties: DataProcessInstancePropertiesClass
+    subscriptionInfo: SubscriptionInfoClass
+    dataHubRetentionConfig: DataHubRetentionConfigClass
+    dataHubRemoteExecutorPoolInfo: RemoteExecutorPoolInfoClass
+    operation: OperationClass
+    share: ShareClass
+    incidentsSummary: IncidentsSummaryClass
+    browsePaths: BrowsePathsClass
+    dataTransformLogic: DataTransformLogicClass
+    subTypes: SubTypesClass
+    displayProperties: DisplayPropertiesClass
+    origin: OriginClass
+    institutionalMemory: InstitutionalMemoryClass
+    deprecation: DeprecationClass
+    assertionsSummary: AssertionsSummaryClass
+    versionProperties: VersionPropertiesClass
+    ownership: OwnershipClass
+    siblings: SiblingsClass
+    dataPlatformInstance: DataPlatformInstanceClass
+    forms: FormsClass
+    browsePathsV2: BrowsePathsV2Class
+    globalTags: GlobalTagsClass
+    glossaryTerms: GlossaryTermsClass
+    access: AccessClass
+    embed: EmbedClass
+    status: StatusClass
+    anomaliesSummary: AnomaliesSummaryClass
+    inputFields: InputFieldsClass
+    cost: CostClass
+    documentation: DocumentationClass
+    proposals: ProposalsClass
+    dataProductKey: DataProductKeyClass
+    dataProductProperties: DataProductPropertiesClass
+    dataHubConnectionDetails: DataHubConnectionDetailsClass
+    platformResourceInfo: PlatformResourceInfoClass
+    platformResourceKey: PlatformResourceKeyClass
+    actors: ActorsClass
+    roleProperties: RolePropertiesClass
+    dataHubRemoteExecutorPoolGlobalConfig: RemoteExecutorPoolGlobalConfigClass
+    assertionRunEvent: AssertionRunEventClass
+    assertionAnalyticsRunEvent: AssertionAnalyticsRunEventClass
+    assertionActions: AssertionActionsClass
+    assertionInferenceDetails: AssertionInferenceDetailsClass
+    assertionInfo: AssertionInfoClass
+    assertionDryRunEvent: AssertionDryRunEventClass
+    assertionSummary: AssertionSummaryClass
+    monitorInfo: MonitorInfoClass
+    monitorSuiteInfo: MonitorSuiteInfoClass
+    monitorTimeseriesState: MonitorTimeseriesStateClass
+    dataHubStepStateProperties: DataHubStepStatePropertiesClass
+    versionSetProperties: VersionSetPropertiesClass
+    schemaFieldsInferredNeighbors: SchemaFieldsInferredNeighborsClass
+    inferredMetadata: InferredMetadataClass
+    inferredNeighbors: InferredNeighborsClass
+    schemaFieldsInferredMetadata: SchemaFieldsInferredMetadataClass
+    incidentNotificationDetails: IncidentNotificationDetailsClass
+    incidentInfo: IncidentInfoClass
+    incidentActivityEvent: IncidentActivityEventClass
+    incidentSource: IncidentSourceClass
+    dataHubPolicyInfo: DataHubPolicyInfoClass
+    dataHubRoleInfo: DataHubRoleInfoClass
+    corpGroupSettings: CorpGroupSettingsClass
+    corpUserInfo: CorpUserInfoClass
+    corpGroupInfo: CorpGroupInfoClass
+    corpGroupEditableInfo: CorpGroupEditableInfoClass
+    inviteToken: InviteTokenClass
+    nativeGroupMembership: NativeGroupMembershipClass
+    roleMembership: RoleMembershipClass
+    groupMembership: GroupMembershipClass
+    corpUserStatus: CorpUserStatusClass
+    corpUserCredentials: CorpUserCredentialsClass
+    corpUserEditableInfo: CorpUserEditableInfoClass
+    corpUserSettings: CorpUserSettingsClass
+    dataHubMetricCubeDefinition: DataHubMetricCubeDefinitionClass
+    dataHubMetricCubeEvent: DataHubMetricCubeEventClass
+    postInfo: PostInfoClass
+    dataContractStatus: DataContractStatusClass
+    dataContractProperties: DataContractPropertiesClass
+    monitorAnomalyEvent: MonitorAnomalyEventClass
+    datasetUsageStatistics: DatasetUsageStatisticsClass
+    editableDatasetProperties: EditableDatasetPropertiesClass
+    upstreamLineage: UpstreamLineageClass
+    viewProperties: ViewPropertiesClass
+    partitionsSummary: PartitionsSummaryClass
+    schemaFieldProfile: SchemaFieldProfileClass
+    datasetProfile: DatasetProfileClass
+    datasetDeprecation: DatasetDeprecationClass
+    datasetUpstreamLineage: DatasetUpstreamLineageClass
+    datasetProperties: DatasetPropertiesClass
+    icebergCatalogInfo: IcebergCatalogInfoClass
+    dataHubPersonaInfo: DataHubPersonaInfoClass
 
 
 KEY_ASPECTS: Dict[str, Type[_Aspect]] = {
-    'dataProduct': DataProductKeyClass,
-    'dataType': DataTypeKeyClass,
-    'businessAttribute': BusinessAttributeKeyClass,
-    'structuredProperty': StructuredPropertyKeyClass,
-    'platformResource': PlatformResourceKeyClass,
     'entityType': EntityTypeKeyClass,
-    'corpGroup': CorpGroupKeyClass,
-    'ownershipType': OwnershipTypeKeyClass,
-    'recommendationModule': RecommendationModuleKeyClass,
-    'dataProcess': DataProcessKeyClass,
-    'mlFeature': MLFeatureKeyClass,
-    'dataHubRemoteExecutorGlobalConfig': RemoteExecutorGlobalConfigKeyClass,
-    'dataHubPersona': DataHubPersonaKeyClass,
-    'dataPlatformInstance': DataPlatformInstanceKeyClass,
-    'erModelRelationship': ERModelRelationshipKeyClass,
-    'container': ContainerKeyClass,
+    'dataHubOpenAPISchema': DataHubOpenAPISchemaKeyClass,
+    'query': QueryKeyClass,
+    'dataJob': DataJobKeyClass,
+    'dataHubAccessToken': DataHubAccessTokenKeyClass,
     'telemetry': TelemetryKeyClass,
-    'dataset': DatasetKeyClass,
-    'dataHubIngestionSource': DataHubIngestionSourceKeyClass,
-    'dataFlow': DataFlowKeyClass,
-    'dataHubSecret': DataHubSecretKeyClass,
-    'dataHubUpgrade': DataHubUpgradeKeyClass,
-    'dataContract': DataContractKeyClass,
-    'inviteToken': InviteTokenKeyClass,
-    'actionRequest': ActionRequestKeyClass,
     'chart': ChartKeyClass,
-    'post': PostKeyClass,
+    'dataHubView': DataHubViewKeyClass,
+    'dataset': DatasetKeyClass,
+    'dataPlatform': DataPlatformKeyClass,
+    'dataPlatformInstance': DataPlatformInstanceKeyClass,
+    'dataHubPolicy': DataHubPolicyKeyClass,
+    'dataHubExecutionRequest': ExecutionRequestKeyClass,
+    'dataHubRetention': DataHubRetentionKeyClass,
+    'dataHubStepState': DataHubStepStateKeyClass,
+    'recommendationModule': RecommendationModuleKeyClass,
+    'container': ContainerKeyClass,
+    'role': RoleKeyClass,
+    'dataProcessInstance': DataProcessInstanceKeyClass,
+    'domain': DomainKeyClass,
+    'erModelRelationship': ERModelRelationshipKeyClass,
+    'dataHubRemoteExecutorPool': RemoteExecutorPoolKeyClass,
+    'dataHubMetricCube': DataHubMetricCubeKeyClass,
     'globalSettings': GlobalSettingsKeyClass,
-    'glossaryTerm': GlossaryTermKeyClass,
+    'actionRequest': ActionRequestKeyClass,
+    'corpuser': CorpUserKeyClass,
+    'glossaryNode': GlossaryNodeKeyClass,
     'subscription': SubscriptionKeyClass,
-    'notebook': NotebookKeyClass,
-    'mlFeatureTable': MLFeatureTableKeyClass,
+    'mlPrimaryKey': MLPrimaryKeyKeyClass,
+    'test': TestKeyClass,
     'mlModelGroup': MLModelGroupKeyClass,
-    'dataHubExecutionRequest': ExecutionRequestKeyClass,
-    'dataProcessInstance': DataProcessInstanceKeyClass,
+    'dataFlow': DataFlowKeyClass,
+    'dataHubPersona': DataHubPersonaKeyClass,
     'monitor': MonitorKeyClass,
-    'incident': IncidentKeyClass,
-    'constraint': ConstraintKeyClass,
-    'role': RoleKeyClass,
-    'dashboard': DashboardKeyClass,
-    'glossaryNode': GlossaryNodeKeyClass,
-    'dataHubView': DataHubViewKeyClass,
-    'dataHubRetention': DataHubRetentionKeyClass,
-    'dataHubRemoteExecutorPool': RemoteExecutorPoolKeyClass,
-    'schemaField': SchemaFieldKeyClass,
     'versionSet': VersionSetKeyClass,
-    'dataJob': DataJobKeyClass,
-    'mlPrimaryKey': MLPrimaryKeyKeyClass,
-    'dataHubAccessToken': DataHubAccessTokenKeyClass,
-    'linkPreview': LinkPreviewKeyClass,
-    'dataHubRole': DataHubRoleKeyClass,
-    'dataHubMetricCube': DataHubMetricCubeKeyClass,
-    'dataHubAction': DataHubActionKeyClass,
-    'dataHubStepState': DataHubStepStateKeyClass,
+    'dataHubUpgrade': DataHubUpgradeKeyClass,
     'assertion': AssertionKeyClass,
-    'query': QueryKeyClass,
-    'mlModelDeployment': MLModelDeploymentKeyClass,
-    'mlModel': MLModelKeyClass,
+    'constraint': ConstraintKeyClass,
+    'corpGroup': CorpGroupKeyClass,
+    'dataHubAction': DataHubActionKeyClass,
+    'dataContract': DataContractKeyClass,
+    'dataHubRemoteExecutorGlobalConfig': RemoteExecutorGlobalConfigKeyClass,
+    'dataHubSecret': DataHubSecretKeyClass,
+    'form': FormKeyClass,
+    'mlFeatureTable': MLFeatureTableKeyClass,
+    'schemaField': SchemaFieldKeyClass,
     'tag': TagKeyClass,
+    'dataProcess': DataProcessKeyClass,
+    'glossaryTerm': GlossaryTermKeyClass,
+    'post': PostKeyClass,
+    'dashboard': DashboardKeyClass,
     'dataHubConnection': DataHubConnectionKeyClass,
+    'mlModel': MLModelKeyClass,
+    'dataHubIngestionSource': DataHubIngestionSourceKeyClass,
     'dataHubRemoteExecutor': RemoteExecutorKeyClass,
-    'domain': DomainKeyClass,
+    'mlFeature': MLFeatureKeyClass,
     'monitorSuite': MonitorSuiteKeyClass,
-    'dataHubPolicy': DataHubPolicyKeyClass,
-    'test': TestKeyClass,
-    'form': FormKeyClass,
-    'corpuser': CorpUserKeyClass,
-    'dataPlatform': DataPlatformKeyClass
+    'mlModelDeployment': MLModelDeploymentKeyClass,
+    'ownershipType': OwnershipTypeKeyClass,
+    'incident': IncidentKeyClass,
+    'inviteToken': InviteTokenKeyClass,
+    'linkPreview': LinkPreviewKeyClass,
+    'notebook': NotebookKeyClass,
+    'dataHubRole': DataHubRoleKeyClass,
+    'businessAttribute': BusinessAttributeKeyClass,
+    'structuredProperty': StructuredPropertyKeyClass,
+    'dataType': DataTypeKeyClass,
+    'dataProduct': DataProductKeyClass,
+    'platformResource': PlatformResourceKeyClass
 }
 
 ENTITY_TYPE_NAMES: List[str] = [
-    'dataProduct',
-    'dataType',
-    'businessAttribute',
-    'structuredProperty',
-    'platformResource',
     'entityType',
-    'corpGroup',
-    'ownershipType',
-    'recommendationModule',
-    'dataProcess',
-    'mlFeature',
-    'dataHubRemoteExecutorGlobalConfig',
-    'dataHubPersona',
-    'dataPlatformInstance',
-    'erModelRelationship',
-    'container',
+    'dataHubOpenAPISchema',
+    'query',
+    'dataJob',
+    'dataHubAccessToken',
     'telemetry',
-    'dataset',
-    'dataHubIngestionSource',
-    'dataFlow',
-    'dataHubSecret',
-    'dataHubUpgrade',
-    'dataContract',
-    'inviteToken',
-    'actionRequest',
     'chart',
-    'post',
+    'dataHubView',
+    'dataset',
+    'dataPlatform',
+    'dataPlatformInstance',
+    'dataHubPolicy',
+    'dataHubExecutionRequest',
+    'dataHubRetention',
+    'dataHubStepState',
+    'recommendationModule',
+    'container',
+    'role',
+    'dataProcessInstance',
+    'domain',
+    'erModelRelationship',
+    'dataHubRemoteExecutorPool',
+    'dataHubMetricCube',
     'globalSettings',
-    'glossaryTerm',
+    'actionRequest',
+    'corpuser',
+    'glossaryNode',
     'subscription',
-    'notebook',
-    'mlFeatureTable',
+    'mlPrimaryKey',
+    'test',
     'mlModelGroup',
-    'dataHubExecutionRequest',
-    'dataProcessInstance',
+    'dataFlow',
+    'dataHubPersona',
     'monitor',
-    'incident',
-    'constraint',
-    'role',
-    'dashboard',
-    'glossaryNode',
-    'dataHubView',
-    'dataHubRetention',
-    'dataHubRemoteExecutorPool',
-    'schemaField',
     'versionSet',
-    'dataJob',
-    'mlPrimaryKey',
-    'dataHubAccessToken',
-    'linkPreview',
-    'dataHubRole',
-    'dataHubMetricCube',
-    'dataHubAction',
-    'dataHubStepState',
+    'dataHubUpgrade',
     'assertion',
-    'query',
-    'mlModelDeployment',
-    'mlModel',
+    'constraint',
+    'corpGroup',
+    'dataHubAction',
+    'dataContract',
+    'dataHubRemoteExecutorGlobalConfig',
+    'dataHubSecret',
+    'form',
+    'mlFeatureTable',
+    'schemaField',
     'tag',
+    'dataProcess',
+    'glossaryTerm',
+    'post',
+    'dashboard',
     'dataHubConnection',
+    'mlModel',
+    'dataHubIngestionSource',
     'dataHubRemoteExecutor',
-    'domain',
+    'mlFeature',
     'monitorSuite',
-    'dataHubPolicy',
-    'test',
-    'form',
-    'corpuser',
-    'dataPlatform'
-]
-EntityTypeName = Literal[
-    'dataProduct',
-    'dataType',
+    'mlModelDeployment',
+    'ownershipType',
+    'incident',
+    'inviteToken',
+    'linkPreview',
+    'notebook',
+    'dataHubRole',
     'businessAttribute',
     'structuredProperty',
-    'platformResource',
+    'dataType',
+    'dataProduct',
+    'platformResource'
+]
+EntityTypeName = Literal[
     'entityType',
-    'corpGroup',
-    'ownershipType',
-    'recommendationModule',
-    'dataProcess',
-    'mlFeature',
-    'dataHubRemoteExecutorGlobalConfig',
-    'dataHubPersona',
-    'dataPlatformInstance',
-    'erModelRelationship',
-    'container',
+    'dataHubOpenAPISchema',
+    'query',
+    'dataJob',
+    'dataHubAccessToken',
     'telemetry',
-    'dataset',
-    'dataHubIngestionSource',
-    'dataFlow',
-    'dataHubSecret',
-    'dataHubUpgrade',
-    'dataContract',
-    'inviteToken',
-    'actionRequest',
     'chart',
-    'post',
+    'dataHubView',
+    'dataset',
+    'dataPlatform',
+    'dataPlatformInstance',
+    'dataHubPolicy',
+    'dataHubExecutionRequest',
+    'dataHubRetention',
+    'dataHubStepState',
+    'recommendationModule',
+    'container',
+    'role',
+    'dataProcessInstance',
+    'domain',
+    'erModelRelationship',
+    'dataHubRemoteExecutorPool',
+    'dataHubMetricCube',
     'globalSettings',
-    'glossaryTerm',
+    'actionRequest',
+    'corpuser',
+    'glossaryNode',
     'subscription',
-    'notebook',
-    'mlFeatureTable',
+    'mlPrimaryKey',
+    'test',
     'mlModelGroup',
-    'dataHubExecutionRequest',
-    'dataProcessInstance',
+    'dataFlow',
+    'dataHubPersona',
     'monitor',
-    'incident',
-    'constraint',
-    'role',
-    'dashboard',
-    'glossaryNode',
-    'dataHubView',
-    'dataHubRetention',
-    'dataHubRemoteExecutorPool',
-    'schemaField',
     'versionSet',
-    'dataJob',
-    'mlPrimaryKey',
-    'dataHubAccessToken',
-    'linkPreview',
-    'dataHubRole',
-    'dataHubMetricCube',
-    'dataHubAction',
-    'dataHubStepState',
+    'dataHubUpgrade',
     'assertion',
-    'query',
-    'mlModelDeployment',
-    'mlModel',
+    'constraint',
+    'corpGroup',
+    'dataHubAction',
+    'dataContract',
+    'dataHubRemoteExecutorGlobalConfig',
+    'dataHubSecret',
+    'form',
+    'mlFeatureTable',
+    'schemaField',
     'tag',
+    'dataProcess',
+    'glossaryTerm',
+    'post',
+    'dashboard',
     'dataHubConnection',
+    'mlModel',
+    'dataHubIngestionSource',
     'dataHubRemoteExecutor',
-    'domain',
+    'mlFeature',
     'monitorSuite',
-    'dataHubPolicy',
-    'test',
-    'form',
-    'corpuser',
-    'dataPlatform'
+    'mlModelDeployment',
+    'ownershipType',
+    'incident',
+    'inviteToken',
+    'linkPreview',
+    'notebook',
+    'dataHubRole',
+    'businessAttribute',
+    'structuredProperty',
+    'dataType',
+    'dataProduct',
+    'platformResource'
 ]
 
 # fmt: on