Merge pull request #57 from gizatechxyz/feature/retrieve-transpilatio…

…n-logs Retrieve version logs
gizatechxyz · May 10, 2024 · 3499804 · 3499804
2 parents ae7bca4 + 4b2836b
commit 3499804
Show file tree

Hide file tree

Showing 16 changed files with 644 additions and 527 deletions.
diff --git a/.github/workflows/onpush.yml b/.github/workflows/onpush.yml
@@ -28,6 +28,9 @@ jobs:
       - name: Lint with ruff
         run: |
           poetry run ruff giza
+      - name: Pre-commit check
+        run: |
+          poetry run pre-commit run --all-files
       - name: Testing
         run: |
           poetry run pytest --cov=giza --cov-report term-missing --cov-fail-under=60
diff --git a/examples/mnist/requirements.txt b/examples/mnist/requirements.txt
@@ -1,5 +1,5 @@
 giza-cli==0.15.1
-onnx==1.14.1
+onnx>=1.16.0
 tf2onnx==1.15.1
 torch==2.1.0
 torchvision==0.16.0
diff --git a/giza/commands/endpoints.py b/giza/commands/endpoints.py
@@ -109,9 +109,13 @@ def list(
         echo.error(f"⛔️Detail -> {info.get('detail')}⛔️")
         echo.error(f"⛔️Status code -> {info.get('status_code')}⛔️")
         echo.error(f"⛔️Error message -> {info.get('content')}⛔️")
-        echo.error(
-            f"⛔️Request ID: Give this to an administrator to trace the error -> {info.get('request_id')}⛔️"
-        ) if info.get("request_id") else None
+        (
+            echo.error(
+                f"⛔️Request ID: Give this to an administrator to trace the error -> {info.get('request_id')}⛔️"
+            )
+            if info.get("request_id")
+            else None
+        )
         if debug:
             raise e
         sys.exit(1)
@@ -155,9 +159,13 @@ def get(
         echo.error(f"⛔️Detail -> {info.get('detail')}⛔️")
         echo.error(f"⛔️Status code -> {info.get('status_code')}⛔️")
         echo.error(f"⛔️Error message -> {info.get('content')}⛔️")
-        echo.error(
-            f"⛔️Request ID: Give this to an administrator to trace the error -> {info.get('request_id')}⛔️"
-        ) if info.get("request_id") else None
+        (
+            echo.error(
+                f"⛔️Request ID: Give this to an administrator to trace the error -> {info.get('request_id')}⛔️"
+            )
+            if info.get("request_id")
+            else None
+        )
         if debug:
             raise e
         sys.exit(1)
@@ -229,9 +237,13 @@ def list_proofs(
         echo.error(f"⛔️Detail -> {info.get('detail')}⛔️")
         echo.error(f"⛔️Status code -> {info.get('status_code')}⛔️")
         echo.error(f"⛔️Error message -> {info.get('content')}⛔️")
-        echo.error(
-            f"⛔️Request ID: Give this to an administrator to trace the error -> {info.get('request_id')}⛔️"
-        ) if info.get("request_id") else None
+        (
+            echo.error(
+                f"⛔️Request ID: Give this to an administrator to trace the error -> {info.get('request_id')}⛔️"
+            )
+            if info.get("request_id")
+            else None
+        )
         if debug:
             raise e
         sys.exit(1)
@@ -278,9 +290,13 @@ def get_proof(
         echo.error(f"⛔️Detail -> {info.get('detail')}⛔️")
         echo.error(f"⛔️Status code -> {info.get('status_code')}⛔️")
         echo.error(f"⛔️Error message -> {info.get('content')}⛔️")
-        echo.error(
-            f"⛔️Request ID: Give this to an administrator to trace the error -> {info.get('request_id')}⛔️"
-        ) if info.get("request_id") else None
+        (
+            echo.error(
+                f"⛔️Request ID: Give this to an administrator to trace the error -> {info.get('request_id')}⛔️"
+            )
+            if info.get("request_id")
+            else None
+        )
         if debug:
             raise e
         sys.exit(1)
@@ -335,9 +351,13 @@ def download_proof(
         echo.error(f"⛔️Detail -> {info.get('detail')}⛔️")
         echo.error(f"⛔️Status code -> {info.get('status_code')}⛔️")
         echo.error(f"⛔️Error message -> {info.get('content')}⛔️")
-        echo.error(
-            f"⛔️Request ID: Give this to an administrator to trace the error -> {info.get('request_id')}⛔️"
-        ) if info.get("request_id") else None
+        (
+            echo.error(
+                f"⛔️Request ID: Give this to an administrator to trace the error -> {info.get('request_id')}⛔️"
+            )
+            if info.get("request_id")
+            else None
+        )
         if debug:
             raise e
         sys.exit(1)

diff --git a/giza/commands/models.py b/giza/commands/models.py
@@ -60,9 +60,13 @@ def get(
         echo.error(f"⛔️Detail -> {info.get('detail')}⛔️")
         echo.error(f"⛔️Status code -> {info.get('status_code')}⛔️")
         echo.error(f"⛔️Error message -> {info.get('content')}⛔️")
-        echo.error(
-            f"⛔️Request ID: Give this to an administrator to trace the error -> {info.get('request_id')}⛔️"
-        ) if info.get("request_id") else None
+        (
+            echo.error(
+                f"⛔️Request ID: Give this to an administrator to trace the error -> {info.get('request_id')}⛔️"
+            )
+            if info.get("request_id")
+            else None
+        )
         if debug:
             raise e
         sys.exit(1)
@@ -110,9 +114,13 @@ def list(
         echo.error(f"⛔️Detail -> {info.get('detail')}⛔️")
         echo.error(f"⛔️Status code -> {info.get('status_code')}⛔️")
         echo.error(f"⛔️Error message -> {info.get('content')}⛔️")
-        echo.error(
-            f"⛔️Request ID: Give this to an administrator to trace the error -> {info.get('request_id')}⛔️"
-        ) if info.get("request_id") else None
+        (
+            echo.error(
+                f"⛔️Request ID: Give this to an administrator to trace the error -> {info.get('request_id')}⛔️"
+            )
+            if info.get("request_id")
+            else None
+        )
         if debug:
             raise e
         sys.exit(1)
@@ -168,9 +176,13 @@ def create(
         echo.error(f"⛔️Detail -> {info.get('detail')}⛔️")
         echo.error(f"⛔️Status code -> {info.get('status_code')}⛔️")
         echo.error(f"⛔️Error message -> {info.get('content')}⛔️")
-        echo.error(
-            f"⛔️Request ID: Give this to an administrator to trace the error -> {info.get('request_id')}⛔️"
-        ) if info.get("request_id") else None
+        (
+            echo.error(
+                f"⛔️Request ID: Give this to an administrator to trace the error -> {info.get('request_id')}⛔️"
+            )
+            if info.get("request_id")
+            else None
+        )
         if debug:
             raise e
         sys.exit(1)

diff --git a/giza/commands/reset_password.py b/giza/commands/reset_password.py
@@ -44,9 +44,13 @@ def handle_http_error(
     echo.error(f"⛔️Detail -> {info.get('detail')}⛔️")
     echo.error(f"⛔️Status code -> {info.get('status_code')}⛔️")
     echo.error(f"⛔️Error message -> {info.get('content')}⛔️")
-    echo.error(
-        f"⛔️Request ID: Give this to an administrator to trace the error -> {info.get('request_id')}⛔️"
-    ) if info.get("request_id") else None
+    (
+        echo.error(
+            f"⛔️Request ID: Give this to an administrator to trace the error -> {info.get('request_id')}⛔️"
+        )
+        if info.get("request_id")
+        else None
+    )
     if debug:
         raise e
     sys.exit(1)
@@ -97,7 +101,9 @@ def reset_password(
     if token is None:
         token = prompt_for_input("Please enter your reset token 🎟️")
 
-    new_password = prompt_for_input("Please enter your new password 🔑", hide_input=True)
+    new_password = prompt_for_input(
+        "Please enter your new password 🔑", hide_input=True
+    )
     confirm_password = prompt_for_input(
         "Please confirm your new password 🔑", hide_input=True
     )

diff --git a/giza/commands/users.py b/giza/commands/users.py
@@ -77,9 +77,13 @@ def create(debug: Optional[bool] = DEBUG_OPTION) -> None:
         echo.error(f"⛔️Detail -> {info.get('detail')}⛔️")
         echo.error(f"⛔️Status code -> {info.get('status_code')}⛔️")
         echo.error(f"⛔️Error message -> {info.get('content')}⛔️")
-        echo.error(
-            f"⛔️Request ID: Give this to an administrator to trace the error -> {info.get('request_id')}⛔️"
-        ) if info.get("request_id") else None
+        (
+            echo.error(
+                f"⛔️Request ID: Give this to an administrator to trace the error -> {info.get('request_id')}⛔️"
+            )
+            if info.get("request_id")
+            else None
+        )
         if debug:
             raise e
         sys.exit(1)
@@ -126,9 +130,13 @@ def login(
         echo.error(f"⛔️Detail -> {info.get('detail')}⛔️")
         echo.error(f"⛔️Status code -> {info.get('status_code')}⛔️")
         echo.error(f"⛔️Error message -> {info.get('content')}⛔️")
-        echo.error(
-            f"⛔️Request ID: Give this to an administrator to trace the error -> {info.get('request_id')}⛔️"
-        ) if info.get("request_id") else None
+        (
+            echo.error(
+                f"⛔️Request ID: Give this to an administrator to trace the error -> {info.get('request_id')}⛔️"
+            )
+            if info.get("request_id")
+            else None
+        )
         if debug:
             raise e
         sys.exit(1)
@@ -170,9 +178,13 @@ def create_api_key(
         echo.error(f"⛔️Detail -> {info.get('detail')}⛔️")
         echo.error(f"⛔️Status code -> {info.get('status_code')}⛔️")
         echo.error(f"⛔️Error message -> {info.get('content')}⛔️")
-        echo.error(
-            f"⛔️Request ID: Give this to an administrator to trace the error -> {info.get('request_id')}⛔️"
-        ) if info.get("request_id") else None
+        (
+            echo.error(
+                f"⛔️Request ID: Give this to an administrator to trace the error -> {info.get('request_id')}⛔️"
+            )
+            if info.get("request_id")
+            else None
+        )
         if debug:
             raise e
         sys.exit(1)
@@ -238,9 +250,13 @@ def resend_email(debug: Optional[bool] = DEBUG_OPTION) -> None:
         echo.error(f"⛔️Detail -> {info.get('detail')}⛔️")
         echo.error(f"⛔️Status code -> {info.get('status_code')}⛔️")
         echo.error(f"⛔️Error message -> {info.get('content')}⛔️")
-        echo.error(
-            f"⛔️Request ID: Give this to an administrator to trace the error -> {info.get('request_id')}⛔️"
-        ) if info.get("request_id") else None
+        (
+            echo.error(
+                f"⛔️Request ID: Give this to an administrator to trace the error -> {info.get('request_id')}⛔️"
+            )
+            if info.get("request_id")
+            else None
+        )
         if debug:
             raise e
         sys.exit(1)

diff --git a/giza/commands/version.py b/giza/commands/version.py
@@ -5,6 +5,7 @@
 from giza.utils.echo import Echo
 
 
+# TODO: Make sure that this works offline, currently it doesn't
 def check_version():
     """
     Check if there is a new version available of the cli in pypi to suggest upgrade

diff --git a/giza/commands/versions.py b/giza/commands/versions.py
@@ -51,7 +51,7 @@ def get(
     with ExceptionHandler(debug=debug):
         client = VersionsClient(API_HOST)
         version: Version = client.get(model_id, version_id)
-    print_json(version.model_dump_json())
+    print_json(version.model_dump_json(exclude={"logs"}))
 
 
 def transpile(

diff --git a/giza/commands/workspaces.py b/giza/commands/workspaces.py
@@ -46,9 +46,13 @@ def get(
         echo.error(f"⛔️Detail -> {info.get('detail')}⛔️")
         echo.error(f"⛔️Status code -> {info.get('status_code')}⛔️")
         echo.error(f"⛔️Error message -> {info.get('content')}⛔️")
-        echo.error(
-            f"⛔️Request ID: Give this to an administrator to trace the error -> {info.get('request_id')}⛔️"
-        ) if info.get("request_id") else None
+        (
+            echo.error(
+                f"⛔️Request ID: Give this to an administrator to trace the error -> {info.get('request_id')}⛔️"
+            )
+            if info.get("request_id")
+            else None
+        )
         if debug:
             raise e
         sys.exit(1)
@@ -114,9 +118,13 @@ def create(
         echo.error(f"⛔️Detail -> {info.get('detail')}⛔️")
         echo.error(f"⛔️Status code -> {info.get('status_code')}⛔️")
         echo.error(f"⛔️Error message -> {info.get('content')}⛔️")
-        echo.error(
-            f"⛔️Request ID: Give this to an administrator to trace the error -> {info.get('request_id')}⛔️"
-        ) if info.get("request_id") else None
+        (
+            echo.error(
+                f"⛔️Request ID: Give this to an administrator to trace the error -> {info.get('request_id')}⛔️"
+            )
+            if info.get("request_id")
+            else None
+        )
         if debug:
             raise e
         sys.exit(1)
@@ -157,9 +165,13 @@ def delete(
         echo.error(f"⛔️Detail -> {info.get('detail')}⛔️")
         echo.error(f"⛔️Status code -> {info.get('status_code')}⛔️")
         echo.error(f"⛔️Error message -> {info.get('content')}⛔️")
-        echo.error(
-            f"⛔️Request ID: Give this to an administrator to trace the error -> {info.get('request_id')}⛔️"
-        ) if info.get("request_id") else None
+        (
+            echo.error(
+                f"⛔️Request ID: Give this to an administrator to trace the error -> {info.get('request_id')}⛔️"
+            )
+            if info.get("request_id")
+            else None
+        )
         if debug:
             raise e
         sys.exit(1)