truefoundry 0.10.5__py3-none-any.whl → 0.10.6__py3-none-any.whl

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/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/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/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/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(

{truefoundry-0.10.5.dist-info → truefoundry-0.10.6.dist-info}/METADATA

@@ -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.dist-info → truefoundry-0.10.6.dist-info}/RECORD

@@ -1,4 +1,4 @@
-truefoundry/__init__.py,sha256=VVpO-Awh1v93VOURe7hank8QpeSPc0dCykwr14GOFsw,967
+truefoundry/__init__.py,sha256=z9iNNI3mqVWkvzBXMRu096jXaYloYSN6rnlTbfmKcJE,1056
 truefoundry/_client.py,sha256=Y3qHi_Lg4Sx6GNvsjAHIoAfFr8PJnqgCrXmpNAI3ECg,1417
 truefoundry/logger.py,sha256=u-YCNjg5HBwE70uQcpjIG64Ghos-K2ulTWaxC03BSj4,714
 truefoundry/pydantic_v1.py,sha256=jSuhGtz0Mbk1qYu8jJ1AcnIDK4oxUsdhALc4spqstmM,345
@@ -51,7 +51,7 @@ truefoundry/common/session.py,sha256=d9l3TEBpqVP4mr4mTGY1qVxc815skzMlNNdw14otg34
 truefoundry/common/storage_provider_utils.py,sha256=yURhMw8k0FLFvaviRHDiifhvc6GnuQwGMC9Qd2uM440,10934
 truefoundry/common/types.py,sha256=BMJFCsR1lPJAw66IQBSvLyV4I6o_x5oj78gVsUa9si8,188
 truefoundry/common/utils.py,sha256=j3QP0uOsaGD_VmDDR68JTwoYE1okkAq6OqpVkzVf48Q,6424
-truefoundry/common/warnings.py,sha256=rs6BHwk7imQYedo07iwh3TWEOywAR3Lqhj0AY4khByg,504
+truefoundry/common/warnings.py,sha256=xDMhR_-ZGC40Ycaj6nlFb5MYPexn8WbKCHd4FlflTXQ,705
 truefoundry/deploy/__init__.py,sha256=PVbGPU9S3-dTFn5LvLwaEnfsp2RrGT9iiM7_15kOV84,2837
 truefoundry/deploy/python_deploy_codegen.py,sha256=k19_m5DGsUyjOUCSKwIVP8vDna2sq01tHABsUfoVpW4,8019
 truefoundry/deploy/_autogen/models.py,sha256=8j_y0Yp8k8Sjj7iVtZDHeuxq9kDvD0xI8-iFnbf0370,73571
@@ -360,11 +360,11 @@ truefoundry/ml/log_types/__init__.py,sha256=g4u4D4Jaj0aBK5GtrLV88-qThKZR9pSZ17vF
 truefoundry/ml/log_types/plot.py,sha256=LDh4uy6z2P_a2oPM2lc85c0lt8utVvunohzeMawFjZw,7572
 truefoundry/ml/log_types/pydantic_base.py,sha256=eBlw_AEyAz4iJKDP4zgJOCFWcldwQqpf7FADW1jzIQY,272
 truefoundry/ml/log_types/utils.py,sha256=xjJ21jdPScvFmw3TbVh5NCzbzJwaqiXJyiiT4xxX1EI,335
-truefoundry/ml/log_types/artifacts/artifact.py,sha256=qaBHO-5ffBvghmhRQ5pperp8rm3T8HYSrkecpCrK_4I,19594
-truefoundry/ml/log_types/artifacts/constants.py,sha256=qKxQ5mMvJE4j83BvGW3qNTKunxCiBg_EEjTdgbgJtyE,1036
+truefoundry/ml/log_types/artifacts/artifact.py,sha256=N1M1a7Oq9fY_7s1PT7uv64s_Ek62Lj-JsPBrWdRKpdY,20050
+truefoundry/ml/log_types/artifacts/constants.py,sha256=uB2JPEqwTbqevkQv2QcEMROsm_4cVAl6s0QU1MLa8SQ,1088
 truefoundry/ml/log_types/artifacts/dataset.py,sha256=UpLXoqhfONqp6YG4N8lDrDe-XhTK6ZZ9Lwg8mI0UZn4,13113
 truefoundry/ml/log_types/artifacts/general_artifact.py,sha256=yr-SQ2fhUR_sE1MB5zoHHYpGC8tizH_-t3lhsxCAULU,2747
-truefoundry/ml/log_types/artifacts/model.py,sha256=kEuqnQhuO9Hmz8bqHEzrbdPn_Vt7H-Tansd0rmplyWE,24696
+truefoundry/ml/log_types/artifacts/model.py,sha256=0_2EGwj8VYRJ3g8Ti2k75s-OiC1Tu-7Ay9U1_QYG_iw,25027
 truefoundry/ml/log_types/artifacts/utils.py,sha256=q_atcGzn3wfxItt3RABxjdris8b3njEFNuC8ihWqUSI,8088
 truefoundry/ml/log_types/image/__init__.py,sha256=fcOq8yQnNj1rkLcPeIjLXBpdA1WIeiPsXOlAAvMxx7M,76
 truefoundry/ml/log_types/image/constants.py,sha256=wLtGEOA4T5fZHSlOXPuNDLX3lpbCtwlvGKPFk_1fah0,255
@@ -381,7 +381,7 @@ truefoundry/workflow/remote_filesystem/__init__.py,sha256=LQ95ViEjJ7Ts4JcCGOxMPs
 truefoundry/workflow/remote_filesystem/logger.py,sha256=em2l7D6sw7xTLDP0kQSLpgfRRCLpN14Qw85TN7ujQcE,1022
 truefoundry/workflow/remote_filesystem/tfy_signed_url_client.py,sha256=xcT0wQmQlgzcj0nP3tJopyFSVWT1uv3nhiTIuwfXYeg,12342
 truefoundry/workflow/remote_filesystem/tfy_signed_url_fs.py,sha256=nSGPZu0Gyd_jz0KsEE-7w_BmnTD8CVF1S8cUJoxaCbc,13305
-truefoundry-0.10.5.dist-info/METADATA,sha256=t8-jiTcLLtOLh7QQQwl_FXqgG0GJdttfh8b06Org1iE,2505
-truefoundry-0.10.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-truefoundry-0.10.5.dist-info/entry_points.txt,sha256=xVjn7RMN-MW2-9f7YU-bBdlZSvvrwzhpX1zmmRmsNPU,98
-truefoundry-0.10.5.dist-info/RECORD,,
+truefoundry-0.10.6.dist-info/METADATA,sha256=CN96HcoQd_5lWHozK46eDlDnN1l_ZIrTXHHKwHS8eKc,2505
+truefoundry-0.10.6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+truefoundry-0.10.6.dist-info/entry_points.txt,sha256=xVjn7RMN-MW2-9f7YU-bBdlZSvvrwzhpX1zmmRmsNPU,98
+truefoundry-0.10.6.dist-info/RECORD,,