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

cognite_toolkit/_cdf_tk/client/api/workflow_versions.py

@@ -0,0 +1,138 @@
+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.resource_classes.identifiers import WorkflowVersionId
+from cognite_toolkit._cdf_tk.client.resource_classes.workflow_version import (
+    WorkflowVersionRequest,
+    WorkflowVersionResponse,
+)
+
+
+class WorkflowVersionsAPI(CDFResourceAPI[WorkflowVersionId, WorkflowVersionRequest, WorkflowVersionResponse]):
+    def __init__(self, http_client: HTTPClient) -> None:
+        super().__init__(
+            http_client=http_client,
+            method_endpoint_map={
+                "upsert": Endpoint(method="POST", path="/workflows/versions", item_limit=1),
+                "retrieve": Endpoint(
+                    method="GET", path="/workflow/{workflowExternalId}/versions/{version}", item_limit=1
+                ),
+                "delete": Endpoint(method="POST", path="/workflows/versions/delete", item_limit=100),
+                "list": Endpoint(method="POST", path="/workflows/versions/list", item_limit=1000),
+            },
+        )
+
+    def _validate_page_response(
+        self, response: SuccessResponse2 | ItemsSuccessResponse2
+    ) -> PagedResponse[WorkflowVersionResponse]:
+        return PagedResponse[WorkflowVersionResponse].model_validate_json(response.body)
+
+    def _reference_response(self, response: SuccessResponse2) -> ResponseItems[WorkflowVersionId]:
+        return ResponseItems[WorkflowVersionId].model_validate_json(response.body)
+
+    def create(self, items: Sequence[WorkflowVersionRequest]) -> list[WorkflowVersionResponse]:
+        """Create or update workflow versions in CDF.
+
+        Args:
+            items: List of WorkflowVersionRequest objects to create or update.
+        Returns:
+            List of created/updated WorkflowVersionResponse objects.
+        """
+        return self._request_item_response(items, "upsert")
+
+    # This is a duplicate of the create method, included to standardize the API interface.
+    def update(self, items: Sequence[WorkflowVersionRequest]) -> list[WorkflowVersionResponse]:
+        """Create or update workflow versions in CDF.
+
+        Args:
+            items: List of WorkflowVersionRequest objects to create or update.
+        Returns:
+            List of created/updated WorkflowVersionResponse objects.
+        """
+        return self.create(items)
+
+    def retrieve(self, items: Sequence[WorkflowVersionId]) -> list[WorkflowVersionResponse]:
+        """Retrieve workflow versions from CDF.
+
+        Args:
+            items: List of WorkflowVersionId objects to retrieve.
+        Returns:
+            List of retrieved WorkflowVersionResponse objects.
+        """
+        result: list[WorkflowVersionResponse] = []
+        for item in items:
+            endpoint = f"/workflows/{item.workflow_external_id}/versions/{item.version}"
+            retrieved = self._request_item_response([item], "retrieve", endpoint=endpoint)
+            result.extend(retrieved)
+        return result
+
+    def delete(self, items: Sequence[WorkflowVersionId]) -> None:
+        """Delete workflow versions from CDF.
+
+        Args:
+            items: List of WorkflowVersionId objects to delete.
+        """
+        self._request_no_response(items, "delete")
+
+    def paginate(
+        self,
+        workflow_external_id: str | None = None,
+        limit: int = 100,
+        cursor: str | None = None,
+    ) -> PagedResponse[WorkflowVersionResponse]:
+        """Iterate over all workflow versions in CDF.
+
+        Args:
+            workflow_external_id: Filter by workflow external ID.
+            limit: Maximum number of items to return.
+            cursor: Cursor for pagination.
+
+        Returns:
+            PagedResponse of WorkflowVersionResponse objects.
+        """
+        body: dict[str, Any] = {}
+        if workflow_external_id:
+            body["workflowExternalId"] = workflow_external_id
+
+        return self._paginate(
+            cursor=cursor,
+            limit=limit,
+            body=body,
+        )
+
+    def iterate(
+        self,
+        workflow_external_id: str | None = None,
+        limit: int = 100,
+    ) -> Iterable[list[WorkflowVersionResponse]]:
+        """Iterate over all workflow versions in CDF.
+
+        Args:
+            workflow_external_id: Filter by workflow external ID.
+            limit: Maximum number of items to return per page.
+
+        Returns:
+            Iterable of lists of WorkflowVersionResponse objects.
+        """
+        body: dict[str, Any] = {}
+        if workflow_external_id:
+            body["workflowExternalId"] = workflow_external_id
+
+        return self._iterate(
+            limit=limit,
+            body=body,
+        )
+
+    def list(
+        self,
+        limit: int | None = 100,
+    ) -> list[WorkflowVersionResponse]:
+        """List all workflow versions in CDF.
+
+        Returns:
+            List of WorkflowVersionResponse objects.
+        """
+        return self._list(limit=limit)

cognite_toolkit/_cdf_tk/client/api/workflows.py

@@ -0,0 +1,119 @@
+from collections.abc import Iterable, Sequence
+
+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 ExternalId
+from cognite_toolkit._cdf_tk.client.resource_classes.workflow import WorkflowRequest, WorkflowResponse
+
+from .workflow_triggers import WorkflowTriggersAPI
+from .workflow_versions import WorkflowVersionsAPI
+
+
+class WorkflowsAPI(CDFResourceAPI[ExternalId, WorkflowRequest, WorkflowResponse]):
+    def __init__(self, http_client: HTTPClient) -> None:
+        super().__init__(
+            http_client=http_client,
+            method_endpoint_map={
+                "upsert": Endpoint(method="POST", path="/workflows", item_limit=1),
+                "retrieve": Endpoint(method="GET", path="/workflows/{workflowExternalId}", item_limit=1),
+                "delete": Endpoint(method="POST", path="/workflows/delete", item_limit=100),
+                "list": Endpoint(method="GET", path="/workflows", item_limit=100),
+            },
+        )
+        self.versions = WorkflowVersionsAPI(http_client)
+        self.triggers = WorkflowTriggersAPI(http_client)
+
+    def _validate_page_response(
+        self, response: SuccessResponse2 | ItemsSuccessResponse2
+    ) -> PagedResponse[WorkflowResponse]:
+        return PagedResponse[WorkflowResponse].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[WorkflowRequest]) -> list[WorkflowResponse]:
+        """Create or update workflows in CDF.
+
+        Args:
+            items: List of WorkflowRequest objects to create or update.
+        Returns:
+            List of created/updated WorkflowResponse objects.
+        """
+        return self._request_item_response(items, "upsert")
+
+    # This is a duplicate of the create method, included to standardize the API interface.
+    def update(self, items: Sequence[WorkflowRequest]) -> list[WorkflowResponse]:
+        """Create or update workflows in CDF.
+
+        Args:
+            items: List of WorkflowRequest objects to create or update.
+        Returns:
+            List of created/updated WorkflowResponse objects.
+        """
+        return self.create(items)
+
+    def retrieve(self, items: Sequence[ExternalId]) -> list[WorkflowResponse]:
+        """Retrieve a workflow from CDF by external ID.
+
+        Args:
+            items: List of ExternalId objects to retrieve.
+
+        Returns:
+            The retrieved WorkflowResponse object, or None if not found.
+        """
+        result: list[WorkflowResponse] = []
+        for item in items:
+            endpoint = f"/workflows/{item.external_id}"
+            retrieved = self._request_item_response([item], "retrieve", endpoint=endpoint)
+            result.extend(retrieved)
+        return result
+
+    def delete(self, items: Sequence[ExternalId]) -> None:
+        """Delete workflows from CDF.
+
+        Args:
+            items: List of ExternalId objects to delete.
+        """
+        self._request_no_response(items, "delete")
+
+    def paginate(
+        self,
+        limit: int = 100,
+        cursor: str | None = None,
+    ) -> PagedResponse[WorkflowResponse]:
+        """Iterate over all workflows in CDF.
+
+        Args:
+            limit: Maximum number of items to return.
+            cursor: Cursor for pagination.
+
+        Returns:
+            PagedResponse of WorkflowResponse objects.
+        """
+        return self._paginate(cursor=cursor, limit=limit)
+
+    def iterate(
+        self,
+        limit: int = 100,
+    ) -> Iterable[list[WorkflowResponse]]:
+        """Iterate over all workflows in CDF.
+
+        Args:
+            limit: Maximum number of items to return per page.
+
+        Returns:
+            Iterable of lists of WorkflowResponse objects.
+        """
+        return self._iterate(limit=limit)
+
+    def list(
+        self,
+        limit: int | None = 100,
+    ) -> list[WorkflowResponse]:
+        """List all workflows in CDF.
+
+        Returns:
+            List of WorkflowResponse objects.
+        """
+        return self._list(limit=limit)

cognite_toolkit/_cdf_tk/client/cdf_client/api.py

@@ -13,12 +13,6 @@ from typing import Any, Generic, Literal, TypeAlias, TypeVar
 
 from pydantic import BaseModel, JsonValue
 
-from cognite_toolkit._cdf_tk.client.data_classes.base import (
-    RequestUpdateable,
-    T_Identifier,
-    T_RequestResource,
-    T_ResponseResource,
-)
 from cognite_toolkit._cdf_tk.client.http_client import (
     HTTPClient,
     ItemsRequest2,
@@ -26,9 +20,15 @@ from cognite_toolkit._cdf_tk.client.http_client import (
     RequestMessage2,
     SuccessResponse2,
 )
+from cognite_toolkit._cdf_tk.client.resource_classes.base import (
+    RequestUpdateable,
+    T_Identifier,
+    T_RequestResource,
+    T_ResponseResource,
+)
 from cognite_toolkit._cdf_tk.utils.collection import chunker_sequence
 
-from .responses import PagedResponse, ResponseItems
+from .responses import PagedResponse
 
 _T_BaseModel = TypeVar("_T_BaseModel", bound=BaseModel)
 
@@ -41,7 +41,7 @@ class Endpoint:
     concurrency_max_workers: int = 1
 
 
-APIMethod: TypeAlias = Literal["create", "retrieve", "update", "delete", "list"]
+APIMethod: TypeAlias = Literal["create", "retrieve", "update", "delete", "list", "upsert"]
 
 
 class CDFResourceAPI(Generic[T_Identifier, T_RequestResource, T_ResponseResource], ABC):
@@ -74,15 +74,12 @@ class CDFResourceAPI(Generic[T_Identifier, T_RequestResource, T_ResponseResource
         return [item.as_update(mode=mode) for item in items]
 
     @abstractmethod
-    def _page_response(self, response: SuccessResponse2 | ItemsSuccessResponse2) -> PagedResponse[T_ResponseResource]:
+    def _validate_page_response(
+        self, response: SuccessResponse2 | ItemsSuccessResponse2
+    ) -> PagedResponse[T_ResponseResource]:
         """Parse a single item response."""
         raise NotImplementedError()
 
-    @abstractmethod
-    def _reference_response(self, response: SuccessResponse2) -> ResponseItems[T_Identifier]:
-        """Parse a reference response."""
-        raise NotImplementedError()
-
     def _make_url(self, path: str = "") -> str:
         """Create the full URL for this resource endpoint."""
         return self._http_client.config.create_api_url(path)
@@ -95,7 +92,7 @@ class CDFResourceAPI(Generic[T_Identifier, T_RequestResource, T_ResponseResource
         for response in self._chunk_requests(
             items, "update", serialization=partial(self._serialize_updates, mode=mode)
         ):
-            response_items.extend(self._page_response(response).items)
+            response_items.extend(self._validate_page_response(response).items)
         return response_items
 
     def _request_item_response(
@@ -108,21 +105,9 @@ class CDFResourceAPI(Generic[T_Identifier, T_RequestResource, T_ResponseResource
     ) -> list[T_ResponseResource]:
         response_items: list[T_ResponseResource] = []
         for response in self._chunk_requests(items, method, self._serialize_items, params, extra_body, endpoint):
-            response_items.extend(self._page_response(response).items)
+            response_items.extend(self._validate_page_response(response).items)
         return response_items
 
-    def _request_reference_response(
-        self,
-        items: Sequence[BaseModel],
-        method: APIMethod,
-        params: dict[str, Any] | None = None,
-        extra_body: dict[str, Any] | None = None,
-    ) -> list[T_Identifier]:
-        all_ids: list[T_Identifier] = []
-        for response in self._chunk_requests(items, method, self._serialize_items, params, extra_body):
-            all_ids.extend(self._reference_response(response).items)
-        return all_ids
-
     def _request_no_response(
         self,
         items: Sequence[BaseModel],
@@ -192,7 +177,7 @@ class CDFResourceAPI(Generic[T_Identifier, T_RequestResource, T_ResponseResource
         """
         response_items: list[T_ResponseResource] = []
         for response in self._chunk_requests_items_split_retries(items, method, params, extra_body):
-            response_items.extend(self._page_response(response).items)
+            response_items.extend(self._validate_page_response(response).items)
         return response_items
 
     def _chunk_requests_items_split_retries(
@@ -254,7 +239,7 @@ class CDFResourceAPI(Generic[T_Identifier, T_RequestResource, T_ResponseResource
             grouped_items[key].append(item)
         return grouped_items
 
-    def _iterate(
+    def _paginate(
         self,
         limit: int,
         cursor: str | None = None,
@@ -305,22 +290,41 @@ class CDFResourceAPI(Generic[T_Identifier, T_RequestResource, T_ResponseResource
         )
         result = self._http_client.request_single_retries(request)
         response = result.get_success_or_raise()
-        return self._page_response(response)
+        return self._validate_page_response(response)
 
-    def _list(
-        self, limit: int | None = None, params: dict[str, Any] | None = None, endpoint_path: str | None = None
-    ) -> list[T_ResponseResource]:
-        """List all resources, handling pagination automatically."""
-        all_items: list[T_ResponseResource] = []
-        next_cursor: str | None = None
+    def _iterate(
+        self,
+        limit: int | None = None,
+        cursor: str | None = None,
+        params: dict[str, Any] | None = None,
+        body: dict[str, Any] | None = None,
+        endpoint_path: str | None = None,
+    ) -> Iterable[list[T_ResponseResource]]:
+        """Iterate over all resources, handling pagination automatically."""
+        next_cursor = cursor
         total = 0
         endpoint = self._method_endpoint_map["list"]
         while True:
             page_limit = endpoint.item_limit if limit is None else min(limit - total, endpoint.item_limit)
-            page = self._iterate(limit=page_limit, cursor=next_cursor, params=params, endpoint_path=endpoint_path)
-            all_items.extend(page.items)
+            page = self._paginate(
+                limit=page_limit, cursor=next_cursor, params=params, body=body, endpoint_path=endpoint_path
+            )
+            yield page.items
             total += len(page.items)
             if page.next_cursor is None or (limit is not None and total >= limit):
                 break
             next_cursor = page.next_cursor
-        return all_items
+
+    def _list(
+        self,
+        limit: int | None = None,
+        params: dict[str, Any] | None = None,
+        endpoint_path: str | None = None,
+        body: dict[str, Any] | None = None,
+    ) -> list[T_ResponseResource]:
+        """List all resources, handling pagination automatically."""
+        return [
+            item
+            for batch in self._iterate(limit=limit, params=params, endpoint_path=endpoint_path, body=body)
+            for item in batch
+        ]

cognite_toolkit/_cdf_tk/client/cdf_client/responses.py

@@ -2,6 +2,8 @@ from typing import Generic, TypeVar
 
 from pydantic import BaseModel, Field, JsonValue
 
+from cognite_toolkit._cdf_tk.client.resource_classes.graphql_data_model import GraphQLDataModelResponse
+
 T = TypeVar("T", bound=BaseModel)
 
 
@@ -25,3 +27,12 @@ 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 GraphQLResponse(BaseModel):
+    data: GraphQLDataModelResponse
+    errors: list[dict[str, JsonValue]] | None = None
+
+
+class GraphQLUpsertResponse(BaseModel):
+    upsert_graph_ql_dml_version: GraphQLResponse = Field(alias="upsertGraphQlDmlVersion")

cognite_toolkit/_cdf_tk/client/http_client/_data_classes.py

@@ -307,6 +307,11 @@ class ItemsRequest(Generic[T_COVARIANT_ID], BodyRequest):
             connect_attempt=self.connect_attempt,
             read_attempt=self.read_attempt,
             status_attempt=status_attempts,
+            api_version=self.api_version,
+            content_type=self.content_type,
+            accept=self.accept,
+            content_length=self.content_length,
+            max_failures_before_abort=self.max_failures_before_abort,
         )
         first_half.tracker = tracker
         second_half = ItemsRequest[T_COVARIANT_ID](
@@ -317,6 +322,11 @@ class ItemsRequest(Generic[T_COVARIANT_ID], BodyRequest):
             connect_attempt=self.connect_attempt,
             read_attempt=self.read_attempt,
             status_attempt=status_attempts,
+            api_version=self.api_version,
+            content_type=self.content_type,
+            accept=self.accept,
+            content_length=self.content_length,
+            max_failures_before_abort=self.max_failures_before_abort,
         )
         second_half.tracker = tracker
         return [first_half, second_half]

cognite_toolkit/_cdf_tk/client/http_client/_data_classes2.py

@@ -1,5 +1,4 @@
 import gzip
-import sys
 from abc import ABC, abstractmethod
 from collections import UserList
 from collections.abc import Sequence
@@ -13,11 +12,6 @@ from cognite_toolkit._cdf_tk.client.http_client._exception import ToolkitAPIErro
 from cognite_toolkit._cdf_tk.client.http_client._tracker import ItemsRequestTracker
 from cognite_toolkit._cdf_tk.utils.useful_types import PrimitiveType
 
-if sys.version_info >= (3, 11):
-    pass
-else:
-    pass
-
 
 class HTTPResult2(BaseModel):
     def get_success_or_raise(self) -> "SuccessResponse2":
@@ -25,7 +19,11 @@ class HTTPResult2(BaseModel):
         if isinstance(self, SuccessResponse2):
             return self
         elif isinstance(self, FailedResponse2):
-            raise ToolkitAPIError(f"Request failed with status code {self.status_code}: {self.error.message}")
+            raise ToolkitAPIError(
+                f"Request failed with status code {self.status_code}: {self.error.message}",
+                missing=self.error.missing,  # type: ignore[arg-type]
+                duplicated=self.error.duplicated,  # type: ignore[arg-type]
+            )
         elif isinstance(self, FailedRequest2):
             raise ToolkitAPIError(f"Request failed with error: {self.error}")
         else:

cognite_toolkit/_cdf_tk/client/http_client/_exception.py

@@ -1,4 +1,13 @@
+from typing import Any
+
+
 class ToolkitAPIError(Exception):
     """Base class for all exceptions raised by the Cognite Toolkit API client."""
 
-    pass
+    def __init__(
+        self, message: str, missing: list[dict[str, Any]] | None = None, duplicated: list[dict[str, Any]] | None = None
+    ) -> None:
+        super().__init__(message)
+        self.message = message
+        self.missing = missing
+        self.duplicated = duplicated

cognite_toolkit/_cdf_tk/client/request_classes/base.py

@@ -0,0 +1,19 @@
+from typing import Any
+
+from pydantic import BaseModel, ConfigDict
+from pydantic.alias_generators import to_camel
+
+
+class BaseModelRequest(BaseModel):
+    """Base class for all object. This includes resources and nested objects."""
+
+    model_config = ConfigDict(alias_generator=to_camel, extra="ignore", populate_by_name=True)
+
+    def dump(self, camel_case: bool = True) -> dict[str, Any]:
+        """Dump the object to a dictionary.
+
+        Args:
+            camel_case (bool): Whether to use camelCase for the keys. Default is True.
+
+        """
+        return self.model_dump(mode="json", by_alias=camel_case, exclude_unset=True)

cognite_toolkit/_cdf_tk/client/request_classes/filters.py

@@ -0,0 +1,69 @@
+import sys
+from typing import Literal
+
+from cognite_toolkit._cdf_tk.client.resource_classes.annotation import AnnotationStatus, AnnotationType
+from cognite_toolkit._cdf_tk.client.resource_classes.identifiers import ExternalId, InternalId
+
+from .base import BaseModelRequest
+
+if sys.version_info >= (3, 11):
+    from typing import Self
+else:
+    from typing_extensions import Self
+
+
+class Filter(BaseModelRequest): ...
+
+
+class ClassicFilter(Filter):
+    asset_subtree_ids: list[ExternalId | InternalId] | None = None
+    data_set_ids: list[ExternalId | InternalId] | None = None
+
+    @classmethod
+    def from_asset_subtree_and_data_sets(
+        cls,
+        asset_subtree_id: str | int | list[str | int] | None = None,
+        data_set_id: str | int | list[str | int] | None = None,
+    ) -> Self:
+        return cls(
+            asset_subtree_ids=cls._as_internal_or_external_id_list(asset_subtree_id),
+            data_set_ids=cls._as_internal_or_external_id_list(data_set_id),
+        )
+
+    @classmethod
+    def _as_internal_or_external_id_list(
+        cls, id: str | int | list[str | int] | None
+    ) -> list[ExternalId | InternalId] | None:
+        if id is None:
+            return None
+        ids = id if isinstance(id, list) else [id]
+        return [ExternalId(external_id=item) if isinstance(item, str) else InternalId(id=item) for item in ids]
+
+
+class DataModelingFilter(Filter):
+    space: str | None = None
+    include_global: bool | None = None
+
+
+class ContainerFilter(DataModelingFilter):
+    used_for: Literal["node", "edge", "record", "all"] | None
+
+
+class ViewFilter(DataModelingFilter):
+    include_inherited_properties: bool | None = None
+    all_versions: bool | None = None
+
+
+class DataModelFilter(DataModelingFilter):
+    inline_views: bool | None = None
+    all_versions: bool | None = None
+
+
+class AnnotationFilter(Filter):
+    annotated_resource_type: Literal["file", "threedmodel"]
+    annotated_resource_ids: list[ExternalId | InternalId]
+    annotation_type: AnnotationType | None = None
+    created_app: str | None = None
+    creating_app_version: str | None = None
+    creating_user: str | None = None
+    status: AnnotationStatus | None = None

cognite_toolkit/_cdf_tk/client/request_classes/graphql.py

@@ -0,0 +1,28 @@
+UPSERT_BODY = """
+mutation UpsertGraphQlDmlVersion($dmCreate: GraphQlDmlVersionUpsert!) {
+    upsertGraphQlDmlVersion(graphQlDmlVersion: $dmCreate) {
+        errors {
+            kind
+            message
+            hint
+            location {
+                start {
+                    line
+                    column
+                }
+            }
+        }
+        result {
+            space
+            externalId
+            version
+            name
+            description
+            graphQlDml
+            isGlobal
+            createdTime
+            lastUpdatedTime
+        }
+    }
+}
+"""

cognite_toolkit/_cdf_tk/client/{data_classes → resource_classes}/agent.py

@@ -2,8 +2,7 @@ from typing import Annotated, Any, Literal
 
 from pydantic import BeforeValidator, Field
 
-from cognite_toolkit._cdf_tk.client.data_classes.base import BaseModelObject, RequestResource, ResponseResource
-from tests.test_unit.test_cdf_tk.test_tk_warnings.test_warnings_metatest import get_all_subclasses
+from cognite_toolkit._cdf_tk.client.resource_classes.base import BaseModelObject, RequestResource, ResponseResource
 
 from .identifiers import ExternalId
 
@@ -74,7 +73,14 @@ class UnknownAgentTool(AgentToolDefinition):
     ...
 
 
-KNOWN_TOOLS = {tool.type: tool for tool in get_all_subclasses(AgentToolDefinition) if hasattr(tool, "type")}
+# Mapping of known agent tool types to their classes
+KNOWN_TOOLS: dict[str, type[AgentToolDefinition]] = {
+    "askDocument": AskDocument,
+    "examineDataSemantically": ExamineDataSemantically,
+    "queryKnowledgeGraph": QueryKnowledgeGraph,
+    "queryTimeSeriesDatapoints": QueryTimeSeriesDatapoints,
+    "summarizeDocument": SummarizeDocument,
+}
 
 
 def _handle_unknown_tool(value: Any) -> Any:

cognite_toolkit/_cdf_tk/client/{data_classes → resource_classes}/annotation.py

@@ -2,7 +2,7 @@ from typing import Any, Literal, TypeAlias
 
 from pydantic import Field, JsonValue
 
-from cognite_toolkit._cdf_tk.client.data_classes.base import BaseModelObject, RequestUpdateable, ResponseResource
+from cognite_toolkit._cdf_tk.client.resource_classes.base import BaseModelObject, RequestUpdateable, ResponseResource
 
 from .identifiers import InternalId
 

cognite_toolkit/_cdf_tk/client/{data_classes → resource_classes}/asset.py

@@ -2,7 +2,7 @@ from typing import ClassVar, Literal
 
 from pydantic import JsonValue
 
-from cognite_toolkit._cdf_tk.client.data_classes.base import BaseModelObject, RequestUpdateable, ResponseResource
+from cognite_toolkit._cdf_tk.client.resource_classes.base import BaseModelObject, RequestUpdateable, ResponseResource
 
 from .identifiers import ExternalId