bizyengine 1.2.21__py3-none-any.whl → 1.2.23__py3-none-any.whl

bizyengine/bizyair_extras/nodes_wan_i2v.py

@@ -1,5 +1,6 @@
 import json
 import time
+import traceback
 import warnings
 
 import requests
@@ -19,14 +20,14 @@ except ModuleNotFoundError as e:
 
 from ..core import pop_api_key_and_prompt_id
 from ..core.common import client
-from ..core.common.env_var import BIZYAIR_DEBUG
+from ..core.common.env_var import BIZYAIR_DEBUG, BIZYAIR_SERVER_ADDRESS
 from ..core.nodes_base import BizyAirBaseNode
 from ..core.path_utils import compose_model_name
 
 
 class WanApiNodeBase:
     MODEL_ENDPOINTS = {
-        "Wan-AI/Wan2.1-I2V-14B-480P-Diffusers": "https://bizyair-api.siliconflow.cn/x/v1/supernode/faas-wan-i2v-14b-480p-server"
+        "Wan-AI/Wan2.1-I2V-14B-480P-Diffusers": f"{BIZYAIR_SERVER_ADDRESS}/supernode/faas-wan-i2v-14b-480p-server"
     }
 
 
@@ -165,29 +166,36 @@ class Wan_ImageToVideoPipeline(WanApiNodeBase, BizyAirBaseNode):
         payload = {"prompt": request_data}
         if "prompt_id" in kwargs:
             payload["prompt_id"] = kwargs["prompt_id"]
-        response = client.send_request(
+
+        resp_body = client.send_request(
             url=endpoint,
             data=json.dumps(payload).encode(),
             headers=headers,
+            callback=lambda x: x,
         )
-        return response["query_url"]
+        request_id = resp_body["request_id"]
+        return f"{BIZYAIR_SERVER_ADDRESS}/supernode/status?request_id={request_id}"
 
     def _poll_for_completion(self, query_url, **kwargs):
         start_time = time.time()
         headers = self._prepare_headers(api_key=kwargs["api_key"])
 
         while time.time() - start_time < self.MAX_POLLING_TIME:
-            response = requests.get(query_url, headers=headers)
             try:
-                response_data = response.json()
-                if response_data["data_status"] == "COMPLETED":
-                    return response_data
+                resp_body = client.send_request(
+                    method="GET", url=query_url, headers=headers
+                )
+                data = resp_body.get("data", {})
+                if data.get("status") == "Succeed":
+                    result_url = data["data_url"]
+                    result_resp = requests.get(result_url)
+                    return result_resp.json()
+                elif data.get("status") == "Failed":
+                    raise RuntimeError(f"Task failed: {data}")
                 time.sleep(self.POLLING_INTERVAL)
-            except (json.JSONDecodeError, KeyError) as e:
+            except Exception as e:
                 if BIZYAIR_DEBUG:
-                    print(
-                        f"Response parsing error: {e} | Raw response: {response.text}"
-                    )
+                    print(f"Response error: {e}\n{traceback.format_exc()}")
                 time.sleep(self.POLLING_INTERVAL)
 
         raise TimeoutError("Task processing timeout")

bizyengine/version.txt

@@ -1 +1 @@
-1.2.21
+1.2.23

{bizyengine-1.2.21.dist-info → bizyengine-1.2.23.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bizyengine
-Version: 1.2.21
+Version: 1.2.23
 Summary: [a/BizyAir](https://github.com/siliconflow/BizyAir) Comfy Nodes that can run in any environment.
 Author-email: SiliconFlow <yaochi@siliconflow.cn>
 Project-URL: Repository, https://github.com/siliconflow/BizyAir

{bizyengine-1.2.21.dist-info → bizyengine-1.2.23.dist-info}/RECORD

@@ -1,5 +1,5 @@
 bizyengine/__init__.py,sha256=GP9V-JM07fz7uv_qTB43QEA2rKdrVJxi5I7LRnn_3ZQ,914
-bizyengine/version.txt,sha256=WVrTloMXahqXpyQ77FyHu13i8taohRDnlzlYlFuXhHg,7
+bizyengine/version.txt,sha256=JGuCN2s9jW7ASnwYZqwTZiJG7zGn_wNDR1LRqsr2jMc,7
 bizyengine/bizy_server/__init__.py,sha256=SP9oSblnPo4KQyh7yOGD26YCskFAcQHAZy04nQBNRIw,200
 bizyengine/bizy_server/api_client.py,sha256=qY1hQ11AeDf5m647qrYvn3YPBzP6Yaw8n2Z9hTk5Wtg,43624
 bizyengine/bizy_server/errno.py,sha256=Q-U96XnZQCuPH_44Om8wnc2-Kh7qFqwLKtox27msU54,16095
@@ -34,7 +34,7 @@ bizyengine/bizyair_extras/nodes_testing_utils.py,sha256=lYmcyCIkTkQ7WOZfpEPU9wUb
 bizyengine/bizyair_extras/nodes_trellis.py,sha256=YFhn8j3jSafH3ZuLqLqHVUkBqVQrUIxOLsu4Y3WrvSU,7432
 bizyengine/bizyair_extras/nodes_ultimatesdupscale.py,sha256=-_SsLTAWAQDv4uw-4Z7IGP2tXTe73BJ3N5D6RqVVAK4,4133
 bizyengine/bizyair_extras/nodes_upscale_model.py,sha256=lrzA1BFI2w5aEPCmNPMh07s-WDzG-xTT49uU6WCnlP8,1151
-bizyengine/bizyair_extras/nodes_wan_i2v.py,sha256=6is3D-xKoqukehzEaXDzrj4QqHAPdAMFNkFQGnsvr_w,8076
+bizyengine/bizyair_extras/nodes_wan_i2v.py,sha256=3XwcxLHmgrihgXDEzcVOjU6VjqnZa3mErINlY014PFA,8435
 bizyengine/bizyair_extras/nodes_wan_video.py,sha256=d1mCcW9jCj-5Oymmymy0Vz-nwWv36FMGE5Gn-E7Rul4,1632
 bizyengine/bizyair_extras/route_bizyair_tools.py,sha256=DMVc7J0oT3H_cdAzoaog-ulhE7zj__w1t-HlWnIpCg0,2080
 bizyengine/bizyair_extras/nodes_ipadapter_plus/__init__.py,sha256=ECKATm_EKi_4G47-FJI4-3rHO3iiF9FVakfSTE-pooE,36
@@ -75,7 +75,7 @@ bizyengine/misc/route_sam.py,sha256=-bMIR2QalfnszipGxSxvDAHGJa5gPSrjkYPb5baaRg4,
 bizyengine/misc/segment_anything.py,sha256=wNKYwlYPMszfwj23524geFZJjZaG4eye65SGaUnh77I,8941
 bizyengine/misc/supernode.py,sha256=STN9gaxfTSErH8OiHeZa47d8z-G9S0I7fXuJvHQOBFM,4532
 bizyengine/misc/utils.py,sha256=deQjBgLAkxIr-NaOMm77TcgBT3ExAp0MFm5ehUJ3CGs,6829
-bizyengine-1.2.21.dist-info/METADATA,sha256=F9uFWNZlmxmlKMBvI9eE4dtq5rjsjqiJqGbjp-asRAw,675
-bizyengine-1.2.21.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-bizyengine-1.2.21.dist-info/top_level.txt,sha256=2zapzqxX-we5cRyJkGf9bd5JinRtXp3-_uDI-xCAnc0,11
-bizyengine-1.2.21.dist-info/RECORD,,
+bizyengine-1.2.23.dist-info/METADATA,sha256=SuF3jmw6QZxmFWQE3nxhWSggrt8WUPCH7l2nEHDjhaU,675
+bizyengine-1.2.23.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+bizyengine-1.2.23.dist-info/top_level.txt,sha256=2zapzqxX-we5cRyJkGf9bd5JinRtXp3-_uDI-xCAnc0,11
+bizyengine-1.2.23.dist-info/RECORD,,