truefoundry 0.3.3__py3-none-any.whl → 0.4.0.dev0__py3-none-any.whl

truefoundry/ml/autogen/client/models/create_multi_part_upload_request_dto.py

@@ -0,0 +1,73 @@
+# coding: utf-8
+
+"""
+FastAPI
+
+No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+The version of the OpenAPI document: 0.1.0
+Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+Do not edit the class manually.
+"""  # noqa: E501
+
+from __future__ import annotations
+
+import json
+import pprint
+import re  # noqa: F401
+
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictInt, StrictStr
+
+
+class CreateMultiPartUploadRequestDto(BaseModel):
+    """
+    CreateMultiPartUploadRequestDto
+    """
+
+    artifact_version_id: StrictStr = Field(...)
+    path: StrictStr = Field(...)
+    num_parts: StrictInt = Field(...)
+    __properties = ["artifact_version_id", "path", "num_parts"]
+
+    class Config:
+        """Pydantic configuration"""
+
+        allow_population_by_field_name = True
+        validate_assignment = True
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.dict(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> CreateMultiPartUploadRequestDto:
+        """Create an instance of CreateMultiPartUploadRequestDto from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self):
+        """Returns the dictionary representation of the model using alias"""
+        _dict = self.dict(by_alias=True, exclude={}, exclude_none=True)
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> CreateMultiPartUploadRequestDto:
+        """Create an instance of CreateMultiPartUploadRequestDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return CreateMultiPartUploadRequestDto.parse_obj(obj)
+
+        _obj = CreateMultiPartUploadRequestDto.parse_obj(
+            {
+                "artifact_version_id": obj.get("artifact_version_id"),
+                "path": obj.get("path"),
+                "num_parts": obj.get("num_parts"),
+            }
+        )
+        return _obj

truefoundry/ml/autogen/client/models/create_python_deployment_config_request_dto.py

@@ -0,0 +1,72 @@
+# coding: utf-8
+
+"""
+FastAPI
+
+No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+The version of the OpenAPI document: 0.1.0
+Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+Do not edit the class manually.
+"""  # noqa: E501
+
+from __future__ import annotations
+
+import json
+import pprint
+import re  # noqa: F401
+from typing import Any, Dict, Optional
+
+from truefoundry.pydantic_v1 import BaseModel, StrictStr
+
+
+class CreatePythonDeploymentConfigRequestDto(BaseModel):
+    """
+    CreatePythonDeploymentConfigRequestDto
+    """
+
+    workspace_fqn: Optional[StrictStr] = None
+    deployment_config: Optional[Dict[str, Any]] = None
+    __properties = ["workspace_fqn", "deployment_config"]
+
+    class Config:
+        """Pydantic configuration"""
+
+        allow_population_by_field_name = True
+        validate_assignment = True
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.dict(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> CreatePythonDeploymentConfigRequestDto:
+        """Create an instance of CreatePythonDeploymentConfigRequestDto from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self):
+        """Returns the dictionary representation of the model using alias"""
+        _dict = self.dict(by_alias=True, exclude={}, exclude_none=True)
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> CreatePythonDeploymentConfigRequestDto:
+        """Create an instance of CreatePythonDeploymentConfigRequestDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return CreatePythonDeploymentConfigRequestDto.parse_obj(obj)
+
+        _obj = CreatePythonDeploymentConfigRequestDto.parse_obj(
+            {
+                "workspace_fqn": obj.get("workspace_fqn"),
+                "deployment_config": obj.get("deployment_config"),
+            }
+        )
+        return _obj

truefoundry/ml/autogen/client/models/create_python_deployment_config_response_dto.py

@@ -0,0 +1,68 @@
+# coding: utf-8
+
+"""
+FastAPI
+
+No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+The version of the OpenAPI document: 0.1.0
+Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+Do not edit the class manually.
+"""  # noqa: E501
+
+from __future__ import annotations
+
+import json
+import pprint
+import re  # noqa: F401
+from typing import Optional
+
+from truefoundry.pydantic_v1 import BaseModel, StrictStr
+
+
+class CreatePythonDeploymentConfigResponseDto(BaseModel):
+    """
+    CreatePythonDeploymentConfigResponseDto
+    """
+
+    deploy_file_content: Optional[StrictStr] = None
+    __properties = ["deploy_file_content"]
+
+    class Config:
+        """Pydantic configuration"""
+
+        allow_population_by_field_name = True
+        validate_assignment = True
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.dict(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> CreatePythonDeploymentConfigResponseDto:
+        """Create an instance of CreatePythonDeploymentConfigResponseDto from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self):
+        """Returns the dictionary representation of the model using alias"""
+        _dict = self.dict(by_alias=True, exclude={}, exclude_none=True)
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> CreatePythonDeploymentConfigResponseDto:
+        """Create an instance of CreatePythonDeploymentConfigResponseDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return CreatePythonDeploymentConfigResponseDto.parse_obj(obj)
+
+        _obj = CreatePythonDeploymentConfigResponseDto.parse_obj(
+            {"deploy_file_content": obj.get("deploy_file_content")}
+        )
+        return _obj

truefoundry/ml/autogen/client/models/create_run_request_dto.py

@@ -0,0 +1,97 @@
+# coding: utf-8
+
+"""
+FastAPI
+
+No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+The version of the OpenAPI document: 0.1.0
+Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+Do not edit the class manually.
+"""  # noqa: E501
+
+from __future__ import annotations
+
+import json
+import pprint
+import re  # noqa: F401
+from typing import Optional
+
+from truefoundry.ml.autogen.client.models.run_tag_dto import RunTagDto
+from truefoundry.pydantic_v1 import BaseModel, StrictInt, StrictStr, conlist
+
+
+class CreateRunRequestDto(BaseModel):
+    """
+    CreateRunRequestDto
+    """
+
+    experiment_id: Optional[StrictStr] = None
+    user_id: Optional[StrictStr] = None
+    start_time: Optional[StrictInt] = None
+    tags: Optional[conlist(RunTagDto)] = None
+    name: Optional[StrictStr] = None
+    description: Optional[StrictStr] = None
+    __properties = [
+        "experiment_id",
+        "user_id",
+        "start_time",
+        "tags",
+        "name",
+        "description",
+    ]
+
+    class Config:
+        """Pydantic configuration"""
+
+        allow_population_by_field_name = True
+        validate_assignment = True
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.dict(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> CreateRunRequestDto:
+        """Create an instance of CreateRunRequestDto from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self):
+        """Returns the dictionary representation of the model using alias"""
+        _dict = self.dict(by_alias=True, exclude={}, exclude_none=True)
+        # override the default output from truefoundry.pydantic_v1 by calling `to_dict()` of each item in tags (list)
+        _items = []
+        if self.tags:
+            for _item in self.tags:
+                if _item:
+                    _items.append(_item.to_dict())
+            _dict["tags"] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> CreateRunRequestDto:
+        """Create an instance of CreateRunRequestDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return CreateRunRequestDto.parse_obj(obj)
+
+        _obj = CreateRunRequestDto.parse_obj(
+            {
+                "experiment_id": obj.get("experiment_id"),
+                "user_id": obj.get("user_id"),
+                "start_time": obj.get("start_time"),
+                "tags": [RunTagDto.from_dict(_item) for _item in obj.get("tags")]
+                if obj.get("tags") is not None
+                else None,
+                "name": obj.get("name"),
+                "description": obj.get("description"),
+            }
+        )
+        return _obj

truefoundry/ml/autogen/client/models/create_run_response_dto.py

@@ -0,0 +1,76 @@
+# coding: utf-8
+
+"""
+FastAPI
+
+No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+The version of the OpenAPI document: 0.1.0
+Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+Do not edit the class manually.
+"""  # noqa: E501
+
+from __future__ import annotations
+
+import json
+import pprint
+import re  # noqa: F401
+from typing import Optional
+
+from truefoundry.ml.autogen.client.models.run_dto import RunDto
+from truefoundry.pydantic_v1 import BaseModel
+
+
+class CreateRunResponseDto(BaseModel):
+    """
+    CreateRunResponseDto
+    """
+
+    run: Optional[RunDto] = None
+    __properties = ["run"]
+
+    class Config:
+        """Pydantic configuration"""
+
+        allow_population_by_field_name = True
+        validate_assignment = True
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.dict(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> CreateRunResponseDto:
+        """Create an instance of CreateRunResponseDto from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self):
+        """Returns the dictionary representation of the model using alias"""
+        _dict = self.dict(by_alias=True, exclude={}, exclude_none=True)
+        # override the default output from truefoundry.pydantic_v1 by calling `to_dict()` of run
+        if self.run:
+            _dict["run"] = self.run.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> CreateRunResponseDto:
+        """Create an instance of CreateRunResponseDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return CreateRunResponseDto.parse_obj(obj)
+
+        _obj = CreateRunResponseDto.parse_obj(
+            {
+                "run": RunDto.from_dict(obj.get("run"))
+                if obj.get("run") is not None
+                else None
+            }
+        )
+        return _obj

truefoundry/ml/autogen/client/models/dataset_dto.py

@@ -0,0 +1,112 @@
+# coding: utf-8
+
+"""
+FastAPI
+
+No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+The version of the OpenAPI document: 0.1.0
+Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+Do not edit the class manually.
+"""  # noqa: E501
+
+from __future__ import annotations
+
+import json
+import pprint
+import re  # noqa: F401
+from datetime import datetime
+from typing import Any, Dict, Optional
+
+from truefoundry.ml.autogen.client.models.experiment_dto import ExperimentDto
+from truefoundry.pydantic_v1 import BaseModel, StrictStr
+
+
+class DatasetDto(BaseModel):
+    """
+    DatasetDto
+    """
+
+    id: Optional[StrictStr] = None
+    experiment_id: Optional[StrictStr] = None
+    experiment: Optional[ExperimentDto] = None
+    name: Optional[StrictStr] = None
+    fqn: Optional[StrictStr] = None
+    description: Optional[StrictStr] = None
+    created_by: Optional[StrictStr] = None
+    created_at: Optional[datetime] = None
+    updated_at: Optional[datetime] = None
+    storage_root: Optional[StrictStr] = None
+    dataset_metadata: Optional[Dict[str, Any]] = None
+    usage_code_snippet: Optional[StrictStr] = None
+    __properties = [
+        "id",
+        "experiment_id",
+        "experiment",
+        "name",
+        "fqn",
+        "description",
+        "created_by",
+        "created_at",
+        "updated_at",
+        "storage_root",
+        "dataset_metadata",
+        "usage_code_snippet",
+    ]
+
+    class Config:
+        """Pydantic configuration"""
+
+        allow_population_by_field_name = True
+        validate_assignment = True
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.dict(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> DatasetDto:
+        """Create an instance of DatasetDto from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self):
+        """Returns the dictionary representation of the model using alias"""
+        _dict = self.dict(by_alias=True, exclude={}, exclude_none=True)
+        # override the default output from truefoundry.pydantic_v1 by calling `to_dict()` of experiment
+        if self.experiment:
+            _dict["experiment"] = self.experiment.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> DatasetDto:
+        """Create an instance of DatasetDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return DatasetDto.parse_obj(obj)
+
+        _obj = DatasetDto.parse_obj(
+            {
+                "id": obj.get("id"),
+                "experiment_id": obj.get("experiment_id"),
+                "experiment": ExperimentDto.from_dict(obj.get("experiment"))
+                if obj.get("experiment") is not None
+                else None,
+                "name": obj.get("name"),
+                "fqn": obj.get("fqn"),
+                "description": obj.get("description"),
+                "created_by": obj.get("created_by"),
+                "created_at": obj.get("created_at"),
+                "updated_at": obj.get("updated_at"),
+                "storage_root": obj.get("storage_root"),
+                "dataset_metadata": obj.get("dataset_metadata"),
+                "usage_code_snippet": obj.get("usage_code_snippet"),
+            }
+        )
+        return _obj

truefoundry/ml/autogen/client/models/dataset_response_dto.py

@@ -0,0 +1,75 @@
+# coding: utf-8
+
+"""
+FastAPI
+
+No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+The version of the OpenAPI document: 0.1.0
+Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+Do not edit the class manually.
+"""  # noqa: E501
+
+from __future__ import annotations
+
+import json
+import pprint
+import re  # noqa: F401
+
+from truefoundry.ml.autogen.client.models.dataset_dto import DatasetDto
+from truefoundry.pydantic_v1 import BaseModel, Field
+
+
+class DatasetResponseDto(BaseModel):
+    """
+    DatasetResponseDto
+    """
+
+    dataset: DatasetDto = Field(...)
+    __properties = ["dataset"]
+
+    class Config:
+        """Pydantic configuration"""
+
+        allow_population_by_field_name = True
+        validate_assignment = True
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.dict(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> DatasetResponseDto:
+        """Create an instance of DatasetResponseDto from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self):
+        """Returns the dictionary representation of the model using alias"""
+        _dict = self.dict(by_alias=True, exclude={}, exclude_none=True)
+        # override the default output from truefoundry.pydantic_v1 by calling `to_dict()` of dataset
+        if self.dataset:
+            _dict["dataset"] = self.dataset.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> DatasetResponseDto:
+        """Create an instance of DatasetResponseDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return DatasetResponseDto.parse_obj(obj)
+
+        _obj = DatasetResponseDto.parse_obj(
+            {
+                "dataset": DatasetDto.from_dict(obj.get("dataset"))
+                if obj.get("dataset") is not None
+                else None
+            }
+        )
+        return _obj

truefoundry/ml/autogen/client/models/delete_artifact_versions_request_dto.py

@@ -0,0 +1,65 @@
+# coding: utf-8
+
+"""
+FastAPI
+
+No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+The version of the OpenAPI document: 0.1.0
+Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+Do not edit the class manually.
+"""  # noqa: E501
+
+from __future__ import annotations
+
+import json
+import pprint
+import re  # noqa: F401
+
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr
+
+
+class DeleteArtifactVersionsRequestDto(BaseModel):
+    """
+    DeleteArtifactVersionsRequestDto
+    """
+
+    id: StrictStr = Field(...)
+    __properties = ["id"]
+
+    class Config:
+        """Pydantic configuration"""
+
+        allow_population_by_field_name = True
+        validate_assignment = True
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.dict(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> DeleteArtifactVersionsRequestDto:
+        """Create an instance of DeleteArtifactVersionsRequestDto from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self):
+        """Returns the dictionary representation of the model using alias"""
+        _dict = self.dict(by_alias=True, exclude={}, exclude_none=True)
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> DeleteArtifactVersionsRequestDto:
+        """Create an instance of DeleteArtifactVersionsRequestDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return DeleteArtifactVersionsRequestDto.parse_obj(obj)
+
+        _obj = DeleteArtifactVersionsRequestDto.parse_obj({"id": obj.get("id")})
+        return _obj