Merge pull request #901 from atlanhq/TTD-105-agentic-types

TTD-105: Add Agentic, Skill, and Context Repository typedefs

Author: Aryamanz29

README.md

@@ -208,8 +208,14 @@ Generate asset models from your Atlan instance:
 # Generate models automatically
 uv run ./generator
 
+# Force re-download typedefs (bypass cache)
+uv run ./generator --override
+
 # Use custom typedefs file
 uv run ./generator ./my-typedefs.json
+
+# Both flags can be combined
+uv run ./generator --override ./my-typedefs.json
 ```
 
 This will:

docs/api/assets/cloud-storage.md

@@ -32,10 +32,6 @@
 
 ::: pyatlan.model.assets.a_d_l_s_object.ADLSObject
 
-## GCS
-
-::: pyatlan.model.assets.g_c_s.GCS
-
 ## GCSBucket
 
 ::: pyatlan.model.assets.g_c_s_bucket.GCSBucket
@@ -52,14 +48,6 @@
 
 ::: pyatlan.model.assets.a_w_s.AWS
 
-## Google
-
-::: pyatlan.model.assets.google.Google
-
 ## Azure
 
 ::: pyatlan.model.assets.azure.Azure
-
-## Cloud
-
-::: pyatlan.model.assets.cloud.Cloud

docs/api/assets/other-bi.md

@@ -348,10 +348,6 @@
 
 ::: pyatlan.model.assets.sisense_widget.SisenseWidget
 
-## DataStudio
-
-::: pyatlan.model.assets.data_studio.DataStudio
-
 ## DataStudioAsset
 
 ::: pyatlan.model.assets.data_studio_asset.DataStudioAsset

docs/api/assets/other-connectors.md

@@ -11,27 +11,3 @@
 ## DataverseEntity
 
 ::: pyatlan.model.assets.dataverse_entity.DataverseEntity
-
-## DremioVirtualDataset
-
-::: pyatlan.model.assets.dremio_virtual_dataset.DremioVirtualDataset
-
-## DremioColumn
-
-::: pyatlan.model.assets.dremio_column.DremioColumn
-
-## DremioSpace
-
-::: pyatlan.model.assets.dremio_space.DremioSpace
-
-## DremioPhysicalDataset
-
-::: pyatlan.model.assets.dremio_physical_dataset.DremioPhysicalDataset
-
-## DremioFolder
-
-::: pyatlan.model.assets.dremio_folder.DremioFolder
-
-## DremioSource
-
-::: pyatlan.model.assets.dremio_source.DremioSource

generator

@@ -4,11 +4,26 @@
 # This script runs both create_typedefs_file.py and class_generator.py
 # It intelligently skips creating typedefs if they already exist and are current
 
-# Usage: ./generator [typedefs_file_path]
+# Usage: ./generator [--override] [typedefs_file_path]
 # If typedefs_file_path is not provided, defaults to /tmp/typedefs.json
+# --override: force re-download of typedefs even if cached file is current
 
 echo "🚀 Starting Atlan Python SDK code generation..."
 
+# Parse flags
+FORCE_OVERRIDE=false
+TYPE_DEF_FILE_ARG=""
+for arg in "$@"; do
+    case "$arg" in
+        --override)
+            FORCE_OVERRIDE=true
+            ;;
+        *)
+            TYPE_DEF_FILE_ARG="$arg"
+            ;;
+    esac
+done
+
 # Check if ATLAN_BASE_URL and ATLAN_API_KEY are set
 if [ -z "$ATLAN_BASE_URL" ] || [ -z "$ATLAN_API_KEY" ]; then
     echo "❌ Error: ATLAN_BASE_URL and ATLAN_API_KEY environment variables must be set."
@@ -19,8 +34,8 @@ if [ -z "$ATLAN_BASE_URL" ] || [ -z "$ATLAN_API_KEY" ]; then
 fi
 
 # Get the typedefs file path from command line argument or use default
-if [ -n "$1" ]; then
-    TYPE_DEF_FILE="$1"
+if [ -n "$TYPE_DEF_FILE_ARG" ]; then
+    TYPE_DEF_FILE="$TYPE_DEF_FILE_ARG"
     echo "📁 Using custom typedefs file: $TYPE_DEF_FILE"
 else
     TMPDIR=${TMPDIR:-/tmp}
@@ -30,7 +45,9 @@ fi
 
 # Check if typedefs file exists and is current (created today)
 SHOULD_CREATE_TYPEDEFS=true
-if [ -f "$TYPE_DEF_FILE" ]; then
+if [ "$FORCE_OVERRIDE" = true ]; then
+    echo "🔄 --override flag set, forcing typedefs re-download"
+elif [ -f "$TYPE_DEF_FILE" ]; then
     # Check if file was created today
     if [ "$(date -r "$TYPE_DEF_FILE" +%Y-%m-%d)" = "$(date +%Y-%m-%d)" ]; then
         echo "✅ Typedefs file already exists and is current: $TYPE_DEF_FILE"

pyatlan/generator/class_generator.py

@@ -269,6 +269,18 @@ class AssetInfo:
         "FabricDataflowEntityColumn",
         "FabricReport",
         "FabricSemanticModel",
+        "Agentic",
+        "Artifact",
+        "Skill",
+        "SkillArtifact",
+        "Context",
+        "ContextRepository",
+        "ContextArtifact",
+        "Dremio",
+        "Cloud",
+        "Google",
+        "GCPDataplex",
+        "GCPDataplexAspectType",
     }
     _IGNORE_ASSETS = {}  # type: ignore[var-annotated]
 

pyatlan/generator/templates/methods/asset/context_artifact.jinja2

@@ -0,0 +1,21 @@
+
+    @classmethod
+    @init_guid
+    def creator(
+        cls,
+        *,
+        name: str,
+        context_repository_qualified_name: str,
+        file_type: FileType,
+    ) -> ContextArtifact:
+        validate_required_fields(
+            ["name", "context_repository_qualified_name", "file_type"],
+            [name, context_repository_qualified_name, file_type],
+        )
+        return ContextArtifact(
+            attributes=ContextArtifact.Attributes.creator(
+                name=name,
+                context_repository_qualified_name=context_repository_qualified_name,
+                file_type=file_type,
+            )
+        )

pyatlan/generator/templates/methods/asset/context_repository.jinja2

@@ -0,0 +1,8 @@
+
+    @classmethod
+    @init_guid
+    def creator(cls, *, name: str) -> ContextRepository:
+        validate_required_fields(["name"], [name])
+        return ContextRepository(
+            attributes=ContextRepository.Attributes.creator(name=name)
+        )

pyatlan/generator/templates/methods/asset/skill.jinja2

@@ -0,0 +1,6 @@
+
+    @classmethod
+    @init_guid
+    def creator(cls, *, name: str) -> Skill:
+        validate_required_fields(["name"], [name])
+        return Skill(attributes=Skill.Attributes.creator(name=name))

pyatlan/generator/templates/methods/asset/skill_artifact.jinja2

@@ -0,0 +1,17 @@
+
+    @classmethod
+    @init_guid
+    def creator(
+        cls, *, name: str, skill_qualified_name: str, file_type: FileType
+    ) -> SkillArtifact:
+        validate_required_fields(
+            ["name", "skill_qualified_name", "file_type"],
+            [name, skill_qualified_name, file_type],
+        )
+        return SkillArtifact(
+            attributes=SkillArtifact.Attributes.creator(
+                name=name,
+                skill_qualified_name=skill_qualified_name,
+                file_type=file_type,
+            )
+        )