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

cognite_toolkit/_cdf_tk/client/api/hosted_extractor_sources.py

@@ -0,0 +1,136 @@
+from collections.abc import Iterable, Sequence
+from typing import Any, Literal
+
+from pydantic import JsonValue, TypeAdapter
+
+from cognite_toolkit._cdf_tk.client.cdf_client import CDFResourceAPI, PagedResponse, ResponseItems
+from cognite_toolkit._cdf_tk.client.cdf_client.api import Endpoint
+from cognite_toolkit._cdf_tk.client.http_client import HTTPClient, ItemsSuccessResponse2, SuccessResponse2
+from cognite_toolkit._cdf_tk.client.resource_classes.hosted_extractor_source import (
+    HostedExtractorSourceRequestUnion,
+    HostedExtractorSourceResponse,
+    HostedExtractorSourceResponseUnion,
+)
+from cognite_toolkit._cdf_tk.client.resource_classes.identifiers import ExternalId
+
+
+class HostedExtractorSourcesAPI(
+    CDFResourceAPI[ExternalId, HostedExtractorSourceRequestUnion, HostedExtractorSourceResponseUnion]
+):
+    def __init__(self, http_client: HTTPClient) -> None:
+        super().__init__(
+            http_client=http_client,
+            method_endpoint_map={
+                "create": Endpoint(method="POST", path="/hostedextractors/sources", item_limit=10),
+                "retrieve": Endpoint(method="POST", path="/hostedextractors/sources/byids", item_limit=100),
+                "update": Endpoint(method="POST", path="/hostedextractors/sources/update", item_limit=10),
+                "delete": Endpoint(method="POST", path="/hostedextractors/sources/delete", item_limit=100),
+                "list": Endpoint(method="GET", path="/hostedextractors/sources", item_limit=100),
+            },
+        )
+
+    def _validate_page_response(
+        self, response: SuccessResponse2 | ItemsSuccessResponse2
+    ) -> PagedResponse[HostedExtractorSourceResponseUnion]:
+        if isinstance(response, SuccessResponse2):
+            data = response.body_json
+        else:
+            data = TypeAdapter(dict[str, JsonValue]).validate_json(response.body)
+        items = [HostedExtractorSourceResponse.validate_python(item) for item in data.get("items", [])]
+        return PagedResponse[HostedExtractorSourceResponseUnion](items=items, nextCursor=data.get("nextCursor"))
+
+    def _reference_response(self, response: SuccessResponse2) -> ResponseItems[ExternalId]:
+        return ResponseItems[ExternalId].model_validate_json(response.body)
+
+    def create(self, items: Sequence[HostedExtractorSourceRequestUnion]) -> list[HostedExtractorSourceResponseUnion]:
+        """Create hosted extractor sources in CDF.
+
+        Args:
+            items: List of source request objects to create.
+        Returns:
+            List of created source response objects.
+        """
+        return self._request_item_response(list(items), "create")
+
+    def retrieve(
+        self, items: Sequence[ExternalId], ignore_unknown_ids: bool = False
+    ) -> list[HostedExtractorSourceResponseUnion]:
+        """Retrieve hosted extractor sources from CDF.
+
+        Args:
+            items: List of ExternalId objects to retrieve.
+            ignore_unknown_ids: Whether to ignore unknown IDs.
+        Returns:
+            List of retrieved source response objects.
+        """
+        return self._request_item_response(
+            items, method="retrieve", extra_body={"ignoreUnknownIds": ignore_unknown_ids}
+        )
+
+    def update(
+        self, items: Sequence[HostedExtractorSourceRequestUnion], mode: Literal["patch", "replace"] = "replace"
+    ) -> list[HostedExtractorSourceResponseUnion]:
+        """Update hosted extractor sources in CDF.
+
+        Args:
+            items: List of source request objects to update.
+            mode: Update mode, either "patch" or "replace".
+
+        Returns:
+            List of updated source response objects.
+        """
+        return self._update(list(items), mode=mode)
+
+    def delete(self, items: Sequence[ExternalId], ignore_unknown_ids: bool = False, force: bool | None = None) -> None:
+        """Delete hosted extractor sources from CDF.
+
+        Args:
+            items: List of ExternalId objects to delete.
+            ignore_unknown_ids: Whether to ignore unknown IDs.
+            force: Delete any jobs associated with each item.
+        """
+        extra_body: dict[str, Any] = {"ignoreUnknownIds": ignore_unknown_ids}
+        if force is not None:
+            extra_body["force"] = force
+        self._request_no_response(items, "delete", extra_body=extra_body)
+
+    def paginate(
+        self,
+        limit: int = 100,
+        cursor: str | None = None,
+    ) -> PagedResponse[HostedExtractorSourceResponseUnion]:
+        """Iterate over hosted extractor sources in CDF.
+
+        Args:
+            limit: Maximum number of items to return.
+            cursor: Cursor for pagination.
+
+        Returns:
+            PagedResponse of source response objects.
+        """
+        return self._paginate(cursor=cursor, limit=limit)
+
+    def iterate(
+        self,
+        limit: int = 100,
+    ) -> Iterable[list[HostedExtractorSourceResponseUnion]]:
+        """Iterate over hosted extractor sources in CDF.
+
+        Args:
+            limit: Maximum number of items to return per page.
+
+        Returns:
+            Iterable of lists of source response objects.
+        """
+        return self._iterate(limit=limit)
+
+    def list(self, limit: int | None = 100) -> list[HostedExtractorSourceResponseUnion]:
+        """List all hosted extractor sources in CDF.
+
+        Args:
+            limit: Maximum number of items to return. None for all items.
+
+        Returns:
+            List of source response objects.
+        """
+        return self._list(limit=limit)

cognite_toolkit/_cdf_tk/client/api/hosted_extractors.py

@@ -0,0 +1,23 @@
+from cognite_toolkit._cdf_tk.client.http_client import HTTPClient
+
+from .hosted_extractor_destinations import HostedExtractorDestinationsAPI
+from .hosted_extractor_jobs import HostedExtractorJobsAPI
+from .hosted_extractor_mappings import HostedExtractorMappingsAPI
+from .hosted_extractor_sources import HostedExtractorSourcesAPI
+
+
+class HostedExtractorsAPI:
+    """API for Hosted Extractors resources.
+
+    This class groups all hosted extractor related APIs:
+    - sources: Manage hosted extractor sources (MQTT, Kafka, EventHub, REST)
+    - jobs: Manage hosted extractor jobs
+    - destinations: Manage hosted extractor destinations
+    - mappings: Manage hosted extractor mappings
+    """
+
+    def __init__(self, http_client: HTTPClient) -> None:
+        self.sources = HostedExtractorSourcesAPI(http_client)
+        self.jobs = HostedExtractorJobsAPI(http_client)
+        self.destinations = HostedExtractorDestinationsAPI(http_client)
+        self.mappings = HostedExtractorMappingsAPI(http_client)

cognite_toolkit/_cdf_tk/client/api/infield.py

@@ -4,23 +4,23 @@ from typing import Any, cast
 from pydantic import TypeAdapter
 from rich.console import Console
 
-from cognite_toolkit._cdf_tk.client.data_classes.api_classes import QueryResponse
-from cognite_toolkit._cdf_tk.client.data_classes.infield import (
+from cognite_toolkit._cdf_tk.client.cdf_client.responses import QueryResponse
+from cognite_toolkit._cdf_tk.client.http_client import (
+    HTTPClient,
+    ItemsRequest2,
+    RequestMessage2,
+)
+from cognite_toolkit._cdf_tk.client.resource_classes.infield import (
     DataExplorationConfig,
     InFieldCDMLocationConfig,
     InfieldLocationConfig,
 )
-from cognite_toolkit._cdf_tk.client.data_classes.instance_api import (
+from cognite_toolkit._cdf_tk.client.resource_classes.instance_api import (
     InstanceResponseItem,
     InstanceResult,
     TypedNodeIdentifier,
 )
 from cognite_toolkit._cdf_tk.tk_warnings import HighSeverityWarning
-from cognite_toolkit._cdf_tk.utils.http_client import (
-    HTTPClient,
-    ItemsRequest2,
-    RequestMessage2,
-)
 
 
 class InfieldConfigAPI:

cognite_toolkit/_cdf_tk/client/api/instances.py

@@ -0,0 +1,139 @@
+from collections.abc import Iterable, Sequence
+from typing import Any, Literal
+
+from cognite_toolkit._cdf_tk.client.cdf_client import CDFResourceAPI, PagedResponse, ResponseItems
+from cognite_toolkit._cdf_tk.client.cdf_client.api import Endpoint
+from cognite_toolkit._cdf_tk.client.http_client import HTTPClient, ItemsSuccessResponse2, SuccessResponse2
+from cognite_toolkit._cdf_tk.client.request_classes.filters import InstanceFilter
+from cognite_toolkit._cdf_tk.client.resource_classes.data_modeling import (
+    InstanceRequest,
+    InstanceResponse,
+    ViewReference,
+)
+from cognite_toolkit._cdf_tk.client.resource_classes.instance_api import TypedInstanceIdentifier
+
+
+class InstancesAPI(CDFResourceAPI[TypedInstanceIdentifier, InstanceRequest, InstanceResponse]):
+    def __init__(self, http_client: HTTPClient) -> None:
+        super().__init__(
+            http_client=http_client,
+            method_endpoint_map={
+                "upsert": Endpoint(method="POST", path="/models/instances", item_limit=1000),
+                "retrieve": Endpoint(method="POST", path="/models/instances/byids", item_limit=1000),
+                "delete": Endpoint(method="POST", path="/models/instances/delete", item_limit=1000),
+                "list": Endpoint(method="POST", path="/models/instances/list", item_limit=1000),
+            },
+        )
+
+    def _validate_page_response(
+        self, response: SuccessResponse2 | ItemsSuccessResponse2
+    ) -> PagedResponse[InstanceResponse]:
+        return PagedResponse[InstanceResponse].model_validate_json(response.body)
+
+    def _validate_response(self, response: SuccessResponse2) -> ResponseItems[TypedInstanceIdentifier]:
+        return ResponseItems[TypedInstanceIdentifier].model_validate_json(response.body)
+
+    def create(self, items: Sequence[InstanceRequest]) -> list[InstanceResponse]:
+        """Create instances in CDF.
+
+        Args:
+            items: List of InstanceRequest objects to create.
+        Returns:
+            List of created InstanceResponse objects.
+        """
+        return self._request_item_response(items, "upsert")
+
+    def retrieve(
+        self, items: Sequence[TypedInstanceIdentifier], source: ViewReference | None = None
+    ) -> list[InstanceResponse]:
+        """Retrieve instances from CDF.
+
+        Args:
+            items: List of ExternalId objects to retrieve.
+            source: Optional ViewReference to specify the source view for the instances.
+        Returns:
+            List of retrieved InstanceResponse objects.
+        """
+        return self._request_item_response(
+            items, method="retrieve", extra_body={"sources": [{"source": source.dump()}]} if source else None
+        )
+
+    def delete(self, items: Sequence[TypedInstanceIdentifier]) -> list[TypedInstanceIdentifier]:
+        """Delete instances from CDF.
+
+        Args:
+            items: List of TypedInstanceIdentifier objects to delete.
+        """
+        response_items: list[TypedInstanceIdentifier] = []
+        for response in self._chunk_requests(items, "delete", self._serialize_items):
+            response_items.extend(self._validate_response(response).items)
+        return response_items
+
+    @staticmethod
+    def _create_sort_body(instance_type: Literal["node", "edge"] | None) -> list[dict]:
+        """We sort by space and externalId to get a stable sort order.
+
+        This is also more performant than sorting by using the default sort, which will sort on
+        internal CDF IDs. This will be slow if you have deleted a lot of instances, as they will be counted.
+        By sorting on space and externalId, we avoid this issue.
+        """
+        instance_type = instance_type or "node"
+        return [
+            {
+                "property": [instance_type, "space"],
+                "direction": "ascending",
+            },
+            {
+                "property": [instance_type, "externalId"],
+                "direction": "ascending",
+            },
+        ]
+
+    @classmethod
+    def _create_body(cls, filter: InstanceFilter | None) -> dict[str, Any]:
+        return {
+            **(filter.model_dump(exclude_none=True) if filter else {}),
+            "sort": cls._create_sort_body(filter.instance_type if filter else "node"),
+        }
+
+    def paginate(
+        self,
+        filter: InstanceFilter | None = None,
+        limit: int = 100,
+        cursor: str | None = None,
+    ) -> PagedResponse[InstanceResponse]:
+        """Iterate over all instances in CDF.
+
+        Args:
+           filter: InstanceFilter to filter instances.
+            limit: Maximum number of items to return.
+            cursor: Cursor for pagination.
+
+        Returns:
+            PagedResponse of InstanceResponse objects.
+        """
+        return self._paginate(
+            cursor=cursor,
+            limit=limit,
+            body=self._create_body(filter),
+        )
+
+    def iterate(self, filter: InstanceFilter | None = None, limit: int = 100) -> Iterable[list[InstanceResponse]]:
+        """Iterate over all instances in CDF.
+
+        Args:
+            filter: InstanceFilter to filter instances.
+            limit: Maximum number of items to return per page.
+
+        Returns:
+            Iterable of lists of InstanceResponse objects.
+        """
+        return self._iterate(limit=limit, body=self._create_body(filter))
+
+    def list(self, filter: InstanceFilter | None = None, limit: int | None = 100) -> list[InstanceResponse]:
+        """List all instances in CDF.
+
+        Returns:
+            List of InstanceResponse objects.
+        """
+        return self._list(limit=limit, body=self._create_body(filter))

cognite_toolkit/_cdf_tk/client/api/labels.py

@@ -0,0 +1,125 @@
+from collections.abc import Iterable, Sequence
+from typing import Any
+
+from cognite_toolkit._cdf_tk.client.cdf_client import CDFResourceAPI, PagedResponse, ResponseItems
+from cognite_toolkit._cdf_tk.client.cdf_client.api import Endpoint
+from cognite_toolkit._cdf_tk.client.http_client import HTTPClient, ItemsSuccessResponse2, SuccessResponse2
+from cognite_toolkit._cdf_tk.client.request_classes.filters import ClassicFilter
+from cognite_toolkit._cdf_tk.client.resource_classes.identifiers import ExternalId
+from cognite_toolkit._cdf_tk.client.resource_classes.label import LabelRequest, LabelResponse
+
+
+class LabelsAPI(CDFResourceAPI[ExternalId, LabelRequest, LabelResponse]):
+    def __init__(self, http_client: HTTPClient) -> None:
+        super().__init__(
+            http_client=http_client,
+            method_endpoint_map={
+                "create": Endpoint(method="POST", path="/labels", item_limit=1000),
+                "retrieve": Endpoint(method="POST", path="/labels/byids", item_limit=1000),
+                "delete": Endpoint(method="POST", path="/labels/delete", item_limit=1000),
+                "list": Endpoint(method="POST", path="/labels/list", item_limit=1000),
+            },
+        )
+
+    def _validate_page_response(
+        self, response: SuccessResponse2 | ItemsSuccessResponse2
+    ) -> PagedResponse[LabelResponse]:
+        return PagedResponse[LabelResponse].model_validate_json(response.body)
+
+    def _reference_response(self, response: SuccessResponse2) -> ResponseItems[ExternalId]:
+        return ResponseItems[ExternalId].model_validate_json(response.body)
+
+    def create(self, items: Sequence[LabelRequest]) -> list[LabelResponse]:
+        """Create labels in CDF.
+
+        Args:
+            items: List of LabelRequest objects to create.
+        Returns:
+            List of created LabelResponse objects.
+        """
+        return self._request_item_response(items, "create")
+
+    def retrieve(self, items: Sequence[ExternalId], ignore_unknown_ids: bool = False) -> list[LabelResponse]:
+        """Retrieve labels from CDF.
+
+        Args:
+            items: List of ExternalId objects to retrieve.
+            ignore_unknown_ids: Whether to ignore unknown IDs.
+        Returns:
+            List of retrieved LabelResponse objects.
+        """
+        return self._request_item_response(
+            items, method="retrieve", extra_body={"ignoreUnknownIds": ignore_unknown_ids}
+        )
+
+    def delete(self, items: Sequence[ExternalId]) -> None:
+        """Delete labels from CDF.
+
+        Args:
+            items: List of ExternalId objects to delete.
+        """
+        self._request_no_response(items, "delete")
+
+    def paginate(
+        self,
+        filter: ClassicFilter | None = None,
+        name: str | None = None,
+        limit: int = 100,
+        cursor: str | None = None,
+    ) -> PagedResponse[LabelResponse]:
+        """Iterate over all labels in CDF.
+
+        Args:
+            filter: Filter by data set IDs.
+            name: Filter by label name (prefix match).
+            limit: Maximum number of items to return.
+            cursor: Cursor for pagination.
+
+        Returns:
+            PagedResponse of LabelResponse objects.
+        """
+        body: dict[str, Any] = filter.dump() if filter else {}
+        if name:
+            body["name"] = name
+
+        return self._paginate(
+            cursor=cursor,
+            limit=limit,
+            body=body,
+        )
+
+    def iterate(
+        self,
+        filter: ClassicFilter | None = None,
+        name: str | None = None,
+        limit: int = 100,
+    ) -> Iterable[list[LabelResponse]]:
+        """Iterate over all labels in CDF.
+
+        Args:
+            filter: Filter by data set IDs.
+            name: Filter by label name (prefix match).
+            limit: Maximum number of items to return per page.
+
+        Returns:
+            Iterable of lists of LabelResponse objects.
+        """
+        body: dict[str, Any] = filter.dump() if filter else {}
+        if name:
+            body["name"] = name
+
+        return self._iterate(
+            limit=limit,
+            body=body,
+        )
+
+    def list(
+        self,
+        limit: int | None = 100,
+    ) -> list[LabelResponse]:
+        """List all labels in CDF.
+
+        Returns:
+            List of LabelResponse objects.
+        """
+        return self._list(limit=limit)

cognite_toolkit/_cdf_tk/client/api/legacy/canvas.py

@@ -16,7 +16,7 @@ from cognite.client.exceptions import CogniteDuplicatedError
 from cognite.client.utils.useful_types import SequenceNotStr
 
 from cognite_toolkit._cdf_tk.client.api.legacy.extended_data_modeling import ExtendedInstancesAPI
-from cognite_toolkit._cdf_tk.client.data_classes.legacy.canvas import (
+from cognite_toolkit._cdf_tk.client.resource_classes.legacy.canvas import (
     ANNOTATION_EDGE_TYPE,
     CANVAS_INSTANCE_SPACE,
     CONTAINER_REFERENCE_EDGE_TYPE,
@@ -30,7 +30,7 @@ from cognite_toolkit._cdf_tk.client.data_classes.legacy.canvas import (
     IndustrialCanvas,
     IndustrialCanvasApply,
 )
-from cognite_toolkit._cdf_tk.client.data_classes.legacy.instances import InstancesApplyResultList
+from cognite_toolkit._cdf_tk.client.resource_classes.legacy.instances import InstancesApplyResultList
 from cognite_toolkit._cdf_tk.exceptions import ToolkitValueError
 
 
@@ -83,7 +83,7 @@ class IndustrialCanvasAPI:
     def create(self, canvas: IndustrialCanvasApply) -> InstancesApplyResultList:
         instance_ids = canvas.as_instance_ids()
         existing = self._instance_api.retrieve(
-            [node for node in instance_ids if isinstance(node, NodeId)],
+            nodes=[node for node in instance_ids if isinstance(node, NodeId)],
             edges=[edge for edge in instance_ids if isinstance(edge, EdgeId)],
         )
         if existing.nodes or existing.edges:

cognite_toolkit/_cdf_tk/client/api/legacy/charts.py

@@ -6,7 +6,7 @@ from cognite.client._api_client import APIClient
 from cognite.client.utils._identifier import IdentifierSequence
 from cognite.client.utils.useful_types import SequenceNotStr
 
-from cognite_toolkit._cdf_tk.client.data_classes.legacy.charts import (
+from cognite_toolkit._cdf_tk.client.resource_classes.legacy.charts import (
     Chart,
     ChartList,
     ChartWrite,

cognite_toolkit/_cdf_tk/client/api/legacy/extended_data_modeling.py

@@ -16,7 +16,7 @@ from cognite.client.utils._concurrency import execute_tasks
 from requests import Response
 
 from cognite_toolkit._cdf_tk.client._constants import DATA_MODELING_MAX_DELETE_WORKERS, DATA_MODELING_MAX_WRITE_WORKERS
-from cognite_toolkit._cdf_tk.client.data_classes.legacy.instances import InstancesApplyResultList
+from cognite_toolkit._cdf_tk.client.resource_classes.legacy.instances import InstancesApplyResultList
 from cognite_toolkit._cdf_tk.client.utils._concurrency import ToolkitConcurrencySettings
 from cognite_toolkit._cdf_tk.client.utils._http_client import ToolkitRetryTracker
 

cognite_toolkit/_cdf_tk/client/api/legacy/extended_files.py

@@ -9,11 +9,11 @@ from cognite.client.utils._concurrency import execute_tasks
 from cognite.client.utils._identifier import IdentifierSequence
 from cognite.client.utils.useful_types import SequenceNotStr
 
-from cognite_toolkit._cdf_tk.client.data_classes.legacy.extended_filemetadata import (
+from cognite_toolkit._cdf_tk.client.resource_classes.legacy.extended_filemetadata import (
     ExtendedFileMetadata,
     ExtendedFileMetadataList,
 )
-from cognite_toolkit._cdf_tk.client.data_classes.legacy.pending_instances_ids import PendingInstanceId
+from cognite_toolkit._cdf_tk.client.resource_classes.legacy.pending_instances_ids import PendingInstanceId
 
 
 class ExtendedFileMetadataAPI(FilesAPI):

cognite_toolkit/_cdf_tk/client/api/legacy/extended_functions.py

@@ -6,8 +6,8 @@ from cognite.client.utils.useful_types import SequenceNotStr
 from rich.console import Console
 
 from cognite_toolkit._cdf_tk.client.config import ToolkitClientConfig
+from cognite_toolkit._cdf_tk.client.http_client import HTTPClient, RequestMessage2
 from cognite_toolkit._cdf_tk.utils.collection import chunker
-from cognite_toolkit._cdf_tk.utils.http_client import HTTPClient, RequestMessage2
 from cognite_toolkit._cdf_tk.utils.useful_types import JsonVal
 
 

cognite_toolkit/_cdf_tk/client/api/legacy/extended_raw.py

@@ -1,6 +1,6 @@
 from cognite.client._api.raw import RawAPI
 
-from cognite_toolkit._cdf_tk.client.data_classes.legacy.raw import RawProfileResults, RawTable
+from cognite_toolkit._cdf_tk.client.resource_classes.legacy.raw import RawProfileResults, RawTable
 
 
 class ExtendedRawAPI(RawAPI):

cognite_toolkit/_cdf_tk/client/api/legacy/extended_timeseries.py

@@ -13,11 +13,11 @@ from cognite.client.utils._identifier import IdentifierSequence
 from cognite.client.utils._validation import prepare_filter_sort, process_asset_subtree_ids, process_data_set_ids
 from cognite.client.utils.useful_types import SequenceNotStr
 
-from cognite_toolkit._cdf_tk.client.data_classes.legacy.extended_timeseries import (
+from cognite_toolkit._cdf_tk.client.resource_classes.legacy.extended_timeseries import (
     ExtendedTimeSeries,
     ExtendedTimeSeriesList,
 )
-from cognite_toolkit._cdf_tk.client.data_classes.legacy.pending_instances_ids import PendingInstanceId
+from cognite_toolkit._cdf_tk.client.resource_classes.legacy.pending_instances_ids import PendingInstanceId
 
 
 class ExtendedTimeSeriesAPI(TimeSeriesAPI):

cognite_toolkit/_cdf_tk/client/api/legacy/location_filters.py

@@ -6,7 +6,7 @@ from cognite.client._api_client import APIClient
 from cognite.client.config import ClientConfig
 from cognite.client.utils._identifier import IdentifierSequence
 
-from cognite_toolkit._cdf_tk.client.data_classes.legacy.location_filters import (
+from cognite_toolkit._cdf_tk.client.resource_classes.legacy.location_filters import (
     LocationFilter,
     LocationFilterList,
     LocationFilterWrite,

cognite_toolkit/_cdf_tk/client/api/legacy/robotics/capabilities.py

@@ -5,7 +5,7 @@ from cognite.client._api_client import APIClient
 from cognite.client.utils._identifier import IdentifierSequence
 from cognite.client.utils.useful_types import SequenceNotStr
 
-from cognite_toolkit._cdf_tk.client.data_classes.legacy.robotics import (
+from cognite_toolkit._cdf_tk.client.resource_classes.legacy.robotics import (
     RobotCapability,
     RobotCapabilityList,
     RobotCapabilityWrite,

cognite_toolkit/_cdf_tk/client/api/legacy/robotics/data_postprocessing.py

@@ -5,7 +5,7 @@ from cognite.client._api_client import APIClient
 from cognite.client.utils._identifier import IdentifierSequence
 from cognite.client.utils.useful_types import SequenceNotStr
 
-from cognite_toolkit._cdf_tk.client.data_classes.legacy.robotics import (
+from cognite_toolkit._cdf_tk.client.resource_classes.legacy.robotics import (
     DataPostProcessing,
     DataPostProcessingList,
     DataPostProcessingWrite,

cognite_toolkit/_cdf_tk/client/api/legacy/robotics/frames.py

@@ -5,7 +5,7 @@ from cognite.client._api_client import APIClient
 from cognite.client.utils._identifier import IdentifierSequence
 from cognite.client.utils.useful_types import SequenceNotStr
 
-from cognite_toolkit._cdf_tk.client.data_classes.legacy.robotics import (
+from cognite_toolkit._cdf_tk.client.resource_classes.legacy.robotics import (
     Frame,
     FrameList,
     FrameWrite,

cognite_toolkit/_cdf_tk/client/api/legacy/robotics/locations.py

@@ -5,7 +5,7 @@ from cognite.client._api_client import APIClient
 from cognite.client.utils._identifier import IdentifierSequence
 from cognite.client.utils.useful_types import SequenceNotStr
 
-from cognite_toolkit._cdf_tk.client.data_classes.legacy.robotics import (
+from cognite_toolkit._cdf_tk.client.resource_classes.legacy.robotics import (
     Location,
     LocationList,
     LocationWrite,

cognite_toolkit/_cdf_tk/client/api/legacy/robotics/maps.py

@@ -5,7 +5,7 @@ from cognite.client._api_client import APIClient
 from cognite.client.utils._identifier import IdentifierSequence
 from cognite.client.utils.useful_types import SequenceNotStr
 
-from cognite_toolkit._cdf_tk.client.data_classes.legacy.robotics import (
+from cognite_toolkit._cdf_tk.client.resource_classes.legacy.robotics import (
     Map,
     MapList,
     MapWrite,

cognite_toolkit/_cdf_tk/client/api/legacy/robotics/robots.py

@@ -4,7 +4,7 @@ from typing import overload
 from cognite.client._api_client import APIClient
 
 from cognite_toolkit._cdf_tk.client.api.legacy.robotics.utlis import tmp_disable_gzip
-from cognite_toolkit._cdf_tk.client.data_classes.legacy.robotics import Robot, RobotList, RobotWrite, _RobotUpdate
+from cognite_toolkit._cdf_tk.client.resource_classes.legacy.robotics import Robot, RobotList, RobotWrite, _RobotUpdate
 
 
 class RobotsAPI(APIClient):

cognite_toolkit/_cdf_tk/client/api/legacy/search_config.py

@@ -4,7 +4,7 @@ from cognite.client import CogniteClient
 from cognite.client._api_client import APIClient
 from cognite.client.config import ClientConfig
 
-from cognite_toolkit._cdf_tk.client.data_classes.legacy.search_config import (
+from cognite_toolkit._cdf_tk.client.resource_classes.legacy.search_config import (
     SearchConfig,
     SearchConfigList,
     SearchConfigWrite,