cosmotech-api 3.2.0.post1__py3-none-any.whl → 3.2.6.dev0__py3-none-any.whl

cosmotech_api/model/scenario_run_status.py

@@ -3,7 +3,7 @@
 
     Cosmo Tech Platform API  # noqa: E501
 
-    The version of the OpenAPI document: 3.1.5
+    The version of the OpenAPI document: 3.1.10
     Contact: platform@cosmotech.com
     Generated by: https://openapi-generator.tech
 """

cosmotech_api/model/scenario_run_status_node.py

@@ -3,7 +3,7 @@
 
     Cosmo Tech Platform API  # noqa: E501
 
-    The version of the OpenAPI document: 3.1.5
+    The version of the OpenAPI document: 3.1.10
     Contact: platform@cosmotech.com
     Generated by: https://openapi-generator.tech
 """

cosmotech_api/model/scenario_run_template_parameter_value.py

@@ -3,7 +3,7 @@
 
     Cosmo Tech Platform API  # noqa: E501
 
-    The version of the OpenAPI document: 3.1.5
+    The version of the OpenAPI document: 3.1.10
     Contact: platform@cosmotech.com
     Generated by: https://openapi-generator.tech
 """

cosmotech_api/model/scenario_security.py

@@ -3,7 +3,7 @@
 
     Cosmo Tech Platform API  # noqa: E501
 
-    The version of the OpenAPI document: 3.1.5
+    The version of the OpenAPI document: 3.1.10
     Contact: platform@cosmotech.com
     Generated by: https://openapi-generator.tech
 """

cosmotech_api/model/scenario_validation_status.py

@@ -3,7 +3,7 @@
 
     Cosmo Tech Platform API  # noqa: E501
 
-    The version of the OpenAPI document: 3.1.5
+    The version of the OpenAPI document: 3.1.10
     Contact: platform@cosmotech.com
     Generated by: https://openapi-generator.tech
 """

cosmotech_api/model/solution.py

@@ -3,7 +3,7 @@
 
     Cosmo Tech Platform API  # noqa: E501
 
-    The version of the OpenAPI document: 3.1.5
+    The version of the OpenAPI document: 3.1.10
     Contact: platform@cosmotech.com
     Generated by: https://openapi-generator.tech
 """

cosmotech_api/model/solution_access_control.py

@@ -3,7 +3,7 @@
 
     Cosmo Tech Platform API  # noqa: E501
 
-    The version of the OpenAPI document: 3.1.5
+    The version of the OpenAPI document: 3.1.10
     Contact: platform@cosmotech.com
     Generated by: https://openapi-generator.tech
 """

cosmotech_api/model/solution_role.py

@@ -3,7 +3,7 @@
 
     Cosmo Tech Platform API  # noqa: E501
 
-    The version of the OpenAPI document: 3.1.5
+    The version of the OpenAPI document: 3.1.10
     Contact: platform@cosmotech.com
     Generated by: https://openapi-generator.tech
 """

cosmotech_api/model/solution_security.py

@@ -3,7 +3,7 @@
 
     Cosmo Tech Platform API  # noqa: E501
 
-    The version of the OpenAPI document: 3.1.5
+    The version of the OpenAPI document: 3.1.10
     Contact: platform@cosmotech.com
     Generated by: https://openapi-generator.tech
 """

cosmotech_api/model/source_info.py

@@ -3,7 +3,7 @@
 
     Cosmo Tech Platform API  # noqa: E501
 
-    The version of the OpenAPI document: 3.1.5
+    The version of the OpenAPI document: 3.1.10
     Contact: platform@cosmotech.com
     Generated by: https://openapi-generator.tech
 """

cosmotech_api/model/sub_dataset_graph_query.py

@@ -3,7 +3,7 @@
 
     Cosmo Tech Platform API  # noqa: E501
 
-    The version of the OpenAPI document: 3.1.5
+    The version of the OpenAPI document: 3.1.10
     Contact: platform@cosmotech.com
     Generated by: https://openapi-generator.tech
 """

cosmotech_api/model/translated_labels.py

@@ -3,7 +3,7 @@
 
     Cosmo Tech Platform API  # noqa: E501
 
-    The version of the OpenAPI document: 3.1.5
+    The version of the OpenAPI document: 3.1.10
     Contact: platform@cosmotech.com
     Generated by: https://openapi-generator.tech
 """

cosmotech_api/model/twin_graph_batch_result.py

@@ -3,7 +3,7 @@
 
     Cosmo Tech Platform API  # noqa: E501
 
-    The version of the OpenAPI document: 3.1.5
+    The version of the OpenAPI document: 3.1.10
     Contact: platform@cosmotech.com
     Generated by: https://openapi-generator.tech
 """

cosmotech_api/model/twin_graph_hash.py

@@ -3,7 +3,7 @@
 
     Cosmo Tech Platform API  # noqa: E501
 
-    The version of the OpenAPI document: 3.1.5
+    The version of the OpenAPI document: 3.1.10
     Contact: platform@cosmotech.com
     Generated by: https://openapi-generator.tech
 """

cosmotech_api/model/twin_graph_query.py

@@ -3,7 +3,7 @@
 
     Cosmo Tech Platform API  # noqa: E501
 
-    The version of the OpenAPI document: 3.1.5
+    The version of the OpenAPI document: 3.1.10
     Contact: platform@cosmotech.com
     Generated by: https://openapi-generator.tech
 """

cosmotech_api/model/validator.py

@@ -3,7 +3,7 @@
 
     Cosmo Tech Platform API  # noqa: E501
 
-    The version of the OpenAPI document: 3.1.5
+    The version of the OpenAPI document: 3.1.10
     Contact: platform@cosmotech.com
     Generated by: https://openapi-generator.tech
 """

cosmotech_api/model/validator_run.py

@@ -3,7 +3,7 @@
 
     Cosmo Tech Platform API  # noqa: E501
 
-    The version of the OpenAPI document: 3.1.5
+    The version of the OpenAPI document: 3.1.10
     Contact: platform@cosmotech.com
     Generated by: https://openapi-generator.tech
 """

cosmotech_api/model/workspace.py

@@ -3,7 +3,7 @@
 
     Cosmo Tech Platform API  # noqa: E501
 
-    The version of the OpenAPI document: 3.1.5
+    The version of the OpenAPI document: 3.1.10
     Contact: platform@cosmotech.com
     Generated by: https://openapi-generator.tech
 """

cosmotech_api/model/workspace_access_control.py

@@ -3,7 +3,7 @@
 
     Cosmo Tech Platform API  # noqa: E501
 
-    The version of the OpenAPI document: 3.1.5
+    The version of the OpenAPI document: 3.1.10
     Contact: platform@cosmotech.com
     Generated by: https://openapi-generator.tech
 """

cosmotech_api/model/workspace_file.py

@@ -3,7 +3,7 @@
 
     Cosmo Tech Platform API  # noqa: E501
 
-    The version of the OpenAPI document: 3.1.5
+    The version of the OpenAPI document: 3.1.10
     Contact: platform@cosmotech.com
     Generated by: https://openapi-generator.tech
 """

cosmotech_api/model/workspace_role.py

@@ -3,7 +3,7 @@
 
     Cosmo Tech Platform API  # noqa: E501
 
-    The version of the OpenAPI document: 3.1.5
+    The version of the OpenAPI document: 3.1.10
     Contact: platform@cosmotech.com
     Generated by: https://openapi-generator.tech
 """

cosmotech_api/model/workspace_secret.py

@@ -3,7 +3,7 @@
 
     Cosmo Tech Platform API  # noqa: E501
 
-    The version of the OpenAPI document: 3.1.5
+    The version of the OpenAPI document: 3.1.10
     Contact: platform@cosmotech.com
     Generated by: https://openapi-generator.tech
 """

cosmotech_api/model/workspace_security.py

@@ -3,7 +3,7 @@
 
     Cosmo Tech Platform API  # noqa: E501
 
-    The version of the OpenAPI document: 3.1.5
+    The version of the OpenAPI document: 3.1.10
     Contact: platform@cosmotech.com
     Generated by: https://openapi-generator.tech
 """

cosmotech_api/model/workspace_solution.py

@@ -3,7 +3,7 @@
 
     Cosmo Tech Platform API  # noqa: E501
 
-    The version of the OpenAPI document: 3.1.5
+    The version of the OpenAPI document: 3.1.10
     Contact: platform@cosmotech.com
     Generated by: https://openapi-generator.tech
 """

cosmotech_api/model/workspace_web_app.py

@@ -3,7 +3,7 @@
 
     Cosmo Tech Platform API  # noqa: E501
 
-    The version of the OpenAPI document: 3.1.5
+    The version of the OpenAPI document: 3.1.10
     Contact: platform@cosmotech.com
     Generated by: https://openapi-generator.tech
 """

cosmotech_api/model_utils.py

@@ -3,7 +3,7 @@
 
     Cosmo Tech Platform API  # noqa: E501
 
-    The version of the OpenAPI document: 3.1.5
+    The version of the OpenAPI document: 3.1.10
     Contact: platform@cosmotech.com
     Generated by: https://openapi-generator.tech
 """

cosmotech_api/models/__init__.py

@@ -6,7 +6,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 3.2.0
+    The version of the OpenAPI document: 4.0.0-onprem.7
     Contact: platform@cosmotech.com
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
@@ -21,6 +21,7 @@ from cosmotech_api.models.connector_parameter import ConnectorParameter
 from cosmotech_api.models.connector_parameter_group import ConnectorParameterGroup
 from cosmotech_api.models.container_resource_size_info import ContainerResourceSizeInfo
 from cosmotech_api.models.container_resource_sizing import ContainerResourceSizing
+from cosmotech_api.models.created_run import CreatedRun
 from cosmotech_api.models.dataset import Dataset
 from cosmotech_api.models.dataset_access_control import DatasetAccessControl
 from cosmotech_api.models.dataset_compatibility import DatasetCompatibility
@@ -37,29 +38,25 @@ from cosmotech_api.models.delete_historical_data import DeleteHistoricalData
 from cosmotech_api.models.file_upload_metadata import FileUploadMetadata
 from cosmotech_api.models.file_upload_validation import FileUploadValidation
 from cosmotech_api.models.graph_properties import GraphProperties
+from cosmotech_api.models.ingestion_status_enum import IngestionStatusEnum
+from cosmotech_api.models.io_types_enum import IoTypesEnum
 from cosmotech_api.models.organization import Organization
 from cosmotech_api.models.organization_access_control import OrganizationAccessControl
 from cosmotech_api.models.organization_role import OrganizationRole
 from cosmotech_api.models.organization_security import OrganizationSecurity
-from cosmotech_api.models.organization_service import OrganizationService
-from cosmotech_api.models.organization_services import OrganizationServices
 from cosmotech_api.models.query_result import QueryResult
 from cosmotech_api.models.resource_size_info import ResourceSizeInfo
 from cosmotech_api.models.run import Run
 from cosmotech_api.models.run_container import RunContainer
-from cosmotech_api.models.run_container_artifact import RunContainerArtifact
-from cosmotech_api.models.run_container_logs import RunContainerLogs
 from cosmotech_api.models.run_data import RunData
 from cosmotech_api.models.run_data_query import RunDataQuery
 from cosmotech_api.models.run_logs import RunLogs
+from cosmotech_api.models.run_logs_entry import RunLogsEntry
 from cosmotech_api.models.run_resource_requested import RunResourceRequested
-from cosmotech_api.models.run_search import RunSearch
-from cosmotech_api.models.run_start_containers import RunStartContainers
 from cosmotech_api.models.run_state import RunState
 from cosmotech_api.models.run_status import RunStatus
 from cosmotech_api.models.run_status_node import RunStatusNode
 from cosmotech_api.models.run_template import RunTemplate
-from cosmotech_api.models.run_template_handler_id import RunTemplateHandlerId
 from cosmotech_api.models.run_template_orchestrator import RunTemplateOrchestrator
 from cosmotech_api.models.run_template_parameter import RunTemplateParameter
 from cosmotech_api.models.run_template_parameter_group import RunTemplateParameterGroup
@@ -68,43 +65,11 @@ from cosmotech_api.models.run_template_resource_sizing import RunTemplateResourc
 from cosmotech_api.models.run_template_step_source import RunTemplateStepSource
 from cosmotech_api.models.runner import Runner
 from cosmotech_api.models.runner_access_control import RunnerAccessControl
-from cosmotech_api.models.runner_changed_parameter_value import RunnerChangedParameterValue
-from cosmotech_api.models.runner_comparison_result import RunnerComparisonResult
-from cosmotech_api.models.runner_data_download_info import RunnerDataDownloadInfo
-from cosmotech_api.models.runner_data_download_job import RunnerDataDownloadJob
-from cosmotech_api.models.runner_job_state import RunnerJobState
-from cosmotech_api.models.runner_last_run import RunnerLastRun
-from cosmotech_api.models.runner_parent_last_run import RunnerParentLastRun
 from cosmotech_api.models.runner_resource_sizing import RunnerResourceSizing
 from cosmotech_api.models.runner_role import RunnerRole
-from cosmotech_api.models.runner_root_last_run import RunnerRootLastRun
 from cosmotech_api.models.runner_run_template_parameter_value import RunnerRunTemplateParameterValue
 from cosmotech_api.models.runner_security import RunnerSecurity
 from cosmotech_api.models.runner_validation_status import RunnerValidationStatus
-from cosmotech_api.models.scenario import Scenario
-from cosmotech_api.models.scenario_access_control import ScenarioAccessControl
-from cosmotech_api.models.scenario_changed_parameter_value import ScenarioChangedParameterValue
-from cosmotech_api.models.scenario_comparison_result import ScenarioComparisonResult
-from cosmotech_api.models.scenario_data_download_info import ScenarioDataDownloadInfo
-from cosmotech_api.models.scenario_data_download_job import ScenarioDataDownloadJob
-from cosmotech_api.models.scenario_job_state import ScenarioJobState
-from cosmotech_api.models.scenario_last_run import ScenarioLastRun
-from cosmotech_api.models.scenario_resource_sizing import ScenarioResourceSizing
-from cosmotech_api.models.scenario_role import ScenarioRole
-from cosmotech_api.models.scenario_run import ScenarioRun
-from cosmotech_api.models.scenario_run_container import ScenarioRunContainer
-from cosmotech_api.models.scenario_run_container_artifact import ScenarioRunContainerArtifact
-from cosmotech_api.models.scenario_run_container_logs import ScenarioRunContainerLogs
-from cosmotech_api.models.scenario_run_logs import ScenarioRunLogs
-from cosmotech_api.models.scenario_run_resource_requested import ScenarioRunResourceRequested
-from cosmotech_api.models.scenario_run_search import ScenarioRunSearch
-from cosmotech_api.models.scenario_run_start_containers import ScenarioRunStartContainers
-from cosmotech_api.models.scenario_run_state import ScenarioRunState
-from cosmotech_api.models.scenario_run_status import ScenarioRunStatus
-from cosmotech_api.models.scenario_run_status_node import ScenarioRunStatusNode
-from cosmotech_api.models.scenario_run_template_parameter_value import ScenarioRunTemplateParameterValue
-from cosmotech_api.models.scenario_security import ScenarioSecurity
-from cosmotech_api.models.scenario_validation_status import ScenarioValidationStatus
 from cosmotech_api.models.send_run_data_request import SendRunDataRequest
 from cosmotech_api.models.solution import Solution
 from cosmotech_api.models.solution_access_control import SolutionAccessControl
@@ -113,15 +78,11 @@ from cosmotech_api.models.solution_security import SolutionSecurity
 from cosmotech_api.models.source_info import SourceInfo
 from cosmotech_api.models.sub_dataset_graph_query import SubDatasetGraphQuery
 from cosmotech_api.models.twin_graph_batch_result import TwinGraphBatchResult
-from cosmotech_api.models.twin_graph_hash import TwinGraphHash
-from cosmotech_api.models.twin_graph_query import TwinGraphQuery
-from cosmotech_api.models.validator import Validator
-from cosmotech_api.models.validator_run import ValidatorRun
+from cosmotech_api.models.twincache_status_enum import TwincacheStatusEnum
 from cosmotech_api.models.workspace import Workspace
 from cosmotech_api.models.workspace_access_control import WorkspaceAccessControl
 from cosmotech_api.models.workspace_file import WorkspaceFile
 from cosmotech_api.models.workspace_role import WorkspaceRole
-from cosmotech_api.models.workspace_secret import WorkspaceSecret
 from cosmotech_api.models.workspace_security import WorkspaceSecurity
 from cosmotech_api.models.workspace_solution import WorkspaceSolution
 from cosmotech_api.models.workspace_web_app import WorkspaceWebApp

cosmotech_api/models/component_role_permissions.py

@@ -5,7 +5,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 3.2.0
+    The version of the OpenAPI document: 4.0.0-onprem.7
     Contact: platform@cosmotech.com
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 

cosmotech_api/models/connector.py

@@ -5,7 +5,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 3.2.0
+    The version of the OpenAPI document: 4.0.0-onprem.7
     Contact: platform@cosmotech.com
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
@@ -18,9 +18,10 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, Field, StrictBool, StrictStr, field_validator
+from pydantic import BaseModel, Field, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
 from cosmotech_api.models.connector_parameter_group import ConnectorParameterGroup
+from cosmotech_api.models.io_types_enum import IoTypesEnum
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -37,22 +38,9 @@ class Connector(BaseModel):
     tags: Optional[List[StrictStr]] = Field(default=None, description="the list of tags")
     owner_id: Optional[StrictStr] = Field(default=None, description="the user id which own this connector version", alias="ownerId")
     url: Optional[StrictStr] = Field(default=None, description="an optional URL link to connector page")
-    azure_managed_identity: Optional[StrictBool] = Field(default=None, description="whether or not the connector uses Azure Managed Identity", alias="azureManagedIdentity")
-    azure_authentication_with_customer_app_registration: Optional[StrictBool] = Field(default=None, description="whether to authenticate against Azure using the app registration credentials provided by the customer", alias="azureAuthenticationWithCustomerAppRegistration")
-    io_types: Optional[List[StrictStr]] = Field(default=None, alias="ioTypes")
+    io_types: Optional[List[IoTypesEnum]] = Field(default=None, alias="ioTypes")
     parameter_groups: Optional[List[ConnectorParameterGroup]] = Field(default=None, description="the list of connector parameters groups", alias="parameterGroups")
-    __properties: ClassVar[List[str]] = ["id", "key", "name", "description", "repository", "version", "tags", "ownerId", "url", "azureManagedIdentity", "azureAuthenticationWithCustomerAppRegistration", "ioTypes", "parameterGroups"]
-
-    @field_validator('io_types')
-    def io_types_validate_enum(cls, value):
-        """Validates the enum"""
-        if value is None:
-            return value
-
-        for i in value:
-            if i not in set(['read', 'write']):
-                raise ValueError("each list item must be one of ('read', 'write')")
-        return value
+    __properties: ClassVar[List[str]] = ["id", "key", "name", "description", "repository", "version", "tags", "ownerId", "url", "ioTypes", "parameterGroups"]
 
     model_config = {
         "populate_by_name": True,
@@ -125,8 +113,6 @@ class Connector(BaseModel):
             "tags": obj.get("tags"),
             "ownerId": obj.get("ownerId"),
             "url": obj.get("url"),
-            "azureManagedIdentity": obj.get("azureManagedIdentity"),
-            "azureAuthenticationWithCustomerAppRegistration": obj.get("azureAuthenticationWithCustomerAppRegistration"),
             "ioTypes": obj.get("ioTypes"),
             "parameterGroups": [ConnectorParameterGroup.from_dict(_item) for _item in obj["parameterGroups"]] if obj.get("parameterGroups") is not None else None
         })

cosmotech_api/models/connector_parameter.py

@@ -5,7 +5,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 3.2.0
+    The version of the OpenAPI document: 4.0.0-onprem.7
     Contact: platform@cosmotech.com
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 

cosmotech_api/models/connector_parameter_group.py

@@ -5,7 +5,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 3.2.0
+    The version of the OpenAPI document: 4.0.0-onprem.7
     Contact: platform@cosmotech.com
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 

cosmotech_api/models/container_resource_size_info.py

@@ -5,7 +5,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 3.2.0
+    The version of the OpenAPI document: 4.0.0-onprem.7
     Contact: platform@cosmotech.com
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 

cosmotech_api/models/container_resource_sizing.py

@@ -5,7 +5,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 3.2.0
+    The version of the OpenAPI document: 4.0.0-onprem.7
     Contact: platform@cosmotech.com
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 

cosmotech_api/models/created_run.py

@@ -0,0 +1,88 @@
+# coding: utf-8
+
+"""
+    Cosmo Tech Platform API
+
+    Cosmo Tech Platform API
+
+    The version of the OpenAPI document: 4.0.0-onprem.7
+    Contact: platform@cosmotech.com
+    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, Field, StrictStr
+from typing import Any, ClassVar, Dict, List
+from typing import Optional, Set
+from typing_extensions import Self
+
+class CreatedRun(BaseModel):
+    """
+    Newle created Run info
+    """ # noqa: E501
+    id: StrictStr = Field(description="Run id")
+    __properties: ClassVar[List[str]] = ["id"]
+
+    model_config = {
+        "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 CreatedRun 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 CreatedRun from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "id": obj.get("id")
+        })
+        return _obj
+
+

cosmotech_api/models/dataset.py

@@ -5,7 +5,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 3.2.0
+    The version of the OpenAPI document: 4.0.0-onprem.7
     Contact: platform@cosmotech.com
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
@@ -18,13 +18,15 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, Field, StrictBool, StrictInt, StrictStr, field_validator
+from pydantic import BaseModel, Field, StrictBool, StrictInt, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
 from cosmotech_api.models.dataset_compatibility import DatasetCompatibility
 from cosmotech_api.models.dataset_connector import DatasetConnector
 from cosmotech_api.models.dataset_security import DatasetSecurity
 from cosmotech_api.models.dataset_source_type import DatasetSourceType
+from cosmotech_api.models.ingestion_status_enum import IngestionStatusEnum
 from cosmotech_api.models.source_info import SourceInfo
+from cosmotech_api.models.twincache_status_enum import TwincacheStatusEnum
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -46,8 +48,8 @@ class Dataset(BaseModel):
     refresh_date: Optional[StrictInt] = Field(default=None, description="the last time a refresh was done", alias="refreshDate")
     source_type: Optional[DatasetSourceType] = Field(default=None, alias="sourceType")
     source: Optional[SourceInfo] = None
-    ingestion_status: Optional[StrictStr] = Field(default=None, description="the Dataset ingestion status", alias="ingestionStatus")
-    twincache_status: Optional[StrictStr] = Field(default=None, description="the twincache data status", alias="twincacheStatus")
+    ingestion_status: Optional[IngestionStatusEnum] = Field(default=None, alias="ingestionStatus")
+    twincache_status: Optional[TwincacheStatusEnum] = Field(default=None, alias="twincacheStatus")
     queries: Optional[List[StrictStr]] = Field(default=None, description="the list of queries")
     tags: Optional[List[StrictStr]] = Field(default=None, description="the list of tags")
     connector: Optional[DatasetConnector] = None
@@ -57,26 +59,6 @@ class Dataset(BaseModel):
     security: Optional[DatasetSecurity] = None
     __properties: ClassVar[List[str]] = ["id", "name", "description", "ownerId", "ownerName", "organizationId", "parentId", "linkedWorkspaceIdList", "twingraphId", "main", "creationDate", "refreshDate", "sourceType", "source", "ingestionStatus", "twincacheStatus", "queries", "tags", "connector", "fragmentsIds", "validatorId", "compatibility", "security"]
 
-    @field_validator('ingestion_status')
-    def ingestion_status_validate_enum(cls, value):
-        """Validates the enum"""
-        if value is None:
-            return value
-
-        if value not in set(['NONE', 'PENDING', 'ERROR', 'SUCCESS']):
-            raise ValueError("must be one of enum values ('NONE', 'PENDING', 'ERROR', 'SUCCESS')")
-        return value
-
-    @field_validator('twincache_status')
-    def twincache_status_validate_enum(cls, value):
-        """Validates the enum"""
-        if value is None:
-            return value
-
-        if value not in set(['EMPTY', 'FULL']):
-            raise ValueError("must be one of enum values ('EMPTY', 'FULL')")
-        return value
-
     model_config = {
         "populate_by_name": True,
         "validate_assignment": True,

cosmotech_api/models/dataset_access_control.py

@@ -5,7 +5,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 3.2.0
+    The version of the OpenAPI document: 4.0.0-onprem.7
     Contact: platform@cosmotech.com
     Generated by OpenAPI Generator (https://openapi-generator.tech)