truefoundry 0.5.9__py3-none-any.whl → 0.5.10__py3-none-any.whl

truefoundry/cli/display_util.py

@@ -38,7 +38,7 @@ def stringify(x):
         return str(x)
 
 
-def display_time_passed(seconds: int):
+def display_time_passed(seconds: float) -> str:
     m, s = divmod(seconds, 60)
     h, m = divmod(m, 60)
     d, h = divmod(h, 24)

truefoundry/common/utils.py

@@ -1,6 +1,7 @@
 import json
 import logging
 import os
+import platform
 import sys
 import time
 from collections import namedtuple
@@ -20,6 +21,7 @@ from truefoundry.common.constants import (
     TFY_INTERNAL_ENV_KEY,
 )
 from truefoundry.pydantic_v1 import BaseSettings
+from truefoundry.version import __version__
 
 logger = logging.getLogger(__name__)
 
@@ -192,3 +194,10 @@ def is_debug_env_set() -> bool:
 
 def is_internal_env_set() -> bool:
     return bool(os.getenv(TFY_INTERNAL_ENV_KEY))
+
+
+def get_user_agent() -> str:
+    try:
+        return f"truefoundry/{__version__} Python/{platform.python_version()} OS/{platform.system()}-{platform.release()} ({platform.architecture()[0]})"
+    except Exception:
+        return f"truefoundry/{__version__}"

truefoundry/deploy/auto_gen/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  application.json
-#   timestamp: 2025-02-19T09:48:05+00:00
+#   timestamp: 2025-02-26T17:51:02+00:00
 
 from __future__ import annotations
 

truefoundry/deploy/io/output_callback.py

@@ -1,23 +1,30 @@
+from typing import Any, List, Optional
+
+
 class OutputCallBack:
-    def print_header(self, line):
+    def print_header(self, line: Any) -> None:
         print(line)
 
-    def _print_separator(self):
+    def _print_separator(self) -> None:
         print("-" * 80)
 
-    def print_line(self, line):
+    def print_line(self, line: str) -> None:
         print(line)
 
-    def print_lines_in_panel(self, lines, header=None):
+    def print_lines_in_panel(
+        self, lines: List[str], header: Optional[str] = None
+    ) -> None:
         self.print_header(header)
         self._print_separator()
         for line in lines:
             self.print_line(line)
         self._print_separator()
 
-    def print_code_in_panel(self, lines, header=None, lang="python"):
+    def print_code_in_panel(
+        self, lines: List[str], header: Optional[str] = None
+    ) -> None:
         self.print_lines_in_panel(lines, header)
 
-    def print(self, line):
+    def print(self, line: Any) -> None:
         # just an alias
         self.print_line(line)

truefoundry/deploy/io/rich_output_callback.py

@@ -1,3 +1,5 @@
+from typing import Any, List, Optional
+
 from rich.console import Console
 from rich.highlighter import ReprHighlighter
 from rich.panel import Panel
@@ -14,14 +16,16 @@ class RichOutputCallBack(OutputCallBack):
     console = Console(soft_wrap=True)
     highlighter = ReprHighlighter()
 
-    def print_header(self, line):
+    def print_header(self, line: Any) -> None:
         self.console.rule(_text(line), style="cyan")
 
-    def print_line(self, line):
+    def print_line(self, line: Any) -> None:
         self.console.print(_text(line))
 
-    def print_lines_in_panel(self, lines, header=None):
+    def print_lines_in_panel(
+        self, lines: List[str], header: Optional[str] = None
+    ) -> None:
         self.console.print(Panel(self.highlighter("\n".join(lines)), title=header))
 
-    def print_code_in_panel(self, lines, header=None, lang="python"):
+    def print_code_in_panel(self, lines: List[str], header: Optional[str] = None):
         self.print_lines_in_panel(lines, header)