scope-client 1.4.991__py3-none-any.whl → 1.4.993__py3-none-any.whl

scope_client/api_bindings/__init__.py

@@ -98,6 +98,7 @@ from scope_client.api_bindings.models.d_type import DType
 from scope_client.api_bindings.models.data_plane import DataPlane
 from scope_client.api_bindings.models.data_plane_association import DataPlaneAssociation
 from scope_client.api_bindings.models.data_plane_association_sort import DataPlaneAssociationSort
+from scope_client.api_bindings.models.data_plane_capabilities import DataPlaneCapabilities
 from scope_client.api_bindings.models.data_plane_sort import DataPlaneSort
 from scope_client.api_bindings.models.data_result_filter import DataResultFilter
 from scope_client.api_bindings.models.data_result_filter_op import DataResultFilterOp
@@ -193,6 +194,7 @@ from scope_client.api_bindings.models.pagination import Pagination
 from scope_client.api_bindings.models.patch_alert_rule import PatchAlertRule
 from scope_client.api_bindings.models.patch_connector_spec import PatchConnectorSpec
 from scope_client.api_bindings.models.patch_data_plane import PatchDataPlane
+from scope_client.api_bindings.models.patch_data_plane_capabilities import PatchDataPlaneCapabilities
 from scope_client.api_bindings.models.patch_dataset import PatchDataset
 from scope_client.api_bindings.models.patch_dataset_locator import PatchDatasetLocator
 from scope_client.api_bindings.models.patch_group import PatchGroup

scope_client/api_bindings/api/data_plane_associations_v1_api.py

@@ -660,6 +660,7 @@ class DataPlaneAssociationsV1Api:
         project_id: StrictStr,
         sort: Annotated[Optional[DataPlaneAssociationSort], Field(description="Override the field used for sorting the returned list. Optional.")] = None,
         order: Annotated[Optional[SortOrder], Field(description="Override the sort order used. Optional.")] = None,
+        gen_ai_enabled: Annotated[Optional[StrictBool], Field(description="Filter to only return engines enabled with the GenAI capability")] = None,
         page: Annotated[Optional[Annotated[int, Field(strict=True, ge=1)]], Field(description="The page to return starting from 1 up to total_pages.")] = None,
         page_size: Annotated[Optional[Annotated[int, Field(le=1000, strict=True, ge=1)]], Field(description="The number of records per page. The max is 1000.")] = None,
         _request_timeout: Union[
@@ -685,6 +686,8 @@ class DataPlaneAssociationsV1Api:
         :type sort: DataPlaneAssociationSort
         :param order: Override the sort order used. Optional.
         :type order: SortOrder
+        :param gen_ai_enabled: Filter to only return engines enabled with the GenAI capability
+        :type gen_ai_enabled: bool
         :param page: The page to return starting from 1 up to total_pages.
         :type page: int
         :param page_size: The number of records per page. The max is 1000.
@@ -715,6 +718,7 @@ class DataPlaneAssociationsV1Api:
             project_id=project_id,
             sort=sort,
             order=order,
+            gen_ai_enabled=gen_ai_enabled,
             page=page,
             page_size=page_size,
             _request_auth=_request_auth,
@@ -746,6 +750,7 @@ class DataPlaneAssociationsV1Api:
         project_id: StrictStr,
         sort: Annotated[Optional[DataPlaneAssociationSort], Field(description="Override the field used for sorting the returned list. Optional.")] = None,
         order: Annotated[Optional[SortOrder], Field(description="Override the sort order used. Optional.")] = None,
+        gen_ai_enabled: Annotated[Optional[StrictBool], Field(description="Filter to only return engines enabled with the GenAI capability")] = None,
         page: Annotated[Optional[Annotated[int, Field(strict=True, ge=1)]], Field(description="The page to return starting from 1 up to total_pages.")] = None,
         page_size: Annotated[Optional[Annotated[int, Field(le=1000, strict=True, ge=1)]], Field(description="The number of records per page. The max is 1000.")] = None,
         _request_timeout: Union[
@@ -771,6 +776,8 @@ class DataPlaneAssociationsV1Api:
         :type sort: DataPlaneAssociationSort
         :param order: Override the sort order used. Optional.
         :type order: SortOrder
+        :param gen_ai_enabled: Filter to only return engines enabled with the GenAI capability
+        :type gen_ai_enabled: bool
         :param page: The page to return starting from 1 up to total_pages.
         :type page: int
         :param page_size: The number of records per page. The max is 1000.
@@ -801,6 +808,7 @@ class DataPlaneAssociationsV1Api:
             project_id=project_id,
             sort=sort,
             order=order,
+            gen_ai_enabled=gen_ai_enabled,
             page=page,
             page_size=page_size,
             _request_auth=_request_auth,
@@ -832,6 +840,7 @@ class DataPlaneAssociationsV1Api:
         project_id: StrictStr,
         sort: Annotated[Optional[DataPlaneAssociationSort], Field(description="Override the field used for sorting the returned list. Optional.")] = None,
         order: Annotated[Optional[SortOrder], Field(description="Override the sort order used. Optional.")] = None,
+        gen_ai_enabled: Annotated[Optional[StrictBool], Field(description="Filter to only return engines enabled with the GenAI capability")] = None,
         page: Annotated[Optional[Annotated[int, Field(strict=True, ge=1)]], Field(description="The page to return starting from 1 up to total_pages.")] = None,
         page_size: Annotated[Optional[Annotated[int, Field(le=1000, strict=True, ge=1)]], Field(description="The number of records per page. The max is 1000.")] = None,
         _request_timeout: Union[
@@ -857,6 +866,8 @@ class DataPlaneAssociationsV1Api:
         :type sort: DataPlaneAssociationSort
         :param order: Override the sort order used. Optional.
         :type order: SortOrder
+        :param gen_ai_enabled: Filter to only return engines enabled with the GenAI capability
+        :type gen_ai_enabled: bool
         :param page: The page to return starting from 1 up to total_pages.
         :type page: int
         :param page_size: The number of records per page. The max is 1000.
@@ -887,6 +898,7 @@ class DataPlaneAssociationsV1Api:
             project_id=project_id,
             sort=sort,
             order=order,
+            gen_ai_enabled=gen_ai_enabled,
             page=page,
             page_size=page_size,
             _request_auth=_request_auth,
@@ -913,6 +925,7 @@ class DataPlaneAssociationsV1Api:
         project_id,
         sort,
         order,
+        gen_ai_enabled,
         page,
         page_size,
         _request_auth,
@@ -947,6 +960,10 @@ class DataPlaneAssociationsV1Api:
             
             _query_params.append(('order', order.value))
             
+        if gen_ai_enabled is not None:
+            
+            _query_params.append(('gen_ai_enabled', gen_ai_enabled))
+            
         if page is not None:
             
             _query_params.append(('page', page))
@@ -1000,6 +1017,7 @@ class DataPlaneAssociationsV1Api:
         order: Annotated[Optional[SortOrder], Field(description="Override the sort order used. Optional.")] = None,
         include_projects: Annotated[Optional[StrictBool], Field(description="Include project details in the response. Defaults to false.")] = None,
         include_data_planes: Annotated[Optional[StrictBool], Field(description="Include data plane details in the response. Defaults to false.")] = None,
+        gen_ai_enabled: Annotated[Optional[StrictBool], Field(description="Filter to only return engines enabled with the GenAI capability")] = None,
         page: Annotated[Optional[Annotated[int, Field(strict=True, ge=1)]], Field(description="The page to return starting from 1 up to total_pages.")] = None,
         page_size: Annotated[Optional[Annotated[int, Field(le=1000, strict=True, ge=1)]], Field(description="The number of records per page. The max is 1000.")] = None,
         _request_timeout: Union[
@@ -1029,6 +1047,8 @@ class DataPlaneAssociationsV1Api:
         :type include_projects: bool
         :param include_data_planes: Include data plane details in the response. Defaults to false.
         :type include_data_planes: bool
+        :param gen_ai_enabled: Filter to only return engines enabled with the GenAI capability
+        :type gen_ai_enabled: bool
         :param page: The page to return starting from 1 up to total_pages.
         :type page: int
         :param page_size: The number of records per page. The max is 1000.
@@ -1061,6 +1081,7 @@ class DataPlaneAssociationsV1Api:
             order=order,
             include_projects=include_projects,
             include_data_planes=include_data_planes,
+            gen_ai_enabled=gen_ai_enabled,
             page=page,
             page_size=page_size,
             _request_auth=_request_auth,
@@ -1094,6 +1115,7 @@ class DataPlaneAssociationsV1Api:
         order: Annotated[Optional[SortOrder], Field(description="Override the sort order used. Optional.")] = None,
         include_projects: Annotated[Optional[StrictBool], Field(description="Include project details in the response. Defaults to false.")] = None,
         include_data_planes: Annotated[Optional[StrictBool], Field(description="Include data plane details in the response. Defaults to false.")] = None,
+        gen_ai_enabled: Annotated[Optional[StrictBool], Field(description="Filter to only return engines enabled with the GenAI capability")] = None,
         page: Annotated[Optional[Annotated[int, Field(strict=True, ge=1)]], Field(description="The page to return starting from 1 up to total_pages.")] = None,
         page_size: Annotated[Optional[Annotated[int, Field(le=1000, strict=True, ge=1)]], Field(description="The number of records per page. The max is 1000.")] = None,
         _request_timeout: Union[
@@ -1123,6 +1145,8 @@ class DataPlaneAssociationsV1Api:
         :type include_projects: bool
         :param include_data_planes: Include data plane details in the response. Defaults to false.
         :type include_data_planes: bool
+        :param gen_ai_enabled: Filter to only return engines enabled with the GenAI capability
+        :type gen_ai_enabled: bool
         :param page: The page to return starting from 1 up to total_pages.
         :type page: int
         :param page_size: The number of records per page. The max is 1000.
@@ -1155,6 +1179,7 @@ class DataPlaneAssociationsV1Api:
             order=order,
             include_projects=include_projects,
             include_data_planes=include_data_planes,
+            gen_ai_enabled=gen_ai_enabled,
             page=page,
             page_size=page_size,
             _request_auth=_request_auth,
@@ -1188,6 +1213,7 @@ class DataPlaneAssociationsV1Api:
         order: Annotated[Optional[SortOrder], Field(description="Override the sort order used. Optional.")] = None,
         include_projects: Annotated[Optional[StrictBool], Field(description="Include project details in the response. Defaults to false.")] = None,
         include_data_planes: Annotated[Optional[StrictBool], Field(description="Include data plane details in the response. Defaults to false.")] = None,
+        gen_ai_enabled: Annotated[Optional[StrictBool], Field(description="Filter to only return engines enabled with the GenAI capability")] = None,
         page: Annotated[Optional[Annotated[int, Field(strict=True, ge=1)]], Field(description="The page to return starting from 1 up to total_pages.")] = None,
         page_size: Annotated[Optional[Annotated[int, Field(le=1000, strict=True, ge=1)]], Field(description="The number of records per page. The max is 1000.")] = None,
         _request_timeout: Union[
@@ -1217,6 +1243,8 @@ class DataPlaneAssociationsV1Api:
         :type include_projects: bool
         :param include_data_planes: Include data plane details in the response. Defaults to false.
         :type include_data_planes: bool
+        :param gen_ai_enabled: Filter to only return engines enabled with the GenAI capability
+        :type gen_ai_enabled: bool
         :param page: The page to return starting from 1 up to total_pages.
         :type page: int
         :param page_size: The number of records per page. The max is 1000.
@@ -1249,6 +1277,7 @@ class DataPlaneAssociationsV1Api:
             order=order,
             include_projects=include_projects,
             include_data_planes=include_data_planes,
+            gen_ai_enabled=gen_ai_enabled,
             page=page,
             page_size=page_size,
             _request_auth=_request_auth,
@@ -1277,6 +1306,7 @@ class DataPlaneAssociationsV1Api:
         order,
         include_projects,
         include_data_planes,
+        gen_ai_enabled,
         page,
         page_size,
         _request_auth,
@@ -1319,6 +1349,10 @@ class DataPlaneAssociationsV1Api:
             
             _query_params.append(('include_data_planes', include_data_planes))
             
+        if gen_ai_enabled is not None:
+            
+            _query_params.append(('gen_ai_enabled', gen_ai_enabled))
+            
         if page is not None:
             
             _query_params.append(('page', page))

scope_client/api_bindings/api/data_planes_v1_api.py

@@ -16,7 +16,7 @@ from pydantic import validate_call, Field, StrictFloat, StrictStr, StrictInt
 from typing import Any, Dict, List, Optional, Tuple, Union
 from typing_extensions import Annotated
 
-from pydantic import Field, StrictStr
+from pydantic import Field, StrictBool, StrictStr
 from typing import Optional
 from typing_extensions import Annotated
 from scope_client.api_bindings.models.data_plane import DataPlane
@@ -595,6 +595,7 @@ class DataPlanesV1Api:
         sort: Annotated[Optional[DataPlaneSort], Field(description="Override the field used for sorting the returned list. Optional.")] = None,
         order: Annotated[Optional[SortOrder], Field(description="Override the sort order used. Optional.")] = None,
         name: Annotated[Optional[StrictStr], Field(description="Search term to filter workspaces by name.")] = None,
+        gen_ai_enabled: Annotated[Optional[StrictBool], Field(description="Filter to only return engines enabled with the GenAI capability")] = None,
         page: Annotated[Optional[Annotated[int, Field(strict=True, ge=1)]], Field(description="The page to return starting from 1 up to total_pages.")] = None,
         page_size: Annotated[Optional[Annotated[int, Field(le=1000, strict=True, ge=1)]], Field(description="The number of records per page. The max is 1000.")] = None,
         _request_timeout: Union[
@@ -622,6 +623,8 @@ class DataPlanesV1Api:
         :type order: SortOrder
         :param name: Search term to filter workspaces by name.
         :type name: str
+        :param gen_ai_enabled: Filter to only return engines enabled with the GenAI capability
+        :type gen_ai_enabled: bool
         :param page: The page to return starting from 1 up to total_pages.
         :type page: int
         :param page_size: The number of records per page. The max is 1000.
@@ -653,6 +656,7 @@ class DataPlanesV1Api:
             sort=sort,
             order=order,
             name=name,
+            gen_ai_enabled=gen_ai_enabled,
             page=page,
             page_size=page_size,
             _request_auth=_request_auth,
@@ -685,6 +689,7 @@ class DataPlanesV1Api:
         sort: Annotated[Optional[DataPlaneSort], Field(description="Override the field used for sorting the returned list. Optional.")] = None,
         order: Annotated[Optional[SortOrder], Field(description="Override the sort order used. Optional.")] = None,
         name: Annotated[Optional[StrictStr], Field(description="Search term to filter workspaces by name.")] = None,
+        gen_ai_enabled: Annotated[Optional[StrictBool], Field(description="Filter to only return engines enabled with the GenAI capability")] = None,
         page: Annotated[Optional[Annotated[int, Field(strict=True, ge=1)]], Field(description="The page to return starting from 1 up to total_pages.")] = None,
         page_size: Annotated[Optional[Annotated[int, Field(le=1000, strict=True, ge=1)]], Field(description="The number of records per page. The max is 1000.")] = None,
         _request_timeout: Union[
@@ -712,6 +717,8 @@ class DataPlanesV1Api:
         :type order: SortOrder
         :param name: Search term to filter workspaces by name.
         :type name: str
+        :param gen_ai_enabled: Filter to only return engines enabled with the GenAI capability
+        :type gen_ai_enabled: bool
         :param page: The page to return starting from 1 up to total_pages.
         :type page: int
         :param page_size: The number of records per page. The max is 1000.
@@ -743,6 +750,7 @@ class DataPlanesV1Api:
             sort=sort,
             order=order,
             name=name,
+            gen_ai_enabled=gen_ai_enabled,
             page=page,
             page_size=page_size,
             _request_auth=_request_auth,
@@ -775,6 +783,7 @@ class DataPlanesV1Api:
         sort: Annotated[Optional[DataPlaneSort], Field(description="Override the field used for sorting the returned list. Optional.")] = None,
         order: Annotated[Optional[SortOrder], Field(description="Override the sort order used. Optional.")] = None,
         name: Annotated[Optional[StrictStr], Field(description="Search term to filter workspaces by name.")] = None,
+        gen_ai_enabled: Annotated[Optional[StrictBool], Field(description="Filter to only return engines enabled with the GenAI capability")] = None,
         page: Annotated[Optional[Annotated[int, Field(strict=True, ge=1)]], Field(description="The page to return starting from 1 up to total_pages.")] = None,
         page_size: Annotated[Optional[Annotated[int, Field(le=1000, strict=True, ge=1)]], Field(description="The number of records per page. The max is 1000.")] = None,
         _request_timeout: Union[
@@ -802,6 +811,8 @@ class DataPlanesV1Api:
         :type order: SortOrder
         :param name: Search term to filter workspaces by name.
         :type name: str
+        :param gen_ai_enabled: Filter to only return engines enabled with the GenAI capability
+        :type gen_ai_enabled: bool
         :param page: The page to return starting from 1 up to total_pages.
         :type page: int
         :param page_size: The number of records per page. The max is 1000.
@@ -833,6 +844,7 @@ class DataPlanesV1Api:
             sort=sort,
             order=order,
             name=name,
+            gen_ai_enabled=gen_ai_enabled,
             page=page,
             page_size=page_size,
             _request_auth=_request_auth,
@@ -860,6 +872,7 @@ class DataPlanesV1Api:
         sort,
         order,
         name,
+        gen_ai_enabled,
         page,
         page_size,
         _request_auth,
@@ -898,6 +911,10 @@ class DataPlanesV1Api:
             
             _query_params.append(('name', name))
             
+        if gen_ai_enabled is not None:
+            
+            _query_params.append(('gen_ai_enabled', gen_ai_enabled))
+            
         if page is not None:
             
             _query_params.append(('page', page))

scope_client/api_bindings/models/__init__.py

@@ -56,6 +56,7 @@ from scope_client.api_bindings.models.d_type import DType
 from scope_client.api_bindings.models.data_plane import DataPlane
 from scope_client.api_bindings.models.data_plane_association import DataPlaneAssociation
 from scope_client.api_bindings.models.data_plane_association_sort import DataPlaneAssociationSort
+from scope_client.api_bindings.models.data_plane_capabilities import DataPlaneCapabilities
 from scope_client.api_bindings.models.data_plane_sort import DataPlaneSort
 from scope_client.api_bindings.models.data_result_filter import DataResultFilter
 from scope_client.api_bindings.models.data_result_filter_op import DataResultFilterOp
@@ -151,6 +152,7 @@ from scope_client.api_bindings.models.pagination import Pagination
 from scope_client.api_bindings.models.patch_alert_rule import PatchAlertRule
 from scope_client.api_bindings.models.patch_connector_spec import PatchConnectorSpec
 from scope_client.api_bindings.models.patch_data_plane import PatchDataPlane
+from scope_client.api_bindings.models.patch_data_plane_capabilities import PatchDataPlaneCapabilities
 from scope_client.api_bindings.models.patch_dataset import PatchDataset
 from scope_client.api_bindings.models.patch_dataset_locator import PatchDatasetLocator
 from scope_client.api_bindings.models.patch_group import PatchGroup

scope_client/api_bindings/models/data_plane.py

@@ -20,6 +20,7 @@ import json
 from datetime import datetime
 from pydantic import BaseModel, ConfigDict, Field, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
+from scope_client.api_bindings.models.data_plane_capabilities import DataPlaneCapabilities
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -35,7 +36,8 @@ class DataPlane(BaseModel):
     description: StrictStr = Field(description="Description of data plane.")
     user_id: StrictStr = Field(description="ID of the data plane's underlying user.")
     last_check_in_time: Optional[datetime]
-    __properties: ClassVar[List[str]] = ["created_at", "updated_at", "id", "name", "workspace_id", "description", "user_id", "last_check_in_time"]
+    capabilities: DataPlaneCapabilities = Field(description="Capabilities for this data plane.")
+    __properties: ClassVar[List[str]] = ["created_at", "updated_at", "id", "name", "workspace_id", "description", "user_id", "last_check_in_time", "capabilities"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -76,6 +78,9 @@ class DataPlane(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
+        # override the default output from pydantic by calling `to_dict()` of capabilities
+        if self.capabilities:
+            _dict['capabilities'] = self.capabilities.to_dict()
         # set to None if last_check_in_time (nullable) is None
         # and model_fields_set contains the field
         if self.last_check_in_time is None and "last_check_in_time" in self.model_fields_set:
@@ -100,7 +105,8 @@ class DataPlane(BaseModel):
             "workspace_id": obj.get("workspace_id"),
             "description": obj.get("description"),
             "user_id": obj.get("user_id"),
-            "last_check_in_time": obj.get("last_check_in_time")
+            "last_check_in_time": obj.get("last_check_in_time"),
+            "capabilities": DataPlaneCapabilities.from_dict(obj["capabilities"]) if obj.get("capabilities") is not None else None
         })
         return _obj
 

scope_client/api_bindings/models/data_plane_capabilities.py

@@ -0,0 +1,87 @@
+# coding: utf-8
+
+"""
+    Arthur Scope
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: 0.1.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from pydantic import BaseModel, ConfigDict, Field, StrictBool
+from typing import Any, ClassVar, Dict, List, Optional
+from typing import Optional, Set
+from typing_extensions import Self
+
+class DataPlaneCapabilities(BaseModel):
+    """
+    DataPlaneCapabilities
+    """ # noqa: E501
+    gen_ai_enabled: Optional[StrictBool] = Field(default=False, description="Field that records if the engine was installed with GenAI capabilities.")
+    __properties: ClassVar[List[str]] = ["gen_ai_enabled"]
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of DataPlaneCapabilities from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of DataPlaneCapabilities from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "gen_ai_enabled": obj.get("gen_ai_enabled") if obj.get("gen_ai_enabled") is not None else False
+        })
+        return _obj
+
+

scope_client/api_bindings/models/patch_data_plane.py

@@ -19,6 +19,7 @@ import json
 
 from pydantic import BaseModel, ConfigDict, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
+from scope_client.api_bindings.models.patch_data_plane_capabilities import PatchDataPlaneCapabilities
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -26,9 +27,10 @@ class PatchDataPlane(BaseModel):
     """
     PatchDataPlane
     """ # noqa: E501
-    name: Optional[StrictStr]
-    description: Optional[StrictStr]
-    __properties: ClassVar[List[str]] = ["name", "description"]
+    name: Optional[StrictStr] = None
+    description: Optional[StrictStr] = None
+    capabilities: Optional[PatchDataPlaneCapabilities] = None
+    __properties: ClassVar[List[str]] = ["name", "description", "capabilities"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -69,6 +71,9 @@ class PatchDataPlane(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
+        # override the default output from pydantic by calling `to_dict()` of capabilities
+        if self.capabilities:
+            _dict['capabilities'] = self.capabilities.to_dict()
         # set to None if name (nullable) is None
         # and model_fields_set contains the field
         if self.name is None and "name" in self.model_fields_set:
@@ -79,6 +84,11 @@ class PatchDataPlane(BaseModel):
         if self.description is None and "description" in self.model_fields_set:
             _dict['description'] = None
 
+        # set to None if capabilities (nullable) is None
+        # and model_fields_set contains the field
+        if self.capabilities is None and "capabilities" in self.model_fields_set:
+            _dict['capabilities'] = None
+
         return _dict
 
     @classmethod
@@ -92,7 +102,8 @@ class PatchDataPlane(BaseModel):
 
         _obj = cls.model_validate({
             "name": obj.get("name"),
-            "description": obj.get("description")
+            "description": obj.get("description"),
+            "capabilities": PatchDataPlaneCapabilities.from_dict(obj["capabilities"]) if obj.get("capabilities") is not None else None
         })
         return _obj
 

scope_client/api_bindings/models/patch_data_plane_capabilities.py

@@ -0,0 +1,92 @@
+# coding: utf-8
+
+"""
+    Arthur Scope
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: 0.1.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from pydantic import BaseModel, ConfigDict, StrictBool
+from typing import Any, ClassVar, Dict, List, Optional
+from typing import Optional, Set
+from typing_extensions import Self
+
+class PatchDataPlaneCapabilities(BaseModel):
+    """
+    PatchDataPlaneCapabilities
+    """ # noqa: E501
+    gen_ai_enabled: Optional[StrictBool] = None
+    __properties: ClassVar[List[str]] = ["gen_ai_enabled"]
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of PatchDataPlaneCapabilities from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        # set to None if gen_ai_enabled (nullable) is None
+        # and model_fields_set contains the field
+        if self.gen_ai_enabled is None and "gen_ai_enabled" in self.model_fields_set:
+            _dict['gen_ai_enabled'] = None
+
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of PatchDataPlaneCapabilities from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "gen_ai_enabled": obj.get("gen_ai_enabled")
+        })
+        return _obj
+
+

scope_client/api_bindings/models/post_data_plane.py

@@ -18,7 +18,8 @@ import re  # noqa: F401
 import json
 
 from pydantic import BaseModel, ConfigDict, Field, StrictStr
-from typing import Any, ClassVar, Dict, List
+from typing import Any, ClassVar, Dict, List, Optional
+from scope_client.api_bindings.models.data_plane_capabilities import DataPlaneCapabilities
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -28,7 +29,8 @@ class PostDataPlane(BaseModel):
     """ # noqa: E501
     name: StrictStr = Field(description="Name of data plane.")
     description: StrictStr = Field(description="Description of data plane.")
-    __properties: ClassVar[List[str]] = ["name", "description"]
+    capabilities: Optional[DataPlaneCapabilities] = Field(default=None, description="Capabilities for this data plane.")
+    __properties: ClassVar[List[str]] = ["name", "description", "capabilities"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -69,6 +71,9 @@ class PostDataPlane(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
+        # override the default output from pydantic by calling `to_dict()` of capabilities
+        if self.capabilities:
+            _dict['capabilities'] = self.capabilities.to_dict()
         return _dict
 
     @classmethod
@@ -82,7 +87,8 @@ class PostDataPlane(BaseModel):
 
         _obj = cls.model_validate({
             "name": obj.get("name"),
-            "description": obj.get("description")
+            "description": obj.get("description"),
+            "capabilities": DataPlaneCapabilities.from_dict(obj["capabilities"]) if obj.get("capabilities") is not None else None
         })
         return _obj
 

scope_client/api_bindings/models/sensitive_data_plane.py

@@ -20,6 +20,7 @@ import json
 from datetime import datetime
 from pydantic import BaseModel, ConfigDict, Field, SecretStr, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
+from scope_client.api_bindings.models.data_plane_capabilities import DataPlaneCapabilities
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -35,9 +36,10 @@ class SensitiveDataPlane(BaseModel):
     description: StrictStr = Field(description="Description of data plane.")
     user_id: StrictStr = Field(description="ID of the data plane's underlying user.")
     last_check_in_time: Optional[datetime]
+    capabilities: DataPlaneCapabilities = Field(description="Capabilities for this data plane.")
     client_id: SecretStr = Field(description="ID of the auth client.")
     client_secret: SecretStr = Field(description="Auth client secret.")
-    __properties: ClassVar[List[str]] = ["created_at", "updated_at", "id", "name", "workspace_id", "description", "user_id", "last_check_in_time", "client_id", "client_secret"]
+    __properties: ClassVar[List[str]] = ["created_at", "updated_at", "id", "name", "workspace_id", "description", "user_id", "last_check_in_time", "capabilities", "client_id", "client_secret"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -78,6 +80,9 @@ class SensitiveDataPlane(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
+        # override the default output from pydantic by calling `to_dict()` of capabilities
+        if self.capabilities:
+            _dict['capabilities'] = self.capabilities.to_dict()
         # set to None if last_check_in_time (nullable) is None
         # and model_fields_set contains the field
         if self.last_check_in_time is None and "last_check_in_time" in self.model_fields_set:
@@ -103,6 +108,7 @@ class SensitiveDataPlane(BaseModel):
             "description": obj.get("description"),
             "user_id": obj.get("user_id"),
             "last_check_in_time": obj.get("last_check_in_time"),
+            "capabilities": DataPlaneCapabilities.from_dict(obj["capabilities"]) if obj.get("capabilities") is not None else None,
             "client_id": obj.get("client_id"),
             "client_secret": obj.get("client_secret")
         })

{scope_client-1.4.991.dist-info → scope_client-1.4.993.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: scope_client
-Version: 1.4.991
+Version: 1.4.993
 Summary: Arthur Python API Client Library
 Author-email: Arthur <info@arthur.ai>
 License: MIT

{scope_client-1.4.991.dist-info → scope_client-1.4.993.dist-info}/RECORD

@@ -1,5 +1,5 @@
 scope_client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-scope_client/api_bindings/__init__.py,sha256=6oJE6wA3E3C2iK0flRPas2vPYibIT0Dr0jGTjbB9Ha0,24234
+scope_client/api_bindings/__init__.py,sha256=Pv9ifClsGZKAHgYXJqZmrUighPauQfRbHNCKd3uuCYA,24427
 scope_client/api_bindings/api_client.py,sha256=nkt88XDyPiuURu94lnI4MtOfMeYXd_jazGUde0T4xVU,27538
 scope_client/api_bindings/api_response.py,sha256=eMxw1mpmJcoGZ3gs9z6jM4oYoZ10Gjk333s9sKxGv7s,652
 scope_client/api_bindings/configuration.py,sha256=a0o2OhNYo7GdZscfNtFKW2kfgoln1mcWE-8E_UeiWv8,15338
@@ -11,8 +11,8 @@ scope_client/api_bindings/api/alerts_v1_api.py,sha256=hAyljzNdtm3ruaN1C5qyTCZJF5
 scope_client/api_bindings/api/authorization_v1_api.py,sha256=0hHddXMvsw87SV-VE87mTp1afVyb8AB2pcCIjie4emg,168384
 scope_client/api_bindings/api/connector_schemas_v1_api.py,sha256=-lkx0uSc0hw4lzdlfiRcZVI-d2dlsq4CbtJ1sdyiu2U,21812
 scope_client/api_bindings/api/connectors_v1_api.py,sha256=DGxBpwqnTo4NYqGG6SERfRq7SUuHt54d-KVS95RexgQ,87522
-scope_client/api_bindings/api/data_plane_associations_v1_api.py,sha256=Y1ow27ypdInRwm2QEV1nOr-7yOcw6BN6va6ZO0Tn3Jc,70350
-scope_client/api_bindings/api/data_planes_v1_api.py,sha256=NmJTxXGRecT6XNZntWXosWDjyWt7Xq5gPou9ZTkrb54,70920
+scope_client/api_bindings/api/data_plane_associations_v1_api.py,sha256=fGUbJC752d45GeSV0ALAeiUOutG9wBw3Q_F2ak4Fuf0,72604
+scope_client/api_bindings/api/data_planes_v1_api.py,sha256=IWbA_RKxpg28ihPAKsIa4y6xCuvZCIoqQo4ntPVmbPg,72059
 scope_client/api_bindings/api/data_retrieval_v1_api.py,sha256=o8yKv-rw3E0KVa2nyePR8jR83FYpPUxqQTq00cCDVyM,97313
 scope_client/api_bindings/api/datasets_v1_api.py,sha256=xV-6MfwUiPo8wp0VOCxOBXrG91LVYrQuHiWMF25Fx54,173726
 scope_client/api_bindings/api/default_api.py,sha256=N2kYEK0fkvWQVIUYyFgRfsdUvpK0lI3G3Yl3Iz4R2oM,10588
@@ -32,7 +32,7 @@ scope_client/api_bindings/api/upsolve_v1_api.py,sha256=Vdep9x-_lIJq_kOwtsGqYqkss
 scope_client/api_bindings/api/users_v1_api.py,sha256=wo-yZd1da0O8NgAuBCo8784NY1vTo4H8J7pf7sFavCc,114570
 scope_client/api_bindings/api/webhooks_v1_api.py,sha256=86tRi7pgwICshf8WVyYA1WrJggThpnsTaON9Mx8R-0M,71286
 scope_client/api_bindings/api/workspaces_v1_api.py,sha256=OjMc-pJe2JAGgv5ZGTD8_TkQxw82vz2e381-5m37vJ4,58653
-scope_client/api_bindings/models/__init__.py,sha256=xKjjY_ZLVhqfR_AzdwOYFGY9UgQ4_lczjIHibrrkww4,21647
+scope_client/api_bindings/models/__init__.py,sha256=BZ8bkeM8X9CD-BLVNo_9h1hT9v6zqGI78OX-rO-5tCY,21840
 scope_client/api_bindings/models/aggregation_metric_type.py,sha256=USKpDqhX4TQ4lqIoHOp3q70mydsf1u_zc1EKus9QhYo,783
 scope_client/api_bindings/models/aggregation_spec.py,sha256=gWFTgLqxNdiIw4iXwbNqKAnAwDpbcgeoFfQ_7VxtKOQ,4109
 scope_client/api_bindings/models/aggregation_spec_schema.py,sha256=MHGlB9eF8NfOasOHiqj1ZzYiBaIl8y8J2SmQysoSWVU,4624
@@ -72,9 +72,10 @@ scope_client/api_bindings/models/create_model_task_job_spec.py,sha256=WqQPt8I5nA
 scope_client/api_bindings/models/created_alerts.py,sha256=HEU8jGJvowCbwZgU7wUn410lNiJAAox5SYRyDd2uglw,3949
 scope_client/api_bindings/models/credentials.py,sha256=z1kk6SelBNpHCC78wW0-feiY6I0Q9sxFD3Bo3C0j_zI,5118
 scope_client/api_bindings/models/d_type.py,sha256=sg1LfWA05rBStEo9VIEgimHRpk8sCcrx63c8WUDCj6Y,869
-scope_client/api_bindings/models/data_plane.py,sha256=Ii81QY9HUGiHLCywnn0YHZLQdZqn4hEYPI5o7kMX2_Y,3740
+scope_client/api_bindings/models/data_plane.py,sha256=zpTmidWWq7nQZn6TY3ziOSoZUqC9iKY0hECtNjFcJlw,4260
 scope_client/api_bindings/models/data_plane_association.py,sha256=GyoPxgFDsHuy8La73VqFl-rtcanteE53fkOlzav0ElY,4373
 scope_client/api_bindings/models/data_plane_association_sort.py,sha256=PaRZV90wjsWum3SDKgWxd-QEJ458Ik-cD9w_fNNoSMo,806
+scope_client/api_bindings/models/data_plane_capabilities.py,sha256=LRRvcAob0be6wDGnwr1vLAzCubeVyOY5AE01t6X_y6U,2736
 scope_client/api_bindings/models/data_plane_sort.py,sha256=UreUIWvBGIDC0yC5Hb4-IvcgE0skQpV_V-pxr92oRw4,791
 scope_client/api_bindings/models/data_result_filter.py,sha256=Q3SQ4Zu40ThylVgYgxEr4iJLKCVQp8JpYz2N-pqyfVc,3187
 scope_client/api_bindings/models/data_result_filter_op.py,sha256=TXBfPsCLA8ErXRf8W5uDxlXPtz13xHEu1NBnro5mcZo,976
@@ -168,7 +169,8 @@ scope_client/api_bindings/models/organization.py,sha256=BQJHXIGcrmcGQrxD4JhIEBla
 scope_client/api_bindings/models/pagination.py,sha256=XWpYUWgFJIYmtJ3bMOqGlu6ganeK4wUxpZ8mCPC1d1Q,3118
 scope_client/api_bindings/models/patch_alert_rule.py,sha256=nAWWly90eTCnFDwuSdxRKmcg4hhMM5zvvnU98_OpPvg,4843
 scope_client/api_bindings/models/patch_connector_spec.py,sha256=EzW-KkOwk0B0hnr5fzsfLcH6ZF5c2sFBoVA9er8o_yk,3754
-scope_client/api_bindings/models/patch_data_plane.py,sha256=hZQGvzjDP4jw_b4GM4XfgdcmAqL9DuV-X3LaPhUPip0,3026
+scope_client/api_bindings/models/patch_data_plane.py,sha256=ZvRlyD4aZlNThEeaZ1MMNPpUXyw6ZypDTPP4Y6Bz2_Q,3772
+scope_client/api_bindings/models/patch_data_plane_capabilities.py,sha256=MVLSPbKUO7kDAJpR6HIrso7rJYofeXjGMsq66OtU7Gk,2833
 scope_client/api_bindings/models/patch_dataset.py,sha256=CTDLrg5BU6T3Zovji-DRufrHp6hMEDFwpOksVRVIezs,3918
 scope_client/api_bindings/models/patch_dataset_locator.py,sha256=Mo1AnQfzup1mXNVgCK2wuKkMNjerKCE9ocl19zROSSM,3244
 scope_client/api_bindings/models/patch_group.py,sha256=q8rDHJsJko5n1zbK--Y4iA1xcSi_cvIaYNGPsz6Oa88,3024
@@ -192,7 +194,7 @@ scope_client/api_bindings/models/post_alert.py,sha256=DYsLeD_zDPpcjROUPThBQn9I8p
 scope_client/api_bindings/models/post_alert_rule.py,sha256=V2jGigltR2X58JbwTcyVmy9RV9YBEklK71KBK_0vLzg,3847
 scope_client/api_bindings/models/post_alerts.py,sha256=6Jd3p9Pi04BK9tq8DR4o8TZ0e8I1hfRQUV7gJ5wANEE,3065
 scope_client/api_bindings/models/post_connector_spec.py,sha256=tCtEBCLSXXLse7x-kD3OWrkCpD0_TuHx2LNDEc8VysU,3917
-scope_client/api_bindings/models/post_data_plane.py,sha256=hiHVE_OO_zmaO6knqxI5s0xIG1zU_Z5QZgJuhZHb0Ms,2666
+scope_client/api_bindings/models/post_data_plane.py,sha256=Y7fobPHVqYRuXM4f69jBKVYjGDG9O-UujG0UAVv4V0Y,3220
 scope_client/api_bindings/models/post_data_plane_association.py,sha256=mEdP-ff6nLfCkj80jW0YGW0RWKJXOySnQRPJTbydX9g,2736
 scope_client/api_bindings/models/post_data_retrieval_operation.py,sha256=3qMB06e4YT73X5v38DkAWUlpWZbnTIVXrPHLFzJSZF0,4305
 scope_client/api_bindings/models/post_dataset.py,sha256=dbPxCV2Cexl9lOpHKlkYNSr8-5Y30bKGDtwy2MdXdWE,5185
@@ -269,7 +271,7 @@ scope_client/api_bindings/models/scalar_type.py,sha256=S5BiTsMG-oM2GC-EmlX7OdY2l
 scope_client/api_bindings/models/schedule_jobs_job_spec.py,sha256=S75JHtQV3VvJ58TPEUoBtfQ68odSCySONyJ2Io5CYpA,3335
 scope_client/api_bindings/models/schema_inspection_job_spec.py,sha256=yKpGEstb0ze_F-jajJd8V9Dfulr4wv5BDUoZ19gnNBk,3352
 scope_client/api_bindings/models/scope_schema_tag.py,sha256=xvPNpX1reIQxcMPPzKZMbJIKRBTjgsTBI3soRfRazHs,982
-scope_client/api_bindings/models/sensitive_data_plane.py,sha256=YhYmBumL2FuTjqWPv0FLz0xkNRAvvAjbEzDcJmKyhjw,4062
+scope_client/api_bindings/models/sensitive_data_plane.py,sha256=R6-8wwLOm_HGYLdsj475kdYTIEOOiurl3NzAupTqsiI,4582
 scope_client/api_bindings/models/sensitive_user.py,sha256=jBBMTZaktUrF0tCjEybOFSD5zBILwfnQBou7yg1g9SM,5526
 scope_client/api_bindings/models/service_account_credentials.py,sha256=jjnqsr-YyTFyOwi8jMx1ygyE1uzvyO83FZSe_5iA8R8,2772
 scope_client/api_bindings/models/sketch_metric.py,sha256=-V2OOnIKS36ucSdAWZQLD3Q3fMUAx5EnLaYFmYHm3P0,3262
@@ -301,7 +303,7 @@ scope_client/auth/device_authorizer.py,sha256=bJMIZRjkwQwoSWTLEp7OoXM2MytO3ADSD9
 scope_client/auth/discovery.py,sha256=hR0MglzRWHdwyi72If5hTnjO50fDJhquP_DD7OzjIQQ,1188
 scope_client/auth/oauth_api_config.py,sha256=wcEslusOFKr0oTzW0Ku2MhM1mvc-nm4BEJU8LHo1uXA,1347
 scope_client/auth/session.py,sha256=wCriib5ajfm1e1WTL_QXVCJmEOrGwQg_0v91e5qrC6g,2649
-scope_client-1.4.991.dist-info/METADATA,sha256=tVwwrWsVd7ToCghiqqHcu_uZxZZf-bWP19AO6HWmP1E,1776
-scope_client-1.4.991.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
-scope_client-1.4.991.dist-info/top_level.txt,sha256=x6MngS09hi-TUDoUGb3SLzmnf8_cf8IVAVNPSqtTzAY,13
-scope_client-1.4.991.dist-info/RECORD,,
+scope_client-1.4.993.dist-info/METADATA,sha256=y5xy4nd4f8ABeo_5Pyp1vmb7EZZEpoIVn3Wk-2rK0qk,1776
+scope_client-1.4.993.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
+scope_client-1.4.993.dist-info/top_level.txt,sha256=x6MngS09hi-TUDoUGb3SLzmnf8_cf8IVAVNPSqtTzAY,13
+scope_client-1.4.993.dist-info/RECORD,,

{scope_client-1.4.991.dist-info → scope_client-1.4.993.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (79.0.0)
+Generator: setuptools (79.0.1)
 Root-Is-Purelib: true
 Tag: py3-none-any