truefoundry 0.11.11rc1__tar.gz → 0.12.0rc1__tar.gz

{truefoundry-0.11.11rc1 → truefoundry-0.12.0rc1}/PKG-INFO

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

{truefoundry-0.11.11rc1 → truefoundry-0.12.0rc1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "truefoundry"
-version = "0.11.11rc1"  # This is a placeholder version, it will be replaced when releasing
+version = "0.12.0rc1"  # 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.11.11rc1 → truefoundry-0.12.0rc1}/truefoundry/common/storage_provider_utils.py

@@ -22,6 +22,23 @@ from truefoundry.pydantic_v1 import BaseModel
 logger = logging.getLogger("truefoundry")
 
 
+def truncate_path_for_progress(
+    path: str, max_length: int, relpath: bool = False
+) -> str:
+    if relpath:
+        path = os.path.relpath(path)
+    if len(path) <= max_length:
+        return path
+    parts = path.split(os.sep)
+    result = parts[-1]  # start with filename
+    i = len(parts) - 2
+    # keep prepending directories until adding more would exceed max_len - 3
+    while i >= 0 and len(result) + len(parts[i]) + 1 <= max_length - 3:
+        result = parts[i] + os.sep + result
+        i -= 1
+    return "..." + os.sep + result
+
+
 class MultiPartUploadStorageProvider(str, Enum):
     S3_COMPATIBLE = "S3_COMPATIBLE"
     AZURE_BLOB = "AZURE_BLOB"
@@ -182,7 +199,7 @@ def _file_part_upload(
                 return response
 
 
-def s3_compatible_multipart_upload(
+def s3_compatible_multipart_upload(  # noqa: C901
     multipart_upload: MultiPartUpload,
     local_file: str,
     multipart_info: _FileMultiPartInfo,
@@ -190,13 +207,15 @@ def s3_compatible_multipart_upload(
     progress_bar: Optional[Progress] = None,
     abort_event: Optional[Event] = None,
     exception_class=HttpRequestException,
-):
+) -> None:
     abort_event = abort_event or Event()
     parts = []
 
-    if progress_bar:
+    if progress_bar is not None:
         multi_part_upload_progress = progress_bar.add_task(
-            f"[green]Uploading {local_file}:", start=True
+            f"[green]⬆ {truncate_path_for_progress(local_file, 64, relpath=True)}",
+            start=True,
+            visible=True,
         )
 
     def upload(part_number: int, seek: int) -> None:
@@ -221,7 +240,7 @@ def s3_compatible_multipart_upload(
             multipart_info.num_parts,
             local_file,
         )
-        if progress_bar:
+        if progress_bar is not None:
             progress_bar.update(
                 multi_part_upload_progress,
                 advance=multipart_info.part_size,
@@ -254,10 +273,12 @@ def s3_compatible_multipart_upload(
         timeout=2 * 60,
     )
     response.raise_for_status()
+    if progress_bar is not None:
+        progress_bar.refresh()
     logger.debug("Multipart upload of %s completed", local_file)
 
 
-def azure_multi_part_upload(
+def azure_multi_part_upload(  # noqa: C901
     multipart_upload: MultiPartUpload,
     local_file: str,
     multipart_info: _FileMultiPartInfo,
@@ -265,12 +286,14 @@ def azure_multi_part_upload(
     progress_bar: Optional[Progress] = None,
     abort_event: Optional[Event] = None,
     exception_class=HttpRequestException,
-):
+) -> None:
     abort_event = abort_event or Event()
 
-    if progress_bar:
+    if progress_bar is not None:
         multi_part_upload_progress = progress_bar.add_task(
-            f"[green]Uploading {local_file}:", start=True
+            f"[green]⬆ {truncate_path_for_progress(local_file, 64, relpath=True)}",
+            start=True,
+            visible=True,
         )
 
     def upload(part_number: int, seek: int):
@@ -289,7 +312,7 @@ def azure_multi_part_upload(
             abort_event=abort_event,
             exception_class=exception_class,
         )
-        if progress_bar:
+        if progress_bar is not None:
             progress_bar.update(
                 multi_part_upload_progress,
                 advance=multipart_info.part_size,
@@ -328,4 +351,6 @@ def azure_multi_part_upload(
             timeout=2 * 60,
         )
         response.raise_for_status()
+    if progress_bar is not None:
+        progress_bar.refresh()
     logger.debug("Multipart upload of %s completed", local_file)

{truefoundry-0.11.11rc1 → truefoundry-0.12.0rc1}/truefoundry/deploy/_autogen/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  application.json
-#   timestamp: 2025-09-09T14:09:56+00:00
+#   timestamp: 2025-09-10T19:32:14+00:00
 
 from __future__ import annotations
 
@@ -621,8 +621,8 @@ class Pip(BaseModel):
 
 class Poetry(BaseModel):
     """
-    Use `poetry` to setup environment and sync `poetry.lock` file
-    Your build context root must contain a `poetry.lock` file
+    Use `poetry` to setup env
+    Your build context root must contain `pyproject.toml` and `poetry.lock
     """
 
     type: Literal["poetry"] = Field(..., description="")
@@ -1060,7 +1060,7 @@ class TaskPythonBuild(BaseModel):
     )
     cuda_version: Optional[
         constr(
-            regex=r"^((\d+\.\d+(\.\d+)?-cudnn\d+-(runtime|devel)-ubuntu\d+\.\d+)|11\.0-cudnn8|11\.1-cudnn8|11\.2-cudnn8|11\.3-cudnn8|11\.4-cudnn8|11\.5-cudnn8|11\.6-cudnn8|11\.7-cudnn8|11\.8-cudnn8|12\.0-cudnn8|12\.1-cudnn8|12\.2-cudnn8|12\.3-cudnn9|12\.4-cudnn9|12\.5-cudnn9|12\.6-cudnn9)$"
+            regex=r"^((\d+\.\d+(\.\d+)?-cudnn\d+-(runtime|devel)-ubuntu\d+\.\d+)|11\.0-cudnn8|11\.1-cudnn8|11\.2-cudnn8|11\.3-cudnn8|11\.4-cudnn8|11\.5-cudnn8|11\.6-cudnn8|11\.7-cudnn8|11\.8-cudnn8|12\.0-cudnn8|12\.1-cudnn8|12\.2-cudnn8|12\.3-cudnn9|12\.4-cudnn9|12\.5-cudnn9|12\.6-cudnn9|12\.8-cudnn9|12\.9-cudnn9)$"
         )
     ] = Field(
         None,
@@ -1094,8 +1094,8 @@ class TrueFoundryInteractiveLogin(BaseModel):
 
 class UV(BaseModel):
     """
-    Use `uv` to setup environment and sync `uv.lock` file
-    Your build context root must contain a `uv.lock` file
+    Use `uv` to setup env.
+    Your build context root must contain `pyproject.toml` and `uv.lock
     """
 
     type: Literal["uv"] = Field(..., description="")
@@ -1448,7 +1448,7 @@ class PythonBuild(BaseModel):
     )
     cuda_version: Optional[
         constr(
-            regex=r"^((\d+\.\d+(\.\d+)?-cudnn\d+-(runtime|devel)-ubuntu\d+\.\d+)|11\.0-cudnn8|11\.1-cudnn8|11\.2-cudnn8|11\.3-cudnn8|11\.4-cudnn8|11\.5-cudnn8|11\.6-cudnn8|11\.7-cudnn8|11\.8-cudnn8|12\.0-cudnn8|12\.1-cudnn8|12\.2-cudnn8|12\.3-cudnn9|12\.4-cudnn9|12\.5-cudnn9|12\.6-cudnn9)$"
+            regex=r"^((\d+\.\d+(\.\d+)?-cudnn\d+-(runtime|devel)-ubuntu\d+\.\d+)|11\.0-cudnn8|11\.1-cudnn8|11\.2-cudnn8|11\.3-cudnn8|11\.4-cudnn8|11\.5-cudnn8|11\.6-cudnn8|11\.7-cudnn8|11\.8-cudnn8|12\.0-cudnn8|12\.1-cudnn8|12\.2-cudnn8|12\.3-cudnn9|12\.4-cudnn9|12\.5-cudnn9|12\.6-cudnn9|12\.8-cudnn9|12\.9-cudnn9)$"
         )
     ] = Field(
         None,

{truefoundry-0.11.11rc1 → truefoundry-0.12.0rc1}/truefoundry/ml/artifact/truefoundry_artifact_repo.py

@@ -40,6 +40,7 @@ from truefoundry.common.storage_provider_utils import (
     azure_multi_part_upload,
     decide_file_parts,
     s3_compatible_multipart_upload,
+    truncate_path_for_progress,
 )
 from truefoundry.ml._autogen.client import (  # type: ignore[attr-defined]
     ApiClient,
@@ -125,14 +126,14 @@ def _signed_url_upload_file(
             augmented_raise_for_status(response, exception_class=MlFoundryException)  # type: ignore
         return
 
-    task_progress_bar = progress_bar.add_task(
-        f"[green]Uploading {local_file}:", start=True
+    task_id = progress_bar.add_task(
+        f"[green]⬆ {truncate_path_for_progress(local_file, 64, relpath=True)}",
+        start=True,
+        visible=True,
     )
 
     def callback(length):
-        progress_bar.update(
-            task_progress_bar, advance=length, total=os.stat(local_file).st_size
-        )
+        progress_bar.update(task_id, advance=length, total=os.stat(local_file).st_size)
         if abort_event and abort_event.is_set():
             raise Exception("aborting upload")
 
@@ -147,6 +148,9 @@ def _signed_url_upload_file(
         ) as response:
             augmented_raise_for_status(response, exception_class=MlFoundryException)  # type: ignore
 
+    if progress_bar is not None:
+        progress_bar.refresh()
+
 
 def _download_file_using_http_uri(
     http_uri,
@@ -167,7 +171,11 @@ def _download_file_using_http_uri(
         exception_class=MlFoundryException,  # type: ignore
     ) as response:
         augmented_raise_for_status(response, exception_class=MlFoundryException)  # type: ignore
-        file_size = int(response.headers.get("Content-Length", 0))
+        file_size = int(response.headers.get("Content-Length", -1))
+        if file_size == 0 and callback:
+            # special case for empty files
+            callback(1, 1)
+        file_size = file_size if file_size > 0 else 0
         with open(download_path, "wb") as output_file:
             for chunk in response.iter_content(chunk_size=chunk_size):
                 if callback:
@@ -658,15 +666,18 @@ class MlFoundryArtifactsRepository:
             logger.info("Downloading %s to %s", remote_file_path, local_path)
 
         if progress_bar is not None:
-            download_progress_bar = progress_bar.add_task(
-                f"[green]Downloading to {remote_file_path}:", start=True
+            task_id = progress_bar.add_task(
+                f"[green]⬇ {truncate_path_for_progress(remote_file_path, 64)}",
+                start=True,
+                visible=True,
             )
 
-        def callback(chunk, total_file_size):
+        def callback(chunk_size: int, total_file_size: int):
+            nonlocal task_id
             if progress_bar is not None:
                 progress_bar.update(
-                    download_progress_bar,
-                    advance=chunk,
+                    task_id,
+                    advance=chunk_size,
                     total=total_file_size,
                 )
             if abort_event and abort_event.is_set():
@@ -677,6 +688,9 @@ class MlFoundryArtifactsRepository:
             download_path=local_path,
             callback=callback,
         )
+
+        if progress_bar is not None:
+            progress_bar.refresh()
         logger.debug("Downloaded %s to %s", remote_file_path, local_path)
 
     def _download_artifact(

{truefoundry-0.11.11rc1 → truefoundry-0.12.0rc1}/truefoundry/ml/log_types/artifacts/artifact.py

@@ -347,10 +347,11 @@ class ArtifactVersion:
             artifact_version.download(path="<your-desired-download-path>")
             ```
         """
-        download_path = self._download(
+        download_info = self._download(
             path=path, overwrite=overwrite, progress=progress
         )
-        return download_path.download_dir
+        logger.info("Downloaded artifact contents to %s", download_info.download_dir)
+        return download_info.download_dir
 
     def delete(self) -> bool:
         """

{truefoundry-0.11.11rc1 → truefoundry-0.12.0rc1}/truefoundry/ml/log_types/artifacts/model.py

@@ -440,6 +440,7 @@ class ModelVersion:
         download_info = self._download(
             path=path, overwrite=overwrite, progress=progress
         )
+        logger.info("Downloaded model contents to %s", download_info.model_dir)
         return download_info
 
     def delete(self) -> bool: