truefoundry 0.3.0rc13__py3-none-any.whl → 0.3.0rc15__py3-none-any.whl

truefoundry/deploy/cli/util.py

@@ -1,12 +1,14 @@
 import functools
 import sys
 import zipfile
+from typing import Dict
 
 import rich_click as click
 from packaging.version import parse as parse_version
 from requests.exceptions import ConnectionError
 from rich.padding import Padding
 from rich.panel import Panel
+from rich.table import Table
 
 from truefoundry.deploy.cli.console import console
 from truefoundry.deploy.lib.exceptions import (
@@ -27,16 +29,29 @@ def handle_exception(exception):
     if is_debug_env_set():
         console.print_exception(show_locals=True)
     if isinstance(exception, BadRequestException):
-        print_error(
-            f"[cyan bold]status_code[/] {exception.status_code}\n"
-            f"[cyan bold]message[/]     {exception.message}"
+        print_dict_as_table_panel(
+            {"Status Code": str(exception.status_code), "Error": exception.message},
+            title="Command Failed",
+            border_style="red",
         )
     elif isinstance(exception, ConnectionError):
-        print_error("Couldn't connect to TrueFoundry.")
+        print_dict_as_table_panel(
+            {"Error": "Couldn't connect to TrueFoundry"},
+            title="Command Failed",
+            border_style="red",
+        )
     elif isinstance(exception, ConfigurationException):
-        print_error(f"[cyan bold]message[/]     {exception.message}")
+        print_dict_as_table_panel(
+            {"Error": exception.message},
+            title="Command Failed",
+            border_style="red",
+        )
     else:
-        console.print(f"[red][bold]Error:[/] {str(exception)}[/]")
+        print_dict_as_table_panel(
+            {"Error": str(exception)},
+            title="Command Failed",
+            border_style="red",
+        )
 
 
 def handle_exception_wrapper(fn):
@@ -77,6 +92,28 @@ def print_message(message):
     )
 
 
+def print_dict_as_table_panel(
+    dct: Dict[str, str],
+    title: str,
+    border_style: str = "cyan",
+    key_color: str = "cyan",
+):
+    table = Table(show_header=False, box=None)
+    table.add_column("Key", style=f"bold {key_color}", width=15)
+    table.add_column("Value")
+    for key, value in dct.items():
+        table.add_row(key, value)
+    console.print(
+        Panel(
+            table,
+            border_style=border_style,
+            title=title,
+            title_align="left",
+            width=click.rich_click.MAX_WIDTH,
+        )
+    )
+
+
 def unzip_package(path_to_package, destination):
     with zipfile.ZipFile(path_to_package, "r") as zip_ref:
         zip_ref.extractall(destination)

truefoundry/deploy/lib/exceptions.py

@@ -1,6 +1,9 @@
+from typing import Optional
+
+
 class BadRequestException(Exception):
-    def __init__(self, status_code, message=None):
-        super(BadRequestException, self).__init__()
+    def __init__(self, status_code: int, message: Optional[str] = None):
+        super().__init__()
         self.status_code = status_code
         self.message = message
 
@@ -12,8 +15,8 @@ class BadRequestException(Exception):
 
 
 class ConfigurationException(Exception):
-    def __init__(self, message=None):
-        super(ConfigurationException, self).__init__()
+    def __init__(self, message: Optional[str] = None):
+        super().__init__()
         self.message = message
 
     def __str__(self):

{truefoundry-0.3.0rc13.dist-info → truefoundry-0.3.0rc15.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: truefoundry
-Version: 0.3.0rc13
+Version: 0.3.0rc15
 Summary: Truefoundry CLI
 Author: Abhishek Choudhary
 Author-email: abhishek@truefoundry.com
@@ -24,7 +24,7 @@ Requires-Dist: flytekit (==1.12.2) ; extra == "workflow"
 Requires-Dist: gitignorefile (>=1.1.2,<1.2.0)
 Requires-Dist: importlib-metadata (>=6.0.1,<8.0.0)
 Requires-Dist: importlib-resources (>=5.2.0,<6.0.0)
-Requires-Dist: mlfoundry (==0.11.4) ; extra == "ml"
+Requires-Dist: mlfoundry (==0.11.5) ; extra == "ml"
 Requires-Dist: openai (>=1.16.2,<2.0.0)
 Requires-Dist: packaging (>=20.0,<25.0)
 Requires-Dist: pydantic (>=1.10.0,<3)

{truefoundry-0.3.0rc13.dist-info → truefoundry-0.3.0rc15.dist-info}/RECORD

@@ -58,7 +58,7 @@ truefoundry/deploy/cli/config.py,sha256=tf8w4UfVzcC6eYkENvuuCPYt_V3sqVpO1bclORV9
 truefoundry/deploy/cli/console.py,sha256=9-dMy4YPisCJQziRKTg8Qa0UJnOGl1soiUnJjsnLDvE,242
 truefoundry/deploy/cli/const.py,sha256=dVHPo1uAiDSSMXwXoT2mR5kNQjExT98QNVRz98Hz_Ts,510
 truefoundry/deploy/cli/display_util.py,sha256=gq8EBdpBMHUzYQp_hxOg9EOYi08FIHgOVFCqXqC2tuo,3044
-truefoundry/deploy/cli/util.py,sha256=HVcVsba5b4K2v4JQHFhNMhwIkVRlMTShQkjpOA38qxI,2526
+truefoundry/deploy/cli/util.py,sha256=tSPYTQLSqEApx7nXxy69BTH69t4LewY1JtMJz5VEz40,3463
 truefoundry/deploy/core/__init__.py,sha256=j61bMWj4BkWihdssKMSFhieo7afJDtpc7qO7zk1rDB4,140
 truefoundry/deploy/core/login.py,sha256=N2VrW3nlBzoyoYulkipxwQvCpjBhi3sfsmhxK1ktWhg,236
 truefoundry/deploy/core/logout.py,sha256=TpWLq4_DsxYS5GX2OJQGDhekNOfiOLb-vO5khQueHXw,80
@@ -92,7 +92,7 @@ truefoundry/deploy/lib/dao/application.py,sha256=uUTFSQkLUrFCtQQgS2Jm9BpyHyhMkN4
 truefoundry/deploy/lib/dao/apply.py,sha256=sXnQY6RVzLVm1fX2BKuWHAoKlKISirrcByHEhY3x4zo,2570
 truefoundry/deploy/lib/dao/version.py,sha256=AtdW_4O1DPUKdfv2qy6iUJsZ_95vM6z0AqeEy3WDKs8,1130
 truefoundry/deploy/lib/dao/workspace.py,sha256=jm8UWytwVajVcrYyHSTCwWYDYl-RHuk0zAf9Caj4GzQ,2356
-truefoundry/deploy/lib/exceptions.py,sha256=ZW44bSwmlcr3GcB10E8Jz-zlgu9wyi8tE7xAFgul_7A,588
+truefoundry/deploy/lib/exceptions.py,sha256=MNYmiEEHAKGVrpO66RBWcvvY6i3tctWJBYuodqAPJvc,604
 truefoundry/deploy/lib/logs_utils.py,sha256=SQxRv3jDDmgHdOUMhlMaAPGYskybnBUMpst7QU_i_sc,1469
 truefoundry/deploy/lib/messages.py,sha256=nhp0bCYf_XpUM68hTq5lBY-__vtEyV2uP7NgnJXJ_Vg,925
 truefoundry/deploy/lib/model/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -130,7 +130,7 @@ truefoundry/workflow/map_task.py,sha256=2m3qGXQ90k9LdS45q8dqCCECc3qr8t2m_LMCVd1m
 truefoundry/workflow/python_task.py,sha256=SRXRLC4vdBqGjhkwuaY39LEWN6iPCpJAuW17URRdWTY,1128
 truefoundry/workflow/task.py,sha256=ToitYiKcNzFCtOVQwz1W8sRjbR97eVS7vQBdbgUQtKg,1779
 truefoundry/workflow/workflow.py,sha256=WaTqUjhwfAXDWu4E5ehuwAxrCbDJkoAf1oWmR2E9Qy0,4575
-truefoundry-0.3.0rc13.dist-info/METADATA,sha256=iWnUhwLSjjS0u5e5vLz1vq94H-8HXnFfnNJkQMpSB9U,2694
-truefoundry-0.3.0rc13.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-truefoundry-0.3.0rc13.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
-truefoundry-0.3.0rc13.dist-info/RECORD,,
+truefoundry-0.3.0rc15.dist-info/METADATA,sha256=31ztTRk4oHRnyjZpc8yXiU_cDG2VfWOV6YfgZF-DRNw,2694
+truefoundry-0.3.0rc15.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+truefoundry-0.3.0rc15.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
+truefoundry-0.3.0rc15.dist-info/RECORD,,