truefoundry 0.4.4rc10__tar.gz → 0.4.4rc11__tar.gz

{truefoundry-0.4.4rc10 → truefoundry-0.4.4rc11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: truefoundry
-Version: 0.4.4rc10
+Version: 0.4.4rc11
 Summary: Truefoundry CLI
 Author: Abhishek Choudhary
 Author-email: abhishek@truefoundry.com

{truefoundry-0.4.4rc10 → truefoundry-0.4.4rc11}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "truefoundry"
-version = "0.4.4rc10"
+version = "0.4.4rc11"
 description = "Truefoundry CLI"
 authors = ["Abhishek Choudhary <abhishek@truefoundry.com>"]
 readme = "README.md"

{truefoundry-0.4.4rc10 → truefoundry-0.4.4rc11}/truefoundry/common/tfy_signed_url_client.py

@@ -92,7 +92,7 @@ class SignedURLClient:
             retries=self.max_retries, backoff_factor=self.retry_backoff_factor
         )
 
-    @log_time
+    @log_time(prefix=LOG_PREFIX)
     def _make_request(
         self,
         endpoint: str,
@@ -111,6 +111,7 @@ class SignedURLClient:
         except RequestException as e:
             raise RuntimeError(f"Error during request to {url}: {e}") from e
 
+    @log_time(prefix=LOG_PREFIX)
     def _make_server_api_call(
         self,
         endpoint: SignedURLServerEndpoint,
@@ -124,7 +125,7 @@ class SignedURLClient:
             endpoint=endpoint_with_params, method="GET", payload=None, headers=headers
         )
 
-    @log_time
+    @log_time(prefix=LOG_PREFIX)
     def _upload_data(self, signed_url: str, data: Any) -> None:
         """
         Upload data to the specified storage path using a signed URL.
@@ -144,6 +145,7 @@ class SignedURLClient:
         except Exception as e:
             raise RuntimeError(f"Failed to upload data: {e}") from e
 
+    @log_time(prefix=LOG_PREFIX)
     def upload_from_bytes(self, data: bytes, storage_uri: str) -> str:
         """Upload bytes to the specified storage path using a signed URL."""
         signed_object = self._make_server_api_call(
@@ -155,7 +157,7 @@ class SignedURLClient:
         self._upload_data(pre_signed_object_dto.signed_url, data)
         return storage_uri
 
-    @log_time
+    @log_time(prefix=LOG_PREFIX)
     def upload(self, file_path: str, storage_uri: str) -> str:
         """Upload a file to the specified storage path using a signed URL."""
         logger.info(f"{LOG_PREFIX} Uploading {file_path} to {storage_uri}")
@@ -169,7 +171,7 @@ class SignedURLClient:
             self._upload_data(pre_signed_object_dto.signed_url, file)
         return storage_uri
 
-    @log_time
+    @log_time(prefix=LOG_PREFIX)
     def _download_file(
         self, signed_url: str, local_path: Optional[str] = None
     ) -> Optional[bytes]:
@@ -191,7 +193,7 @@ class SignedURLClient:
         except RequestException as e:
             raise RuntimeError(f"Failed to download file from {signed_url}: {e}") from e
 
-    @log_time
+    @log_time(prefix=LOG_PREFIX)
     def download(self, storage_uri: str, local_path: str) -> Optional[str]:
         """Download a file from the specified storage path to a local path using a signed URL."""
         logger.info(f"{LOG_PREFIX} Downloading {storage_uri} to {local_path}")
@@ -204,6 +206,7 @@ class SignedURLClient:
         self._download_file(presigned_object.signed_url, local_path)
         return local_path
 
+    @log_time(prefix=LOG_PREFIX)
     def download_to_bytes(self, storage_uri: str) -> bytes:
         """Download a file from the specified storage path and return it as bytes."""
         response = self._make_server_api_call(
@@ -214,6 +217,7 @@ class SignedURLClient:
         presigned_object = SignedURLAPIResponseDto.parse_obj(response)
         return self._download_file(presigned_object.signed_url)
 
+    @log_time(prefix=LOG_PREFIX)
     def exists(self, uri: str) -> bool:
         """Check if a file exists at the specified path."""
         response = self._make_server_api_call(
@@ -223,6 +227,7 @@ class SignedURLClient:
         )
         return SignedURLExistsAPIResponseDto.parse_obj(response).exists
 
+    @log_time(prefix=LOG_PREFIX)
     def is_directory(self, uri: str) -> bool:
         """Check if the specified URI is a directory."""
         response = self._make_server_api_call(
@@ -236,7 +241,7 @@ class SignedURLClient:
         logger.info(f"{LOG_PREFIX} Path {uri} is a directory: {is_directory}")
         return is_directory
 
-    @log_time
+    @log_time(prefix=LOG_PREFIX)
     def list_files(
         self, path: str, detail: bool = False, max_results: int = 1000
     ) -> Union[List[FileInfo], List[str]]:

{truefoundry-0.4.4rc10 → truefoundry-0.4.4rc11}/truefoundry/common/utils.py

@@ -10,6 +10,7 @@ from truefoundry.common.constants import (
     MLFOUNDRY_SERVER_RELATIVE_PATH,
     TFY_HOST_ENV_KEY,
 )
+from truefoundry.logger import logger
 from truefoundry.pydantic_v1 import BaseSettings
 
 T = TypeVar("T")
@@ -104,7 +105,7 @@ def resolve_tfy_host(tfy_host: Optional[str] = None) -> str:
     return tfy_host
 
 
-def log_time(func):
+def log_time(func, prefix: str = ""):
     """Decorator to log the time taken by I/O operations."""
 
     @wraps(func)
@@ -112,7 +113,7 @@ def log_time(func):
         start_time = time.time()
         result = func(*args, **kwargs)
         elapsed_time = time.time() - start_time
-        print(f"{func.__name__} took {elapsed_time:.2f} seconds")
+        logger.info(f"{prefix} {func.__name__} took {elapsed_time:.2f} seconds")
         return result
 
     return wrapper