scope-client 1.4.991__tar.gz → 1.4.992__tar.gz

{scope_client-1.4.991 → scope_client-1.4.992}/PKG-INFO

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

{scope_client-1.4.991 → scope_client-1.4.992}/scope_client/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-1.4.991 → scope_client-1.4.992}/scope_client/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-1.4.991 → scope_client-1.4.992}/scope_client/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-1.4.991 → scope_client-1.4.992}/scope_client/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-1.4.991 → scope_client-1.4.992}/scope_client/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-1.4.992/scope_client/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-1.4.991 → scope_client-1.4.992}/scope_client/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-1.4.992/scope_client/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-1.4.991 → scope_client-1.4.992}/scope_client/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