cosmotech-api 4.0.0.dev9__py3-none-any.whl → 5.0.0b4__py3-none-any.whl

cosmotech_api/models/run_container_logs.py

@@ -5,7 +5,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 3.2.10
+    The version of the OpenAPI document: 3.3.4
     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: 4.0.0-onprem.9
+    The version of the OpenAPI document: 5.0.0-beta4
     Contact: platform@cosmotech.com
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
@@ -25,7 +25,7 @@ from typing_extensions import Self
 
 class RunData(BaseModel):
     """
-    Run Data to send
+    Run Data stored
     """ # noqa: E501
     database_name: Optional[StrictStr] = Field(default=None, description="Database name")
     table_name: Optional[StrictStr] = Field(default=None, description="Table name")

cosmotech_api/models/run_data_query.py

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

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-beta4
+    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: 4.0.0-onprem.9
+    The version of the OpenAPI document: 4.1.3-onprem
     Contact: platform@cosmotech.com
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 

cosmotech_api/models/run_logs_entry.py

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

cosmotech_api/models/run_resource_requested.py

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

cosmotech_api/models/run_search.py

@@ -5,7 +5,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 3.2.10
+    The version of the OpenAPI document: 3.3.4
     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_search_state import RunSearchState
 from typing import Optional, Set
@@ -38,11 +38,11 @@ class RunSearch(BaseModel):
     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"]
 
-    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_state.py

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

cosmotech_api/models/run_start_containers.py

@@ -5,7 +5,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 3.2.10
+    The version of the OpenAPI document: 3.3.4
     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 import RunContainer
 from typing import Optional, Set
@@ -35,11 +35,11 @@ class RunStartContainers(BaseModel):
     containers: List[RunContainer] = Field(description="the containerslist")
     __properties: ClassVar[List[str]] = ["generateName", "csmSimulationId", "nodeLabel", "labels", "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:
@@ -77,9 +77,9 @@ class RunStartContainers(BaseModel):
         # 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
 

cosmotech_api/models/run_state.py

@@ -5,7 +5,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 4.0.0-onprem.9
+    The version of the OpenAPI document: 5.0.0-beta4
     Contact: platform@cosmotech.com
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
@@ -31,6 +31,7 @@ class RunState(str, Enum):
     SUCCESSFUL = 'Successful'
     FAILED = 'Failed'
     UNKNOWN = 'Unknown'
+    NOTSTARTED = 'NotStarted'
 
     @classmethod
     def from_json(cls, json_str: str) -> Self:

cosmotech_api/models/run_status.py

@@ -5,7 +5,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 4.0.0-onprem.9
+    The version of the OpenAPI document: 5.0.0-beta4
     Contact: platform@cosmotech.com
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
@@ -20,6 +20,7 @@ import json
 
 from pydantic import BaseModel, ConfigDict, Field, StrictInt, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
+from cosmotech_api.models.run_edit_info import RunEditInfo
 from cosmotech_api.models.run_state import RunState
 from cosmotech_api.models.run_status_node import RunStatusNode
 from typing import Optional, Set
@@ -35,6 +36,7 @@ class RunStatus(BaseModel):
     runner_id: Optional[StrictStr] = Field(default=None, description="the Runner id", alias="runnerId")
     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")
+    create_info: Optional[RunEditInfo] = Field(default=None, description="The timestamp of the Run creation in milliseconds", alias="createInfo")
     start_time: Optional[StrictStr] = Field(default=None, description="the Run start Date Time", alias="startTime")
     end_time: Optional[StrictStr] = Field(default=None, description="the Run end Date Time", alias="endTime")
     phase: Optional[StrictStr] = Field(default=None, description="high-level summary of where the workflow is in its lifecycle")
@@ -43,7 +45,7 @@ class RunStatus(BaseModel):
     estimated_duration: Optional[StrictInt] = Field(default=None, description="estimatedDuration in seconds", alias="estimatedDuration")
     nodes: Optional[List[RunStatusNode]] = Field(default=None, description="status of Run nodes")
     state: Optional[RunState] = None
-    __properties: ClassVar[List[str]] = ["id", "organizationId", "workspaceId", "runnerId", "workflowId", "workflowName", "startTime", "endTime", "phase", "progress", "message", "estimatedDuration", "nodes", "state"]
+    __properties: ClassVar[List[str]] = ["id", "organizationId", "workspaceId", "runnerId", "workflowId", "workflowName", "createInfo", "startTime", "endTime", "phase", "progress", "message", "estimatedDuration", "nodes", "state"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -84,6 +86,9 @@ class RunStatus(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 nodes (list)
         _items = []
         if self.nodes:
@@ -109,6 +114,7 @@ class RunStatus(BaseModel):
             "runnerId": obj.get("runnerId"),
             "workflowId": obj.get("workflowId"),
             "workflowName": obj.get("workflowName"),
+            "createInfo": RunEditInfo.from_dict(obj["createInfo"]) if obj.get("createInfo") is not None else None,
             "startTime": obj.get("startTime"),
             "endTime": obj.get("endTime"),
             "phase": obj.get("phase"),

cosmotech_api/models/run_status_node.py

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

cosmotech_api/models/run_template.py

@@ -5,7 +5,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 4.0.0-onprem.9
+    The version of the OpenAPI document: 5.0.0-beta4
     Contact: platform@cosmotech.com
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
@@ -18,54 +18,27 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictInt, StrictStr
+from pydantic import BaseModel, ConfigDict, Field, StrictInt, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
-from cosmotech_api.models.delete_historical_data import DeleteHistoricalData
-from cosmotech_api.models.run_template_orchestrator import RunTemplateOrchestrator
+from typing_extensions import Annotated
 from cosmotech_api.models.run_template_resource_sizing import RunTemplateResourceSizing
-from cosmotech_api.models.run_template_step_source import RunTemplateStepSource
 from typing import Optional, Set
 from typing_extensions import Self
 
 class RunTemplate(BaseModel):
     """
-    a Solution Run Template
+    A Solution Run Template
     """ # noqa: E501
-    id: StrictStr = Field(description="the Solution Run Template id")
-    name: Optional[StrictStr] = Field(default=None, description="the Run Template name")
-    labels: Optional[Dict[str, StrictStr]] = Field(default=None, description="a translated label with key as ISO 639-1 code")
-    description: Optional[StrictStr] = Field(default=None, description="the Run Template description")
-    csm_simulation: Optional[StrictStr] = Field(default=None, description="the Cosmo Tech simulation name. This information is send to the Engine. Mandatory information if no Engine is defined", alias="csmSimulation")
-    tags: Optional[List[StrictStr]] = Field(default=None, description="the list of Run Template tags")
-    compute_size: Optional[StrictStr] = Field(default=None, description="the compute size needed for this Run Template. Standard sizes are basic and highcpu. Default is basic", alias="computeSize")
+    id: Annotated[str, Field(min_length=1, strict=True, max_length=50)] = Field(description="The Solution Run Template id")
+    name: Optional[Annotated[str, Field(min_length=1, strict=True, max_length=50)]] = Field(default=None, description="The Run Template name")
+    labels: Optional[Dict[str, StrictStr]] = Field(default=None, description="A translated label with key as ISO 639-1 code")
+    description: Optional[StrictStr] = Field(default=None, description="The Run Template description")
+    tags: Optional[List[StrictStr]] = Field(default=None, description="The list of Run Template tags")
+    compute_size: Optional[StrictStr] = Field(default=None, description="The compute size needed for this Run Template", alias="computeSize")
     run_sizing: Optional[RunTemplateResourceSizing] = Field(default=None, alias="runSizing")
-    no_data_ingestion_state: Optional[StrictBool] = Field(default=None, description="set to true if the run template does not want to check data ingestion state (no probes or not control plane)", alias="noDataIngestionState")
-    fetch_datasets: Optional[StrictBool] = Field(default=None, description="whether or not the fetch dataset step is done", alias="fetchDatasets")
-    scenario_data_download_transform: Optional[StrictBool] = Field(default=None, description="whether or not the scenario data download transform step step is done", alias="scenarioDataDownloadTransform")
-    fetch_scenario_parameters: Optional[StrictBool] = Field(default=None, description="whether or not the fetch parameters step is done", alias="fetchScenarioParameters")
-    apply_parameters: Optional[StrictBool] = Field(default=None, description="whether or not the apply parameter step is done", alias="applyParameters")
-    validate_data: Optional[StrictBool] = Field(default=None, description="whether or not the validate step is done", alias="validateData")
-    send_datasets_to_data_warehouse: Optional[StrictBool] = Field(default=None, description="whether or not the Datasets values are send to the DataWarehouse prior to Simulation Run. If not set follow the Workspace setting", alias="sendDatasetsToDataWarehouse")
-    send_input_parameters_to_data_warehouse: Optional[StrictBool] = Field(default=None, description="whether or not the input parameters values are send to the DataWarehouse prior to Simulation Run. If not set follow the Workspace setting", alias="sendInputParametersToDataWarehouse")
-    pre_run: Optional[StrictBool] = Field(default=None, description="whether or not the pre-run step is done", alias="preRun")
-    run: Optional[StrictBool] = Field(default=None, description="whether or not the run step is done")
-    post_run: Optional[StrictBool] = Field(default=None, description="whether or not the post-run step is done", alias="postRun")
-    parameters_json: Optional[StrictBool] = Field(default=None, description="whether or not to store the scenario parameters in json instead of csv", alias="parametersJson")
-    parameters_handler_source: Optional[RunTemplateStepSource] = Field(default=None, alias="parametersHandlerSource")
-    dataset_validator_source: Optional[RunTemplateStepSource] = Field(default=None, alias="datasetValidatorSource")
-    pre_run_source: Optional[RunTemplateStepSource] = Field(default=None, alias="preRunSource")
-    run_source: Optional[RunTemplateStepSource] = Field(default=None, alias="runSource")
-    post_run_source: Optional[RunTemplateStepSource] = Field(default=None, alias="postRunSource")
-    scenariodata_transform_source: Optional[RunTemplateStepSource] = Field(default=None, alias="scenariodataTransformSource")
-    parameter_groups: Optional[List[StrictStr]] = Field(default=None, description="the ordered list of parameters groups for the Run Template", alias="parameterGroups")
-    stack_steps: Optional[StrictBool] = Field(default=None, description="whether or not to stack adjacent scenario run steps in one container run which will chain steps", alias="stackSteps")
-    git_repository_url: Optional[StrictStr] = Field(default=None, description="an optional URL to the git repository", alias="gitRepositoryUrl")
-    git_branch_name: Optional[StrictStr] = Field(default=None, description="an optional git branch name", alias="gitBranchName")
-    run_template_source_dir: Optional[StrictStr] = Field(default=None, description="an optional directory where to find the run template source", alias="runTemplateSourceDir")
-    orchestrator_type: Optional[RunTemplateOrchestrator] = Field(default=None, alias="orchestratorType")
-    execution_timeout: Optional[StrictInt] = Field(default=None, description="an optional duration in seconds in which a workflow is allowed to run", alias="executionTimeout")
-    delete_historical_data: Optional[DeleteHistoricalData] = Field(default=None, alias="deleteHistoricalData")
-    __properties: ClassVar[List[str]] = ["id", "name", "labels", "description", "csmSimulation", "tags", "computeSize", "runSizing", "noDataIngestionState", "fetchDatasets", "scenarioDataDownloadTransform", "fetchScenarioParameters", "applyParameters", "validateData", "sendDatasetsToDataWarehouse", "sendInputParametersToDataWarehouse", "preRun", "run", "postRun", "parametersJson", "parametersHandlerSource", "datasetValidatorSource", "preRunSource", "runSource", "postRunSource", "scenariodataTransformSource", "parameterGroups", "stackSteps", "gitRepositoryUrl", "gitBranchName", "runTemplateSourceDir", "orchestratorType", "executionTimeout", "deleteHistoricalData"]
+    parameter_groups: List[StrictStr] = Field(description="The ordered list of parameters groups for the Run Template", alias="parameterGroups")
+    execution_timeout: Optional[StrictInt] = Field(default=None, description="An optional duration in seconds in which a workflow is allowed to run", alias="executionTimeout")
+    __properties: ClassVar[List[str]] = ["id", "name", "labels", "description", "tags", "computeSize", "runSizing", "parameterGroups", "executionTimeout"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -109,9 +82,6 @@ class RunTemplate(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 delete_historical_data
-        if self.delete_historical_data:
-            _dict['deleteHistoricalData'] = self.delete_historical_data.to_dict()
         return _dict
 
     @classmethod
@@ -128,36 +98,11 @@ class RunTemplate(BaseModel):
             "name": obj.get("name"),
             "labels": obj.get("labels"),
             "description": obj.get("description"),
-            "csmSimulation": obj.get("csmSimulation"),
             "tags": obj.get("tags"),
             "computeSize": obj.get("computeSize"),
             "runSizing": RunTemplateResourceSizing.from_dict(obj["runSizing"]) if obj.get("runSizing") is not None else None,
-            "noDataIngestionState": obj.get("noDataIngestionState"),
-            "fetchDatasets": obj.get("fetchDatasets"),
-            "scenarioDataDownloadTransform": obj.get("scenarioDataDownloadTransform"),
-            "fetchScenarioParameters": obj.get("fetchScenarioParameters"),
-            "applyParameters": obj.get("applyParameters"),
-            "validateData": obj.get("validateData"),
-            "sendDatasetsToDataWarehouse": obj.get("sendDatasetsToDataWarehouse"),
-            "sendInputParametersToDataWarehouse": obj.get("sendInputParametersToDataWarehouse"),
-            "preRun": obj.get("preRun"),
-            "run": obj.get("run"),
-            "postRun": obj.get("postRun"),
-            "parametersJson": obj.get("parametersJson"),
-            "parametersHandlerSource": obj.get("parametersHandlerSource"),
-            "datasetValidatorSource": obj.get("datasetValidatorSource"),
-            "preRunSource": obj.get("preRunSource"),
-            "runSource": obj.get("runSource"),
-            "postRunSource": obj.get("postRunSource"),
-            "scenariodataTransformSource": obj.get("scenariodataTransformSource"),
             "parameterGroups": obj.get("parameterGroups"),
-            "stackSteps": obj.get("stackSteps"),
-            "gitRepositoryUrl": obj.get("gitRepositoryUrl"),
-            "gitBranchName": obj.get("gitBranchName"),
-            "runTemplateSourceDir": obj.get("runTemplateSourceDir"),
-            "orchestratorType": obj.get("orchestratorType"),
-            "executionTimeout": obj.get("executionTimeout"),
-            "deleteHistoricalData": DeleteHistoricalData.from_dict(obj["deleteHistoricalData"]) if obj.get("deleteHistoricalData") is not None else None
+            "executionTimeout": obj.get("executionTimeout")
         })
         return _obj
 

cosmotech_api/models/run_template_create_request.py

@@ -0,0 +1,109 @@
+# coding: utf-8
+
+"""
+    Cosmo Tech Platform API
+
+    Cosmo Tech Platform API
+
+    The version of the OpenAPI document: 5.0.0-beta4
+    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, Optional
+from typing_extensions import Annotated
+from cosmotech_api.models.run_template_resource_sizing import RunTemplateResourceSizing
+from typing import Optional, Set
+from typing_extensions import Self
+
+class RunTemplateCreateRequest(BaseModel):
+    """
+    A Solution Run Template Create Request
+    """ # noqa: E501
+    id: Annotated[str, Field(min_length=1, strict=True, max_length=50)] = Field(description="The Solution Run Template id")
+    name: Optional[Annotated[str, Field(min_length=1, strict=True, max_length=50)]] = Field(default=None, description="The Run Template name")
+    labels: Optional[Dict[str, StrictStr]] = Field(default=None, description="A translated label with key as ISO 639-1 code")
+    description: Optional[StrictStr] = Field(default=None, description="The Run Template description")
+    tags: Optional[List[StrictStr]] = Field(default=None, description="The list of Run Template tags")
+    compute_size: Optional[StrictStr] = Field(default=None, description="The compute size needed for this Run Template", alias="computeSize")
+    run_sizing: Optional[RunTemplateResourceSizing] = Field(default=None, alias="runSizing")
+    parameter_groups: Optional[List[StrictStr]] = Field(default=None, description="The ordered list of parameters groups for the Run Template", alias="parameterGroups")
+    execution_timeout: Optional[StrictInt] = Field(default=None, description="An optional duration in seconds in which a workflow is allowed to run", alias="executionTimeout")
+    __properties: ClassVar[List[str]] = ["id", "name", "labels", "description", "tags", "computeSize", "runSizing", "parameterGroups", "executionTimeout"]
+
+    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 RunTemplateCreateRequest 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,
+        )
+        # override the default output from pydantic by calling `to_dict()` of run_sizing
+        if self.run_sizing:
+            _dict['runSizing'] = self.run_sizing.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of RunTemplateCreateRequest 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"),
+            "name": obj.get("name"),
+            "labels": obj.get("labels"),
+            "description": obj.get("description"),
+            "tags": obj.get("tags"),
+            "computeSize": obj.get("computeSize"),
+            "runSizing": RunTemplateResourceSizing.from_dict(obj["runSizing"]) if obj.get("runSizing") is not None else None,
+            "parameterGroups": obj.get("parameterGroups"),
+            "executionTimeout": obj.get("executionTimeout")
+        })
+        return _obj
+
+

cosmotech_api/models/run_template_handler_id.py

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

cosmotech_api/models/run_template_orchestrator.py

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