cognite-toolkit 0.7.30__py3-none-any.whl → 0.7.39__py3-none-any.whl

cognite_toolkit/_cdf_tk/client/api/three_d.py

@@ -1,19 +1,80 @@
+from collections import defaultdict
+from collections.abc import Iterable, Sequence
+from typing import Any, TypeVar
+
+from pydantic import TypeAdapter
 from rich.console import Console
 
-from cognite_toolkit._cdf_tk.client.data_classes.api_classes import PagedResponse
-from cognite_toolkit._cdf_tk.client.data_classes.three_d import ThreeDModelResponse
-from cognite_toolkit._cdf_tk.utils.http_client import HTTPClient, ParamRequest
+from cognite_toolkit._cdf_tk.client.data_classes.api_classes import InternalIdRequest, PagedResponse
+from cognite_toolkit._cdf_tk.client.data_classes.three_d import (
+    AssetMappingClassicRequest,
+    AssetMappingDMRequest,
+    AssetMappingResponse,
+    ThreeDModelClassicRequest,
+    ThreeDModelResponse,
+)
+from cognite_toolkit._cdf_tk.utils.collection import chunker_sequence
+from cognite_toolkit._cdf_tk.utils.http_client import (
+    HTTPClient,
+    ItemsRequest2,
+    RequestMessage2,
+)
 from cognite_toolkit._cdf_tk.utils.useful_types import PrimitiveType
 
 
 class ThreeDModelAPI:
     ENDPOINT = "/3d/models"
+    MAX_CLASSIC_MODELS_PER_CREATE_REQUEST = 1000
+    MAX_MODELS_PER_DELETE_REQUEST = 1000
+    _LIST_REQUEST_MAX_LIMIT = 1000
 
     def __init__(self, http_client: HTTPClient, console: Console) -> None:
         self._http_client = http_client
         self._console = console
         self._config = http_client.config
 
+    def create(self, models: Sequence[ThreeDModelClassicRequest]) -> list[ThreeDModelResponse]:
+        """Create 3D models in classic format.
+
+        Args:
+            models (Sequence[ThreeDModelClassicRequest]): The 3D model(s) to create.
+
+        Returns:
+            list[ThreeDModelResponse]: The created 3D model(s).
+        """
+        if not models:
+            return []
+        if len(models) > self.MAX_CLASSIC_MODELS_PER_CREATE_REQUEST:
+            raise ValueError("Cannot create more than 1000 3D models in a single request.")
+        responses = self._http_client.request_items_retries(
+            ItemsRequest2(
+                endpoint_url=self._config.create_api_url(self.ENDPOINT),
+                method="POST",
+                items=models,
+            )
+        )
+        responses.raise_for_status()
+        return TypeAdapter(list[ThreeDModelResponse]).validate_python(responses.get_items())
+
+    def delete(self, ids: Sequence[int]) -> None:
+        """Delete 3D models by their IDs.
+
+        Args:
+            ids (Sequence[int]): The IDs of the 3D models to delete.
+        """
+        if not ids:
+            return None
+        if len(ids) > self.MAX_MODELS_PER_DELETE_REQUEST:
+            raise ValueError("Cannot delete more than 1000 3D models in a single request.")
+        responses = self._http_client.request_items_retries(
+            ItemsRequest2(
+                endpoint_url=self._config.create_api_url(self.ENDPOINT + "/delete"),
+                method="POST",
+                items=InternalIdRequest.from_ids(list(ids)),
+            )
+        )
+        responses.raise_for_status()
+
     def iterate(
         self,
         published: bool | None = None,
@@ -21,8 +82,8 @@ class ThreeDModelAPI:
         limit: int = 100,
         cursor: str | None = None,
     ) -> PagedResponse[ThreeDModelResponse]:
-        if not (0 < limit <= 1000):
-            raise ValueError("Limit must be between 1 and 1000.")
+        if not (0 < limit <= self._LIST_REQUEST_MAX_LIMIT):
+            raise ValueError(f"Limit must be between 1 and {self._LIST_REQUEST_MAX_LIMIT}, got {limit}.")
         parameters: dict[str, PrimitiveType] = {
             # There is a bug in the API. The parameter includeRevisionInfo is expected to be lower case and not
             # camel case as documented. You get error message: Unrecognized query parameter includeRevisionInfo,
@@ -34,17 +95,290 @@ class ThreeDModelAPI:
             parameters["published"] = published
         if cursor is not None:
             parameters["cursor"] = cursor
-        responses = self._http_client.request_with_retries(
-            ParamRequest(
+        responses = self._http_client.request_single_retries(
+            RequestMessage2(
                 endpoint_url=self._config.create_api_url(self.ENDPOINT),
                 method="GET",
                 parameters=parameters,
             )
         )
-        responses.raise_for_status()
-        return PagedResponse[ThreeDModelResponse].model_validate(responses.get_first_body())
+        success_response = responses.get_success_or_raise()
+        return PagedResponse[ThreeDModelResponse].model_validate(success_response.body_json)
+
+    def list(
+        self,
+        published: bool | None = None,
+        include_revision_info: bool = False,
+        limit: int | None = 100,
+        cursor: str | None = None,
+    ) -> list[ThreeDModelResponse]:
+        results: list[ThreeDModelResponse] = []
+        while True:
+            request_limit = (
+                self._LIST_REQUEST_MAX_LIMIT
+                if limit is None
+                else min(limit - len(results), self._LIST_REQUEST_MAX_LIMIT)
+            )
+            if request_limit <= 0:
+                break
+            page = self.iterate(
+                published=published,
+                include_revision_info=include_revision_info,
+                limit=request_limit,
+                cursor=cursor,
+            )
+            results.extend(page.items)
+            if page.next_cursor is None:
+                break
+            cursor = page.next_cursor
+        return results
+
+
+T_RequestMapping = TypeVar("T_RequestMapping", bound=AssetMappingClassicRequest | AssetMappingDMRequest)
+
+
+class ThreeDAssetMappingAPI:
+    ENDPOINT = "/3d/models/{modelId}/revisions/{revisionId}/mappings"
+    CREATE_CLASSIC_MAX_MAPPINGS_PER_REQUEST = 1000
+    CREATE_DM_MAX_MAPPINGS_PER_REQUEST = 100
+    DELETE_CLASSIC_MAX_MAPPINGS_PER_REQUEST = 1000
+    DELETE_DM_MAX_MAPPINGS_PER_REQUEST = 100
+    LIST_REQUEST_MAX_LIMIT = 1000
+
+    def __init__(self, http_client: HTTPClient, console: Console) -> None:
+        self._http_client = http_client
+        self._console = console
+        self._config = http_client.config
+
+    def create(self, mappings: Sequence[AssetMappingClassicRequest]) -> list[AssetMappingResponse]:
+        """Create 3D asset mappings.
+
+        Args:
+            mappings (Sequence[AssetMappingClassicRequest]):
+                The 3D asset mapping(s) to create.
+
+        Returns:
+            list[AssetMappingResponse]: The created 3D asset mapping(s).
+        """
+        results: list[AssetMappingResponse] = []
+        for endpoint, model_id, revision_id, revision_mappings in self._chunk_mappings_by_endpoint(
+            mappings, self.CREATE_CLASSIC_MAX_MAPPINGS_PER_REQUEST
+        ):
+            responses = self._http_client.request_items_retries(
+                ItemsRequest2(
+                    endpoint_url=self._config.create_api_url(endpoint),
+                    method="POST",
+                    items=revision_mappings,
+                )
+            )
+            responses.raise_for_status()
+            items = responses.get_items()
+            for item in items:
+                # We append modelId and revisionId to each item since the API does not return them
+                # this is needed to fully populate the AssetMappingResponse data class
+                item["modelId"] = model_id
+                item["revisionId"] = revision_id
+            results.extend(TypeAdapter(list[AssetMappingResponse]).validate_python(items))
+        return results
+
+    def create_dm(
+        self, mappings: Sequence[AssetMappingDMRequest], object_3d_space: str, cad_node_space: str
+    ) -> list[AssetMappingResponse]:
+        """Create 3D asset mappings in Data Modeling format.
+
+        Args:
+            mappings (Sequence[AssetMappingDMRequest]):
+                The 3D asset mapping(s) to create
+            object_3d_space (str):
+                The instance space where the Cognite3DObject are located.
+            cad_node_space (str):
+                The instance space where the CogniteCADNode are located.
+        Returns:
+            list[AssetMappingResponse]: The created 3D asset mapping(s).
+        """
+        results: list[AssetMappingResponse] = []
+        for endpoint, model_id, revision_id, revision_mappings in self._chunk_mappings_by_endpoint(
+            mappings, self.CREATE_DM_MAX_MAPPINGS_PER_REQUEST
+        ):
+            responses = self._http_client.request_items_retries(
+                ItemsRequest2(
+                    endpoint_url=self._config.create_api_url(endpoint),
+                    method="POST",
+                    items=revision_mappings,
+                    extra_body_fields={
+                        "dmsContextualizationConfig": {
+                            "object3DSpace": object_3d_space,
+                            "cadNodeSpace": cad_node_space,
+                        }
+                    },
+                )
+            )
+            responses.raise_for_status()
+            items = responses.get_items()
+            for item in items:
+                # We append modelId and revisionId to each item since the API does not return them
+                # this is needed to fully populate the AssetMappingResponse data class
+                item["modelId"] = model_id
+                item["revisionId"] = revision_id
+            results.extend(TypeAdapter(list[AssetMappingResponse]).validate_python(items))
+        return results
+
+    @classmethod
+    def _chunk_mappings_by_endpoint(
+        cls, mappings: Sequence[T_RequestMapping], chunk_size: int
+    ) -> Iterable[tuple[str, int, int, list[T_RequestMapping]]]:
+        chunked_mappings: dict[tuple[int, int], list[T_RequestMapping]] = defaultdict(list)
+        for mapping in mappings:
+            key = mapping.model_id, mapping.revision_id
+            chunked_mappings[key].append(mapping)
+        for (model_id, revision_id), revision_mappings in chunked_mappings.items():
+            endpoint = cls.ENDPOINT.format(modelId=model_id, revisionId=revision_id)
+            for chunk in chunker_sequence(revision_mappings, chunk_size):
+                yield endpoint, model_id, revision_id, chunk
+
+    def delete(self, mappings: Sequence[AssetMappingClassicRequest]) -> None:
+        """Delete 3D asset mappings.
+
+        Args:
+            mappings (Sequence[AssetMappingClassicRequest]):
+                The 3D asset mapping(s) to delete.
+        """
+        for endpoint, *_, revision_mappings in self._chunk_mappings_by_endpoint(
+            mappings, self.DELETE_CLASSIC_MAX_MAPPINGS_PER_REQUEST
+        ):
+            responses = self._http_client.request_items_retries(
+                ItemsRequest2(
+                    endpoint_url=self._config.create_api_url(f"{endpoint}/delete"),
+                    method="DELETE",
+                    items=revision_mappings,
+                )
+            )
+            responses.raise_for_status()
+        return None
+
+    def delete_dm(self, mappings: Sequence[AssetMappingDMRequest], object_3d_space: str, cad_node_space: str) -> None:
+        """Delete 3D asset mappings in Data Modeling format.
+
+        Args:
+            mappings (Sequence[AssetMappingDMRequest]):
+                The 3D asset mapping(s) to delete.
+            object_3d_space (str):
+                The instance space where the Cognite3DObject are located.
+            cad_node_space (str):
+                The instance space where the CogniteCADNode are located.
+        """
+        for endpoint, *_, revision_mappings in self._chunk_mappings_by_endpoint(
+            mappings, self.DELETE_DM_MAX_MAPPINGS_PER_REQUEST
+        ):
+            responses = self._http_client.request_items_retries(
+                ItemsRequest2(
+                    endpoint_url=self._config.create_api_url(f"{endpoint}/delete"),
+                    method="DELETE",
+                    items=revision_mappings,
+                    extra_body_fields={
+                        "dmsContextualizationConfig": {
+                            "object3DSpace": object_3d_space,
+                            "cadNodeSpace": cad_node_space,
+                        }
+                    },
+                )
+            )
+            responses.raise_for_status()
+        return None
+
+    def iterate(
+        self,
+        model_id: int,
+        revision_id: int,
+        asset_ids: list[int] | None = None,
+        asset_instance_ids: list[str] | None = None,
+        node_ids: list[int] | None = None,
+        tree_indexes: list[int] | None = None,
+        get_dms_instances: bool = False,
+        limit: int = 100,
+        cursor: str | None = None,
+    ) -> PagedResponse[AssetMappingResponse]:
+        if not (0 < limit <= self.LIST_REQUEST_MAX_LIMIT):
+            raise ValueError(f"Limit must be between 1 and {self.LIST_REQUEST_MAX_LIMIT}, got {limit}.")
+        if sum(param is not None for param in [asset_ids, asset_instance_ids, node_ids, tree_indexes]) > 1:
+            raise ValueError("Only one of asset_ids, asset_instance_ids, node_ids, or tree_indexes can be provided.")
+        body: dict[str, Any] = {
+            "getDmsInstances": get_dms_instances,
+            "limit": limit,
+        }
+        if asset_ids is not None:
+            if not (0 < len(asset_ids) <= 100):
+                raise ValueError("asset_ids must contain between 1 and 100 IDs.")
+            body["filter"] = {"assetIds": asset_ids}
+        elif asset_instance_ids is not None:
+            if not (0 < len(asset_instance_ids) <= 100):
+                raise ValueError("asset_instance_ids must contain between 1 and 100 IDs.")
+            body["filter"] = {"assetInstanceIds": asset_instance_ids}
+        elif node_ids is not None:
+            if not (0 < len(node_ids) <= 100):
+                raise ValueError("node_ids must contain between 1 and 100 IDs.")
+            body["filter"] = {"nodeIds": node_ids}
+        elif tree_indexes is not None:
+            if not (0 < len(tree_indexes) <= 100):
+                raise ValueError("tree_indexes must contain between 1 and 100 indexes.")
+            body["filter"] = {"treeIndexes": tree_indexes}
+        if cursor is not None:
+            body["cursor"] = cursor
+
+        endpoint = self.ENDPOINT.format(modelId=model_id, revisionId=revision_id)
+        responses = self._http_client.request_single_retries(
+            RequestMessage2(
+                endpoint_url=self._config.create_api_url(f"{endpoint}/list"),
+                method="POST",
+                body_content=body,
+            )
+        )
+        success_response = responses.get_success_or_raise()
+        body_json = success_response.body_json
+        # Add modelId and revisionId to items since the API does not return them
+        for item in body_json.get("items", []):
+            item["modelId"] = model_id
+            item["revisionId"] = revision_id
+        return PagedResponse[AssetMappingResponse].model_validate(body_json)
+
+    def list(
+        self,
+        model_id: int,
+        revision_id: int,
+        asset_ids: list[int] | None = None,
+        asset_instance_ids: list[str] | None = None,
+        node_ids: list[int] | None = None,
+        tree_indexes: list[int] | None = None,
+        get_dms_instances: bool = False,
+        limit: int | None = 100,
+    ) -> list[AssetMappingResponse]:
+        results: list[AssetMappingResponse] = []
+        cursor: str | None = None
+        while True:
+            request_limit = (
+                self.LIST_REQUEST_MAX_LIMIT if limit is None else min(limit - len(results), self.LIST_REQUEST_MAX_LIMIT)
+            )
+            if request_limit <= 0:
+                break
+            page = self.iterate(
+                model_id=model_id,
+                revision_id=revision_id,
+                asset_ids=asset_ids,
+                asset_instance_ids=asset_instance_ids,
+                node_ids=node_ids,
+                tree_indexes=tree_indexes,
+                get_dms_instances=get_dms_instances,
+                limit=request_limit,
+                cursor=cursor,
+            )
+            results.extend(page.items)
+            if page.next_cursor is None:
+                break
+            cursor = page.next_cursor
+        return results
 
 
 class ThreeDAPI:
     def __init__(self, http_client: HTTPClient, console: Console) -> None:
         self.models = ThreeDModelAPI(http_client, console)
+        self.asset_mappings = ThreeDAssetMappingAPI(http_client, console)

cognite_toolkit/_cdf_tk/client/data_classes/api_classes.py

@@ -2,6 +2,8 @@ from typing import Generic, TypeVar
 
 from pydantic import BaseModel, Field, JsonValue
 
+from cognite_toolkit._cdf_tk.utils.http_client._data_classes2 import RequestResource
+
 T = TypeVar("T", bound=BaseModel)
 
 
@@ -15,3 +17,14 @@ class QueryResponse(BaseModel, Generic[T]):
     typing: dict[str, JsonValue] | None = None
     next_cursor: dict[str, str] = Field(alias="nextCursor")
     debug: dict[str, JsonValue] | None = None
+
+
+class InternalIdRequest(RequestResource):
+    id: int
+
+    def as_id(self) -> int:
+        return self.id
+
+    @classmethod
+    def from_ids(cls, ids: list[int]) -> list["InternalIdRequest"]:
+        return [cls(id=id_) for id_ in ids]

cognite_toolkit/_cdf_tk/client/data_classes/base.py

@@ -3,9 +3,11 @@ from abc import ABC, abstractmethod
 from collections import UserList
 from typing import TYPE_CHECKING, Any, Generic, TypeVar
 
-from pydantic import BaseModel, ConfigDict
+from pydantic import ConfigDict
 from pydantic.alias_generators import to_camel
 
+from cognite_toolkit._cdf_tk.utils.http_client._data_classes2 import BaseModelObject, RequestResource
+
 if TYPE_CHECKING:
     from cognite.client import CogniteClient
 
@@ -15,28 +17,6 @@ else:
     from typing_extensions import Self
 
 
-class BaseModelObject(BaseModel):
-    """Base class for all object. This includes resources and nested objects."""
-
-    # We allow extra fields to support forward compatibility.
-    model_config = ConfigDict(alias_generator=to_camel, extra="allow")
-
-    def dump(self, camel_case: bool = True) -> dict[str, Any]:
-        """Dump the resource to a dictionary.
-
-        This is the default serialization method for request resources.
-        """
-        return self.model_dump(mode="json", by_alias=camel_case)
-
-    @classmethod
-    def _load(cls, resource: dict[str, Any]) -> "Self":
-        """Load method to match CogniteResource signature."""
-        return cls.model_validate(resource)
-
-
-class RequestResource(BaseModelObject): ...
-
-
 T_RequestResource = TypeVar("T_RequestResource", bound=RequestResource)
 
 
@@ -46,22 +26,22 @@ class ResponseResource(BaseModelObject, Generic[T_RequestResource], ABC):
         """Convert the response resource to a request resource."""
         ...
 
+    def as_write(self) -> T_RequestResource:
+        """Alias for as_request_resource to match protocol signature."""
+        return self.as_request_resource()
 
-class Identifier(BaseModel):
+
+class Identifier(RequestResource, ABC):
     """Base class for all identifier classes."""
 
     model_config = ConfigDict(alias_generator=to_camel, extra="ignore", populate_by_name=True, frozen=True)
 
-    def dump(self, include_type: bool = True) -> dict[str, Any]:
-        """Dump the identifier to a dictionary.
-
-        Args:
-            include_type (bool): Whether to include the type of the identifier in the output.
+    def dump(self, camel_case: bool = True, include_type: bool = True) -> dict[str, Any]:
+        """Dump the resource to a dictionary.
 
-        Returns:
-            dict[str, Any]: The dumped identifier.
+        This is the default serialization method for request resources.
         """
-        return self.model_dump(mode="json", by_alias=True, exclude_defaults=not include_type)
+        return self.model_dump(mode="json", by_alias=camel_case, exclude_unset=not include_type)
 
     def as_id(self) -> Self:
         return self

cognite_toolkit/_cdf_tk/client/data_classes/instance_api.py

@@ -7,7 +7,7 @@ from .base import BaseModelObject, Identifier, RequestResource
 InstanceType: TypeAlias = Literal["node", "edge"]
 
 
-class InstanceIdentifier(Identifier):
+class TypedInstanceIdentifier(Identifier):
     """Identifier for an Instance instance."""
 
     instance_type: InstanceType
@@ -15,14 +15,19 @@ class InstanceIdentifier(Identifier):
     external_id: str
 
 
-class NodeIdentifier(InstanceIdentifier):
+class TypedNodeIdentifier(TypedInstanceIdentifier):
     instance_type: Literal["node"] = "node"
 
 
-class EdgeIdentifier(InstanceIdentifier):
+class TypedEdgeIdentifier(TypedInstanceIdentifier):
     instance_type: Literal["edge"] = "edge"
 
 
+class InstanceIdentifier(Identifier):
+    space: str
+    external_id: str
+
+
 class InstanceResult(BaseModelObject):
     instance_type: InstanceType
     version: int
@@ -32,8 +37,8 @@ class InstanceResult(BaseModelObject):
     created_time: int
     last_updated_time: int
 
-    def as_id(self) -> InstanceIdentifier:
-        return InstanceIdentifier(
+    def as_id(self) -> TypedInstanceIdentifier:
+        return TypedInstanceIdentifier(
             instance_type=self.instance_type,
             space=self.space,
             external_id=self.external_id,
@@ -62,8 +67,8 @@ class InstanceRequestResource(RequestResource):
     space: str
     external_id: str
 
-    def as_id(self) -> InstanceIdentifier:
-        return InstanceIdentifier(
+    def as_id(self) -> TypedInstanceIdentifier:
+        return TypedInstanceIdentifier(
             instance_type=self.instance_type,
             space=self.space,
             external_id=self.external_id,
@@ -107,7 +112,7 @@ class InstanceSource(BaseModelObject):
         return value
 
 
-class InstanceRequestItem(BaseModelObject):
+class InstanceRequestItem(RequestResource):
     model_config = ConfigDict(populate_by_name=True)
     instance_type: InstanceType
     space: str
@@ -115,8 +120,8 @@ class InstanceRequestItem(BaseModelObject):
     existing_version: int | None = None
     sources: list[InstanceSource] | None = None
 
-    def as_id(self) -> InstanceIdentifier:
-        return InstanceIdentifier(
+    def as_id(self) -> TypedInstanceIdentifier:
+        return TypedInstanceIdentifier(
             instance_type=self.instance_type,
             space=self.space,
             external_id=self.external_id,
@@ -128,7 +133,7 @@ class InstanceResponseItem(BaseModelObject):
     space: str
     external_id: str
     version: int
-    type: InstanceIdentifier | None = None
+    type: TypedInstanceIdentifier | None = None
     created_time: int
     last_updated_time: int
     deleted_time: int | None = None
@@ -149,8 +154,8 @@ class InstanceResponseItem(BaseModelObject):
         output.update(space_properties.get(view_version, {}))
         return output
 
-    def as_id(self) -> InstanceIdentifier:
-        return InstanceIdentifier(
+    def as_id(self) -> TypedInstanceIdentifier:
+        return TypedInstanceIdentifier(
             instance_type=self.instance_type,
             space=self.space,
             external_id=self.external_id,

cognite_toolkit/_cdf_tk/client/data_classes/{canvas.py → legacy/canvas.py}

@@ -30,7 +30,7 @@ from cognite.client.data_classes.data_modeling.instances import (
     TypedNodeApply,
 )
 
-from cognite_toolkit._cdf_tk.client.data_classes.migration import AssetCentricId
+from cognite_toolkit._cdf_tk.client.data_classes.legacy.migration import AssetCentricId
 from cognite_toolkit._cdf_tk.utils.useful_types import JsonVal
 
 if sys.version_info >= (3, 11):

cognite_toolkit/_cdf_tk/client/data_classes/three_d.py

@@ -1,5 +1,8 @@
+from collections.abc import Hashable
 from typing import Literal
 
+from pydantic import Field
+
 from .base import BaseModelObject, RequestResource, ResponseResource
 
 
@@ -19,11 +22,17 @@ class RevisionStatus(BaseModelObject):
 class ThreeDModelRequest(RequestResource):
     name: str
 
+    def as_id(self) -> str:
+        return self.name
+
 
 class ThreeDModelClassicRequest(ThreeDModelRequest):
     data_set_id: int | None = None
     metadata: dict[str, str] | None = None
 
+    def as_id(self) -> str:
+        return self.name
+
 
 class ThreeDModelDMSRequest(ThreeDModelRequest):
     space: str
@@ -45,3 +54,59 @@ class ThreeDModelResponse(ResponseResource[ThreeDModelRequest]):
             return ThreeDModelClassicRequest._load(self.dump())
         else:
             return ThreeDModelDMSRequest._load(self.dump())
+
+
+class AssetMappingDMRequest(RequestResource):
+    node_id: int
+    asset_instance_id: NodeReference
+    # These fields are part of the path request and not the body schema.
+    model_id: int = Field(exclude=True)
+    revision_id: int = Field(exclude=True)
+
+    def as_id(self) -> Hashable:
+        return (
+            self.model_id,
+            self.revision_id,
+            self.node_id,
+            self.asset_instance_id.space,
+            self.asset_instance_id.external_id,
+        )
+
+
+class AssetMappingClassicRequest(RequestResource):
+    node_id: int
+    asset_id: int | None = None
+    asset_instance_id: NodeReference | None = None
+    # These fields are part of the path request and not the body schema.
+    model_id: int = Field(exclude=True)
+    revision_id: int = Field(exclude=True)
+
+    def as_id(self) -> Hashable:
+        if self.asset_id:
+            return self.model_id, self.revision_id, self.node_id, self.asset_id
+        elif self.asset_instance_id:
+            return (
+                self.model_id,
+                self.revision_id,
+                self.node_id,
+                self.asset_instance_id.space,
+                self.asset_instance_id.external_id,
+            )
+        else:
+            raise AttributeError("asset_id or asset_instance_id is required")
+
+
+class AssetMappingResponse(ResponseResource[AssetMappingClassicRequest]):
+    node_id: int
+    asset_id: int | None = None
+    asset_instance_id: NodeReference | None = None
+    tree_index: int | None = None
+    subtree_size: int | None = None
+    # These fields are part of the path request and response, but they are included here for convenience.
+    model_id: int = Field(exclude=True)
+    revision_id: int = Field(exclude=True)
+
+    def as_request_resource(self) -> AssetMappingClassicRequest:
+        return AssetMappingClassicRequest.model_validate(
+            {**self.dump(), "modelId": self.model_id, "revisionId": self.revision_id}
+        )