truefoundry 0.10.5__tar.gz → 0.10.6__tar.gz

{truefoundry-0.10.5 → truefoundry-0.10.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: truefoundry
-Version: 0.10.5
+Version: 0.10.6
 Summary: TrueFoundry CLI
 Author-email: TrueFoundry Team <abhishek@truefoundry.com>
 Requires-Python: <3.14,>=3.8.1

{truefoundry-0.10.5 → truefoundry-0.10.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "truefoundry"
-version = "0.10.5"  # This is a placeholder version, it will be replaced when releasing
+version = "0.10.6"  # This is a placeholder version, it will be replaced when releasing
 requires-python = ">=3.8.1,<3.14"
 description = "TrueFoundry CLI"
 authors = [{ name = "TrueFoundry Team", email = "abhishek@truefoundry.com" }]

{truefoundry-0.10.5 → truefoundry-0.10.6}/truefoundry/__init__.py

@@ -16,11 +16,13 @@ from truefoundry_sdk import (
 from truefoundry._client import client
 from truefoundry.common.warnings import (
     suppress_truefoundry_deprecation_warnings,
+    suppress_truefoundry_ml_autogen_warnings,
     surface_truefoundry_deprecation_warnings,
 )
 from truefoundry.deploy.core import login, logout
 from truefoundry.ml.prompt_utils import render_prompt
 
+suppress_truefoundry_ml_autogen_warnings()
 surface_truefoundry_deprecation_warnings()
 
 __all__ = [

{truefoundry-0.10.5 → truefoundry-0.10.6}/truefoundry/common/warnings.py

@@ -19,3 +19,11 @@ def suppress_truefoundry_deprecation_warnings() -> None:
         "ignore",
         category=TrueFoundryDeprecationWarning,
     )
+
+
+def suppress_truefoundry_ml_autogen_warnings() -> None:
+    """Mute TrueFoundry ML autogen warnings."""
+    warnings.filterwarnings(
+        "ignore",
+        module="truefoundry.ml._autogen",
+    )

{truefoundry-0.10.5 → truefoundry-0.10.6}/truefoundry/ml/log_types/artifacts/artifact.py

@@ -29,7 +29,10 @@ from truefoundry.ml.artifact.truefoundry_artifact_repo import (
     MlFoundryArtifactsRepository,
 )
 from truefoundry.ml.exceptions import MlFoundryException
-from truefoundry.ml.log_types.artifacts.constants import INTERNAL_METADATA_PATH
+from truefoundry.ml.log_types.artifacts.constants import (
+    ARTIFACT_METADATA_TRUEFOUNDRY_KEY,
+    INTERNAL_METADATA_PATH,
+)
 from truefoundry.ml.log_types.artifacts.utils import (
     _get_src_dest_pairs,
     _validate_artifact_metadata,
@@ -455,6 +458,13 @@ def _log_artifact_version_helper(
             experiment_name=ml_repo
         ).experiment.experiment_id
 
+    metadata = metadata or {}
+    if ARTIFACT_METADATA_TRUEFOUNDRY_KEY not in metadata:
+        metadata[ARTIFACT_METADATA_TRUEFOUNDRY_KEY] = {}
+
+    _validate_description(description)
+    _validate_artifact_metadata(metadata)
+
     assert mlfoundry_artifacts_api is not None
     _create_artifact_response = mlfoundry_artifacts_api.create_artifact_version_post(
         create_artifact_version_request_dto=CreateArtifactVersionRequestDto(
@@ -465,7 +475,8 @@ def _log_artifact_version_helper(
     )
     version_id = _create_artifact_response.id
     artifact_storage_root = _create_artifact_response.artifact_storage_root
-    total_size = 0
+    total_size = None
+
     if isinstance(artifact_dir, tempfile.TemporaryDirectory):
         # Source is of type TrueFoundryManagedSource
         source = TrueFoundryManagedSource(type="truefoundry", uri=artifact_storage_root)
@@ -502,6 +513,9 @@ def _log_artifact_version_helper(
     else:
         raise MlFoundryException("Invalid artifact_dir provided")
 
+    if total_size is not None and total_size > 0:
+        metadata[ARTIFACT_METADATA_TRUEFOUNDRY_KEY]["artifact_size"] = total_size
+
     artifact_manifest = None
     if artifact_type == ArtifactType.ARTIFACT:
         _source_cls = get_autogen_type(ArtifactManifest, "source")
@@ -509,11 +523,13 @@ def _log_artifact_version_helper(
             name=name,
             ml_repo=ml_repo,
             description=description,
-            metadata=metadata or {},
+            metadata=metadata,
             source=_source_cls.from_dict(source.dict()),
             step=step,
+            run_id=run.run_id if run else None,
         )
     _manifest_cls = get_autogen_type(FinalizeArtifactVersionRequestDto, "manifest")
+
     finalize_artifact_version_request_dto = FinalizeArtifactVersionRequestDto(
         id=version_id,
         run_uuid=run.run_id if run else None,

{truefoundry-0.10.5 → truefoundry-0.10.6}/truefoundry/ml/log_types/artifacts/constants.py

@@ -31,3 +31,5 @@ model_version = ModelSchema(...) # or schema in dictionary format {{"features":
 model_version.update()
 ```
 """
+
+ARTIFACT_METADATA_TRUEFOUNDRY_KEY = ".truefoundry"

{truefoundry-0.10.5 → truefoundry-0.10.6}/truefoundry/ml/log_types/artifacts/model.py

@@ -36,6 +36,7 @@ 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 (
+    ARTIFACT_METADATA_TRUEFOUNDRY_KEY,
     INTERNAL_METADATA_PATH,
 )
 from truefoundry.ml.log_types.artifacts.utils import (
@@ -547,6 +548,8 @@ def _log_model_version(  # noqa: C901
     step = step or 0
     total_size = None
     metadata = metadata or {}
+    if ARTIFACT_METADATA_TRUEFOUNDRY_KEY not in metadata:
+        metadata[ARTIFACT_METADATA_TRUEFOUNDRY_KEY] = {}
 
     _validate_description(description)
     _validate_artifact_metadata(metadata)
@@ -617,6 +620,9 @@ def _log_model_version(  # noqa: C901
     else:
         raise MlFoundryException("Invalid model_file_or_folder provided")
 
+    if total_size is not None and total_size > 0:
+        metadata[ARTIFACT_METADATA_TRUEFOUNDRY_KEY]["artifact_size"] = total_size
+
     _source_cls = get_autogen_type(ModelManifest, "source")
     # Auto fetch the framework & environment details if not provided
     framework = _ModelFramework.to_model_framework_type(framework)
@@ -636,6 +642,7 @@ def _log_model_version(  # noqa: C901
         framework=Framework.from_dict(framework.dict()) if framework else None,
         environment=environment,
         step=step,
+        run_id=run.run_id if run else None,
     )
     _manifest_cls = get_autogen_type(FinalizeArtifactVersionRequestDto, "manifest")
     artifact_version_response = mlfoundry_artifacts_api.finalize_artifact_version_post(