cosmotech-api 3.2.0.post1__py3-none-any.whl → 5.0.0rc3__py3-none-any.whl

cosmotech_api/models/run.py

@@ -5,7 +5,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 3.2.0
+    The version of the OpenAPI document: 5.0.0-rc3
     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, StrictStr
+from pydantic import BaseModel, ConfigDict, Field, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
 from cosmotech_api.models.run_container import RunContainer
+from cosmotech_api.models.run_edit_info import RunEditInfo
 from cosmotech_api.models.run_state import RunState
 from cosmotech_api.models.run_template_parameter_value import RunTemplateParameterValue
 from typing import Optional, Set
@@ -33,29 +34,28 @@ class Run(BaseModel):
     id: Optional[StrictStr] = Field(default=None, description="the Run")
     state: Optional[RunState] = None
     organization_id: Optional[StrictStr] = Field(default=None, description="the Organization id", alias="organizationId")
+    create_info: RunEditInfo = Field(description="The details of the Run creation", alias="createInfo")
     workflow_id: Optional[StrictStr] = Field(default=None, description="the Cosmo Tech compute cluster Argo Workflow Id to search", alias="workflowId")
     csm_simulation_run: Optional[StrictStr] = Field(default=None, description="the Cosmo Tech Simulation Run Id", alias="csmSimulationRun")
     generate_name: Optional[StrictStr] = Field(default=None, description="the base name for workflow name generation", alias="generateName")
     workflow_name: Optional[StrictStr] = Field(default=None, description="the Cosmo Tech compute cluster Argo Workflow Name", alias="workflowName")
-    owner_id: Optional[StrictStr] = Field(default=None, description="the user id which own this run", alias="ownerId")
     workspace_id: Optional[StrictStr] = Field(default=None, description="the Workspace Id", alias="workspaceId")
     workspace_key: Optional[StrictStr] = Field(default=None, description="technical key for resource name convention and version grouping. Must be unique", alias="workspaceKey")
     runner_id: Optional[StrictStr] = Field(default=None, description="the Runner Id", alias="runnerId")
     solution_id: Optional[StrictStr] = Field(default=None, description="the Solution Id", alias="solutionId")
     run_template_id: Optional[StrictStr] = Field(default=None, description="the Solution Run Template id", alias="runTemplateId")
     compute_size: Optional[StrictStr] = Field(default=None, description="the compute size needed for this Analysis. Standard sizes are basic and highcpu. Default is basic", alias="computeSize")
-    created_at: Optional[StrictStr] = Field(default=None, description="the Run creation date", alias="createdAt")
     dataset_list: Optional[List[StrictStr]] = Field(default=None, description="the list of Dataset Id associated to this Run", alias="datasetList")
     parameters_values: Optional[List[RunTemplateParameterValue]] = Field(default=None, description="the list of Run Template parameters values", alias="parametersValues")
     node_label: Optional[StrictStr] = Field(default=None, description="the node label request", alias="nodeLabel")
     containers: Optional[List[RunContainer]] = Field(default=None, description="the containers list. This information is not returned by the API.")
-    __properties: ClassVar[List[str]] = ["id", "state", "organizationId", "workflowId", "csmSimulationRun", "generateName", "workflowName", "ownerId", "workspaceId", "workspaceKey", "runnerId", "solutionId", "runTemplateId", "computeSize", "createdAt", "datasetList", "parametersValues", "nodeLabel", "containers"]
+    __properties: ClassVar[List[str]] = ["id", "state", "organizationId", "createInfo", "workflowId", "csmSimulationRun", "generateName", "workflowName", "workspaceId", "workspaceKey", "runnerId", "solutionId", "runTemplateId", "computeSize", "datasetList", "parametersValues", "nodeLabel", "containers"]
 
-    model_config = {
-        "populate_by_name": True,
-        "validate_assignment": True,
-        "protected_namespaces": (),
-    }
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
 
 
     def to_str(self) -> str:
@@ -92,20 +92,16 @@ class Run(BaseModel):
         * OpenAPI `readOnly` fields are excluded.
         * OpenAPI `readOnly` fields are excluded.
         * OpenAPI `readOnly` fields are excluded.
-        * OpenAPI `readOnly` fields are excluded.
-        * OpenAPI `readOnly` fields are excluded.
         """
         excluded_fields: Set[str] = set([
             "id",
             "csm_simulation_run",
-            "owner_id",
             "workspace_id",
             "workspace_key",
             "runner_id",
             "solution_id",
             "run_template_id",
             "compute_size",
-            "created_at",
             "dataset_list",
             "parameters_values",
             "node_label",
@@ -116,19 +112,22 @@ class Run(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
+        # override the default output from pydantic by calling `to_dict()` of create_info
+        if self.create_info:
+            _dict['createInfo'] = self.create_info.to_dict()
         # override the default output from pydantic by calling `to_dict()` of each item in parameters_values (list)
         _items = []
         if self.parameters_values:
-            for _item in self.parameters_values:
-                if _item:
-                    _items.append(_item.to_dict())
+            for _item_parameters_values in self.parameters_values:
+                if _item_parameters_values:
+                    _items.append(_item_parameters_values.to_dict())
             _dict['parametersValues'] = _items
         # override the default output from pydantic by calling `to_dict()` of each item in containers (list)
         _items = []
         if self.containers:
-            for _item in self.containers:
-                if _item:
-                    _items.append(_item.to_dict())
+            for _item_containers in self.containers:
+                if _item_containers:
+                    _items.append(_item_containers.to_dict())
             _dict['containers'] = _items
         return _dict
 
@@ -145,18 +144,17 @@ class Run(BaseModel):
             "id": obj.get("id"),
             "state": obj.get("state"),
             "organizationId": obj.get("organizationId"),
+            "createInfo": RunEditInfo.from_dict(obj["createInfo"]) if obj.get("createInfo") is not None else None,
             "workflowId": obj.get("workflowId"),
             "csmSimulationRun": obj.get("csmSimulationRun"),
             "generateName": obj.get("generateName"),
             "workflowName": obj.get("workflowName"),
-            "ownerId": obj.get("ownerId"),
             "workspaceId": obj.get("workspaceId"),
             "workspaceKey": obj.get("workspaceKey"),
             "runnerId": obj.get("runnerId"),
             "solutionId": obj.get("solutionId"),
             "runTemplateId": obj.get("runTemplateId"),
             "computeSize": obj.get("computeSize"),
-            "createdAt": obj.get("createdAt"),
             "datasetList": obj.get("datasetList"),
             "parametersValues": [RunTemplateParameterValue.from_dict(_item) for _item in obj["parametersValues"]] if obj.get("parametersValues") is not None else None,
             "nodeLabel": obj.get("nodeLabel"),

cosmotech_api/models/run_container.py

@@ -5,7 +5,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 3.2.0
+    The version of the OpenAPI document: 5.0.0-rc3
     Contact: platform@cosmotech.com
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
@@ -18,10 +18,9 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, Field, StrictBool, StrictStr
+from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
 from cosmotech_api.models.container_resource_sizing import ContainerResourceSizing
-from cosmotech_api.models.run_container_artifact import RunContainerArtifact
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -40,14 +39,13 @@ class RunContainer(BaseModel):
     solution_container: Optional[StrictBool] = Field(default=None, description="whether or not this container is a Cosmo Tech solution container", alias="solutionContainer")
     node_label: Optional[StrictStr] = Field(default=None, description="the node label request", alias="nodeLabel")
     run_sizing: Optional[ContainerResourceSizing] = Field(default=None, alias="runSizing")
-    artifacts: Optional[List[RunContainerArtifact]] = Field(default=None, description="the list of artifacts")
-    __properties: ClassVar[List[str]] = ["id", "name", "labels", "envVars", "image", "entrypoint", "runArgs", "dependencies", "solutionContainer", "nodeLabel", "runSizing", "artifacts"]
+    __properties: ClassVar[List[str]] = ["id", "name", "labels", "envVars", "image", "entrypoint", "runArgs", "dependencies", "solutionContainer", "nodeLabel", "runSizing"]
 
-    model_config = {
-        "populate_by_name": True,
-        "validate_assignment": True,
-        "protected_namespaces": (),
-    }
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
 
 
     def to_str(self) -> str:
@@ -89,13 +87,6 @@ class RunContainer(BaseModel):
         # override the default output from pydantic by calling `to_dict()` of run_sizing
         if self.run_sizing:
             _dict['runSizing'] = self.run_sizing.to_dict()
-        # override the default output from pydantic by calling `to_dict()` of each item in artifacts (list)
-        _items = []
-        if self.artifacts:
-            for _item in self.artifacts:
-                if _item:
-                    _items.append(_item.to_dict())
-            _dict['artifacts'] = _items
         return _dict
 
     @classmethod
@@ -118,8 +109,7 @@ class RunContainer(BaseModel):
             "dependencies": obj.get("dependencies"),
             "solutionContainer": obj.get("solutionContainer"),
             "nodeLabel": obj.get("nodeLabel"),
-            "runSizing": ContainerResourceSizing.from_dict(obj["runSizing"]) if obj.get("runSizing") is not None else None,
-            "artifacts": [RunContainerArtifact.from_dict(_item) for _item in obj["artifacts"]] if obj.get("artifacts") is not None else None
+            "runSizing": ContainerResourceSizing.from_dict(obj["runSizing"]) if obj.get("runSizing") is not None else None
         })
         return _obj
 

cosmotech_api/models/run_container_artifact.py

@@ -5,7 +5,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 3.2.0
+    The version of the OpenAPI document: 3.3.6
     Contact: platform@cosmotech.com
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
@@ -18,7 +18,7 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, Field, StrictStr
+from pydantic import BaseModel, ConfigDict, Field, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
 from typing import Optional, Set
 from typing_extensions import Self
@@ -31,11 +31,11 @@ class RunContainerArtifact(BaseModel):
     path: Optional[StrictStr] = Field(default=None, description="the artifact path (relative to /var/csmoutput)")
     __properties: ClassVar[List[str]] = ["name", "path"]
 
-    model_config = {
-        "populate_by_name": True,
-        "validate_assignment": True,
-        "protected_namespaces": (),
-    }
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
 
 
     def to_str(self) -> str:

cosmotech_api/models/run_container_logs.py

@@ -5,7 +5,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 3.2.0
+    The version of the OpenAPI document: 3.3.6
     Contact: platform@cosmotech.com
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
@@ -18,7 +18,7 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, Field, StrictStr
+from pydantic import BaseModel, ConfigDict, Field, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
 from typing import Optional, Set
 from typing_extensions import Self
@@ -33,11 +33,11 @@ class RunContainerLogs(BaseModel):
     logs: Optional[StrictStr] = Field(default=None, description="the node logs in plain text")
     __properties: ClassVar[List[str]] = ["nodeId", "containerName", "children", "logs"]
 
-    model_config = {
-        "populate_by_name": True,
-        "validate_assignment": True,
-        "protected_namespaces": (),
-    }
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
 
 
     def to_str(self) -> str:

cosmotech_api/models/run_data.py

@@ -5,7 +5,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 3.2.0
+    The version of the OpenAPI document: 3.3.6
     Contact: platform@cosmotech.com
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
@@ -18,7 +18,7 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, Field, StrictStr
+from pydantic import BaseModel, ConfigDict, Field, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
 from typing import Optional, Set
 from typing_extensions import Self
@@ -32,11 +32,11 @@ class RunData(BaseModel):
     data: Optional[List[Dict[str, Any]]] = None
     __properties: ClassVar[List[str]] = ["database_name", "table_name", "data"]
 
-    model_config = {
-        "populate_by_name": True,
-        "validate_assignment": True,
-        "protected_namespaces": (),
-    }
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
 
 
     def to_str(self) -> str:

cosmotech_api/models/run_data_query.py

@@ -5,7 +5,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 3.2.0
+    The version of the OpenAPI document: 3.3.6
     Contact: platform@cosmotech.com
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
@@ -18,7 +18,7 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, Field, StrictStr
+from pydantic import BaseModel, ConfigDict, Field, StrictStr
 from typing import Any, ClassVar, Dict, List
 from typing import Optional, Set
 from typing_extensions import Self
@@ -30,11 +30,11 @@ class RunDataQuery(BaseModel):
     query: StrictStr = Field(description="the query in SQL")
     __properties: ClassVar[List[str]] = ["query"]
 
-    model_config = {
-        "populate_by_name": True,
-        "validate_assignment": True,
-        "protected_namespaces": (),
-    }
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
 
 
     def to_str(self) -> str:

cosmotech_api/models/run_edit_info.py

@@ -0,0 +1,90 @@
+# coding: utf-8
+
+"""
+    Cosmo Tech Platform API
+
+    Cosmo Tech Platform API
+
+    The version of the OpenAPI document: 5.0.0-rc3
+    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, ConfigDict, Field, StrictInt, StrictStr
+from typing import Any, ClassVar, Dict, List
+from typing import Optional, Set
+from typing_extensions import Self
+
+class RunEditInfo(BaseModel):
+    """
+    RunEditInfo
+    """ # noqa: E501
+    timestamp: StrictInt = Field(description="The timestamp of the modification in millisecond")
+    user_id: StrictStr = Field(description="The id of the user who did the modification", alias="userId")
+    __properties: ClassVar[List[str]] = ["timestamp", "userId"]
+
+    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 RunEditInfo 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 RunEditInfo from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "timestamp": obj.get("timestamp"),
+            "userId": obj.get("userId")
+        })
+        return _obj
+
+

cosmotech_api/models/run_logs.py

@@ -5,7 +5,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 3.2.0
+    The version of the OpenAPI document: 3.3.6
     Contact: platform@cosmotech.com
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
@@ -18,7 +18,7 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, Field, StrictStr
+from pydantic import BaseModel, ConfigDict, Field, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
 from cosmotech_api.models.run_container_logs import RunContainerLogs
 from typing import Optional, Set
@@ -32,11 +32,11 @@ class RunLogs(BaseModel):
     containers: Optional[Dict[str, RunContainerLogs]] = Field(default=None, description="the container map of logs")
     __properties: ClassVar[List[str]] = ["runId", "containers"]
 
-    model_config = {
-        "populate_by_name": True,
-        "validate_assignment": True,
-        "protected_namespaces": (),
-    }
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
 
 
     def to_str(self) -> str:
@@ -78,9 +78,9 @@ class RunLogs(BaseModel):
         # override the default output from pydantic by calling `to_dict()` of each value in containers (dict)
         _field_dict = {}
         if self.containers:
-            for _key in self.containers:
-                if self.containers[_key]:
-                    _field_dict[_key] = self.containers[_key].to_dict()
+            for _key_containers in self.containers:
+                if self.containers[_key_containers]:
+                    _field_dict[_key_containers] = self.containers[_key_containers].to_dict()
             _dict['containers'] = _field_dict
         return _dict
 

cosmotech_api/models/run_logs_entry.py

@@ -0,0 +1,88 @@
+# coding: utf-8
+
+"""
+    Cosmo Tech Platform API
+
+    Cosmo Tech Platform API
+
+    The version of the OpenAPI document: 4.2.1-onprem
+    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, ConfigDict, Field, StrictStr
+from typing import Any, ClassVar, Dict, List
+from typing import Optional, Set
+from typing_extensions import Self
+
+class RunLogsEntry(BaseModel):
+    """
+    single run log entry
+    """ # noqa: E501
+    line: StrictStr = Field(description="log line data")
+    __properties: ClassVar[List[str]] = ["line"]
+
+    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 RunLogsEntry 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 RunLogsEntry from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "line": obj.get("line")
+        })
+        return _obj
+
+

cosmotech_api/models/run_resource_requested.py

@@ -5,7 +5,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 3.2.0
+    The version of the OpenAPI document: 5.0.0-rc3
     Contact: platform@cosmotech.com
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
@@ -18,7 +18,7 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, Field, StrictInt
+from pydantic import BaseModel, ConfigDict, Field, StrictInt
 from typing import Any, ClassVar, Dict, List, Optional
 from typing import Optional, Set
 from typing_extensions import Self
@@ -31,11 +31,11 @@ class RunResourceRequested(BaseModel):
     memory: Optional[StrictInt] = Field(default=None, description="the memory requested")
     __properties: ClassVar[List[str]] = ["cpu", "memory"]
 
-    model_config = {
-        "populate_by_name": True,
-        "validate_assignment": True,
-        "protected_namespaces": (),
-    }
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
 
 
     def to_str(self) -> str:

cosmotech_api/models/run_search.py

@@ -5,7 +5,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 3.2.0
+    The version of the OpenAPI document: 3.3.6
     Contact: platform@cosmotech.com
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
@@ -18,8 +18,9 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, Field, StrictStr, field_validator
+from pydantic import BaseModel, ConfigDict, Field, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
+from cosmotech_api.models.run_search_state import RunSearchState
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -31,27 +32,17 @@ class RunSearch(BaseModel):
     run_template_id: Optional[StrictStr] = Field(default=None, description="the Solution Analysis Id to search", alias="runTemplateId")
     workspace_id: Optional[StrictStr] = Field(default=None, description="the Workspace Id to search", alias="workspaceId")
     runner_id: Optional[StrictStr] = Field(default=None, description="the Runner Id to search", alias="runnerId")
-    state: Optional[StrictStr] = Field(default=None, description="the state to search")
+    state: Optional[RunSearchState] = None
     workflow_id: Optional[StrictStr] = Field(default=None, description="the Cosmo Tech compute cluster Argo Workflow Id to search", alias="workflowId")
     workflow_name: Optional[StrictStr] = Field(default=None, description="the Cosmo Tech compute cluster Argo Workflow Name", alias="workflowName")
     owner_id: Optional[StrictStr] = Field(default=None, description="the owner Id to search", alias="ownerId")
     __properties: ClassVar[List[str]] = ["solutionId", "runTemplateId", "workspaceId", "runnerId", "state", "workflowId", "workflowName", "ownerId"]
 
-    @field_validator('state')
-    def state_validate_enum(cls, value):
-        """Validates the enum"""
-        if value is None:
-            return value
-
-        if value not in set(['FetchingDatasets', 'FetchingScenarioParameters', 'ApplyingScenarioParameters', 'ValidatingScenarioData', 'SendingScenarioDataToDataWarehouse', 'PreRun', 'Running', 'PostRun', 'Success', 'Failed']):
-            raise ValueError("must be one of enum values ('FetchingDatasets', 'FetchingScenarioParameters', 'ApplyingScenarioParameters', 'ValidatingScenarioData', 'SendingScenarioDataToDataWarehouse', 'PreRun', 'Running', 'PostRun', 'Success', 'Failed')")
-        return value
-
-    model_config = {
-        "populate_by_name": True,
-        "validate_assignment": True,
-        "protected_namespaces": (),
-    }
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
 
 
     def to_str(self) -> str: