Handle error

bizyair_extras/nodes_trellis.py

@@ -17,12 +17,7 @@ class BizyAir_IF_TrellisCheckpointLoader(BizyAirBaseNode):
     @classmethod
     def INPUT_TYPES(cls):
         """Define input types with device-specific options."""
-        device_options = []
-        if torch.cuda.is_available():
-            device_options.append("cuda")
-        if hasattr(torch.backends, "mps") and torch.backends.mps.is_available():
-            device_options.append("mps")
-        device_options.append("cpu")
+        device_options = ["cuda"]
 
         return {
             "required": {
@@ -201,5 +196,5 @@ def main(self, url, file_name):
         return (output,)
 
     @classmethod
-    def IS_CHANGED(self, url, file_name):
+    def IS_CHANGED(self, url, file_name, *args, **kwargs):
         return uuid.uuid4().hex

src/bizyair/common/client.py

@@ -133,12 +133,34 @@ def send_request(
             response_data = response.read().decode("utf-8")
     except urllib.error.URLError as e:
         error_message = str(e)
+        response_body = e.read().decode("utf-8") if hasattr(e, "read") else "N/A"
         if verbose:
             print(f"URLError encountered: {error_message}")
+            print(f"Response Body: {response_data}")
+        code, message = "N/A", "N/A"
+        try:
+            response_dict = json.loads(response_body)
+            if isinstance(response_dict, dict):
+                code = response_dict.get("code", "N/A")
+                message = response_dict.get("message", "N/A")
+
+        except json.JSONDecodeError:
+            if verbose:
+                print("Failed to decode response body as JSON.")
+
         if "Unauthorized" in error_message:
             raise PermissionError(
                 "Key is invalid, please refer to https://cloud.siliconflow.cn to get the API key.\n"
-                "If you have the key, please click the 'BizyAir Key' button at the bottom right to set the key."
+                "If you have the key, please click the 'API Key' button at the bottom right to set the key."
+            )
+        elif code != "N/A" and message != "N/A":
+            raise ConnectionError(
+                f"Failed to handle your request: {error_message}.\n"
+                + f"    Error code: {code}.\n"
+                + f"    Error message: {message}\n."
+                + "The cause of this issue may be incorrect parameter status or ongoing background tasks. \n"
+                + "If retrying after waiting for a while still does not resolve the issue, please report it to "
+                + "Bizyair's official support."
             )
         else:
             raise ConnectionError(