cognite-toolkit 0.7.45__py3-none-any.whl → 0.7.47__py3-none-any.whl

cognite_toolkit/_cdf_tk/client/api/raw.py

@@ -1,8 +1,8 @@
-from collections.abc import Sequence
+from collections.abc import Iterable, Sequence
 
 from cognite_toolkit._cdf_tk.client.cdf_client import CDFResourceAPI, Endpoint, PagedResponse, ResponseItems
-from cognite_toolkit._cdf_tk.client.data_classes.raw import RAWDatabase, RAWTable
 from cognite_toolkit._cdf_tk.client.http_client import HTTPClient, ItemsSuccessResponse2, SuccessResponse2
+from cognite_toolkit._cdf_tk.client.resource_classes.raw import RAWDatabase, RAWTable
 
 
 class RawDatabasesAPI(CDFResourceAPI[RAWDatabase, RAWDatabase, RAWDatabase]):
@@ -21,7 +21,7 @@ class RawDatabasesAPI(CDFResourceAPI[RAWDatabase, RAWDatabase, RAWDatabase]):
             },
         )
 
-    def _page_response(self, response: SuccessResponse2 | ItemsSuccessResponse2) -> PagedResponse[RAWDatabase]:
+    def _validate_page_response(self, response: SuccessResponse2 | ItemsSuccessResponse2) -> PagedResponse[RAWDatabase]:
         return PagedResponse[RAWDatabase].model_validate_json(response.body)
 
     def _reference_response(self, response: SuccessResponse2) -> ResponseItems[RAWDatabase]:
@@ -47,7 +47,7 @@ class RawDatabasesAPI(CDFResourceAPI[RAWDatabase, RAWDatabase, RAWDatabase]):
         """
         self._request_no_response(list(items), "delete", extra_body={"recursive": recursive})
 
-    def iterate(
+    def paginate(
         self,
         limit: int = 100,
         cursor: str | None = None,
@@ -61,7 +61,21 @@ class RawDatabasesAPI(CDFResourceAPI[RAWDatabase, RAWDatabase, RAWDatabase]):
         Returns:
             PagedResponse of RAWDatabase objects.
         """
-        return self._iterate(limit=limit, cursor=cursor)
+        return self._paginate(limit=limit, cursor=cursor)
+
+    def iterate(
+        self,
+        limit: int = 100,
+    ) -> Iterable[list[RAWDatabase]]:
+        """Iterate over all databases in CDF.
+
+        Args:
+            limit: Maximum number of items to return per page.
+
+        Returns:
+            Iterable of lists of RAWDatabase objects.
+        """
+        return self._iterate(limit=limit)
 
     def list(self, limit: int | None = None) -> list[RAWDatabase]:
         """List all databases in CDF.
@@ -95,7 +109,7 @@ class RawTablesAPI(CDFResourceAPI[RAWTable, RAWTable, RAWTable]):
             },
         )
 
-    def _page_response(self, response: SuccessResponse2 | ItemsSuccessResponse2) -> PagedResponse[RAWTable]:
+    def _validate_page_response(self, response: SuccessResponse2 | ItemsSuccessResponse2) -> PagedResponse[RAWTable]:
         """Parse a page response. Note: db_name must be injected separately."""
         return PagedResponse[RAWTable].model_validate_json(response.body)
 
@@ -137,7 +151,7 @@ class RawTablesAPI(CDFResourceAPI[RAWTable, RAWTable, RAWTable]):
             endpoint = f"/raw/dbs/{db_name}/tables/delete"
             self._request_no_response(list(group), "delete", endpoint=endpoint)
 
-    def iterate(
+    def paginate(
         self,
         db_name: str,
         limit: int = 100,
@@ -153,7 +167,23 @@ class RawTablesAPI(CDFResourceAPI[RAWTable, RAWTable, RAWTable]):
         Returns:
             PagedResponse of RAWTable objects.
         """
-        return self._iterate(cursor=cursor, limit=limit, endpoint_path=f"/raw/dbs/{db_name}/tables")
+        return self._paginate(cursor=cursor, limit=limit, endpoint_path=f"/raw/dbs/{db_name}/tables")
+
+    def iterate(
+        self,
+        db_name: str,
+        limit: int = 100,
+    ) -> Iterable[list[RAWTable]]:
+        """Iterate over all tables in a database in CDF.
+
+        Args:
+            db_name: The name of the database to list tables from.
+            limit: Maximum number of items to return per page.
+
+        Returns:
+            Iterable of lists of RAWTable objects.
+        """
+        return self._iterate(limit=limit, endpoint_path=f"/raw/dbs/{db_name}/tables")
 
     def list(self, db_name: str, limit: int | None = None) -> list[RAWTable]:
         """List all tables in a database in CDF.

cognite_toolkit/_cdf_tk/client/api/relationships.py

@@ -0,0 +1,133 @@
+"""Relationships API for managing CDF relationships.
+
+Based on the API specification at:
+https://api-docs.cognite.com/20230101/tag/Relationships/operation/createRelationships
+"""
+
+from collections.abc import Iterable, Sequence
+from typing import Literal
+
+from cognite_toolkit._cdf_tk.client.cdf_client import CDFResourceAPI, Endpoint, PagedResponse
+from cognite_toolkit._cdf_tk.client.http_client import HTTPClient, ItemsSuccessResponse2, SuccessResponse2
+from cognite_toolkit._cdf_tk.client.resource_classes.identifiers import ExternalId
+from cognite_toolkit._cdf_tk.client.resource_classes.relationship import RelationshipRequest, RelationshipResponse
+
+
+class RelationshipsAPI(CDFResourceAPI[ExternalId, RelationshipRequest, RelationshipResponse]):
+    """API for managing CDF relationships."""
+
+    def __init__(self, http_client: HTTPClient) -> None:
+        super().__init__(
+            http_client=http_client,
+            method_endpoint_map={
+                "create": Endpoint(method="POST", path="/relationships", item_limit=1000),
+                "retrieve": Endpoint(method="POST", path="/relationships/byids", item_limit=1000),
+                "update": Endpoint(method="POST", path="/relationships/update", item_limit=1000),
+                "delete": Endpoint(method="POST", path="/relationships/delete", item_limit=1000),
+                "list": Endpoint(method="POST", path="/relationships/list", item_limit=1000),
+            },
+        )
+
+    def _validate_page_response(
+        self, response: SuccessResponse2 | ItemsSuccessResponse2
+    ) -> PagedResponse[RelationshipResponse]:
+        return PagedResponse[RelationshipResponse].model_validate_json(response.body)
+
+    def create(self, items: Sequence[RelationshipRequest]) -> list[RelationshipResponse]:
+        """Create relationships in CDF.
+
+        Args:
+            items: List of RelationshipRequest objects to create.
+
+        Returns:
+            List of created RelationshipResponse objects.
+        """
+        return self._request_item_response(items, "create")
+
+    def retrieve(
+        self, items: Sequence[ExternalId], ignore_unknown_ids: bool = False, fetch_resources: bool = False
+    ) -> list[RelationshipResponse]:
+        """Retrieve relationships from CDF.
+
+        Args:
+            items: List of ExternalId objects to retrieve.
+            ignore_unknown_ids: Whether to ignore unknown IDs.
+            fetch_resources: If true, will try to fetch the resources referred to in the relationship,
+                based on the users access rights. Will silently fail to attach the resources
+                if the user lacks access to some of them.
+
+        Returns:
+            List of retrieved RelationshipResponse objects.
+        """
+        return self._request_item_response(
+            items,
+            method="retrieve",
+            extra_body={"ignoreUnknownIds": ignore_unknown_ids, "fetchResources": fetch_resources},
+        )
+
+    def update(
+        self, items: Sequence[RelationshipRequest], mode: Literal["patch", "replace"] = "replace"
+    ) -> list[RelationshipResponse]:
+        """Update relationships in CDF.
+
+        Args:
+            items: List of RelationshipRequest objects to update.
+            mode: Update mode, either "patch" or "replace".
+
+        Returns:
+            List of updated RelationshipResponse objects.
+        """
+        return self._update(items, mode=mode)
+
+    def delete(self, items: Sequence[ExternalId], ignore_unknown_ids: bool = False) -> None:
+        """Delete relationships from CDF.
+
+        Args:
+            items: List of ExternalId objects to delete.
+            ignore_unknown_ids: Whether to ignore unknown IDs.
+        """
+        self._request_no_response(items, "delete", extra_body={"ignoreUnknownIds": ignore_unknown_ids})
+
+    def paginate(
+        self,
+        limit: int = 100,
+        cursor: str | None = None,
+    ) -> PagedResponse[RelationshipResponse]:
+        """Get a page of relationships from CDF.
+
+        Args:
+            limit: Maximum number of relationships to return.
+            cursor: Cursor for pagination.
+
+        Returns:
+            PagedResponse of RelationshipResponse objects.
+        """
+        return self._paginate(
+            cursor=cursor,
+            limit=limit,
+        )
+
+    def iterate(
+        self,
+        limit: int | None = None,
+    ) -> Iterable[list[RelationshipResponse]]:
+        """Iterate over all relationships in CDF.
+
+        Args:
+            limit: Maximum total number of relationships to return.
+
+        Returns:
+            Iterable of lists of RelationshipResponse objects.
+        """
+        return self._iterate(limit=limit)
+
+    def list(self, limit: int | None = None) -> list[RelationshipResponse]:
+        """List all relationships in CDF.
+
+        Args:
+            limit: Maximum total number of relationships to return.
+
+        Returns:
+            List of RelationshipResponse objects.
+        """
+        return self._list(limit=limit)

cognite_toolkit/_cdf_tk/client/api/security_categories.py

@@ -0,0 +1,94 @@
+from collections.abc import Iterable, Sequence
+from typing import 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.resource_classes.identifiers import InternalId
+from cognite_toolkit._cdf_tk.client.resource_classes.securitycategory import (
+    SecurityCategoryRequest,
+    SecurityCategoryResponse,
+)
+
+
+class SecurityCategoriesAPI(CDFResourceAPI[InternalId, SecurityCategoryRequest, SecurityCategoryResponse]):
+    def __init__(self, http_client: HTTPClient) -> None:
+        super().__init__(
+            http_client=http_client,
+            method_endpoint_map={
+                "create": Endpoint(method="POST", path="/securitycategories", item_limit=1000),
+                "delete": Endpoint(method="POST", path="/securitycategories/delete", item_limit=1000),
+                "list": Endpoint(method="GET", path="/securitycategories", item_limit=1000),
+            },
+        )
+
+    def _validate_page_response(
+        self, response: SuccessResponse2 | ItemsSuccessResponse2
+    ) -> PagedResponse[SecurityCategoryResponse]:
+        return PagedResponse[SecurityCategoryResponse].model_validate_json(response.body)
+
+    def _reference_response(self, response: SuccessResponse2) -> ResponseItems[InternalId]:
+        return ResponseItems[InternalId].model_validate_json(response.body)
+
+    def create(self, items: Sequence[SecurityCategoryRequest]) -> list[SecurityCategoryResponse]:
+        """Create security categories in CDF.
+
+        Args:
+            items: List of SecurityCategoryRequest objects to create.
+        Returns:
+            List of created SecurityCategoryResponse objects.
+        """
+        return self._request_item_response(items, "create")
+
+    def delete(self, items: Sequence[InternalId]) -> None:
+        """Delete security categories from CDF.
+
+        Args:
+            items: List of InternalId objects to delete.
+        """
+        self._request_no_response(items, "delete")
+
+    def paginate(
+        self,
+        sort: Literal["ASC", "DESC"] = "ASC",
+        limit: int = 100,
+        cursor: str | None = None,
+    ) -> PagedResponse[SecurityCategoryResponse]:
+        """Iterate over all security categories in CDF.
+
+        Args:
+            sort: Sort descending or ascending.
+            limit: Maximum number of items to return.
+            cursor: Cursor for pagination.
+
+        Returns:
+            PagedResponse of SecurityCategoryResponse objects.
+        """
+        return self._paginate(cursor=cursor, limit=limit, params={"sort": sort})
+
+    def iterate(
+        self,
+        sort: Literal["ASC", "DESC"] = "ASC",
+        limit: int = 100,
+    ) -> Iterable[list[SecurityCategoryResponse]]:
+        """Iterate over all security categories in CDF.
+
+        Args:
+            sort: Sort descending or ascending.
+            limit: Maximum number of items to return per page.
+
+        Returns:
+            Iterable of lists of SecurityCategoryResponse objects.
+        """
+        return self._iterate(limit=limit, params={"sort": sort})
+
+    def list(
+        self,
+        limit: int | None = 100,
+    ) -> list[SecurityCategoryResponse]:
+        """List all security categories in CDF.
+
+        Returns:
+            List of SecurityCategoryResponse objects.
+        """
+        return self._list(limit=limit)

cognite_toolkit/_cdf_tk/client/api/sequences.py

@@ -0,0 +1,133 @@
+from collections.abc import Iterable, Sequence
+from typing import 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 ClassicFilter
+from cognite_toolkit._cdf_tk.client.resource_classes.identifiers import InternalOrExternalId
+from cognite_toolkit._cdf_tk.client.resource_classes.sequence import SequenceRequest, SequenceResponse
+
+
+class SequencesAPI(CDFResourceAPI[InternalOrExternalId, SequenceRequest, SequenceResponse]):
+    def __init__(self, http_client: HTTPClient) -> None:
+        super().__init__(
+            http_client=http_client,
+            method_endpoint_map={
+                "create": Endpoint(method="POST", path="/sequences", item_limit=1000, concurrency_max_workers=1),
+                "retrieve": Endpoint(
+                    method="POST", path="/sequences/byids", item_limit=1000, concurrency_max_workers=1
+                ),
+                "update": Endpoint(method="POST", path="/sequences/update", item_limit=1000, concurrency_max_workers=1),
+                "delete": Endpoint(method="POST", path="/sequences/delete", item_limit=1000, concurrency_max_workers=1),
+                "list": Endpoint(method="POST", path="/sequences/list", item_limit=1000),
+            },
+        )
+
+    def _validate_page_response(
+        self, response: SuccessResponse2 | ItemsSuccessResponse2
+    ) -> PagedResponse[SequenceResponse]:
+        return PagedResponse[SequenceResponse].model_validate_json(response.body)
+
+    def _reference_response(self, response: SuccessResponse2) -> ResponseItems[InternalOrExternalId]:
+        return ResponseItems[InternalOrExternalId].model_validate_json(response.body)
+
+    def create(self, items: Sequence[SequenceRequest]) -> list[SequenceResponse]:
+        """Create sequences in CDF.
+
+        Args:
+            items: List of SequenceRequest objects to create.
+        Returns:
+            List of created SequenceResponse objects.
+        """
+        return self._request_item_response(items, "create")
+
+    def retrieve(
+        self, items: Sequence[InternalOrExternalId], ignore_unknown_ids: bool = False
+    ) -> list[SequenceResponse]:
+        """Retrieve sequences from CDF.
+
+        Args:
+            items: List of InternalOrExternalId objects to retrieve.
+            ignore_unknown_ids: Whether to ignore unknown IDs.
+        Returns:
+            List of retrieved SequenceResponse objects.
+        """
+        return self._request_item_response(
+            items, method="retrieve", extra_body={"ignoreUnknownIds": ignore_unknown_ids}
+        )
+
+    def update(
+        self, items: Sequence[SequenceRequest], mode: Literal["patch", "replace"] = "replace"
+    ) -> list[SequenceResponse]:
+        """Update sequences in CDF.
+
+        Args:
+            items: List of SequenceRequest objects to update.
+            mode: Update mode, either "patch" or "replace".
+
+        Returns:
+            List of updated SequenceResponse objects.
+        """
+        return self._update(items, mode=mode)
+
+    def delete(self, items: Sequence[InternalOrExternalId], ignore_unknown_ids: bool = False) -> None:
+        """Delete sequences from CDF.
+
+        Args:
+            items: List of InternalOrExternalId objects to delete.
+            ignore_unknown_ids: Whether to ignore unknown IDs.
+        """
+        self._request_no_response(items, "delete", extra_body={"ignoreUnknownIds": ignore_unknown_ids})
+
+    def paginate(
+        self,
+        filter: ClassicFilter | None = None,
+        limit: int = 100,
+        cursor: str | None = None,
+    ) -> PagedResponse[SequenceResponse]:
+        """Iterate over all sequences in CDF.
+
+        Args:
+            filter: Filter by data set IDs and/or asset subtree IDs.
+            limit: Maximum number of items to return.
+            cursor: Cursor for pagination.
+
+        Returns:
+            PagedResponse of SequenceResponse objects.
+        """
+        return self._paginate(
+            cursor=cursor,
+            limit=limit,
+            body={"filter": filter.dump() if filter else None},
+        )
+
+    def iterate(
+        self,
+        filter: ClassicFilter | None = None,
+        limit: int = 100,
+    ) -> Iterable[list[SequenceResponse]]:
+        """Iterate over all sequences in CDF.
+
+        Args:
+            filter: Filter by data set IDs and/or asset subtree IDs.
+            limit: Maximum number of items to return per page.
+
+        Returns:
+            Iterable of lists of SequenceResponse objects.
+        """
+        return self._iterate(
+            limit=limit,
+            body={"filter": filter.dump() if filter else None},
+        )
+
+    def list(
+        self,
+        limit: int | None = 100,
+    ) -> list[SequenceResponse]:
+        """List all sequences in CDF.
+
+        Returns:
+            List of SequenceResponse objects.
+        """
+        return self._list(limit=limit)

cognite_toolkit/_cdf_tk/client/api/simulator_models.py

@@ -1,11 +1,14 @@
-from collections.abc import Sequence
+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.data_classes.identifiers import InternalOrExternalId
-from cognite_toolkit._cdf_tk.client.data_classes.simulator_model import SimulatorModelRequest, SimulatorModelResponse
 from cognite_toolkit._cdf_tk.client.http_client import HTTPClient, ItemsSuccessResponse2, SuccessResponse2
+from cognite_toolkit._cdf_tk.client.resource_classes.identifiers import InternalOrExternalId
+from cognite_toolkit._cdf_tk.client.resource_classes.simulator_model import (
+    SimulatorModelRequest,
+    SimulatorModelResponse,
+)
 
 
 class SimulatorModelsAPI(CDFResourceAPI[InternalOrExternalId, SimulatorModelRequest, SimulatorModelResponse]):
@@ -27,7 +30,7 @@ class SimulatorModelsAPI(CDFResourceAPI[InternalOrExternalId, SimulatorModelRequ
             },
         )
 
-    def _page_response(
+    def _validate_page_response(
         self, response: SuccessResponse2 | ItemsSuccessResponse2
     ) -> PagedResponse[SimulatorModelResponse]:
         return PagedResponse[SimulatorModelResponse].model_validate_json(response.body)
@@ -87,7 +90,7 @@ class SimulatorModelsAPI(CDFResourceAPI[InternalOrExternalId, SimulatorModelRequ
         """
         self._request_no_response(items, "delete")
 
-    def iterate(
+    def paginate(
         self,
         simulator_external_ids: list[str] | None = None,
         limit: int = 100,
@@ -107,12 +110,35 @@ class SimulatorModelsAPI(CDFResourceAPI[InternalOrExternalId, SimulatorModelRequ
         if simulator_external_ids:
             filter_["simulatorExternalIds"] = simulator_external_ids
 
-        return self._iterate(
+        return self._paginate(
             cursor=cursor,
             limit=limit,
             body={"filter": filter_ or None},
         )
 
+    def iterate(
+        self,
+        simulator_external_ids: list[str] | None = None,
+        limit: int = 100,
+    ) -> Iterable[list[SimulatorModelResponse]]:
+        """Iterate over simulator models in CDF.
+
+        Args:
+            simulator_external_ids: Filter by simulator external IDs.
+            limit: Maximum number of items to return per page.
+
+        Returns:
+            Iterable of lists of SimulatorModelResponse objects.
+        """
+        filter_: dict[str, Any] = {}
+        if simulator_external_ids:
+            filter_["simulatorExternalIds"] = simulator_external_ids
+
+        return self._iterate(
+            limit=limit,
+            body={"filter": filter_ or None},
+        )
+
     def list(
         self,
         limit: int | None = 100,

cognite_toolkit/_cdf_tk/client/api/spaces.py

@@ -0,0 +1,117 @@
+"""Spaces API for managing CDF data modeling spaces.
+
+Based on the API specification at:
+https://api-docs.cognite.com/20230101/tag/Spaces/operation/ApplySpaces
+"""
+
+from collections.abc import Iterable, Sequence
+
+from cognite_toolkit._cdf_tk.client.cdf_client import CDFResourceAPI, Endpoint, PagedResponse
+from cognite_toolkit._cdf_tk.client.http_client import HTTPClient, ItemsSuccessResponse2, SuccessResponse2
+from cognite_toolkit._cdf_tk.client.resource_classes.data_modeling import SpaceReference, SpaceRequest, SpaceResponse
+
+
+class SpacesAPI(CDFResourceAPI[SpaceReference, SpaceRequest, SpaceResponse]):
+    """API for managing CDF data modeling spaces.
+
+    Spaces use an apply/upsert pattern for create and update operations.
+    """
+
+    def __init__(self, http_client: HTTPClient) -> None:
+        super().__init__(
+            http_client=http_client,
+            method_endpoint_map={
+                "upsert": Endpoint(method="POST", path="/models/spaces", item_limit=100),
+                "retrieve": Endpoint(method="POST", path="/models/spaces/byids", item_limit=100),
+                "delete": Endpoint(method="POST", path="/models/spaces/delete", item_limit=100),
+                "list": Endpoint(method="GET", path="/models/spaces", item_limit=1000),
+            },
+        )
+
+    def _validate_page_response(
+        self, response: SuccessResponse2 | ItemsSuccessResponse2
+    ) -> PagedResponse[SpaceResponse]:
+        return PagedResponse[SpaceResponse].model_validate_json(response.body)
+
+    def apply(self, items: Sequence[SpaceRequest]) -> list[SpaceResponse]:
+        """Apply (create or update) spaces in CDF.
+
+        Args:
+            items: List of SpaceRequest objects to apply.
+
+        Returns:
+            List of applied SpaceResponse objects.
+        """
+        return self._request_item_response(items, "upsert")
+
+    def retrieve(self, items: Sequence[SpaceReference]) -> list[SpaceResponse]:
+        """Retrieve spaces from CDF.
+
+        Args:
+            items: List of SpaceReference objects to retrieve.
+
+        Returns:
+            List of retrieved SpaceResponse objects.
+        """
+        return self._request_item_response(items, method="retrieve")
+
+    def delete(self, items: Sequence[SpaceReference]) -> None:
+        """Delete spaces from CDF.
+
+        Args:
+            items: List of SpaceReference objects to delete.
+        """
+        self._request_no_response(items, "delete")
+
+    def paginate(
+        self,
+        include_global: bool = False,
+        limit: int = 100,
+        cursor: str | None = None,
+    ) -> PagedResponse[SpaceResponse]:
+        """Get a page of spaces from CDF.
+
+        Args:
+            include_global: Whether to include global spaces.
+            limit: Maximum number of spaces to return.
+            cursor: Cursor for pagination.
+
+        Returns:
+            PagedResponse of SpaceResponse objects.
+        """
+        return self._paginate(
+            cursor=cursor,
+            limit=limit,
+            params={"includeGlobal": include_global},
+        )
+
+    def iterate(
+        self,
+        include_global: bool = False,
+        limit: int | None = None,
+    ) -> Iterable[list[SpaceResponse]]:
+        """Iterate over all spaces in CDF.
+
+        Args:
+            include_global: Whether to include global spaces.
+            limit: Maximum total number of spaces to return.
+
+        Returns:
+            Iterable of lists of SpaceResponse objects.
+        """
+        return self._iterate(
+            limit=limit,
+            params={"includeGlobal": include_global},
+        )
+
+    def list(self, include_global: bool = False, limit: int | None = None) -> list[SpaceResponse]:
+        """List all spaces in CDF.
+
+        Args:
+            include_global: Whether to include global spaces.
+            limit: Maximum total number of spaces to return.
+
+        Returns:
+            List of SpaceResponse objects.
+        """
+        return self._list(limit=limit, params={"includeGlobal": include_global})

cognite_toolkit/_cdf_tk/client/api/streams.py

@@ -4,12 +4,12 @@ from pydantic import TypeAdapter
 from rich.console import Console
 
 from cognite_toolkit._cdf_tk.client.cdf_client.responses import PagedResponse
-from cognite_toolkit._cdf_tk.client.data_classes.streams import StreamRequest, StreamResponse
 from cognite_toolkit._cdf_tk.client.http_client import (
     HTTPClient,
     ItemsRequest2,
     RequestMessage2,
 )
+from cognite_toolkit._cdf_tk.client.resource_classes.streams import StreamRequest, StreamResponse
 
 
 class StreamsAPI: