Merge pull request #920 from atlanhq/sync-models-BLDX-1206

feat(models): sync pyatlan_v9 models from models@BLDX-1206 + fix 4 collection errors [BLDX-1206]

Author: Aryamanz29

pyatlan_v9/model/aio/translators.py

@@ -60,7 +60,7 @@ async def translate(self, data: dict[str, Any]) -> dict[str, Any]:
                 tag_names = []
                 for tag_id in raw_json[key]:
                     name = await self.client.atlan_tag_cache.get_name_for_id(tag_id)
-                    tag_names.append(AtlanTagName(name or DELETED_))
+                    tag_names.append(name or DELETED_)
                 raw_json[key] = tag_names
 
         for key in self._CLASSIFICATION_KEYS:
@@ -82,11 +82,10 @@ async def translate(self, data: dict[str, Any]) -> dict[str, Any]:
                 if not attr_id:
                     continue
                 attributes = classification.get("attributes")
-                if not attributes or not attributes.get(attr_id):
-                    continue
+                source_tags = attributes.get(attr_id) if attributes else None
                 classification[self._SOURCE_ATTACHMENTS] = [
                     msgspec.convert(source_tag["attributes"], type=SourceTagAttachment)
-                    for source_tag in attributes.get(attr_id)
+                    for source_tag in (source_tags or [])
                     if isinstance(source_tag, dict) and source_tag.get("attributes")
                 ]
 

pyatlan_v9/model/assets/__init__.py

@@ -45,6 +45,7 @@
         "RelatedADLSContainer",
         "RelatedADLSObject",
     ],
+    "_init_agentic": ["Agentic", "RelatedAgentic"],
     "_init_ai": [
         "AI",
         "AIApplication",
@@ -116,6 +117,7 @@
         "RelatedApplicationField",
     ],
     "_init_app_workflow_run": ["AppWorkflowRun", "RelatedAppWorkflowRun"],
+    "_init_artifact": ["Artifact", "RelatedArtifact"],
     "_init_asset": [
         "Asset",
         "DataSet",
@@ -254,6 +256,14 @@
         "RelatedCognosReport",
     ],
     "_init_connection": ["Connection", "RelatedConnection"],
+    "_init_context": [
+        "Context",
+        "ContextArtifact",
+        "ContextRepository",
+        "RelatedContext",
+        "RelatedContextArtifact",
+        "RelatedContextRepository",
+    ],
     "_init_cosmos_mongo_db": [
         "CosmosMongoDB",
         "CosmosMongoDBAccount",
@@ -473,6 +483,12 @@
         "RelatedFlowReusableUnit",
     ],
     "_init_form": ["Form", "RelatedForm", "RelatedResponse"],
+    "_init_gcp_dataplex": [
+        "GCPDataplex",
+        "GCPDataplexAspectType",
+        "RelatedGCPDataplex",
+        "RelatedGCPDataplexAspectType",
+    ],
     "_init_gcs": [
         "GCS",
         "GCSBucket",
@@ -515,6 +531,14 @@
         "RelatedKafkaField",
         "RelatedKafkaTopic",
     ],
+    "_init_knowledge": [
+        "Knowledge",
+        "KnowledgeFile",
+        "KnowledgeFolder",
+        "RelatedKnowledge",
+        "RelatedKnowledgeFile",
+        "RelatedKnowledgeFolder",
+    ],
     "_init_looker": [
         "Looker",
         "LookerDashboard",
@@ -904,6 +928,8 @@
         "SisenseFolder",
         "SisenseWidget",
     ],
+    "_init_skill": ["RelatedSkill", "Skill"],
+    "_init_skill_artifact": ["RelatedSkillArtifact", "SkillArtifact"],
     "_init_snowflake": [
         "RelatedSnowflake",
         "RelatedSnowflakeAIModelContext",
@@ -1047,6 +1073,16 @@
         "ThoughtspotView",
         "ThoughtspotWorksheet",
     ],
+    "_init_unstructured_v2": [
+        "RelatedUnstructuredV2",
+        "RelatedUnstructuredV2Container",
+        "RelatedUnstructuredV2Folder",
+        "RelatedUnstructuredV2Object",
+        "UnstructuredV2",
+        "UnstructuredV2Container",
+        "UnstructuredV2Folder",
+        "UnstructuredV2Object",
+    ],
     "_init_workflow": ["RelatedWorkflow", "RelatedWorkflowRun", "Workflow"],
 }
 

pyatlan_v9/model/assets/__init__.pyi

@@ -38,6 +38,8 @@ __all__ = [
     "RelatedADLSAccount",
     "RelatedADLSContainer",
     "RelatedADLSObject",
+    "Agentic",
+    "RelatedAgentic",
     "AI",
     "AIApplication",
     "AIModel",
@@ -98,6 +100,8 @@ __all__ = [
     "RelatedApplicationField",
     "AppWorkflowRun",
     "RelatedAppWorkflowRun",
+    "Artifact",
+    "RelatedArtifact",
     "Asset",
     "DataSet",
     "Incident",
@@ -215,6 +219,12 @@ __all__ = [
     "RelatedCognosReport",
     "Connection",
     "RelatedConnection",
+    "Context",
+    "ContextArtifact",
+    "ContextRepository",
+    "RelatedContext",
+    "RelatedContextArtifact",
+    "RelatedContextRepository",
     "CosmosMongoDB",
     "CosmosMongoDBAccount",
     "CosmosMongoDBCollection",
@@ -405,6 +415,10 @@ __all__ = [
     "Form",
     "RelatedForm",
     "RelatedResponse",
+    "GCPDataplex",
+    "GCPDataplexAspectType",
+    "RelatedGCPDataplex",
+    "RelatedGCPDataplexAspectType",
     "GCS",
     "GCSBucket",
     "GCSObject",
@@ -439,6 +453,12 @@ __all__ = [
     "RelatedKafkaConsumerGroup",
     "RelatedKafkaField",
     "RelatedKafkaTopic",
+    "Knowledge",
+    "KnowledgeFile",
+    "KnowledgeFolder",
+    "RelatedKnowledge",
+    "RelatedKnowledgeFile",
+    "RelatedKnowledgeFolder",
     "Looker",
     "LookerDashboard",
     "LookerExplore",
@@ -776,6 +796,10 @@ __all__ = [
     "SisenseDatamodelTable",
     "SisenseFolder",
     "SisenseWidget",
+    "RelatedSkill",
+    "Skill",
+    "RelatedSkillArtifact",
+    "SkillArtifact",
     "RelatedSnowflake",
     "RelatedSnowflakeAIModelContext",
     "RelatedSnowflakeAIModelVersion",
@@ -900,6 +924,14 @@ __all__ = [
     "ThoughtspotTable",
     "ThoughtspotView",
     "ThoughtspotWorksheet",
+    "RelatedUnstructuredV2",
+    "RelatedUnstructuredV2Container",
+    "RelatedUnstructuredV2Folder",
+    "RelatedUnstructuredV2Object",
+    "UnstructuredV2",
+    "UnstructuredV2Container",
+    "UnstructuredV2Folder",
+    "UnstructuredV2Object",
     "RelatedWorkflow",
     "RelatedWorkflowRun",
     "Workflow",
@@ -931,6 +963,8 @@ from .adls_related import RelatedADLS as RelatedADLS
 from .adls_related import RelatedADLSAccount as RelatedADLSAccount
 from .adls_related import RelatedADLSContainer as RelatedADLSContainer
 from .adls_related import RelatedADLSObject as RelatedADLSObject
+from .agentic import Agentic as Agentic
+from .agentic_related import RelatedAgentic as RelatedAgentic
 from .ai import AI as AI
 from .ai_application import AIApplication as AIApplication
 from .ai_model import AIModel as AIModel
@@ -993,6 +1027,8 @@ from .app_workflow_run import AppWorkflowRun as AppWorkflowRun
 from .app_workflow_run_related import RelatedAppWorkflowRun as RelatedAppWorkflowRun
 from .application import Application as Application
 from .application_field import ApplicationField as ApplicationField
+from .artifact import Artifact as Artifact
+from .artifact_related import RelatedArtifact as RelatedArtifact
 from .asset import Asset as Asset
 from .asset_grouping import AssetGrouping as AssetGrouping
 from .asset_grouping_collection import (
@@ -1138,6 +1174,12 @@ from .column import Column as Column
 from .column_process import ColumnProcess as ColumnProcess
 from .connection import Connection as Connection
 from .connection_related import RelatedConnection as RelatedConnection
+from .context import Context as Context
+from .context_artifact import ContextArtifact as ContextArtifact
+from .context_related import RelatedContext as RelatedContext
+from .context_related import RelatedContextArtifact as RelatedContextArtifact
+from .context_related import RelatedContextRepository as RelatedContextRepository
+from .context_repository import ContextRepository as ContextRepository
 from .cosmos_mongo_db import CosmosMongoDB as CosmosMongoDB
 from .cosmos_mongo_db_account import CosmosMongoDBAccount as CosmosMongoDBAccount
 from .cosmos_mongo_db_collection import (
@@ -1391,6 +1433,12 @@ from .form import Form as Form
 from .form_related import RelatedForm as RelatedForm
 from .form_related import RelatedResponse as RelatedResponse
 from .function import Function as Function
+from .gcp_dataplex import GCPDataplex as GCPDataplex
+from .gcp_dataplex_aspect_type import GCPDataplexAspectType as GCPDataplexAspectType
+from .gcp_dataplex_related import RelatedGCPDataplex as RelatedGCPDataplex
+from .gcp_dataplex_related import (
+    RelatedGCPDataplexAspectType as RelatedGCPDataplexAspectType,
+)
 from .gcs import GCS as GCS
 from .gcs_bucket import GCSBucket as GCSBucket
 from .gcs_object import GCSObject as GCSObject
@@ -1428,6 +1476,12 @@ from .kafka_related import RelatedKafkaConsumerGroup as RelatedKafkaConsumerGrou
 from .kafka_related import RelatedKafkaField as RelatedKafkaField
 from .kafka_related import RelatedKafkaTopic as RelatedKafkaTopic
 from .kafka_topic import KafkaTopic as KafkaTopic
+from .knowledge import Knowledge as Knowledge
+from .knowledge_file import KnowledgeFile as KnowledgeFile
+from .knowledge_folder import KnowledgeFolder as KnowledgeFolder
+from .knowledge_related import RelatedKnowledge as RelatedKnowledge
+from .knowledge_related import RelatedKnowledgeFile as RelatedKnowledgeFile
+from .knowledge_related import RelatedKnowledgeFolder as RelatedKnowledgeFolder
 from .link import Link as Link
 from .looker import Looker as Looker
 from .looker_dashboard import LookerDashboard as LookerDashboard
@@ -1833,6 +1887,10 @@ from .sisense_related import (
 from .sisense_related import RelatedSisenseFolder as RelatedSisenseFolder
 from .sisense_related import RelatedSisenseWidget as RelatedSisenseWidget
 from .sisense_widget import SisenseWidget as SisenseWidget
+from .skill import Skill as Skill
+from .skill_artifact import SkillArtifact as SkillArtifact
+from .skill_artifact_related import RelatedSkillArtifact as RelatedSkillArtifact
+from .skill_related import RelatedSkill as RelatedSkill
 from .snowflake import Snowflake as Snowflake
 from .snowflake_ai_model_context import (
     SnowflakeAIModelContext as SnowflakeAIModelContext,
@@ -1993,6 +2051,22 @@ from .thoughtspot_related import (
 from .thoughtspot_table import ThoughtspotTable as ThoughtspotTable
 from .thoughtspot_view import ThoughtspotView as ThoughtspotView
 from .thoughtspot_worksheet import ThoughtspotWorksheet as ThoughtspotWorksheet
+from .unstructured_v2 import UnstructuredV2 as UnstructuredV2
+from .unstructured_v2_container import (
+    UnstructuredV2Container as UnstructuredV2Container,
+)
+from .unstructured_v2_folder import UnstructuredV2Folder as UnstructuredV2Folder
+from .unstructured_v2_object import UnstructuredV2Object as UnstructuredV2Object
+from .unstructured_v2_related import RelatedUnstructuredV2 as RelatedUnstructuredV2
+from .unstructured_v2_related import (
+    RelatedUnstructuredV2Container as RelatedUnstructuredV2Container,
+)
+from .unstructured_v2_related import (
+    RelatedUnstructuredV2Folder as RelatedUnstructuredV2Folder,
+)
+from .unstructured_v2_related import (
+    RelatedUnstructuredV2Object as RelatedUnstructuredV2Object,
+)
 from .view import View as View
 from .workflow import Workflow as Workflow
 from .workflow_related import RelatedWorkflow as RelatedWorkflow

pyatlan_v9/model/assets/_init_agentic.py

@@ -0,0 +1,17 @@
+# Auto-generated by PythonMsgspecRenderer.pkl - DO NOT EDIT
+# SPDX-License-Identifier: Apache-2.0
+# Copyright 2024 Atlan Pte. Ltd.
+
+"""
+Agentic module exports.
+
+This module provides convenient imports for all Agentic types and their Related variants.
+"""
+
+from .agentic import Agentic
+from .agentic_related import RelatedAgentic
+
+__all__ = [
+    "Agentic",
+    "RelatedAgentic",
+]

pyatlan_v9/model/assets/_init_artifact.py

@@ -0,0 +1,17 @@
+# Auto-generated by PythonMsgspecRenderer.pkl - DO NOT EDIT
+# SPDX-License-Identifier: Apache-2.0
+# Copyright 2024 Atlan Pte. Ltd.
+
+"""
+Artifact module exports.
+
+This module provides convenient imports for all Artifact types and their Related variants.
+"""
+
+from .artifact import Artifact
+from .artifact_related import RelatedArtifact
+
+__all__ = [
+    "Artifact",
+    "RelatedArtifact",
+]

pyatlan_v9/model/assets/_init_context.py

@@ -0,0 +1,27 @@
+# Auto-generated by PythonMsgspecRenderer.pkl - DO NOT EDIT
+# SPDX-License-Identifier: Apache-2.0
+# Copyright 2024 Atlan Pte. Ltd.
+
+"""
+Context module exports.
+
+This module provides convenient imports for all Context types and their Related variants.
+"""
+
+from .context import Context
+from .context_artifact import ContextArtifact
+from .context_related import (
+    RelatedContext,
+    RelatedContextArtifact,
+    RelatedContextRepository,
+)
+from .context_repository import ContextRepository
+
+__all__ = [
+    "Context",
+    "ContextArtifact",
+    "ContextRepository",
+    "RelatedContext",
+    "RelatedContextArtifact",
+    "RelatedContextRepository",
+]

pyatlan_v9/model/assets/_init_gcp_dataplex.py

@@ -0,0 +1,20 @@
+# Auto-generated by PythonMsgspecRenderer.pkl - DO NOT EDIT
+# SPDX-License-Identifier: Apache-2.0
+# Copyright 2024 Atlan Pte. Ltd.
+
+"""
+GCPDataplex module exports.
+
+This module provides convenient imports for all GCPDataplex types and their Related variants.
+"""
+
+from .gcp_dataplex import GCPDataplex
+from .gcp_dataplex_aspect_type import GCPDataplexAspectType
+from .gcp_dataplex_related import RelatedGCPDataplex, RelatedGCPDataplexAspectType
+
+__all__ = [
+    "GCPDataplex",
+    "GCPDataplexAspectType",
+    "RelatedGCPDataplex",
+    "RelatedGCPDataplexAspectType",
+]

pyatlan_v9/model/assets/_init_knowledge.py

@@ -0,0 +1,27 @@
+# Auto-generated by PythonMsgspecRenderer.pkl - DO NOT EDIT
+# SPDX-License-Identifier: Apache-2.0
+# Copyright 2024 Atlan Pte. Ltd.
+
+"""
+Knowledge module exports.
+
+This module provides convenient imports for all Knowledge types and their Related variants.
+"""
+
+from .knowledge import Knowledge
+from .knowledge_file import KnowledgeFile
+from .knowledge_folder import KnowledgeFolder
+from .knowledge_related import (
+    RelatedKnowledge,
+    RelatedKnowledgeFile,
+    RelatedKnowledgeFolder,
+)
+
+__all__ = [
+    "Knowledge",
+    "KnowledgeFile",
+    "KnowledgeFolder",
+    "RelatedKnowledge",
+    "RelatedKnowledgeFile",
+    "RelatedKnowledgeFolder",
+]