truefoundry 0.3.4__py3-none-any.whl → 0.4.0__py3-none-any.whl

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

@@ -0,0 +1,34 @@
+# 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
+
+import json
+import re  # noqa: F401
+
+from aenum import Enum
+
+
+class PredictionType(str, Enum):
+    """
+    An enumeration.
+    """
+
+    """
+    allowed enum values
+    """
+    CATEGORICAL = "categorical"
+    NUMERIC = "numeric"
+
+    @classmethod
+    def from_json(cls, json_str: str) -> PredictionType:
+        """Create an instance of PredictionType from a JSON string"""
+        return PredictionType(json.loads(json_str))

truefoundry/ml/autogen/client/models/resolve_agent_app_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.agent_app import AgentApp
+from truefoundry.pydantic_v1 import BaseModel, Field
+
+
+class ResolveAgentAppResponseDto(BaseModel):
+    """
+    ResolveAgentAppResponseDto
+    """
+
+    agent_app_manifest: AgentApp = Field(...)
+    __properties = ["agent_app_manifest"]
+
+    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) -> ResolveAgentAppResponseDto:
+        """Create an instance of ResolveAgentAppResponseDto 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 agent_app_manifest
+        if self.agent_app_manifest:
+            _dict["agent_app_manifest"] = self.agent_app_manifest.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> ResolveAgentAppResponseDto:
+        """Create an instance of ResolveAgentAppResponseDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return ResolveAgentAppResponseDto.parse_obj(obj)
+
+        _obj = ResolveAgentAppResponseDto.parse_obj(
+            {
+                "agent_app_manifest": AgentApp.from_dict(obj.get("agent_app_manifest"))
+                if obj.get("agent_app_manifest") is not None
+                else None
+            }
+        )
+        return _obj

truefoundry/ml/autogen/client/models/restore_run_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 RestoreRunRequestDto(BaseModel):
+    """
+    RestoreRunRequestDto
+    """
+
+    run_id: StrictStr = Field(...)
+    __properties = ["run_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) -> RestoreRunRequestDto:
+        """Create an instance of RestoreRunRequestDto 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) -> RestoreRunRequestDto:
+        """Create an instance of RestoreRunRequestDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return RestoreRunRequestDto.parse_obj(obj)
+
+        _obj = RestoreRunRequestDto.parse_obj({"run_id": obj.get("run_id")})
+        return _obj

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

@@ -0,0 +1,104 @@
+# 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.metric_dto import MetricDto
+from truefoundry.ml.autogen.client.models.param_dto import ParamDto
+from truefoundry.ml.autogen.client.models.run_tag_dto import RunTagDto
+from truefoundry.pydantic_v1 import BaseModel, conlist
+
+
+class RunDataDto(BaseModel):
+    """
+    RunDataDto
+    """
+
+    metrics: Optional[conlist(MetricDto)] = None
+    params: Optional[conlist(ParamDto)] = None
+    tags: Optional[conlist(RunTagDto)] = None
+    __properties = ["metrics", "params", "tags"]
+
+    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) -> RunDataDto:
+        """Create an instance of RunDataDto 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 metrics (list)
+        _items = []
+        if self.metrics:
+            for _item in self.metrics:
+                if _item:
+                    _items.append(_item.to_dict())
+            _dict["metrics"] = _items
+        # override the default output from truefoundry.pydantic_v1 by calling `to_dict()` of each item in params (list)
+        _items = []
+        if self.params:
+            for _item in self.params:
+                if _item:
+                    _items.append(_item.to_dict())
+            _dict["params"] = _items
+        # 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) -> RunDataDto:
+        """Create an instance of RunDataDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return RunDataDto.parse_obj(obj)
+
+        _obj = RunDataDto.parse_obj(
+            {
+                "metrics": [MetricDto.from_dict(_item) for _item in obj.get("metrics")]
+                if obj.get("metrics") is not None
+                else None,
+                "params": [ParamDto.from_dict(_item) for _item in obj.get("params")]
+                if obj.get("params") is not None
+                else None,
+                "tags": [RunTagDto.from_dict(_item) for _item in obj.get("tags")]
+                if obj.get("tags") is not None
+                else None,
+            }
+        )
+        return _obj

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

@@ -0,0 +1,84 @@
+# 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_data_dto import RunDataDto
+from truefoundry.ml.autogen.client.models.run_info_dto import RunInfoDto
+from truefoundry.pydantic_v1 import BaseModel, Field
+
+
+class RunDto(BaseModel):
+    """
+    RunDto
+    """
+
+    info: RunInfoDto = Field(...)
+    data: Optional[RunDataDto] = None
+    __properties = ["info", "data"]
+
+    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) -> RunDto:
+        """Create an instance of RunDto 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 info
+        if self.info:
+            _dict["info"] = self.info.to_dict()
+        # override the default output from truefoundry.pydantic_v1 by calling `to_dict()` of data
+        if self.data:
+            _dict["data"] = self.data.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> RunDto:
+        """Create an instance of RunDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return RunDto.parse_obj(obj)
+
+        _obj = RunDto.parse_obj(
+            {
+                "info": RunInfoDto.from_dict(obj.get("info"))
+                if obj.get("info") is not None
+                else None,
+                "data": RunDataDto.from_dict(obj.get("data"))
+                if obj.get("data") is not None
+                else None,
+            }
+        )
+        return _obj

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

@@ -0,0 +1,105 @@
+# 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, Field, StrictInt, StrictStr
+
+
+class RunInfoDto(BaseModel):
+    """
+    RunInfoDto
+    """
+
+    run_uuid: StrictStr = Field(...)
+    run_id: StrictStr = Field(...)
+    experiment_id: StrictStr = Field(...)
+    name: StrictStr = Field(...)
+    fqn: StrictStr = Field(...)
+    user_id: Optional[StrictStr] = None
+    status: Optional[StrictStr] = None
+    start_time: Optional[StrictInt] = None
+    end_time: Optional[StrictInt] = None
+    artifact_uri: Optional[StrictStr] = None
+    lifecycle_stage: Optional[StrictStr] = None
+    description: Optional[StrictStr] = None
+    __properties = [
+        "run_uuid",
+        "run_id",
+        "experiment_id",
+        "name",
+        "fqn",
+        "user_id",
+        "status",
+        "start_time",
+        "end_time",
+        "artifact_uri",
+        "lifecycle_stage",
+        "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) -> RunInfoDto:
+        """Create an instance of RunInfoDto 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) -> RunInfoDto:
+        """Create an instance of RunInfoDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return RunInfoDto.parse_obj(obj)
+
+        _obj = RunInfoDto.parse_obj(
+            {
+                "run_uuid": obj.get("run_uuid"),
+                "run_id": obj.get("run_id"),
+                "experiment_id": obj.get("experiment_id"),
+                "name": obj.get("name"),
+                "fqn": obj.get("fqn"),
+                "user_id": obj.get("user_id"),
+                "status": obj.get("status"),
+                "start_time": obj.get("start_time"),
+                "end_time": obj.get("end_time"),
+                "artifact_uri": obj.get("artifact_uri"),
+                "lifecycle_stage": obj.get("lifecycle_stage"),
+                "description": obj.get("description"),
+            }
+        )
+        return _obj

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

@@ -0,0 +1,90 @@
+# 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, StrictInt, StrictStr
+
+
+class RunLogDto(BaseModel):
+    """
+    RunLogDto
+    """
+
+    key: Optional[StrictStr] = None
+    timestamp: Optional[StrictInt] = None
+    step: Optional[StrictInt] = None
+    log_type: Optional[StrictStr] = None
+    artifact_path: Optional[StrictStr] = None
+    value: Optional[StrictStr] = None
+    artifact_signed_uri: Optional[StrictStr] = None
+    __properties = [
+        "key",
+        "timestamp",
+        "step",
+        "log_type",
+        "artifact_path",
+        "value",
+        "artifact_signed_uri",
+    ]
+
+    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) -> RunLogDto:
+        """Create an instance of RunLogDto 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) -> RunLogDto:
+        """Create an instance of RunLogDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return RunLogDto.parse_obj(obj)
+
+        _obj = RunLogDto.parse_obj(
+            {
+                "key": obj.get("key"),
+                "timestamp": obj.get("timestamp"),
+                "step": obj.get("step"),
+                "log_type": obj.get("log_type"),
+                "artifact_path": obj.get("artifact_path"),
+                "value": obj.get("value"),
+                "artifact_signed_uri": obj.get("artifact_signed_uri"),
+            }
+        )
+        return _obj

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

@@ -0,0 +1,80 @@
+# 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, Field, StrictInt, StrictStr
+
+
+class RunLogInputDto(BaseModel):
+    """
+    RunLogInputDto
+    """
+
+    key: StrictStr = Field(...)
+    value: Optional[StrictStr] = None
+    timestamp: StrictInt = Field(...)
+    step: Optional[StrictInt] = 0
+    log_type: StrictStr = Field(...)
+    artifact_path: Optional[StrictStr] = None
+    __properties = ["key", "value", "timestamp", "step", "log_type", "artifact_path"]
+
+    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) -> RunLogInputDto:
+        """Create an instance of RunLogInputDto 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) -> RunLogInputDto:
+        """Create an instance of RunLogInputDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return RunLogInputDto.parse_obj(obj)
+
+        _obj = RunLogInputDto.parse_obj(
+            {
+                "key": obj.get("key"),
+                "value": obj.get("value"),
+                "timestamp": obj.get("timestamp"),
+                "step": obj.get("step") if obj.get("step") is not None else 0,
+                "log_type": obj.get("log_type"),
+                "artifact_path": obj.get("artifact_path"),
+            }
+        )
+        return _obj