[change] Remove duplicated dataclass/enum models, re-export from legacy

- search.py: delete 25 duplicated dataclass/ABC/Enum classes and ABC
  registration block; re-export from pyatlan.model.search. Keep only
  msgspec DSL/IndexSearchRequest/IndexSearchRequestMetadata + v9 helpers.
- lineage.py: delete duplicated DirectedPair/LineageGraph; re-export
  from pyatlan.model.lineage.
- audit.py: delete duplicated AuditActionType; re-export from
  pyatlan.model.audit.
- pyatlan/model/search.py: TermAttributes/TextAttributes use plain
  str/bool/float instead of Pydantic StrictStr/StrictBool/StrictFloat.
- pyatlan/validate.py: add _is_model_instance helper for cross-boundary
  Pydantic/msgspec isinstance checks.
- pyatlan/client/common/asset.py: register msgspec.Struct in Pydantic
  ENCODERS_BY_TYPE for JSON serialization.
- core.py: add to_dict() on BulkRequest for nested serialization.
- entity.py: add semantic field to Entity base class.
- asset.py: ref_by_guid/ref_by_qualified_name accept semantic param.
- tests: update VALUES_BY_TYPE to plain types, use _is_model_instance
  for cross-boundary assertions, clean up test_client.py imports.

Author: Aryamanz29

pyatlan/client/common/asset.py

@@ -112,7 +112,7 @@ def _ensure_type_filter_present(criteria: IndexSearchRequest) -> None:
             criteria
             and criteria.dsl
             and criteria.dsl.query
-            and isinstance(criteria.dsl.query, Bool)
+            and _is_model_instance(criteria.dsl.query, Bool)
         ):
             return
 
@@ -126,7 +126,8 @@ def _ensure_type_filter_present(criteria: IndexSearchRequest) -> None:
 
         def needs_type_filter(clause: Optional[List]) -> bool:
             return not any(
-                isinstance(f, (Term, Terms)) and f.field in (type_name_fields)
+                (_is_model_instance(f, Term) or _is_model_instance(f, Terms))
+                and f.field in (type_name_fields)
                 for f in clause or []
             )
 

pyatlan/model/search.py

@@ -50,41 +50,41 @@ def __new__(cls, value: str, attribute_type: type) -> "Attributes":
 
 class TermAttributes(Attributes):
     CONNECTOR_NAME = ("connectorName", AtlanConnectorType)
-    CATEGORIES = ("__categories", StrictStr)
+    CATEGORIES = ("__categories", str)
     CREATE_TIME_AS_TIMESTAMP = ("__timestamp", datetime)
-    CREATED_BY = ("__createdBy", StrictStr)
-    GLOSSARY = ("__glossary", StrictStr)
-    GUID = ("__guid", StrictStr)
-    HAS_LINEAGE = ("__hasLineage", StrictBool)
-    MEANINGS = ("__meanings", StrictStr)
-    MODIFIED_BY = ("__modifiedBy", StrictStr)
-    NAME = ("name.keyword", StrictStr)
-    OWNER_GROUPS = ("ownerGroups", StrictStr)
-    OWNER_USERS = ("ownerUsers", StrictStr)
-    PARENT_CATEGORY = ("__parentCategory", StrictStr)
+    CREATED_BY = ("__createdBy", str)
+    GLOSSARY = ("__glossary", str)
+    GUID = ("__guid", str)
+    HAS_LINEAGE = ("__hasLineage", bool)
+    MEANINGS = ("__meanings", str)
+    MODIFIED_BY = ("__modifiedBy", str)
+    NAME = ("name.keyword", str)
+    OWNER_GROUPS = ("ownerGroups", str)
+    OWNER_USERS = ("ownerUsers", str)
+    PARENT_CATEGORY = ("__parentCategory", str)
     POPULARITY_SCORE = ("popularityScore", float)
-    QUALIFIED_NAME = ("qualifiedName", StrictStr)
+    QUALIFIED_NAME = ("qualifiedName", str)
     STATE = ("__state", Literal["ACTIVE", "DELETED", "PURGED"])
-    SUPER_TYPE_NAMES = ("__superTypeNames.keyword", StrictStr)
-    TYPE_NAME = ("__typeName.keyword", StrictStr)
+    SUPER_TYPE_NAMES = ("__superTypeNames.keyword", str)
+    TYPE_NAME = ("__typeName.keyword", str)
     UPDATE_TIME_AS_TIMESTAMP = ("__modificationTimestamp", datetime)
     CERTIFICATE_STATUS = ("certificateStatus", CertificateStatus)
 
 
 class TextAttributes(Attributes):
-    CLASSIFICATION_NAMES = ("__classificationNames", StrictStr)
-    CLASSIFICATIONS_TEXT = ("__classificationsText", StrictStr)
-    CREATE_TIME_AS_DATE = ("__timestamp.date", StrictStr)
-    DESCRIPTION = ("description", StrictStr)
-    MEANINGS_TEXT = ("__meaningsText", StrictStr)
-    NAME = ("name", StrictStr)
-    QUALIFIED_NAME = ("qualifiedName.text", StrictStr)
-    PROPAGATED_CLASSIFICATION_NAMES = ("__propagatedClassificationNames", StrictStr)
-    PROPAGATED_TRAIT_NAMES = ("__propagatedTraitNames", StrictStr)
-    SUPER_TYPE_NAMES = ("__superTypeNames", StrictStr)
-    TRAIT_NAMES = ("__traitNames", StrictStr)
-    UPDATE_TIME_AS_DATE = ("__modificationTimestamp.date", StrictStr)
-    USER_DESCRIPTION = ("userDescription", StrictStr)
+    CLASSIFICATION_NAMES = ("__classificationNames", str)
+    CLASSIFICATIONS_TEXT = ("__classificationsText", str)
+    CREATE_TIME_AS_DATE = ("__timestamp.date", str)
+    DESCRIPTION = ("description", str)
+    MEANINGS_TEXT = ("__meaningsText", str)
+    NAME = ("name", str)
+    QUALIFIED_NAME = ("qualifiedName.text", str)
+    PROPAGATED_CLASSIFICATION_NAMES = ("__propagatedClassificationNames", str)
+    PROPAGATED_TRAIT_NAMES = ("__propagatedTraitNames", str)
+    SUPER_TYPE_NAMES = ("__superTypeNames", str)
+    TRAIT_NAMES = ("__traitNames", str)
+    UPDATE_TIME_AS_DATE = ("__modificationTimestamp.date", str)
+    USER_DESCRIPTION = ("userDescription", str)
 
 
 def get_with_string(attribute: TermAttributes):

pyatlan/validate.py

@@ -84,6 +84,13 @@ class with the same name as *expected_type*.
         if expected_type.__name__ in spec_mro_names:
             return True
 
+    # Generic fallback: match by class name in MRO for any class
+    # (handles dataclasses, plain classes, etc. with identical names
+    # across legacy and v9 modules)
+    value_mro_names = {cls.__name__ for cls in type(value).__mro__}
+    if expected_type.__name__ in value_mro_names:
+        return True
+
     return False
 
 

pyatlan_v9/model/assets/asset.py

@@ -18,6 +18,7 @@
 import msgspec
 from msgspec import UNSET, UnsetType
 
+from pyatlan_v9.model.assets.related_entity import SaveSemantic
 from pyatlan_v9.model.conversion_utils import (
     categorize_relationships,
     merge_relationships,
@@ -667,7 +668,9 @@ def from_json(json_data: Union[str, bytes], serde: Serde | None = None) -> "Asse
     # =========================================================================
 
     @classmethod
-    def ref_by_guid(cls, guid: str, semantic: str = "REPLACE") -> "Asset":
+    def ref_by_guid(
+        cls, guid: str, semantic: "SaveSemantic | str" = SaveSemantic.REPLACE
+    ) -> "Asset":
         """
         Create a minimal reference to this asset type by its GUID.
 
@@ -678,11 +681,13 @@ def ref_by_guid(cls, guid: str, semantic: str = "REPLACE") -> "Asset":
         Returns:
             Asset reference instance
         """
-        return cls(guid=guid, type_name=cls.__name__)
+        if isinstance(semantic, str):
+            semantic = SaveSemantic(semantic)
+        return cls(guid=guid, type_name=cls.__name__, semantic=semantic)
 
     @classmethod
     def ref_by_qualified_name(
-        cls, qualified_name: str, semantic: str = "REPLACE"
+        cls, qualified_name: str, semantic: "SaveSemantic | str" = SaveSemantic.REPLACE
     ) -> "Asset":
         """
         Create a minimal reference to this asset type by its qualifiedName.
@@ -694,7 +699,9 @@ def ref_by_qualified_name(
         Returns:
             Asset reference instance
         """
-        return cls(qualified_name=qualified_name, type_name=cls.__name__)
+        if isinstance(semantic, str):
+            semantic = SaveSemantic(semantic)
+        return cls(qualified_name=qualified_name, type_name=cls.__name__, semantic=semantic)
 
     # =========================================================================
     # Asset Mutation Convenience Methods

pyatlan_v9/model/assets/entity.py

@@ -18,6 +18,8 @@
 import msgspec
 from msgspec import UNSET, UnsetType
 
+from pyatlan_v9.model.assets.related_entity import SaveSemantic
+
 
 class AtlasClassification(
     msgspec.Struct, kw_only=True, omit_defaults=True, rename="camel"
@@ -174,6 +176,11 @@ class Entity(msgspec.Struct, kw_only=True, omit_defaults=True, rename="camel"):
     home_id: Union[str, UnsetType] = UNSET
     """Home identifier for distributed Atlas systems."""
 
+    semantic: Union[SaveSemantic, None, UnsetType] = UNSET
+    """Save semantic for relationship operations (REPLACE, APPEND, REMOVE).
+    Not serialized to JSON — used internally to control how relationship
+    attributes are categorized during bulk save operations."""
+
     # =========================================================================
     # Compatibility Methods (legacy API surface)
     # =========================================================================

pyatlan_v9/model/audit.py

@@ -4,7 +4,6 @@
 
 import json as json_lib
 from datetime import datetime
-from enum import Enum
 from typing import Any, Dict, Generator, Iterable, List, Optional, Set, Union
 
 import msgspec
@@ -18,6 +17,11 @@
 from pyatlan_v9.model.aggregation import Aggregation
 from pyatlan_v9.model.search import DSL, Bool, Query, Range, SortItem, Term
 
+# ---------------------------------------------------------------------------
+# Re-export plain Enum from legacy — no migration needed.
+# ---------------------------------------------------------------------------
+from pyatlan.model.audit import AuditActionType  # noqa: F401
+
 TOTAL_COUNT = "totalCount"
 
 ENTITY_AUDITS = "entityAudits"
@@ -29,24 +33,6 @@
 LATEST_FIRST = [SortItem("created", order=SortOrder.DESCENDING)]
 
 
-class AuditActionType(str, Enum):
-    ENTITY_CREATE = "ENTITY_CREATE"
-    ENTITY_UPDATE = "ENTITY_UPDATE"
-    ENTITY_DELETE = "ENTITY_DELETE"
-    CUSTOM_METADATA_UPDATE = "BUSINESS_ATTRIBUTE_UPDATE"
-    ATLAN_TAG_ADD = "CLASSIFICATION_ADD"
-    PROPAGATED_ATLAN_TAG_ADD = "PROPAGATED_CLASSIFICATION_ADD"
-    ATLAN_TAG_DELETE = "CLASSIFICATION_DELETE"
-    PROPAGATED_ATLAN_TAG_DELETE = "PROPAGATED_CLASSIFICATION_DELETE"
-    ENTITY_IMPORT_CREATE = "ENTITY_IMPORT_CREATE"
-    ENTITY_IMPORT_UPDATE = "ENTITY_IMPORT_UPDATE"
-    ENTITY_IMPORT_DELETE = "ENTITY_IMPORT_DELETE"
-    ATLAN_TAG_UPDATE = "CLASSIFICATION_UPDATE"
-    PROPAGATED_ATLAN_TAG_UPDATE = "PROPAGATED_CLASSIFICATION_UPDATE"
-    TERM_ADD = "TERM_ADD"
-    TERM_DELETE = "TERM_DELETE"
-
-
 class AuditSearchRequest(msgspec.Struct, kw_only=True):
     """Class from which to configure a search against Atlan's activity log."""
 

pyatlan_v9/model/conversion_utils.py

@@ -43,6 +43,9 @@ def categorize_relationships(
 
         # Handle list of related entities
         if isinstance(value, list):
+            if len(value) == 0:
+                # Empty list means "replace with nothing" (clear the relationship)
+                replace_kwargs[field_name] = []
             for item in value:
                 semantic = getattr(item, "semantic", UNSET)
                 if semantic is UNSET or semantic == SaveSemantic.REPLACE:

pyatlan_v9/model/core.py

@@ -359,10 +359,29 @@ class BulkRequest(msgspec.Struct, kw_only=True, rename="camel"):
     """
     Wrapper for bulk asset API requests.
 
-    Note: The original Pydantic version included a @validator that processed
-    relationship attributes (append/remove semantics). In v9 that processing
-    logic is handled separately outside this struct.
+    In v9, relationship categorization (replace/append/remove semantics) is
+    handled by each entity's ``to_json(nested=True)`` conversion, which calls
+    ``categorize_relationships()`` under the hood.
     """
 
     entities: list[Any]
     """List of asset entities to send to the API in bulk."""
+
+    def to_dict(self) -> dict:
+        """
+        Convert to a dict in the API nested format.
+
+        Each entity is converted to its nested representation (with
+        ``attributes``, ``relationshipAttributes``,
+        ``appendRelationshipAttributes``, ``removeRelationshipAttributes``).
+
+        Returns:
+            Dict suitable for JSON serialization and API submission.
+        """
+        entity_dicts = []
+        for entity in self.entities:
+            if hasattr(entity, "to_json"):
+                entity_dicts.append(json.loads(entity.to_json(nested=True)))
+            else:
+                entity_dicts.append(msgspec.to_builtins(entity))
+        return {"entities": entity_dicts}