cosmotech-api 3.1.0.dev0__py3-none-any.whl → 3.1.13__py3-none-any.whl

cosmotech_api/models/run_template.py

@@ -0,0 +1,164 @@
+# coding: utf-8
+
+"""
+    Cosmo Tech Platform API
+
+    Cosmo Tech Platform API
+
+    The version of the OpenAPI document: 4.0.1-SNAPSHOT
+    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, StrictBool, 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 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
+    """ # 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")
+    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"]
+
+    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 RunTemplate 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()
+        # 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
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of RunTemplate 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"),
+            "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
+        })
+        return _obj
+
+

cosmotech_api/models/run_template_handler_id.py

@@ -0,0 +1,42 @@
+# coding: utf-8
+
+"""
+    Cosmo Tech Platform API
+
+    Cosmo Tech Platform API
+
+    The version of the OpenAPI document: 3.3.1
+    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 json
+from enum import Enum
+from typing_extensions import Self
+
+
+class RunTemplateHandlerId(str, Enum):
+    """
+    the Run Template step handler identifier
+    """
+
+    """
+    allowed enum values
+    """
+    PARAMETERS_HANDLER = 'parameters_handler'
+    VALIDATOR = 'validator'
+    PRERUN = 'prerun'
+    ENGINE = 'engine'
+    POSTRUN = 'postrun'
+    SCENARIODATA_TRANSFORM = 'scenariodata_transform'
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of RunTemplateHandlerId from a JSON string"""
+        return cls(json.loads(json_str))
+
+

cosmotech_api/models/run_template_orchestrator.py

@@ -0,0 +1,38 @@
+# coding: utf-8
+
+"""
+    Cosmo Tech Platform API
+
+    Cosmo Tech Platform API
+
+    The version of the OpenAPI document: 4.0.1-SNAPSHOT
+    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 json
+from enum import Enum
+from typing_extensions import Self
+
+
+class RunTemplateOrchestrator(str, Enum):
+    """
+    the Orchestrator to use for the Run Template
+    """
+
+    """
+    allowed enum values
+    """
+    ARGOWORKFLOW = 'argoWorkflow'
+    CSMORC = 'csmOrc'
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of RunTemplateOrchestrator from a JSON string"""
+        return cls(json.loads(json_str))
+
+

cosmotech_api/models/run_template_parameter.py

@@ -0,0 +1,102 @@
+# coding: utf-8
+
+"""
+    Cosmo Tech Platform API
+
+    Cosmo Tech Platform API
+
+    The version of the OpenAPI document: 4.0.1-SNAPSHOT
+    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, Optional
+from typing import Optional, Set
+from typing_extensions import Self
+
+class RunTemplateParameter(BaseModel):
+    """
+    a Run Template Parameter
+    """ # noqa: E501
+    id: StrictStr = Field(description="the Parameter id")
+    labels: Optional[Dict[str, StrictStr]] = Field(default=None, description="a translated label with key as ISO 639-1 code")
+    var_type: Optional[StrictStr] = Field(default=None, description="the variable type for the parameter. Basic types or special type %DATASETID%", alias="varType")
+    default_value: Optional[StrictStr] = Field(default=None, description="the default value for this parameter", alias="defaultValue")
+    min_value: Optional[StrictStr] = Field(default=None, description="the minimum value for this parameter", alias="minValue")
+    max_value: Optional[StrictStr] = Field(default=None, description="the maximum value for this parameter", alias="maxValue")
+    regex_validation: Optional[StrictStr] = Field(default=None, description="a regex to validate the value", alias="regexValidation")
+    options: Optional[Dict[str, Any]] = Field(default=None, description="freeform options")
+    __properties: ClassVar[List[str]] = ["id", "labels", "varType", "defaultValue", "minValue", "maxValue", "regexValidation", "options"]
+
+    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 RunTemplateParameter 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 RunTemplateParameter 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"),
+            "labels": obj.get("labels"),
+            "varType": obj.get("varType"),
+            "defaultValue": obj.get("defaultValue"),
+            "minValue": obj.get("minValue"),
+            "maxValue": obj.get("maxValue"),
+            "regexValidation": obj.get("regexValidation"),
+            "options": obj.get("options")
+        })
+        return _obj
+
+

cosmotech_api/models/run_template_parameter_group.py

@@ -0,0 +1,98 @@
+# coding: utf-8
+
+"""
+    Cosmo Tech Platform API
+
+    Cosmo Tech Platform API
+
+    The version of the OpenAPI document: 4.0.1-SNAPSHOT
+    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, StrictBool, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional
+from typing import Optional, Set
+from typing_extensions import Self
+
+class RunTemplateParameterGroup(BaseModel):
+    """
+    a Parameter Group for a Run Template
+    """ # noqa: E501
+    id: StrictStr = Field(description="the Parameter Group id")
+    labels: Optional[Dict[str, StrictStr]] = Field(default=None, description="a translated label with key as ISO 639-1 code")
+    is_table: Optional[StrictBool] = Field(default=None, description="does the group define a table", alias="isTable")
+    options: Optional[Dict[str, Any]] = Field(default=None, description="freeform options")
+    parent_id: Optional[StrictStr] = Field(default=None, description="the Run Template Group parent Id", alias="parentId")
+    parameters: Optional[List[StrictStr]] = Field(default=None, description="an ordered list of Run Template Parameters")
+    __properties: ClassVar[List[str]] = ["id", "labels", "isTable", "options", "parentId", "parameters"]
+
+    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 RunTemplateParameterGroup 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 RunTemplateParameterGroup 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"),
+            "labels": obj.get("labels"),
+            "isTable": obj.get("isTable"),
+            "options": obj.get("options"),
+            "parentId": obj.get("parentId"),
+            "parameters": obj.get("parameters")
+        })
+        return _obj
+
+

cosmotech_api/models/run_template_parameter_value.py

@@ -0,0 +1,94 @@
+# coding: utf-8
+
+"""
+    Cosmo Tech Platform API
+
+    Cosmo Tech Platform API
+
+    The version of the OpenAPI document: 4.0.1-SNAPSHOT
+    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, Optional
+from typing import Optional, Set
+from typing_extensions import Self
+
+class RunTemplateParameterValue(BaseModel):
+    """
+    the value of Analysis parameter for a Runner for this Run
+    """ # noqa: E501
+    parameter_id: StrictStr = Field(description="the parameter Id", alias="parameterId")
+    var_type: Optional[StrictStr] = Field(default=None, description="the parameter value type", alias="varType")
+    value: StrictStr = Field(description="the parameter value")
+    __properties: ClassVar[List[str]] = ["parameterId", "varType", "value"]
+
+    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 RunTemplateParameterValue 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.
+        * OpenAPI `readOnly` fields are excluded.
+        """
+        excluded_fields: Set[str] = set([
+            "var_type",
+        ])
+
+        _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 RunTemplateParameterValue from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "parameterId": obj.get("parameterId"),
+            "varType": obj.get("varType"),
+            "value": obj.get("value")
+        })
+        return _obj
+
+

cosmotech_api/models/run_template_resource_sizing.py

@@ -0,0 +1,97 @@
+# coding: utf-8
+
+"""
+    Cosmo Tech Platform API
+
+    Cosmo Tech Platform API
+
+    The version of the OpenAPI document: 4.0.1-SNAPSHOT
+    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
+from typing import Any, ClassVar, Dict, List
+from cosmotech_api.models.resource_size_info import ResourceSizeInfo
+from typing import Optional, Set
+from typing_extensions import Self
+
+class RunTemplateResourceSizing(BaseModel):
+    """
+    a description object for resource requests and limits (default same configuration as basic sizing)
+    """ # noqa: E501
+    requests: ResourceSizeInfo
+    limits: ResourceSizeInfo
+    __properties: ClassVar[List[str]] = ["requests", "limits"]
+
+    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 RunTemplateResourceSizing 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 requests
+        if self.requests:
+            _dict['requests'] = self.requests.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of limits
+        if self.limits:
+            _dict['limits'] = self.limits.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of RunTemplateResourceSizing from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "requests": ResourceSizeInfo.from_dict(obj["requests"]) if obj.get("requests") is not None else None,
+            "limits": ResourceSizeInfo.from_dict(obj["limits"]) if obj.get("limits") is not None else None
+        })
+        return _obj
+
+

cosmotech_api/models/run_template_step_source.py

@@ -0,0 +1,40 @@
+# coding: utf-8
+
+"""
+    Cosmo Tech Platform API
+
+    Cosmo Tech Platform API
+
+    The version of the OpenAPI document: 4.0.1-SNAPSHOT
+    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 json
+from enum import Enum
+from typing_extensions import Self
+
+
+class RunTemplateStepSource(str, Enum):
+    """
+    the source local, cloud or git repository
+    """
+
+    """
+    allowed enum values
+    """
+    CLOUD = 'cloud'
+    GIT = 'git'
+    LOCAL = 'local'
+    PLATFORM = 'platform'
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of RunTemplateStepSource from a JSON string"""
+        return cls(json.loads(json_str))
+
+