truefoundry 0.5.0rc3__py3-none-any.whl → 0.5.0rc5__py3-none-any.whl

truefoundry/ml/log_types/artifacts/model.py

@@ -4,6 +4,7 @@ import json
 import logging
 import os.path
 import tempfile
+import typing
 import uuid
 import warnings
 from pathlib import Path
@@ -15,25 +16,23 @@ from truefoundry.ml.artifact.truefoundry_artifact_repo import (
 )
 from truefoundry.ml.autogen.client import (  # type: ignore[attr-defined]
     ArtifactType,
-    ArtifactVersionSerializationFormat,
     CreateArtifactVersionRequestDto,
     DeleteArtifactVersionsRequestDto,
-    ExternalModelSource,
+    ExternalArtifactSource,
     FinalizeArtifactVersionRequestDto,
     Framework,
+    Manifest,
     MlfoundryArtifactsApi,
     ModelDto,
     ModelVersionDto,
     ModelVersionManifest,
     NotifyArtifactVersionFailureDto,
-    TruefoundryModelSource,
+    TrueFoundryArtifactSource,
     UpdateModelVersionRequestDto,
 )
-from truefoundry.ml.autogen.client import (
-    Source as ModelVersionSource,
-)
 from truefoundry.ml.enums import ModelFramework
 from truefoundry.ml.exceptions import MlFoundryException
+from truefoundry.ml.log_types.artifacts.artifact import BlobStorageDirectory
 from truefoundry.ml.log_types.artifacts.constants import (
     INTERNAL_METADATA_PATH,
 )
@@ -46,7 +45,7 @@ from truefoundry.ml.log_types.artifacts.utils import (
 )
 from truefoundry.ml.model_framework import ModelFrameworkType, _ModelFramework
 from truefoundry.ml.session import _get_api_client
-from truefoundry.pydantic_v1 import BaseModel, Extra, StrictStr
+from truefoundry.pydantic_v1 import BaseModel, Extra
 
 if TYPE_CHECKING:
     from truefoundry.ml.mlfoundry_run import MlFoundryRun
@@ -75,10 +74,6 @@ class ModelVersionInternalMetadata(BaseModel):
         return dct
 
 
-class BlobStorageModelDirectory(BaseModel):
-    uri: StrictStr
-
-
 class ModelVersionDownloadInfo(BaseModel):
     download_dir: str
     model_dir: str
@@ -101,7 +96,8 @@ class ModelVersion:
         self._deleted = False
         self._description: str = ""
         self._metadata: Dict[str, Any] = {}
-        self._framework: Optional["Framework"] = None
+        self._model_schema: Optional[Dict[str, Any]] = None
+        self._framework: Optional[ModelFrameworkType] = None
         self._set_mutable_attrs()
 
     @classmethod
@@ -140,20 +136,33 @@ class ModelVersion:
             )
 
     def _set_mutable_attrs(self):
-        self._description = self._model_version.description or ""
-        self._metadata = copy.deepcopy(self._model_version.artifact_metadata)
-        self._framework = (
-            copy.deepcopy(self._model_version.manifest.framework)
-            if self._model_version.manifest
-            else None
-        )
+        if self._model_version.manifest:
+            self._description = self._model_version.manifest.description or ""
+            self._metadata = copy.deepcopy(self._model_version.manifest.metadata)
+            self._model_schema = copy.deepcopy(
+                self._model_version.manifest.model_schema
+            )
+            if self._model_version.manifest.framework:
+                self._framework = copy.deepcopy(
+                    self._model_version.manifest.framework.actual_instance
+                )
+            else:
+                self._framework = None
+        else:
+            self._description = self._model_version.description or ""
+            self._metadata = copy.deepcopy(self._model_version.artifact_metadata)
+            self._framework = _ModelFramework.to_model_framework_type(
+                self._model_version.model_framework
+            )
+            self._model_schema = None
 
-    def _refetch_model_version(self):
+    def _refetch_model_version(self, reset_mutable_attrs: bool = True):
         _model_version = self._mlfoundry_artifacts_api.get_model_version_get(
             id=self._model_version.id
         )
         self._model_version = _model_version.model_version
-        self._set_mutable_attrs()
+        if reset_mutable_attrs:
+            self._set_mutable_attrs()
 
     def __repr__(self):
         return f"{self.__class__.__name__}(fqn={self.fqn!r})"
@@ -187,12 +196,14 @@ class ModelVersion:
         return self._model_version.fqn
 
     @property
-    def step(self) -> int:
+    def step(self) -> Optional[int]:
         """Get the step in which model was created"""
+        if self._model_version.manifest:
+            return self._model_version.manifest.step
         return self._model_version.step
 
     @property
-    def description(self) -> Optional[str]:
+    def description(self) -> str:
         """Get description of the model"""
         return self._description
 
@@ -201,8 +212,6 @@ class ModelVersion:
         """set the description of the model"""
         _validate_description(value)
         self._description = value
-        if self._model_version.manifest:
-            self._model_version.manifest.description = value
 
     @property
     def metadata(self) -> Dict[str, Any]:
@@ -213,52 +222,51 @@ class ModelVersion:
     def metadata(self, value: Dict[str, Any]):
         """set the metadata for current model"""
         _validate_artifact_metadata(value)
-        self._metadata = value
-        if self._model_version.manifest:
-            self._model_version.manifest.metadata = value
+        self._metadata = copy.deepcopy(value)
+
+    @property
+    def model_schema(self) -> Optional[Dict[str, Any]]:
+        """Get model_schema for the current model"""
+        return self._model_schema
+
+    @model_schema.setter
+    def model_schema(self, value: Optional[Dict[str, Any]]):
+        """set the model_schema for current model"""
+        if not self._model_version.manifest:
+            warnings.warn(
+                message="This model version was created using an older serialization format. model_schema will not be updated",
+                category=DeprecationWarning,
+                stacklevel=2,
+            )
+            return
+        self._model_schema = copy.deepcopy(value)
 
     @property
     def framework(self) -> Optional["ModelFrameworkType"]:
         """Get the framework of the model"""
-        return (
-            _ModelFramework.from_dict(self._framework.actual_instance.dict())
-            if self._framework and self._framework.actual_instance
-            else None
-        )
+        return self._framework
 
     @framework.setter
     def framework(
         self, value: Optional[Union[str, ModelFramework, "ModelFrameworkType"]]
     ):
-        """set the framework of the model"""
-        model_framework_type: Optional[ModelFrameworkType] = None
-
-        if value is not None:
-            model_framework_type = _ModelFramework.to_model_framework_type(value)
-
-        if (
-            self._model_version.serialization_format
-            == ArtifactVersionSerializationFormat.V2
-        ):
-            self._model_version.manifest.framework = self._framework = (
-                Framework.from_dict(model_framework_type.dict())
-                if model_framework_type
-                else None
-            )
-        else:
+        """Set the framework of the model"""
+        if not self._model_version.manifest:
             warnings.warn(
                 message="This model version was created using an older serialization format. Framework will not be updated",
                 category=DeprecationWarning,
                 stacklevel=2,
             )
+            return
+        self._framework = _ModelFramework.to_model_framework_type(value)
 
     @property
-    def created_at(self) -> datetime.datetime:
+    def created_at(self) -> Optional[datetime.datetime]:
         """Get the time at which model version was created"""
         return self._model_version.created_at
 
     @property
-    def updated_at(self) -> datetime.datetime:
+    def updated_at(self) -> Optional[datetime.datetime]:
         """Get the information about when the model version was updated"""
         return self._model_version.updated_at
 
@@ -311,13 +319,11 @@ class ModelVersion:
             path=path, overwrite=overwrite, progress=progress
         )
 
-        if (
-            self._model_version.serialization_format
-            == ArtifactVersionSerializationFormat.V2
-        ):
+        if self._model_version.manifest:
+            _framework = self._model_version.manifest.framework
             model_framework = (
-                ModelFramework(self.framework.type)
-                if self.framework
+                ModelFramework(_framework.actual_instance.type)
+                if _framework
                 else ModelFramework.UNKNOWN
             )
             download_info = ModelVersionDownloadInfo(
@@ -429,22 +435,35 @@ class ModelVersion:
             ```
         """
         self._ensure_not_deleted()
-        _model_version = self._mlfoundry_artifacts_api.update_model_version_post(
-            update_model_version_request_dto=UpdateModelVersionRequestDto(
-                id=self._model_version.id,
-                description=self.description,
-                artifact_metadata=self.metadata,
-                manifest=self._model_version.manifest,
+        if self._model_version.manifest:
+            self._model_version.manifest.description = self.description
+            self._model_version.manifest.metadata = self.metadata
+            self._model_version.manifest.model_schema = self.model_schema
+            self._model_version.manifest.framework = (
+                Framework.from_dict(self.framework.dict()) if self.framework else None
             )
-        )
-        self._model_version = _model_version.model_version
-        self._set_mutable_attrs()
+        try:
+            _model_version = self._mlfoundry_artifacts_api.update_model_version_post(
+                update_model_version_request_dto=UpdateModelVersionRequestDto(
+                    id=self._model_version.id,
+                    description=self.description,
+                    artifact_metadata=self.metadata,
+                    manifest=self._model_version.manifest,
+                )
+            )
+        except Exception:
+            # rollback edits to internal object
+            self._refetch_model_version(reset_mutable_attrs=False)
+            raise
+        else:
+            self._model_version = _model_version.model_version
+            self._set_mutable_attrs()
 
 
 def _log_model_version(  # noqa: C901
     run: Optional["MlFoundryRun"],
     name: str,
-    model_file_or_folder: Union[str, BlobStorageModelDirectory],
+    model_file_or_folder: Union[str, BlobStorageDirectory],
     mlfoundry_artifacts_api: Optional[MlfoundryArtifactsApi] = None,
     ml_repo_id: Optional[str] = None,
     additional_files: Sequence[Tuple[Union[str, Path], Optional[str]]] = (),
@@ -453,15 +472,16 @@ def _log_model_version(  # noqa: C901
     step: Optional[int] = 0,
     progress: Optional[bool] = None,
     framework: Optional[Union[str, ModelFramework, "ModelFrameworkType"]] = None,
+    model_schema: Optional[Dict[str, Any]] = None,
 ) -> ModelVersion:
     if (run and mlfoundry_artifacts_api) or (not run and not mlfoundry_artifacts_api):
         raise MlFoundryException(
             "Exactly one of run, mlfoundry_artifacts_api should be passed"
         )
 
-    if not isinstance(model_file_or_folder, (str, BlobStorageModelDirectory)):
+    if not isinstance(model_file_or_folder, (str, BlobStorageDirectory)):
         raise MlFoundryException(
-            "model_file_or_folder should be of type str or BlobStorageModelDirectory"
+            "model_file_or_folder should be of type str or BlobStorageDirectory"
         )
 
     if mlfoundry_artifacts_api and not ml_repo_id:
@@ -522,11 +542,9 @@ def _log_model_version(  # noqa: C901
     version_id = _create_artifact_version_response.id
     artifact_storage_root = _create_artifact_version_response.artifact_storage_root
     if isinstance(model_file_or_folder, str):
-        # Source is of type TruefoundryModelSource
-        source = ModelVersionSource.from_json(
-            TruefoundryModelSource(
-                type="truefoundry", uri=artifact_storage_root
-            ).to_json()
+        # Source is of type TrueFoundryArtifactSource
+        source = TrueFoundryArtifactSource(
+            type="truefoundry", uri=artifact_storage_root
         )
         artifacts_repo = MlFoundryArtifactsRepository(
             artifact_identifier=ArtifactIdentifier(
@@ -555,20 +573,20 @@ def _log_model_version(  # noqa: C901
             raise MlFoundryException("Failed to log model") from e
         finally:
             temp_dir.cleanup()
-    elif isinstance(model_file_or_folder, BlobStorageModelDirectory):
-        source = ModelVersionSource.from_json(
-            ExternalModelSource(type="external", uri=model_file_or_folder.uri).to_json()
-        )
+    elif isinstance(model_file_or_folder, BlobStorageDirectory):
+        source = ExternalArtifactSource(type="external", uri=model_file_or_folder.uri)
     else:
         raise MlFoundryException("Invalid model_file_or_folder provided")
 
     _framework = _ModelFramework.to_model_framework_type(framework)
+    _source_cls = typing.get_type_hints(ModelVersionManifest)["source"]
     model_manifest = ModelVersionManifest(
         description=description,
         metadata=metadata,
-        source=source,
+        source=_source_cls.from_dict(source.dict()),
         framework=Framework.from_dict(_framework.dict()) if _framework else None,
-        step=step if run else 0,
+        step=step,
+        model_schema=model_schema,
     )
     artifact_version_response = mlfoundry_artifacts_api.finalize_artifact_version_post(
         finalize_artifact_version_request_dto=FinalizeArtifactVersionRequestDto(
@@ -578,7 +596,7 @@ def _log_model_version(  # noqa: C901
             artifact_metadata=metadata,
             internal_metadata=None,
             step=model_manifest.step,
-            manifest=model_manifest,
+            manifest=Manifest.from_dict(model_manifest.to_dict()),
         )
     )
     return ModelVersion.from_fqn(fqn=artifact_version_response.artifact_version.fqn)

truefoundry/ml/mlfoundry_api.py

@@ -48,12 +48,12 @@ from truefoundry.ml.internal_namespace import NAMESPACE
 from truefoundry.ml.log_types.artifacts.artifact import (
     ArtifactPath,
     ArtifactVersion,
+    BlobStorageDirectory,
     ChatPromptVersion,
 )
 from truefoundry.ml.log_types.artifacts.dataset import DataDirectory
 from truefoundry.ml.log_types.artifacts.general_artifact import _log_artifact_version
 from truefoundry.ml.log_types.artifacts.model import (
-    BlobStorageModelDirectory,
     ModelVersion,
     _log_model_version,
 )
@@ -1226,12 +1226,13 @@ class MlFoundry:
         *,
         ml_repo: str,
         name: str,
-        model_file_or_folder: Union[str, BlobStorageModelDirectory],
+        model_file_or_folder: Union[str, BlobStorageDirectory],
         additional_files: Sequence[Tuple[Union[str, Path], Optional[str]]] = (),
         description: Optional[str] = None,
         metadata: Optional[Dict[str, Any]] = None,
         progress: Optional[bool] = None,
         framework: Optional[Union[str, ModelFramework, "ModelFrameworkType"]] = None,
+        model_schema: Optional[Dict[str, Any]] = None,
     ) -> ModelVersion:
         """
         Serialize and log a versioned model under the current ml_repo. Each logged model generates a new version
@@ -1244,12 +1245,12 @@ class MlFoundry:
                 the logged model will be added as a new version under that `name`. If no models exist with the given
                 `name`, the given model will be logged as version 1.
 
-            model_file_or_folder (Union[str, BlobStorageModelDirectory]):
+            model_file_or_folder (Union[str, BlobStorageDirectory]):
                 str:
                     Path to either a single file or a folder containing model files.
                     This folder is typically created using serialization methods from libraries or frameworks,
                     e.g., `joblib.dump`, `model.save_pretrained(...)`, `torch.save(...)`, or `model.save(...)`.
-                BlobStorageModelDirectory:
+                BlobStorageDirectory:
                     uri (str): URI to the model file or folder in a storage integration associated with the specified ML Repo.
                         The model files or folder must reside within the same storage integration as the specified ML Repo.
                         Accepted URI formats include `s3://integration-bucket-name/prefix/path/to/model` or `gs://integration-bucket-name/prefix/path/to/model`.
@@ -1391,6 +1392,7 @@ class MlFoundry:
             step=None,
             progress=progress,
             framework=framework,
+            model_schema=model_schema,
         )
         logger.info(f"Logged model successfully with fqn {model_version.fqn!r}")
         return model_version

truefoundry/ml/mlfoundry_run.py

@@ -44,10 +44,13 @@ from truefoundry.ml.enums import ModelFramework, RunStatus
 from truefoundry.ml.exceptions import MlFoundryException
 from truefoundry.ml.internal_namespace import NAMESPACE
 from truefoundry.ml.log_types import Image, Plot
-from truefoundry.ml.log_types.artifacts.artifact import ArtifactPath, ArtifactVersion
+from truefoundry.ml.log_types.artifacts.artifact import (
+    ArtifactPath,
+    ArtifactVersion,
+    BlobStorageDirectory,
+)
 from truefoundry.ml.log_types.artifacts.general_artifact import _log_artifact_version
 from truefoundry.ml.log_types.artifacts.model import (
-    BlobStorageModelDirectory,
     ModelVersion,
     _log_model_version,
 )
@@ -926,13 +929,14 @@ class MlFoundryRun:
         self,
         *,
         name: str,
-        model_file_or_folder: Union[str, BlobStorageModelDirectory],
+        model_file_or_folder: Union[str, BlobStorageDirectory],
         additional_files: Sequence[Tuple[Union[str, Path], Optional[str]]] = (),
         description: Optional[str] = None,
         metadata: Optional[Dict[str, Any]] = None,
         step: int = 0,
         progress: Optional[bool] = None,
         framework: Optional[Union[str, ModelFramework, "ModelFrameworkType"]] = None,
+        model_schema: Optional[Dict[str, Any]] = None,
     ) -> ModelVersion:
         # TODO (chiragjn): Document mapping of framework to list of valid model save kwargs
         # TODO (chiragjn): Add more examples
@@ -945,12 +949,12 @@ class MlFoundryRun:
             name (str): Name of the model. If a model with this name already exists under the current ML Repo,
                 the logged model will be added as a new version under that `name`. If no models exist with the given
                 `name`, the given model will be logged as version 1.
-            model_file_or_folder (Union[str, BlobStorageModelDirectory]):
+            model_file_or_folder (Union[str, BlobStorageDirectory]):
                 str:
                     Path to either a single file or a folder containing model files.
                     This folder is typically created using serialization methods from libraries or frameworks,
                     e.g., `joblib.dump`, `model.save_pretrained(...)`, `torch.save(...)`, or `model.save(...)`.
-                BlobStorageModelDirectory:
+                BlobStorageDirectory:
                     uri (str): URI to the model file or folder in a storage integration associated with the specified ML Repo.
                         The model files or folder must reside within the same storage integration as the specified ML Repo.
                         Accepted URI formats include `s3://integration-bucket-name/prefix/path/to/model` or `gs://integration-bucket-name/prefix/path/to/model`.
@@ -1085,6 +1089,7 @@ class MlFoundryRun:
             step=step,
             progress=progress,
             framework=framework,
+            model_schema=model_schema,
         )
         logger.info(f"Logged model successfully with fqn {model_version.fqn!r}")
         return model_version

truefoundry/ml/model_framework.py

@@ -1,5 +1,5 @@
 import warnings
-from typing import Literal, Optional, Union, get_args
+from typing import Any, Dict, Literal, Optional, Union, get_args
 
 from truefoundry.ml import ModelFramework
 from truefoundry.ml.autogen.entities import artifacts as autogen_artifacts
@@ -161,7 +161,7 @@ class _ModelFramework(BaseModel):
         )
 
     @classmethod
-    def from_dict(cls, obj: dict) -> ModelFrameworkType:
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[ModelFrameworkType]:
         """Create an instance of ModelFramework from a dict"""
         if obj is None:
             return None

{truefoundry-0.5.0rc3.dist-info → truefoundry-0.5.0rc5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: truefoundry
-Version: 0.5.0rc3
+Version: 0.5.0rc5
 Summary: Truefoundry CLI
 Author: Abhishek Choudhary
 Author-email: abhishek@truefoundry.com
@@ -21,7 +21,7 @@ Requires-Dist: coolname (>=1.1.0,<2.0.0)
 Requires-Dist: docker (>=6.1.2,<8.0.0)
 Requires-Dist: fastapi (>=0.56.0,<0.200.0)
 Requires-Dist: filelock (>=3.8.0,<4.0.0)
-Requires-Dist: flytekit (==1.12.2) ; extra == "workflow"
+Requires-Dist: flytekit (==1.13.13) ; extra == "workflow"
 Requires-Dist: gitignorefile (>=1.1.2,<2.0.0)
 Requires-Dist: importlib-metadata (>=4.11.3,<9.0.0)
 Requires-Dist: importlib-resources (>=5.2.0,<7.0.0)

{truefoundry-0.5.0rc3.dist-info → truefoundry-0.5.0rc5.dist-info}/RECORD

@@ -111,7 +111,7 @@ truefoundry/deploy/python_deploy_codegen.py,sha256=qJHH1BJQII9e6PhkcRFYiE_3De7_V
 truefoundry/deploy/v2/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/deploy/v2/lib/__init__.py,sha256=WEiVMZXOVljzEE3tpGJil14liIn_PCDoACJ6b3tZ6sI,188
 truefoundry/deploy/v2/lib/deploy.py,sha256=HIcY3SzQ5lWl7avuuKi3J0Z-PBES6Sf4hgMK-m6_53U,11990
-truefoundry/deploy/v2/lib/deploy_workflow.py,sha256=WhlrBuRf7r83qoQdTZSQzHt635fw9D4_qQIqusFWGag,12372
+truefoundry/deploy/v2/lib/deploy_workflow.py,sha256=hgAhd1EGwFLz319Vs-WNXHDJmbKjdgkGPzDnBD1Up1k,12579
 truefoundry/deploy/v2/lib/deployable_patched_models.py,sha256=MROgMxhn9hDEAKwJSWl3iz12tUVvRKzEtqF2QUT6dAk,3343
 truefoundry/deploy/v2/lib/models.py,sha256=pSolLMTArDuYpeNsmeeS5DWliloN_iCDfZSpRllMHUg,1120
 truefoundry/deploy/v2/lib/patched_models.py,sha256=sokVDUdnhe3qx6dXlHM0shbf6HvSlF72-mvi8Lzt_Y8,13968
@@ -123,11 +123,11 @@ truefoundry/langchain/truefoundry_embeddings.py,sha256=8nRaZ7W1ao1WF0LHk6nNel1Lu
 truefoundry/langchain/truefoundry_llm.py,sha256=CJXyCgXIMbDsVRuuvEA5PKJsf6aRyVlYuG7zC4qtZXE,3802
 truefoundry/langchain/utils.py,sha256=PGLDe9chZ3BuUjakexOGpIqZRFoHEgu-zJ9yKdpLLmM,1329
 truefoundry/logger.py,sha256=u-YCNjg5HBwE70uQcpjIG64Ghos-K2ulTWaxC03BSj4,714
-truefoundry/ml/__init__.py,sha256=ZMT9C7q9YnwfnfruI0xipizSo1OO__f8Ao-oE65DZyg,1836
+truefoundry/ml/__init__.py,sha256=tcFeEcIgBLR92pUDiu7zDSeI1vsSshlCzuIgnSbyE8E,1839
 truefoundry/ml/artifact/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/ml/artifact/truefoundry_artifact_repo.py,sha256=VU8i3jnY62MLfzA3rxXuUjdqLz8Yaw4zqqPWSsf0mBg,45850
 truefoundry/ml/autogen/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/ml/autogen/client/__init__.py,sha256=0-I7KuPudFU1I1Y4qPbkvAK54E7wzjwIdjnVB57fWGA,18958
+truefoundry/ml/autogen/client/__init__.py,sha256=OD4hsjMELkJNFkpVzcRGX_0tEIfWxL2n_0XYuy8mMoo,19211
 truefoundry/ml/autogen/client/api/__init__.py,sha256=NyMBxBmIzB1o5LzZZwz9LiydHB3-hPqc_sevsnY6Jrw,746
 truefoundry/ml/autogen/client/api/auth_api.py,sha256=zpWzJhUmW6HHMY_atlUf0B25k77E1kue2hmix5I5Ih0,7017
 truefoundry/ml/autogen/client/api/deprecated_api.py,sha256=0qooHNzm5mgweKUEDjnESHPeUqrlzuNEsDJwW9Cm9Vs,24232
@@ -142,7 +142,7 @@ truefoundry/ml/autogen/client/api_client.py,sha256=8qg-WpadDuKgbRt5yABJ4wVS4IRxd
 truefoundry/ml/autogen/client/api_response.py,sha256=KRyvecPMXF05PaxILHZ8JHoP4rgKBjKONMgG83aU-rM,844
 truefoundry/ml/autogen/client/configuration.py,sha256=V1oaEnxt-NfpaNmp-EZpf2glovzVhM2coWYt8HBNB4M,15723
 truefoundry/ml/autogen/client/exceptions.py,sha256=XbCbDHhYT3BVejdoGNPgEa4oS56ypkwFdxk1iOc_tFY,5355
-truefoundry/ml/autogen/client/models/__init__.py,sha256=ZNjoIgoxY6uxmChlRP7q5ea1g2EZvNARDvHKIYujGXU,17580
+truefoundry/ml/autogen/client/models/__init__.py,sha256=IH4Cx1qYoYds5nkThxh9yMShRYZ5mGwn5fFFyPvnG6Y,17833
 truefoundry/ml/autogen/client/models/add_custom_metrics_to_model_version_request_dto.py,sha256=_ISDspicTGjBCYYXubKfRYYSSQVyW3AvG-jFh47-Zfc,2163
 truefoundry/ml/autogen/client/models/add_features_to_model_version_request_dto.py,sha256=rU0h96pEE8K1Ukw2pzDSjq0e6BgtDEuOctI-aZMrpUY,2653
 truefoundry/ml/autogen/client/models/agent.py,sha256=fnMWdEPe5Iw50WKydtu7QAxU419j3ju1IukpChUnGqY,3871
@@ -153,7 +153,8 @@ truefoundry/ml/autogen/client/models/agent_with_fqn.py,sha256=_LFRmdT0KNe1CRNBfE
 truefoundry/ml/autogen/client/models/artifact_dto.py,sha256=JWL8_sITJJXbxywcfRj4xqt8jR9pyBmA4TfwNTdX6CE,4243
 truefoundry/ml/autogen/client/models/artifact_response_dto.py,sha256=T_HLjkHnqiO1bnSXiCUOlwDEHGz89mlUNhBq0SJTapo,2226
 truefoundry/ml/autogen/client/models/artifact_type.py,sha256=LBXGYUoesmL5J-0xzCR8jBST3bjb6HY86RmqSGvytRs,828
-truefoundry/ml/autogen/client/models/artifact_version_dto.py,sha256=oGWsRSxLDy8aGMI9dfaaNZH9JpPnqLevWSRmMCsHo44,6233
+truefoundry/ml/autogen/client/models/artifact_version_dto.py,sha256=GXjjKycpsEgJ60V2hAfL4NMxM2gZKLP2qV2yBb9WgdM,6174
+truefoundry/ml/autogen/client/models/artifact_version_manifest.py,sha256=xqznaihmb9voR_-GqQ3vlDG5ZxvYhjjWCP9nR-WmIIE,3531
 truefoundry/ml/autogen/client/models/artifact_version_response_dto.py,sha256=JmiSYdPYG-ki6LCBeF7NtihdQ2LPiHld8cSPivY2GmQ,2421
 truefoundry/ml/autogen/client/models/artifact_version_serialization_format.py,sha256=2Qfgq1nykn0_mKNEJzOxhQwHm_95WIr9wJcHLeFWXQM,774
 truefoundry/ml/autogen/client/models/artifact_version_status.py,sha256=iIcomqhuaJkZH2rT3e6IjQutM79t9hEshW-RCIMKHEY,780
@@ -198,12 +199,12 @@ truefoundry/ml/autogen/client/models/experiment_dto.py,sha256=bl4-Hp1EHl8CEKgWq6
 truefoundry/ml/autogen/client/models/experiment_id_request_dto.py,sha256=ZMH827n_UTpDI30UnkuOam-4ANBKCDgocIzI8StxFR8,1891
 truefoundry/ml/autogen/client/models/experiment_response_dto.py,sha256=wuflV6_f8PQq061-wU2GzNY4BZi8SG8ARCIbSQN1oT4,2268
 truefoundry/ml/autogen/client/models/experiment_tag_dto.py,sha256=nEpCkeZ9ficIDkjmmLfkJeNNokd-Rhgr-cepPWG6L3M,1902
-truefoundry/ml/autogen/client/models/external_model_source.py,sha256=WE3JoHmaCDIeoHpcSOtkDsNocBoaTMzqiIuAiq6qIro,2336
+truefoundry/ml/autogen/client/models/external_artifact_source.py,sha256=s7cDavSpmlCblMVV1DjSWur258YHYPxVjWze3uQTjXY,2360
 truefoundry/ml/autogen/client/models/fast_ai_framework.py,sha256=pl--nyTJF687SNmX0v1ICISmUCVc6ekF8VH_Rl1kHEQ,2170
 truefoundry/ml/autogen/client/models/feature_dto.py,sha256=XM7fF71P_bYP3UlO1ILjnD3DJNpf2tGwAWbrkQyvyA4,1924
 truefoundry/ml/autogen/client/models/feature_value_type.py,sha256=3ZKxxEfrrWk8ePlV7n11wa-FdDH0SRLyHRqSvIU7F-M,732
 truefoundry/ml/autogen/client/models/file_info_dto.py,sha256=7oc8venicsFVk8zT9wHNhHnZGtFkFlqimFnS7ozGL9k,2156
-truefoundry/ml/autogen/client/models/finalize_artifact_version_request_dto.py,sha256=tArvzRPp80PpfJ4CoMwwaKqPAfzonq-8C-yKBEjs9co,3877
+truefoundry/ml/autogen/client/models/finalize_artifact_version_request_dto.py,sha256=tkPcNjI0_qhX-D3m6AXqLzdmo9_K0tFBtbPDeg0Zo7U,3818
 truefoundry/ml/autogen/client/models/framework.py,sha256=W_KrqH1qKRvlrssCCWFIlmPaeAw0eiM6VqS3pbdlMXU,14720
 truefoundry/ml/autogen/client/models/get_experiment_response_dto.py,sha256=FhJpoUeRbZV1s1KR3Y07Kv6U4enLbbgdpGbguT9us0A,2923
 truefoundry/ml/autogen/client/models/get_latest_run_log_response_dto.py,sha256=g5NG252VzjGHiA-w5nhxmZxGrowJO9XCHDQTjf8y2KE,2266
@@ -251,6 +252,7 @@ truefoundry/ml/autogen/client/models/list_run_logs_response_dto.py,sha256=ckQBBY
 truefoundry/ml/autogen/client/models/log_batch_request_dto.py,sha256=xDas-pmJP5T-8zKd17BbSnglZuX15ei3n9blBZUxa-Y,3702
 truefoundry/ml/autogen/client/models/log_metric_request_dto.py,sha256=QED1aIeaZJG-8iKFxT3NX9ozlUz_wom1xDsyZgL-a-4,2442
 truefoundry/ml/autogen/client/models/log_param_request_dto.py,sha256=Hd7AqrVnSkwfa9CcDbATERaim5Hf8rM59WZQNxSlJzk,2160
+truefoundry/ml/autogen/client/models/manifest.py,sha256=PosK8HiHt54GARPhLecmMcDQrzCgPJhReV4UHArj19w,5078
 truefoundry/ml/autogen/client/models/method.py,sha256=NepwKtC72Wn0b4kkYM9uek6lZ55erqFEtlASOn1fYbU,734
 truefoundry/ml/autogen/client/models/metric_collection_dto.py,sha256=02ZWKm2_8DjGBhrPMysTACE2SQpyz4BSVMY01-k_qEA,2507
 truefoundry/ml/autogen/client/models/metric_dto.py,sha256=Xrcc8XRY0sZeg_cnb6gyf-AUUIS0FPxSGTCbDzcBzXQ,2174
@@ -260,7 +262,7 @@ truefoundry/ml/autogen/client/models/model_dto.py,sha256=TUO74MDqe8XCVJBB2O7Tiqi
 truefoundry/ml/autogen/client/models/model_response_dto.py,sha256=osrTxfygkuhxWj6SkRBALrSnFVPH4LSK6qTufTeZuJg,2163
 truefoundry/ml/autogen/client/models/model_schema_dto.py,sha256=ElEPK7fwuf3eiohNBBHJyl-nhNqzGnDLzZVzuQl-xC0,2620
 truefoundry/ml/autogen/client/models/model_version_dto.py,sha256=6kI1Z2czYAieJZBaVvcV8Y2eeNEUDyjPuTLHTdyWnOE,7282
-truefoundry/ml/autogen/client/models/model_version_manifest.py,sha256=Jd8oJj9-awzg-jP_hhLEMH3PSXir2p5Kc5o0YEB9ZRA,3957
+truefoundry/ml/autogen/client/models/model_version_manifest.py,sha256=jPE3JYVvv8Wmqc4-w4Ff10HlqqxXaK9EQVkIgN7h9j4,4262
 truefoundry/ml/autogen/client/models/model_version_response_dto.py,sha256=D6XOCyggxqTkbePuypqYSHYh1PYeDv7R_J32q61sDM0,2320
 truefoundry/ml/autogen/client/models/multi_part_upload_dto.py,sha256=Ckq405vud8RQmMyKCJQJBlW5iXO7Y2mlgo8eVkiMvxg,3738
 truefoundry/ml/autogen/client/models/multi_part_upload_response_dto.py,sha256=VjH0kvl7rMjgDHjYGHnsh7KsZ5-qn-k3ksdGLJ49nIM,2431
@@ -288,7 +290,8 @@ truefoundry/ml/autogen/client/models/set_experiment_tag_request_dto.py,sha256=nr
 truefoundry/ml/autogen/client/models/set_tag_request_dto.py,sha256=IRgAdMcWBxmjNV6nZJej4pcNfLmZwrelEZ3otwt7eeE,2144
 truefoundry/ml/autogen/client/models/signed_url_dto.py,sha256=9oHoXBj07xTdc04rqOqJO3eOjQWXCyWPhfHg-6qX60w,1897
 truefoundry/ml/autogen/client/models/sklearn_framework.py,sha256=8-GvfwN5fHVDiOXPvVa9rBeG0VEavwwTsfEp3Nl0uO8,2187
-truefoundry/ml/autogen/client/models/source.py,sha256=8vQ6Fcky8kYAQiATuphxuhxKGRhH4WmcTiUa9N8iZQM,5095
+truefoundry/ml/autogen/client/models/source.py,sha256=ZG2-3oIs_M8mZNPjoIRPzhHqlm63vLCVQC9idUxQ0XU,5180
+truefoundry/ml/autogen/client/models/source1.py,sha256=io35b9uVzqSvOHw6RlCXxM30thN-vHm1upaSiN2i0II,5183
 truefoundry/ml/autogen/client/models/spa_cy_framework.py,sha256=JM1DLF8JL4c7-JZN-dQX_7NF8jIS6RP-mEk1uoWQpaU,2149
 truefoundry/ml/autogen/client/models/stats_models_framework.py,sha256=INndD-nZFAfRbKZXss4erx4xtotb_hlRFlEvitKrIzQ,2230
 truefoundry/ml/autogen/client/models/stop.py,sha256=UlAjRWI6Gjc92D2UME6qRAIfd6GoDI3VBWYkJqgixmU,4760
@@ -299,11 +302,11 @@ truefoundry/ml/autogen/client/models/system_message.py,sha256=yxRCJPl36FGv2zrsuH
 truefoundry/ml/autogen/client/models/tensor_flow_framework.py,sha256=lvABZqmVED7Hg7aDZ7bovN6usG_TZCWI4B8Ew6NAymw,2218
 truefoundry/ml/autogen/client/models/text.py,sha256=HAIK-w_TxB1kXGvVNK1d2cweoQi-GJ9d-0QSZl21u68,4862
 truefoundry/ml/autogen/client/models/text_content_part.py,sha256=fRQu22tiV3rDKmTgVZSLHX-cMcha3q53W8qWNCk5wmU,2492
-truefoundry/ml/autogen/client/models/transformers_framework.py,sha256=ialS6K3F64iLSe19qVXeuXNctEq6IzMFi7cBhP00bdg,2903
+truefoundry/ml/autogen/client/models/transformers_framework.py,sha256=0lbFboK0hfHxZGLRiLAhX0imn-EAJHunFWDORF6-zSY,3203
 truefoundry/ml/autogen/client/models/trigger_job_run_config_request_dto.py,sha256=1HtO6zfW6Ulq8Gm_UL0YqVrse0795vZkMxgCQ0GKD9w,2864
 truefoundry/ml/autogen/client/models/trigger_job_run_config_response_dto.py,sha256=P35xzE8MyS_HtxCcXC9fZORPX2wbf93LbCZCRggkQxw,2144
-truefoundry/ml/autogen/client/models/truefoundry_model_source.py,sha256=Jc3sP2n05kQz92nHaU5RCcwP9bfeIBFUuxfvwdojub8,2418
-truefoundry/ml/autogen/client/models/update_artifact_version_request_dto.py,sha256=AhXj2afmrKskBi5uMT0-mSywZyz3g17lFE3uMRQ5pEM,2246
+truefoundry/ml/autogen/client/models/true_foundry_artifact_source.py,sha256=Qwtowy8XSqMgQQ9yEk9AUL1ghaETLbwQW9sGNQfh6D0,2456
+truefoundry/ml/autogen/client/models/update_artifact_version_request_dto.py,sha256=nBcEe5aCIvqKz_iIeegkDKsOedSTq6-ahlC_GAwxkO4,2766
 truefoundry/ml/autogen/client/models/update_dataset_request_dto.py,sha256=MF-rjJGUP6sDZVH3xP4Q8AEC9CV2PXEO9cXrR6yK57A,2182
 truefoundry/ml/autogen/client/models/update_experiment_request_dto.py,sha256=QE_kZzTDdiXKtQ2U-zbI_gb5vxS5Yv9YupYvD22RkFs,2198
 truefoundry/ml/autogen/client/models/update_model_version_request_dto.py,sha256=0xWqHHflnpO8FyT79psPiPBIGW6Aiovs-b6H7sjMSDQ,3589
@@ -315,8 +318,8 @@ truefoundry/ml/autogen/client/models/validation_error.py,sha256=mFjwoc8g2-Usu1HX
 truefoundry/ml/autogen/client/models/validation_error_loc_inner.py,sha256=nThJ5Gmy8W2Wok-ZOI4sK7uRe1BAkLS0qzq-XZbq8zs,4915
 truefoundry/ml/autogen/client/models/xg_boost_framework.py,sha256=kXijWcmEd5mc6Rks0mYQLqgYp1v7bxp31HQAmaJLZ3s,2182
 truefoundry/ml/autogen/client/rest.py,sha256=9goba8qHjQuVx5O_yRaTKu7PvBnb7r7swfy3dwuTEgk,14281
-truefoundry/ml/autogen/client_README.md,sha256=h8EUUiMl1vHri3Jrt0PjkFD-CoXAf5WXSgG4iRqlouc,35564
-truefoundry/ml/autogen/entities/artifacts.py,sha256=qc4XW3X8BR-zE32tNbCBMJ5kc_aPehDOBhyugvAYY-8,16646
+truefoundry/ml/autogen/client_README.md,sha256=nIyCT1OXBMLV_cadu0OIqQCtXmGm3frJnO0gbJ_7b9Q,35790
+truefoundry/ml/autogen/entities/artifacts.py,sha256=EbGDmD0npfa5BSHau67Y8-dSk6qPJJlJX3I3N4kqAgA,17326
 truefoundry/ml/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/ml/cli/cli.py,sha256=ckBcjUpqfhgrPE1okqT_G2iouOLt-0KjpLhHp2YdVFU,256
 truefoundry/ml/cli/commands/__init__.py,sha256=diDUiRUX4l6TtNLI4iF-ZblczkELM7FRViJ-8gGNJQY,82
@@ -332,11 +335,11 @@ truefoundry/ml/exceptions.py,sha256=8aJm2NYtAWWsRLu4MbzaoOqHsQZ6RjOFwBWQWqb6qrc,
 truefoundry/ml/git_info.py,sha256=jvAVm9ilqivnGq8qJdUvYdd8Siv0PLtqurB-PXsS5ho,2023
 truefoundry/ml/internal_namespace.py,sha256=QcqMHp6-C2im2H_02hlhi01EIcr1HhNaZprszs13EMU,1790
 truefoundry/ml/log_types/__init__.py,sha256=g4u4D4Jaj0aBK5GtrLV88-qThKZR9pSZ17vFEkN-LmM,125
-truefoundry/ml/log_types/artifacts/artifact.py,sha256=n2EwAupOigDcUXQQFIIESRqp7zipLZieec0vLQ2DVRQ,17593
+truefoundry/ml/log_types/artifacts/artifact.py,sha256=6X4lO23DwM7yTgUOY4RRoYMrf9m2AbiBBo2WftS2-dc,20457
 truefoundry/ml/log_types/artifacts/constants.py,sha256=qKxQ5mMvJE4j83BvGW3qNTKunxCiBg_EEjTdgbgJtyE,1036
 truefoundry/ml/log_types/artifacts/dataset.py,sha256=a4dxd2EN8p7Ci-cLGGiDOboN3t0395_XhWE1dmTw1Q4,13112
-truefoundry/ml/log_types/artifacts/general_artifact.py,sha256=_EOlNGMg2PFjlevlD6LEOsaQBBwRWs60PkzJases6tE,3927
-truefoundry/ml/log_types/artifacts/model.py,sha256=fcMc63BKqcbjUygpA6w0SFuoWZDvyxJvlPHyh7bNKv4,21161
+truefoundry/ml/log_types/artifacts/general_artifact.py,sha256=B4XErLr-m6RmQWtxMTu3wlFRFcqSwPYp6J0OL4Ng6L0,3179
+truefoundry/ml/log_types/artifacts/model.py,sha256=PZgqRO8NImvU7yfdjldqJ4Z3ow47GMnSL0A1DeE2DYk,22511
 truefoundry/ml/log_types/artifacts/model_extras.py,sha256=TIE73bLKfwIVzNiVcjmaZ841A70BHBwu4XAM6ZAQRFI,1045
 truefoundry/ml/log_types/artifacts/utils.py,sha256=3FVOYlfcvZiW8vTbMe7Ft1cjNPR2GW7J69M3dyAY-qc,6299
 truefoundry/ml/log_types/image/__init__.py,sha256=fcOq8yQnNj1rkLcPeIjLXBpdA1WIeiPsXOlAAvMxx7M,76
@@ -348,9 +351,9 @@ truefoundry/ml/log_types/plot.py,sha256=HuYvvRA5r8V0xAIuuqMME2IHb9d3SfGHUiuEkOP3
 truefoundry/ml/log_types/pydantic_base.py,sha256=eBlw_AEyAz4iJKDP4zgJOCFWcldwQqpf7FADW1jzIQY,272
 truefoundry/ml/log_types/utils.py,sha256=xjJ21jdPScvFmw3TbVh5NCzbzJwaqiXJyiiT4xxX1EI,335
 truefoundry/ml/logger.py,sha256=VT-BF3BnBYTWVq87O58F0c8uXMu94gYzsiFlGY3_7Ao,458
-truefoundry/ml/mlfoundry_api.py,sha256=eLzbXPSaUWBuIZQKld0r30StOHD-dXiS9zQL_Wq4Zk4,62037
-truefoundry/ml/mlfoundry_run.py,sha256=FDTEBflyKRA7nP8qCm834NyYjpyy9540J_WnagB25oI,46195
-truefoundry/ml/model_framework.py,sha256=YCdRD31rQbPQK-ehNLf1YHuOcPgserFjdPgrrB2Twqg,4737
+truefoundry/ml/mlfoundry_api.py,sha256=_LAXCBu1A-xg8fD3qzjQv9YlghEbSgYe6rQB0YgvXGQ,62111
+truefoundry/ml/mlfoundry_run.py,sha256=jLeDz9raUPKjXGH4PPAFel3Zg030iUWB_QP7FeW0DE0,46282
+truefoundry/ml/model_framework.py,sha256=TosB73eEpTRlsh6v1AY_LnnW_4OoaSlzENnBXqkVRl4,4778
 truefoundry/ml/run_utils.py,sha256=0W208wSLUrbdfk2pjNcZlkUi9bNxG2JORqoe-5rVqHI,2423
 truefoundry/ml/session.py,sha256=F83GTC5WwGBjnJ69Ct8MqMnlutYc56JCc6YhEY1Wl-A,5394
 truefoundry/ml/validation_utils.py,sha256=XBSUd9OoyriWJpT3M5LKz17iWY3yVMr3hM5vdaVjtf0,12082
@@ -371,7 +374,7 @@ truefoundry/workflow/remote_filesystem/tfy_signed_url_client.py,sha256=5mBCIc-ON
 truefoundry/workflow/remote_filesystem/tfy_signed_url_fs.py,sha256=Hf6Dk6Fu6P7DqsK5ULgraf9DStjgigf-kjaRAMBW-RU,8680
 truefoundry/workflow/task.py,sha256=ToitYiKcNzFCtOVQwz1W8sRjbR97eVS7vQBdbgUQtKg,1779
 truefoundry/workflow/workflow.py,sha256=WaTqUjhwfAXDWu4E5ehuwAxrCbDJkoAf1oWmR2E9Qy0,4575
-truefoundry-0.5.0rc3.dist-info/METADATA,sha256=NztODLVhdKXFCu6xs3BKqMuKiLFe9FBw98dKtIwmm5Q,3101
-truefoundry-0.5.0rc3.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-truefoundry-0.5.0rc3.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
-truefoundry-0.5.0rc3.dist-info/RECORD,,
+truefoundry-0.5.0rc5.dist-info/METADATA,sha256=WD9-zZ_Rxg1qbBVQZm_GDVlNvWPzuM2HNC78-Rbg-mk,3102
+truefoundry-0.5.0rc5.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+truefoundry-0.5.0rc5.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
+truefoundry-0.5.0rc5.dist-info/RECORD,,