truefoundry 0.3.0rc13__tar.gz → 0.3.0rc15__tar.gz

{truefoundry-0.3.0rc13 → truefoundry-0.3.0rc15}/PKG-INFO

@@ -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 → truefoundry-0.3.0rc15}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "truefoundry"
-version = "0.3.0rc13"
+version = "0.3.0rc15"
 description = "Truefoundry CLI"
 authors = ["Abhishek Choudhary <abhishek@truefoundry.com>"]
 readme = "README.md"
@@ -17,7 +17,7 @@ GitPython = ">=3.1.43,<3.2.0"
 importlib-metadata = ">=6.0.1,<8.0.0"
 importlib-resources = "^5.2.0"
 Mako = "^1.1.6"
-mlfoundry = { version = "0.11.4", optional = true }
+mlfoundry = { version = "0.11.5", optional = true }
 openai = ">=1.16.2,<2.0.0"
 packaging = ">=20.0,<25.0"
 pydantic = ">=1.10.0,<3"
@@ -48,6 +48,15 @@ pre-commit = ">=3.5.0,<4.0.0"
 pytest = ">=8.2.2,<9.0.0"
 # This is only added to resolve linting in truefoundry.langchain package
 langchain = ">=0.2.0,<0.3.0"
+# These are only for mlfoundry tests
+matplotlib = ">=3.0.0,<4.0.0"
+plotly = ">=5.8.2,<6.0.0"
+seaborn = ">=0.13.0,<0.14.0"
+shap =  [
+    { version = ">=0.40.0", python = "<3.12" },
+    { version = ">=0.45.0", python = ">=3.12" },
+]
+Pillow = ">=9.1.1,<11.0.0"
 
 [tool.poetry-dynamic-versioning]
 enable = false

{truefoundry-0.3.0rc13 → truefoundry-0.3.0rc15}/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-0.3.0rc13 → truefoundry-0.3.0rc15}/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):