truefoundry 0.11.7__py3-none-any.whl → 0.11.8rc1__py3-none-any.whl

truefoundry/ml/log_types/artifacts/artifact.py

@@ -39,6 +39,7 @@ from truefoundry.ml.log_types.artifacts.utils import (
     _validate_description,
     calculate_total_size,
     get_autogen_type,
+    set_tfy_internal_metadata,
 )
 from truefoundry.ml.logger import logger
 from truefoundry.ml.session import _get_api_client
@@ -455,8 +456,7 @@ def _log_artifact_version_helper(
         ml_repo_id = result.id
 
     metadata = metadata or {}
-    if ARTIFACT_METADATA_TRUEFOUNDRY_KEY not in metadata:
-        metadata[ARTIFACT_METADATA_TRUEFOUNDRY_KEY] = {}
+    metadata = set_tfy_internal_metadata(metadata)
 
     _validate_description(description)
     _validate_artifact_metadata(metadata)

truefoundry/ml/log_types/artifacts/constants.py

@@ -33,3 +33,8 @@ model_version.update()
 """
 
 ARTIFACT_METADATA_TRUEFOUNDRY_KEY = ".truefoundry"
+TFY_INTERNAL_APPLICATION_ID_ENV_VAR = "TFY_INTERNAL_APPLICATION_ID"
+TFY_INTERNAL_APPLICATION_NAME_ENV_VAR = "TFY_INTERNAL_APPLICATION_NAME"
+TFY_INTERNAL_APPLICATION_FQN_ENV_VAR = "TFY_INTERNAL_APPLICATION_FQN"
+TFY_INTERNAL_APPLICATION_VERSION_ENV_VAR = "TFY_INTERNAL_APPLICATION_VERSION"
+TFY_INTERNAL_JOB_RUN_NAME_ENV_VAR = "TFY_INTERNAL_JOB_RUN_NAME"

truefoundry/ml/log_types/artifacts/model.py

@@ -46,6 +46,7 @@ from truefoundry.ml.log_types.artifacts.utils import (
     _validate_description,
     calculate_total_size,
     get_autogen_type,
+    set_tfy_internal_metadata,
 )
 from truefoundry.ml.model_framework import (
     ModelFrameworkType,
@@ -544,8 +545,7 @@ 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] = {}
+    metadata = set_tfy_internal_metadata(metadata)
 
     _validate_description(description)
     _validate_artifact_metadata(metadata)

truefoundry/ml/log_types/artifacts/utils.py

@@ -7,7 +7,15 @@ from pathlib import Path, PureWindowsPath
 from typing import Any, Dict, List, Optional, Sequence, Tuple, Type, Union
 
 from truefoundry.ml.exceptions import MlFoundryException
-from truefoundry.ml.log_types.artifacts.constants import DESCRIPTION_MAX_LENGTH
+from truefoundry.ml.log_types.artifacts.constants import (
+    ARTIFACT_METADATA_TRUEFOUNDRY_KEY,
+    DESCRIPTION_MAX_LENGTH,
+    TFY_INTERNAL_APPLICATION_FQN_ENV_VAR,
+    TFY_INTERNAL_APPLICATION_ID_ENV_VAR,
+    TFY_INTERNAL_APPLICATION_NAME_ENV_VAR,
+    TFY_INTERNAL_APPLICATION_VERSION_ENV_VAR,
+    TFY_INTERNAL_JOB_RUN_NAME_ENV_VAR,
+)
 from truefoundry.pydantic_v1 import BaseModel
 
 logger = logging.getLogger(__name__)
@@ -228,3 +236,30 @@ def get_autogen_type(parent_type: Type[BaseModel], field_name: str) -> Type[Base
             return args[0]
 
     raise NotImplementedError(f"Cannot extract main type from {type_}")
+
+
+def set_tfy_internal_metadata(metadata: Dict[str, Any]) -> Dict[str, Any]:
+    if ARTIFACT_METADATA_TRUEFOUNDRY_KEY not in metadata:
+        metadata[ARTIFACT_METADATA_TRUEFOUNDRY_KEY] = {}
+    elif not isinstance(metadata[ARTIFACT_METADATA_TRUEFOUNDRY_KEY], dict):
+        return metadata
+
+    tfy_metadata = metadata[ARTIFACT_METADATA_TRUEFOUNDRY_KEY]
+    created_by_key = "created_by"
+    if created_by_key not in tfy_metadata:
+        tfy_metadata[created_by_key] = {}
+    elif not isinstance(tfy_metadata[created_by_key], dict):
+        return metadata
+
+    tfy_internal_metadata = {
+        "application_id": os.environ.get(TFY_INTERNAL_APPLICATION_ID_ENV_VAR),
+        "application_fqn": os.environ.get(TFY_INTERNAL_APPLICATION_FQN_ENV_VAR),
+        "application_name": os.environ.get(TFY_INTERNAL_APPLICATION_NAME_ENV_VAR),
+        "application_version": os.environ.get(TFY_INTERNAL_APPLICATION_VERSION_ENV_VAR),
+        "job_run_name": os.environ.get(TFY_INTERNAL_JOB_RUN_NAME_ENV_VAR),
+    }
+    created_by_metadata = tfy_metadata[created_by_key]
+    for key, value in tfy_internal_metadata.items():
+        if key not in created_by_metadata and value is not None:
+            created_by_metadata[key] = value
+    return metadata

{truefoundry-0.11.7.dist-info → truefoundry-0.11.8rc1.dist-info}/METADATA

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

{truefoundry-0.11.7.dist-info → truefoundry-0.11.8rc1.dist-info}/RECORD

@@ -364,12 +364,13 @@ 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=dZj9UkRvvU5EN-HUiuLCPsmosSS0yyHM5JNWzaaIxMA,19887
-truefoundry/ml/log_types/artifacts/constants.py,sha256=uB2JPEqwTbqevkQv2QcEMROsm_4cVAl6s0QU1MLa8SQ,1088
+truefoundry/ml/log_types/artifacts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+truefoundry/ml/log_types/artifacts/artifact.py,sha256=JiqdWp6XQe2S_zP96urXCjyfdxFlx6-GNBzH9Te7hTQ,19854
+truefoundry/ml/log_types/artifacts/constants.py,sha256=LmY4syH4PzZShrFgSg04Pse5rQqkPf9Jdsm7daUwZb0,1440
 truefoundry/ml/log_types/artifacts/dataset.py,sha256=OgWIoT59AhMw8P01FfvUKbJ3EL6HQf_Xw8X4E3Ff5Sg,13172
 truefoundry/ml/log_types/artifacts/general_artifact.py,sha256=yr-SQ2fhUR_sE1MB5zoHHYpGC8tizH_-t3lhsxCAULU,2747
-truefoundry/ml/log_types/artifacts/model.py,sha256=kAiq7TDJ8RHG1Z4IN9mQD1dzKgmLCP1p0v_Yta9GHlM,24864
-truefoundry/ml/log_types/artifacts/utils.py,sha256=q_atcGzn3wfxItt3RABxjdris8b3njEFNuC8ihWqUSI,8088
+truefoundry/ml/log_types/artifacts/model.py,sha256=oRX6bKb5tN4LjDFifeDpjv4z1-vHXf3Ma6-nkp7mN4M,24831
+truefoundry/ml/log_types/artifacts/utils.py,sha256=FrnsumSB3SIqC_xpPp3RZijbIxgOFCQyuhSfcDaZoXI,9589
 truefoundry/ml/log_types/image/__init__.py,sha256=fcOq8yQnNj1rkLcPeIjLXBpdA1WIeiPsXOlAAvMxx7M,76
 truefoundry/ml/log_types/image/constants.py,sha256=wLtGEOA4T5fZHSlOXPuNDLX3lpbCtwlvGKPFk_1fah0,255
 truefoundry/ml/log_types/image/image.py,sha256=sa0tBHdyluC8bELXY16E0HgFrUDnDBxHrteix4BFXcs,12479
@@ -386,7 +387,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.11.7.dist-info/METADATA,sha256=ydC2lFQf5iICc9uFTslsw4zS96saw6rWGDbcEup0zSc,2760
-truefoundry-0.11.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-truefoundry-0.11.7.dist-info/entry_points.txt,sha256=xVjn7RMN-MW2-9f7YU-bBdlZSvvrwzhpX1zmmRmsNPU,98
-truefoundry-0.11.7.dist-info/RECORD,,
+truefoundry-0.11.8rc1.dist-info/METADATA,sha256=X6K9nwRmNf159lyDFhJjugm5o0eurCHD0ISxo8D61Vs,2763
+truefoundry-0.11.8rc1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+truefoundry-0.11.8rc1.dist-info/entry_points.txt,sha256=xVjn7RMN-MW2-9f7YU-bBdlZSvvrwzhpX1zmmRmsNPU,98
+truefoundry-0.11.8rc1.dist-info/RECORD,,