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

cognite_toolkit/_cdf_tk/client/cdf_client/api.py

@@ -51,15 +51,20 @@ class CDFResourceAPI(Generic[T_Identifier, T_RequestResource, T_ResponseResource
     including creating, retrieving, deleting, and listing resources.
     """
 
-    def __init__(self, http_client: HTTPClient, method_endpoint_map: dict[APIMethod, Endpoint]) -> None:
+    def __init__(
+        self, http_client: HTTPClient, method_endpoint_map: dict[APIMethod, Endpoint], disable_gzip: bool = False
+    ) -> None:
         """Initialize the resource API.
 
         Args:
             http_client: The HTTP client to use for API requests.
             method_endpoint_map: A mapping of endpoint suffixes to their properties.
+            disable_gzip: Whether to disable gzip compression for requests. Defaults to False.
+                This is only used by the robotics API. If that API is dropped, this parameter can be removed.
         """
         self._http_client = http_client
         self._method_endpoint_map = method_endpoint_map
+        self._disable_gzip = disable_gzip
 
     @classmethod
     def _serialize_items(cls, items: Sequence[BaseModel]) -> list[dict[str, JsonValue]]:
@@ -151,6 +156,7 @@ class CDFResourceAPI(Generic[T_Identifier, T_RequestResource, T_ResponseResource
                 method=endpoint.method,
                 body_content={"items": serialization(chunk), **(extra_body or {})},  # type: ignore[dict-item]
                 parameters=request_params,
+                disable_gzip=self._disable_gzip,
             )
             response = self._http_client.request_single_retries(request)
             yield response.get_success_or_raise()
@@ -180,6 +186,27 @@ class CDFResourceAPI(Generic[T_Identifier, T_RequestResource, T_ResponseResource
             response_items.extend(self._validate_page_response(response).items)
         return response_items
 
+    def _request_item_split_retries_no_response(
+        self,
+        items: Sequence[_T_BaseModel],
+        method: APIMethod,
+        params: dict[str, Any] | None = None,
+        extra_body: dict[str, Any] | None = None,
+    ) -> None:
+        """Request items with retries, splitting on failures, without returning any response.
+
+        This method handles large batches of items by chunking them according to the endpoint's item limit.
+        If a single item fails, it splits the request into individual item requests to isolate the failure.
+
+        Args:
+            items: Sequence of items to request.
+            method: API method to use for the request.
+            params: Optional query parameters for the request.
+            extra_body: Optional additional body fields for the request.
+        """
+        list(self._chunk_requests_items_split_retries(items, method, params, extra_body))
+        return None
+
     def _chunk_requests_items_split_retries(
         self,
         items: Sequence[_T_BaseModel],
@@ -215,6 +242,7 @@ class CDFResourceAPI(Generic[T_Identifier, T_RequestResource, T_ResponseResource
                 parameters=request_params,
                 items=chunk,
                 extra_body_fields=extra_body,
+                disable_gzip=self._disable_gzip,
             )
             responses = self._http_client.request_items_retries(request)
             for response in responses:
@@ -230,12 +258,12 @@ class CDFResourceAPI(Generic[T_Identifier, T_RequestResource, T_ResponseResource
 
     @classmethod
     def _group_items_by_text_field(
-        cls, items: Sequence[_T_BaseModel], field_name: str
-    ) -> dict[str, list[_T_BaseModel]]:
+        cls, items: Sequence[_T_BaseModel], *field_names: str
+    ) -> dict[tuple[str, ...], list[_T_BaseModel]]:
         """Group items by a text field."""
-        grouped_items: dict[str, list[_T_BaseModel]] = defaultdict(list)
+        grouped_items: dict[tuple[str, ...], list[_T_BaseModel]] = defaultdict(list)
         for item in items:
-            key = str(getattr(item, field_name))
+            key = tuple(str(getattr(item, field_name)) for field_name in field_names)
             grouped_items[key].append(item)
         return grouped_items
 
@@ -287,6 +315,7 @@ class CDFResourceAPI(Generic[T_Identifier, T_RequestResource, T_ResponseResource
             method=endpoint.method,
             parameters=request_params,
             body_content=body,
+            disable_gzip=self._disable_gzip,
         )
         result = self._http_client.request_single_retries(request)
         response = result.get_success_or_raise()

cognite_toolkit/_cdf_tk/client/http_client/_client.py

@@ -172,6 +172,7 @@ class HTTPClient:
         content_type: str = "application/json",
         accept: str = "application/json",
         content_length: int | None = None,
+        disable_gzip: bool = False,
     ) -> MutableMapping[str, str]:
         headers: MutableMapping[str, str] = {}
         headers["User-Agent"] = f"httpx/{httpx.__version__} {get_user_agent()}"
@@ -184,7 +185,7 @@ class HTTPClient:
         headers["x-cdp-sdk"] = f"CogniteToolkit:{get_current_toolkit_version()}"
         headers["x-cdp-app"] = self.config.client_name
         headers["cdf-version"] = api_version or self.config.api_subversion
-        if not global_config.disable_gzip and content_length is None:
+        if not global_config.disable_gzip and content_length is None and not disable_gzip:
             headers["Content-Encoding"] = "gzip"
         return headers
 
@@ -332,7 +333,9 @@ class HTTPClient:
                 raise TypeError(f"Unexpected result type: {type(result)}")
 
     def _make_request2(self, message: BaseRequestMessage) -> httpx.Response:
-        headers = self._create_headers(message.api_version, message.content_type, message.accept)
+        headers = self._create_headers(
+            message.api_version, message.content_type, message.accept, disable_gzip=message.disable_gzip
+        )
         return self.session.request(
             method=message.method,
             url=message.endpoint_url,

cognite_toolkit/_cdf_tk/client/http_client/_data_classes2.py

@@ -77,6 +77,7 @@ class BaseRequestMessage(BaseModel, ABC):
     read_attempt: int = 0
     status_attempt: int = 0
     api_version: str | None = None
+    disable_gzip: bool = False
     content_type: str = "application/json"
     accept: str = "application/json"
 
@@ -106,13 +107,13 @@ class RequestMessage2(BaseRequestMessage):
         data: str | bytes | None = None
         if self.data_content is not None:
             data = self.data_content
-            if not global_config.disable_gzip:
+            if not global_config.disable_gzip and not self.disable_gzip:
                 data = gzip.compress(data)
         elif self.body_content is not None:
             # We serialize using pydantic instead of json.dumps. This is because pydantic is faster
             # and handles more complex types such as datetime, float('nan'), etc.
             data = _BODY_SERIALIZER.dump_json(self.body_content)
-            if not global_config.disable_gzip and isinstance(data, bytes):
+            if not global_config.disable_gzip and not self.disable_gzip and isinstance(data, bytes):
                 data = gzip.compress(data)
         return data
 
@@ -161,7 +162,7 @@ class ItemsRequest2(BaseRequestMessage):
         if self.extra_body_fields:
             body.update(self.extra_body_fields)
         res = _BODY_SERIALIZER.dump_json(body)
-        if not global_config.disable_gzip and isinstance(res, bytes):
+        if not global_config.disable_gzip and not self.disable_gzip and isinstance(res, bytes):
             return gzip.compress(res)
         return res
 

cognite_toolkit/_cdf_tk/client/request_classes/filters.py

@@ -1,7 +1,10 @@
 import sys
-from typing import Literal
+from typing import Any, Literal
+
+from pydantic import Field
 
 from cognite_toolkit._cdf_tk.client.resource_classes.annotation import AnnotationStatus, AnnotationType
+from cognite_toolkit._cdf_tk.client.resource_classes.data_modeling import NodeReference, ViewReference
 from cognite_toolkit._cdf_tk.client.resource_classes.identifiers import ExternalId, InternalId
 
 from .base import BaseModelRequest
@@ -59,6 +62,28 @@ class DataModelFilter(DataModelingFilter):
     all_versions: bool | None = None
 
 
+class InstanceFilter(Filter):
+    instance_type: Literal["node", "edge"] | None = None
+    source: ViewReference | None = None
+    space: list[str] | None = None
+
+    def dump(self, camel_case: bool = True) -> dict[str, Any]:
+        body: dict[str, Any] = {}
+        if self.instance_type is not None:
+            body["instanceType"] = self.instance_type
+        if self.source is not None:
+            body["sources"] = [{"source": self.source.dump()}]
+        if self.space is not None:
+            instance_type = self.instance_type or "node"
+            body["filter"] = {
+                "in": {
+                    "property": [instance_type, "space"],
+                    "values": self.space,
+                }
+            }
+        return body
+
+
 class AnnotationFilter(Filter):
     annotated_resource_type: Literal["file", "threedmodel"]
     annotated_resource_ids: list[ExternalId | InternalId]
@@ -67,3 +92,22 @@ class AnnotationFilter(Filter):
     creating_app_version: str | None = None
     creating_user: str | None = None
     status: AnnotationStatus | None = None
+
+
+class ThreeDAssetMappingFilter(Filter): ...
+
+
+class ThreeDAssetMappingAssetIdFilter(ThreeDAssetMappingFilter):
+    asset_ids: list[int] = Field(max_length=100)
+
+
+class ThreeDAssetMappingAssetInstanceIdFilter(ThreeDAssetMappingFilter):
+    asset_instance_ids: list[NodeReference] = Field(max_length=100)
+
+
+class ThreeDAssetMapping3DNodeFilter(ThreeDAssetMappingFilter):
+    node_ids: list[int] = Field(max_length=100)
+
+
+class ThreeDAssetMappingTreeIndexFilter(ThreeDAssetMappingFilter):
+    tree_indexes: list[int] = Field(max_length=100)

cognite_toolkit/_cdf_tk/client/resource_classes/apm_config.py

@@ -0,0 +1,128 @@
+from typing import ClassVar, Literal
+
+from cognite_toolkit._cdf_tk.client.resource_classes.base import (
+    BaseModelObject,
+    RequestResource,
+    ResponseResource,
+)
+
+from .data_modeling import ViewReference
+from .identifiers import ExternalId
+
+
+class ThreeDModelIdentifier(BaseModelObject):
+    revision_id: int | None = None
+    model_id: int | None = None
+    name: str | None = None
+
+
+class ThreeDConfiguration(BaseModelObject):
+    full_weight_models: list[ThreeDModelIdentifier] | None = None
+    light_weight_models: list[ThreeDModelIdentifier] | None = None
+
+
+class ResourceFilters(BaseModelObject):
+    data_set_ids: list[int] | None = None
+    asset_subtree_external_ids: list[str] | None = None
+    root_asset_external_ids: list[str] | None = None
+    external_id_prefix: str | None = None
+    spaces: list[str] | None = None
+
+
+class RootLocationDataFilters(BaseModelObject):
+    general: ResourceFilters | None = None
+    assets: ResourceFilters | None = None
+    files: ResourceFilters | None = None
+    timeseries: ResourceFilters | None = None
+
+
+class ObservationFeatureToggles(BaseModelObject):
+    is_enabled: bool | None = None
+    is_write_back_enabled: bool | None = None
+    notifications_endpoint_external_id: str | None = None
+    attachments_endpoint_external_id: str | None = None
+
+
+class RootLocationFeatureToggles(BaseModelObject):
+    three_d: bool | None = None
+    trends: bool | None = None
+    documents: bool | None = None
+    workorders: bool | None = None
+    notifications: bool | None = None
+    media: bool | None = None
+    template_checklist_flow: bool | None = None
+    workorder_checklist_flow: bool | None = None
+    observations: ObservationFeatureToggles | None = None
+
+
+class ObservationConfigFieldProperty(BaseModelObject):
+    display_title: str | None = None
+    display_description: str | None = None
+    is_required: bool | None = None
+
+
+class ObservationConfigDropdownPropertyOption(BaseModelObject):
+    id: str | None = None
+    value: str | None = None
+    label: str | None = None
+
+
+class ObservationConfigDropdownProperty(ObservationConfigFieldProperty):
+    options: list[ObservationConfigDropdownPropertyOption] | None = None
+
+
+class ObservationsConfig(BaseModelObject):
+    files: ObservationConfigFieldProperty | None = None
+    description: ObservationConfigFieldProperty | None = None
+    asset: ObservationConfigFieldProperty | None = None
+    troubleshooting: ObservationConfigFieldProperty | None = None
+    type: ObservationConfigDropdownProperty | None = None
+    priority: ObservationConfigDropdownProperty | None = None
+
+
+class RootLocationConfiguration(BaseModelObject):
+    asset_external_id: str | None = None
+    external_id: str | None = None
+    display_name: str | None = None
+    three_d_configuration: ThreeDConfiguration | None = None
+    data_set_id: int | None = None
+    template_admins: list[str] | None = None  # list of Group Names
+    checklist_admins: list[str] | None = None  # list of Group Names
+    app_data_instance_space: str | None = None
+    source_data_instance_space: str | None = None
+    data_filters: RootLocationDataFilters | None = None
+    feature_toggles: RootLocationFeatureToggles | None = None
+    observations: ObservationsConfig | None = None
+
+
+class FeatureConfiguration(BaseModelObject):
+    root_location_configurations: list[RootLocationConfiguration] | None = None
+
+
+class APMConfig(BaseModelObject):
+    space: ClassVar[str] = "APM_Config"
+    view_ref: ClassVar[ViewReference] = ViewReference(space="APM_Config", external_id="APM_Config", version="1")
+    instance_type: Literal["node"] = "node"
+    external_id: str
+    name: str | None = None
+    app_data_space_id: str | None = None
+    app_data_space_version: str | None = None
+    customer_data_space_id: str | None = None
+    customer_data_space_version: str | None = None
+    feature_configuration: FeatureConfiguration | None = None
+
+    def as_id(self) -> ExternalId:
+        return ExternalId(external_id=self.external_id)
+
+
+class APMConfigRequest(APMConfig, RequestResource):
+    existing_version: int | None = None
+
+
+class APMConfigResponse(APMConfig, ResponseResource[APMConfigRequest]):
+    version: int
+    created_time: int
+    last_updated_time: int
+
+    def as_request_resource(self) -> APMConfigRequest:
+        return APMConfigRequest.model_validate(self.dump(), extra="ignore")

cognite_toolkit/_cdf_tk/client/resource_classes/cognite_file.py

@@ -0,0 +1,53 @@
+from datetime import datetime
+from typing import ClassVar, Literal
+
+from cognite_toolkit._cdf_tk.client.resource_classes.base import (
+    BaseModelObject,
+    RequestResource,
+    ResponseResource,
+)
+
+from .data_modeling import ViewReference
+from .instance_api import NodeReference, TypedNodeIdentifier
+
+
+class CogniteFile(BaseModelObject):
+    view_ref: ClassVar[ViewReference] = ViewReference(space="cdf_cdm", external_id="CogniteFile", version="v1")
+    instance_type: Literal["node"] = "node"
+    space: str
+    external_id: str
+    name: str | None = None
+    description: str | None = None
+    tags: list[str] | None = None
+    aliases: list[str] | None = None
+    source_id: str | None = None
+    source_context: str | None = None
+    source: NodeReference | None = None
+    source_created_time: datetime | None = None
+    source_updated_time: datetime | None = None
+    source_created_user: str | None = None
+    source_updated_user: str | None = None
+    assets: list[NodeReference] | None = None
+    mime_type: str | None = None
+    directory: str | None = None
+    category: NodeReference | None = None
+    type: NodeReference | None = None
+
+    def as_id(self) -> TypedNodeIdentifier:
+        return TypedNodeIdentifier(space=self.space, external_id=self.external_id)
+
+
+class CogniteFileRequest(CogniteFile, RequestResource):
+    existing_version: int | None = None
+
+
+class CogniteFileResponse(CogniteFile, ResponseResource[CogniteFileRequest]):
+    version: int
+    created_time: int
+    last_updated_time: int
+    is_uploaded: bool | None = None
+    uploaded_time: datetime | None = None
+    deleted_time: int | None = None
+
+    def as_request_resource(self) -> CogniteFileRequest:
+        return CogniteFileRequest.model_validate(self.dump(), extra="ignore")

cognite_toolkit/_cdf_tk/client/resource_classes/data_modeling/__init__.py

@@ -46,6 +46,8 @@ from ._instance import (
     EdgeRequest,
     EdgeResponse,
     InstanceDefinition,
+    InstanceRequest,
+    InstanceResponse,
     InstanceResponseDefinition,
     InstanceSource,
     NodeRequest,
@@ -120,6 +122,8 @@ __all__ = [
     "IndexAdapter",
     "IndexDefinition",
     "InstanceDefinition",
+    "InstanceRequest",
+    "InstanceResponse",
     "InstanceResponseDefinition",
     "InstanceSource",
     "Int32Property",

cognite_toolkit/_cdf_tk/client/resource_classes/data_modeling/_instance.py

@@ -1,7 +1,7 @@
 from abc import ABC
-from typing import Any, Generic, Literal
+from typing import Annotated, Any, Generic, Literal, TypeAlias
 
-from pydantic import JsonValue, field_serializer, field_validator
+from pydantic import Field, JsonValue, field_serializer, field_validator
 
 from cognite_toolkit._cdf_tk.client.resource_classes.base import (
     BaseModelObject,
@@ -9,8 +9,9 @@ from cognite_toolkit._cdf_tk.client.resource_classes.base import (
     ResponseResource,
     T_RequestResource,
 )
+from cognite_toolkit._cdf_tk.client.resource_classes.instance_api import TypedEdgeIdentifier, TypedNodeIdentifier
 
-from ._references import ContainerReference, EdgeReference, NodeReference, ViewReference
+from ._references import ContainerReference, NodeReference, ViewReference
 
 
 class InstanceDefinition(BaseModelObject, ABC):
@@ -87,8 +88,8 @@ class NodeRequest(InstanceRequestDefinition):
     instance_type: Literal["node"] = "node"
     type: NodeReference | None = None
 
-    def as_id(self) -> NodeReference:
-        return NodeReference(space=self.space, external_id=self.external_id)
+    def as_id(self) -> TypedNodeIdentifier:
+        return TypedNodeIdentifier(space=self.space, external_id=self.external_id)
 
 
 class EdgeRequest(InstanceRequestDefinition):
@@ -99,8 +100,8 @@ class EdgeRequest(InstanceRequestDefinition):
     start_node: NodeReference
     end_node: NodeReference
 
-    def as_id(self) -> EdgeReference:
-        return EdgeReference(space=self.space, external_id=self.external_id)
+    def as_id(self) -> TypedEdgeIdentifier:
+        return TypedEdgeIdentifier(space=self.space, external_id=self.external_id)
 
 
 class NodeResponse(InstanceResponseDefinition[NodeRequest]):
@@ -109,8 +110,8 @@ class NodeResponse(InstanceResponseDefinition[NodeRequest]):
     instance_type: Literal["node"] = "node"
     type: NodeReference | None = None
 
-    def as_id(self) -> NodeReference:
-        return NodeReference(space=self.space, external_id=self.external_id)
+    def as_id(self) -> TypedNodeIdentifier:
+        return TypedNodeIdentifier(space=self.space, external_id=self.external_id)
 
     def as_request_resource(self) -> NodeRequest:
         dumped = self.dump()
@@ -130,8 +131,8 @@ class EdgeResponse(InstanceResponseDefinition[EdgeRequest]):
     start_node: NodeReference
     end_node: NodeReference
 
-    def as_id(self) -> EdgeReference:
-        return EdgeReference(space=self.space, external_id=self.external_id)
+    def as_id(self) -> TypedEdgeIdentifier:
+        return TypedEdgeIdentifier(space=self.space, external_id=self.external_id)
 
     def as_request_resource(self) -> EdgeRequest:
         dumped = self.dump()
@@ -141,3 +142,13 @@ class EdgeResponse(InstanceResponseDefinition[EdgeRequest]):
             ]
         dumped["existingVersion"] = dumped.pop("version", None)
         return EdgeRequest.model_validate(dumped, extra="ignore")
+
+
+InstanceRequest: TypeAlias = Annotated[
+    NodeRequest | EdgeRequest,
+    Field(discriminator="instance_type"),
+]
+InstanceResponse: TypeAlias = Annotated[
+    NodeResponse | EdgeResponse,
+    Field(discriminator="instance_type"),
+]

cognite_toolkit/_cdf_tk/client/resource_classes/identifiers.py

@@ -50,3 +50,10 @@ class WorkflowVersionId(Identifier):
 
     def __str__(self) -> str:
         return f"workflowExternalId='{self.workflow_external_id}', version='{self.version}'"
+
+
+class DataSetId(Identifier):
+    data_set_id: int
+
+    def __str__(self) -> str:
+        return f"dataSetId={self.data_set_id}"

cognite_toolkit/_cdf_tk/client/resource_classes/location_filter.py

@@ -1,5 +1,7 @@
 from typing import Literal
 
+from pydantic import Field
+
 from cognite_toolkit._cdf_tk.client.resource_classes.base import BaseModelObject, RequestResource, ResponseResource
 
 from .data_modeling import DataModelReference
@@ -61,8 +63,13 @@ class LocationFilter(BaseModelObject):
 class LocationFilterRequest(LocationFilter, RequestResource):
     """Request resource for creating/updating location filters."""
 
-    def as_id(self) -> ExternalId:
-        return ExternalId(external_id=self.external_id)
+    # This is not part of the request payload, but we need it to identify existing resources for updates.
+    id: int | None = Field(default=None, exclude=True)
+
+    def as_id(self) -> InternalId:
+        if self.id is None:
+            raise ValueError("Cannot get ID for LocationFilterRequest without 'id' set.")
+        return InternalId(id=self.id)
 
 
 class LocationFilterResponse(LocationFilter, ResponseResource[LocationFilterRequest]):

cognite_toolkit/_cdf_tk/client/resource_classes/resource_view_mapping.py

@@ -0,0 +1,38 @@
+from typing import ClassVar, Literal
+
+from cognite_toolkit._cdf_tk.client.resource_classes.base import (
+    BaseModelObject,
+    RequestResource,
+    ResponseResource,
+)
+
+from .data_modeling import ViewReference
+from .identifiers import ExternalId
+
+
+class ResourceViewMapping(BaseModelObject):
+    space: ClassVar[str] = "cognite_migration"
+    view_ref: ClassVar[ViewReference] = ViewReference(
+        space="cognite_migration", external_id="ResourceViewMapping", version="v1"
+    )
+    instance_type: Literal["node"] = "node"
+    external_id: str
+    resource_type: str
+    view_id: ViewReference
+    property_mapping: dict[str, str]
+
+    def as_id(self) -> ExternalId:
+        return ExternalId(external_id=self.external_id)
+
+
+class ResourceViewMappingRequest(ResourceViewMapping, RequestResource):
+    existing_version: int | None = None
+
+
+class ResourceViewMappingResponse(ResourceViewMapping, ResponseResource[ResourceViewMappingRequest]):
+    version: int
+    created_time: int
+    last_updated_time: int
+
+    def as_request_resource(self) -> ResourceViewMappingRequest:
+        return ResourceViewMappingRequest.model_validate(self.dump(), extra="ignore")

cognite_toolkit/_cdf_tk/client/resource_classes/robotics/_map.py

@@ -1,4 +1,4 @@
-from typing import ClassVar
+from typing import ClassVar, Literal
 
 from pydantic import JsonValue
 
@@ -49,6 +49,11 @@ class RobotMapRequest(RobotMap, RequestUpdateable):
         {"description", "data", "frame_external_id", "location_external_id", "scale"}
     )
 
+    def as_update(self, mode: Literal["patch", "replace"]) -> dict[str, JsonValue]:
+        update = super().as_update(mode)
+        update["update"].pop("mapType", None)
+        return update
+
 
 class RobotMapResponse(RobotMap, ResponseResource[RobotMapRequest]):
     """Response resource for a RobotMap."""

cognite_toolkit/_cdf_tk/client/resource_classes/robotics/_robot.py

@@ -1,11 +1,11 @@
-from typing import ClassVar
+from typing import Any, ClassVar, Literal
 
 from cognite_toolkit._cdf_tk.client.resource_classes.base import (
     BaseModelObject,
     RequestUpdateable,
     ResponseResource,
 )
-from cognite_toolkit._cdf_tk.client.resource_classes.identifiers import NameId
+from cognite_toolkit._cdf_tk.client.resource_classes.identifiers import DataSetId
 
 from ._common import RobotType
 
@@ -32,15 +32,20 @@ class Robot(BaseModelObject):
     metadata: dict[str, str] | None = None
     location_external_id: str | None = None
 
-    def as_id(self) -> NameId:
-        return NameId(name=self.name)
+    def as_id(self) -> DataSetId:
+        return DataSetId(data_set_id=self.data_set_id)
 
 
 class RobotRequest(Robot, RequestUpdateable):
     """Request resource for creating or updating a Robot."""
 
     container_fields: ClassVar[frozenset[str]] = frozenset({"metadata"})
-    non_nullable_fields: ClassVar[frozenset[str]] = frozenset({"location_external_id"})
+    non_nullable_fields: ClassVar[frozenset[str]] = frozenset({"location_external_id", "description"})
+
+    def as_update(self, mode: Literal["patch", "replace"]) -> dict[str, Any]:
+        update = super().as_update(mode)
+        update["update"].pop("capabilities", None)
+        return update
 
 
 class RobotResponse(Robot, ResponseResource[RobotRequest]):

cognite_toolkit/_cdf_tk/client/resource_classes/streams.py

@@ -1,12 +1,8 @@
 from typing import Literal
 
 from cognite_toolkit._cdf_tk.constants import StreamTemplateName
-from cognite_toolkit._cdf_tk.protocols import (
-    ResourceRequestListProtocol,
-    ResourceResponseListProtocol,
-)
 
-from .base import BaseModelObject, BaseResourceList, RequestResource, ResponseResource
+from .base import BaseModelObject, RequestResource, ResponseResource
 from .identifiers import ExternalId
 
 
@@ -23,12 +19,6 @@ class StreamRequest(Stream, RequestResource):
         return ExternalId(external_id=self.external_id)
 
 
-class StreamRequestList(BaseResourceList[StreamRequest], ResourceRequestListProtocol):
-    """List of Stream request resources."""
-
-    _RESOURCE = StreamRequest
-
-
 class LifecycleObject(BaseModelObject):
     """Lifecycle object."""
 
@@ -81,12 +71,3 @@ class StreamResponse(Stream, ResponseResource[StreamRequest]):
                 "settings": {"template": {"name": self.created_from_template}},
             }
         )
-
-
-class StreamResponseList(BaseResourceList[StreamResponse], ResourceResponseListProtocol):
-    """List of Stream response resources."""
-
-    _RESOURCE = StreamResponse
-
-    def as_write(self) -> StreamRequestList:
-        return StreamRequestList([item.as_write() for item in self.data])