cosmotech-api 3.2.6.dev0__py3-none-any.whl → 3.3.4__py3-none-any.whl

cosmotech_api/models/organization.py

@@ -5,7 +5,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 4.0.0-onprem.7
+    The version of the OpenAPI document: 3.3.4
     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.organization_security import OrganizationSecurity
+from cosmotech_api.models.organization_services import OrganizationServices
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -31,14 +32,15 @@ class Organization(BaseModel):
     id: Optional[StrictStr] = Field(default=None, description="the Organization unique identifier")
     name: Optional[StrictStr] = Field(default=None, description="the Organization name")
     owner_id: Optional[StrictStr] = Field(default=None, description="the Owner User Id", alias="ownerId")
+    services: Optional[OrganizationServices] = None
     security: Optional[OrganizationSecurity] = None
-    __properties: ClassVar[List[str]] = ["id", "name", "ownerId", "security"]
+    __properties: ClassVar[List[str]] = ["id", "name", "ownerId", "services", "security"]
 
-    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,6 +79,9 @@ class Organization(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
+        # override the default output from pydantic by calling `to_dict()` of services
+        if self.services:
+            _dict['services'] = self.services.to_dict()
         # override the default output from pydantic by calling `to_dict()` of security
         if self.security:
             _dict['security'] = self.security.to_dict()
@@ -95,6 +100,7 @@ class Organization(BaseModel):
             "id": obj.get("id"),
             "name": obj.get("name"),
             "ownerId": obj.get("ownerId"),
+            "services": OrganizationServices.from_dict(obj["services"]) if obj.get("services") is not None else None,
             "security": OrganizationSecurity.from_dict(obj["security"]) if obj.get("security") is not None else None
         })
         return _obj

cosmotech_api/models/organization_access_control.py

@@ -5,7 +5,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 4.0.0-onprem.7
+    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
 from typing import Optional, Set
 from typing_extensions import Self
@@ -31,11 +31,11 @@ class OrganizationAccessControl(BaseModel):
     role: StrictStr = Field(description="a role")
     __properties: ClassVar[List[str]] = ["id", "role"]
 
-    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/organization_create_request.py

@@ -0,0 +1,95 @@
+# 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
+from typing import Any, ClassVar, Dict, List, Optional
+from typing_extensions import Annotated
+from cosmotech_api.models.organization_security import OrganizationSecurity
+from typing import Optional, Set
+from typing_extensions import Self
+
+class OrganizationCreateRequest(BaseModel):
+    """
+    Request object for creating a new organization
+    """ # noqa: E501
+    name: Annotated[str, Field(min_length=1, strict=True, max_length=50)] = Field(description="the Organization name")
+    security: Optional[OrganizationSecurity] = None
+    __properties: ClassVar[List[str]] = ["name", "security"]
+
+    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 OrganizationCreateRequest 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 security
+        if self.security:
+            _dict['security'] = self.security.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of OrganizationCreateRequest from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "name": obj.get("name"),
+            "security": OrganizationSecurity.from_dict(obj["security"]) if obj.get("security") is not None else None
+        })
+        return _obj
+
+

cosmotech_api/models/organization_role.py

@@ -5,7 +5,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 4.0.0-onprem.7
+    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
 from typing import Optional, Set
 from typing_extensions import Self
@@ -30,11 +30,11 @@ class OrganizationRole(BaseModel):
     role: StrictStr = Field(description="the Organization Role")
     __properties: ClassVar[List[str]] = ["role"]
 
-    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/organization_security.py

@@ -5,7 +5,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 4.0.0-onprem.7
+    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
 from cosmotech_api.models.organization_access_control import OrganizationAccessControl
 from typing import Optional, Set
@@ -32,11 +32,11 @@ class OrganizationSecurity(BaseModel):
     access_control_list: List[OrganizationAccessControl] = Field(description="the list which can access this Organization with detailed access control information", alias="accessControlList")
     __properties: ClassVar[List[str]] = ["default", "accessControlList"]
 
-    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:
@@ -74,9 +74,9 @@ class OrganizationSecurity(BaseModel):
         # override the default output from pydantic by calling `to_dict()` of each item in access_control_list (list)
         _items = []
         if self.access_control_list:
-            for _item in self.access_control_list:
-                if _item:
-                    _items.append(_item.to_dict())
+            for _item_access_control_list in self.access_control_list:
+                if _item_access_control_list:
+                    _items.append(_item_access_control_list.to_dict())
             _dict['accessControlList'] = _items
         return _dict
 

cosmotech_api/models/organization_service.py

@@ -5,7 +5,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 3.2.9
+    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
@@ -34,11 +34,11 @@ class OrganizationService(BaseModel):
     credentials: Optional[Dict[str, Any]] = Field(default=None, description="a freeform credentials object. Structure depends on service")
     __properties: ClassVar[List[str]] = ["cloudService", "baseUri", "platformService", "resourceUri", "credentials"]
 
-    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/organization_services.py

@@ -5,7 +5,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 3.2.9
+    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
+from pydantic import BaseModel, ConfigDict, Field
 from typing import Any, ClassVar, Dict, List, Optional
 from cosmotech_api.models.organization_service import OrganizationService
 from typing import Optional, Set
@@ -33,11 +33,11 @@ class OrganizationServices(BaseModel):
     solutions_container_registry: Optional[OrganizationService] = Field(default=None, alias="solutionsContainerRegistry")
     __properties: ClassVar[List[str]] = ["tenantCredentials", "storage", "solutionsContainerRegistry"]
 
-    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/organization_update_request.py

@@ -0,0 +1,89 @@
+# 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
+from typing import Any, ClassVar, Dict, List, Optional
+from typing_extensions import Annotated
+from typing import Optional, Set
+from typing_extensions import Self
+
+class OrganizationUpdateRequest(BaseModel):
+    """
+    Request object for updating an organization
+    """ # noqa: E501
+    name: Optional[Annotated[str, Field(min_length=1, strict=True, max_length=50)]] = Field(default=None, description="the Organization name")
+    __properties: ClassVar[List[str]] = ["name"]
+
+    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 OrganizationUpdateRequest 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 OrganizationUpdateRequest from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "name": obj.get("name")
+        })
+        return _obj
+
+

cosmotech_api/models/query_result.py

@@ -5,7 +5,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 4.0.0-onprem.7
+    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
+from pydantic import BaseModel, ConfigDict, Field
 from typing import Any, ClassVar, Dict, List, Optional
 from typing import Optional, Set
 from typing_extensions import Self
@@ -30,11 +30,11 @@ class QueryResult(BaseModel):
     result: Optional[List[Dict[str, Any]]] = Field(default=None, description="the list of results")
     __properties: ClassVar[List[str]] = ["result"]
 
-    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/resource_size_info.py

@@ -5,7 +5,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 4.0.0-onprem.7
+    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
 from typing import Optional, Set
 from typing_extensions import Self
@@ -31,11 +31,11 @@ class ResourceSizeInfo(BaseModel):
     memory: StrictStr = Field(description="define memory needs")
     __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.py

@@ -5,7 +5,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 4.0.0-onprem.7
+    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 cosmotech_api.models.run_state import RunState
@@ -51,11 +51,11 @@ class Run(BaseModel):
     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"]
 
-    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:
@@ -119,16 +119,16 @@ class Run(BaseModel):
         # 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
 

cosmotech_api/models/run_container.py

@@ -5,7 +5,7 @@
 
     Cosmo Tech Platform API
 
-    The version of the OpenAPI document: 4.0.0-onprem.7
+    The version of the OpenAPI document: 3.3.4
     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
+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
 
@@ -39,13 +40,14 @@ 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")
-    __properties: ClassVar[List[str]] = ["id", "name", "labels", "envVars", "image", "entrypoint", "runArgs", "dependencies", "solutionContainer", "nodeLabel", "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"]
 
-    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:
@@ -87,6 +89,13 @@ 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_artifacts in self.artifacts:
+                if _item_artifacts:
+                    _items.append(_item_artifacts.to_dict())
+            _dict['artifacts'] = _items
         return _dict
 
     @classmethod
@@ -109,7 +118,8 @@ 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
+            "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
         })
         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.9
+    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
@@ -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: