bizyengine 1.2.5__py3-none-any.whl → 1.2.7__py3-none-any.whl

bizyengine/bizy_server/api_client.py

@@ -65,13 +65,16 @@ class APIClient:
             async with session.get(url, headers=headers) as response:
                 resp_json = await response.json()
                 if response.status != 200:
+                    isJson = isinstance(resp_json, dict)
                     return None, ErrorNo(
                         response.status,
-                        resp_json.get("code", response.status),
+                        resp_json.get("code", response.status) if isJson else resp_json,
                         None,
                         {
-                            user_profile.getLang(): resp_json.get(
-                                "message", await response.text()
+                            user_profile.getLang(): (
+                                resp_json.get("message", await response.text())
+                                if isJson
+                                else resp_json
                             )
                         },
                     )
@@ -82,13 +85,16 @@ class APIClient:
             async with session.post(url, json=data, headers=headers) as response:
                 resp_json = await response.json()
                 if response.status != 200:
+                    isJson = isinstance(resp_json, dict)
                     return None, ErrorNo(
                         response.status,
-                        resp_json.get("code", response.status),
+                        resp_json.get("code", response.status) if isJson else resp_json,
                         None,
                         {
-                            user_profile.getLang(): resp_json.get(
-                                "message", await response.text()
+                            user_profile.getLang(): (
+                                resp_json.get("message", await response.text())
+                                if isJson
+                                else resp_json
                             )
                         },
                     )
@@ -99,13 +105,16 @@ class APIClient:
             async with session.put(url, json=data, headers=headers) as response:
                 resp_json = await response.json()
                 if response.status != 200:
+                    isJson = isinstance(resp_json, dict)
                     return None, ErrorNo(
                         response.status,
-                        resp_json.get("code", response.status),
+                        resp_json.get("code", response.status) if isJson else resp_json,
                         None,
                         {
-                            user_profile.getLang(): resp_json.get(
-                                "message", await response.text()
+                            user_profile.getLang(): (
+                                resp_json.get("message", await response.text())
+                                if isJson
+                                else resp_json
                             )
                         },
                     )
@@ -116,13 +125,16 @@ class APIClient:
             async with session.delete(url, json=data, headers=headers) as response:
                 resp_json = await response.json()
                 if response.status != 200:
+                    isJson = isinstance(resp_json, dict)
                     return None, ErrorNo(
                         response.status,
-                        resp_json.get("code", response.status),
+                        resp_json.get("code", response.status) if isJson else resp_json,
                         None,
                         {
-                            user_profile.getLang(): resp_json.get(
-                                "message", await response.text()
+                            user_profile.getLang(): (
+                                resp_json.get("message", await response.text())
+                                if isJson
+                                else resp_json
                             )
                         },
                     )

bizyengine/bizyair_extras/__init__.py

@@ -22,5 +22,6 @@ from .nodes_testing_utils import *
 from .nodes_trellis import *
 from .nodes_ultimatesdupscale import *
 from .nodes_upscale_model import *
+from .nodes_wan_i2v import *
 from .nodes_wan_video import *
 from .route_bizyair_tools import *

bizyengine/bizyair_extras/nodes_wan_i2v.py

@@ -0,0 +1,222 @@
+import json
+import time
+import warnings
+
+import requests
+
+try:
+    from comfy_api_nodes.apinode_utils import (
+        download_url_to_video_output,
+        tensor_to_base64_string,
+    )
+except ModuleNotFoundError as e:
+    download_url_to_video_output = None
+    tensor_to_base64_string = None
+
+    ERROR_MSG = f"Error {e} ComfyUI API nodes module not found. Please ensure you have ComfyUI version 0.3.36 or later installed."
+
+    warnings.warn(ERROR_MSG)
+
+
+from ..core.common import client
+from ..core.common.env_var import BIZYAIR_DEBUG
+from ..core.nodes_base import BizyAirBaseNode
+
+
+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"
+    }
+
+
+class Wan_LoraLoader(BizyAirBaseNode):
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "lora_name": (
+                    "STRING",
+                    {
+                        "multiline": True,
+                        "default": "https://huggingface.co/Remade-AI/Squish/resolve/main/squish_18.safetensors",
+                        "tooltip": "LoRA 模型下载地址",
+                    },
+                ),
+            },
+            "optional": {
+                "lora_weight": (
+                    "FLOAT",
+                    {
+                        "default": 1,
+                        "min": 0.0,
+                        "max": 1.0,
+                        "step": 0.05,
+                        "tooltip": "LoRA权重强度",
+                    },
+                )
+            },
+        }
+
+    RETURN_TYPES = ("LORA_CONFIG",)
+    RETURN_NAMES = ("lora_config",)
+    FUNCTION = "apply_lora"
+    CATEGORY = "Diffusers/WAN Video Generation"
+
+    def apply_lora(self, lora_name, lora_weight=0.75):
+        return ([(lora_name, lora_weight)],)
+
+
+class Wan_ImageToVideoPipeline(WanApiNodeBase, BizyAirBaseNode):
+
+    POLLING_INTERVAL = 10  # sec
+    MAX_POLLING_TIME = 60 * 20  # sec
+
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "image": (
+                    "IMAGE",
+                    {
+                        "default": None,
+                        "tooltip": "Optional reference image to guide video generation",
+                    },
+                ),
+                "model_id": (["Wan-AI/Wan2.1-I2V-14B-480P-Diffusers"],),
+                "prompt": (
+                    "STRING",
+                    {
+                        "multiline": True,
+                        "default": "",
+                        "tooltip": "Text description of the video",
+                    },
+                ),
+            },
+            "optional": {
+                "negative_prompt": (
+                    "STRING",
+                    {
+                        "multiline": True,
+                        "default": "",
+                        "tooltip": "Negative text prompt to guide what to avoid in the video",
+                    },
+                ),
+                "steps": ("INT", {"default": 30, "min": 1, "max": 40}),
+                "cfg": (
+                    "FLOAT",
+                    {
+                        "default": 6.0,
+                        "min": 0.0,
+                        "max": 100.0,
+                        "step": 0.1,
+                        "round": 0.01,
+                    },
+                ),
+                "seed": (
+                    "INT",
+                    {
+                        "default": 0,
+                        "min": 0,
+                        "max": 0xFFFFFFFF,
+                        "step": 1,
+                        "display": "number",
+                        "control_after_generate": True,
+                        "tooltip": "Seed for video generation (0 for random)",
+                    },
+                ),
+                "use_teacache": (["enable", "disable"],),
+                "lora_config": ("LORA_CONFIG", {}),
+            },
+        }
+
+    RETURN_TYPES = ("VIDEO",)
+
+    FUNCTION = "generate_video"
+    CATEGORY = "Diffusers/WAN Video Generation"
+
+    def _encode_image(self, image_tensor):
+        # https://docs.comfy.org/custom-nodes/backend/snippets
+        tensor_to_base64_string(image_tensor=image_tensor, mime_type="image/webp")
+        base64_str = "data:image/webp;base64," + tensor_to_base64_string(
+            image_tensor=image_tensor, mime_type="image/webp"
+        )
+        return base64_str
+
+    def _prepare_headers(self):
+        headers = client._headers()
+        headers["X-Fn-Task-Mode"] = "non-blocking"
+        return headers
+
+    def _send_initial_request(self, endpoint, request_data):
+        headers = self._prepare_headers()
+        response = client.send_request(
+            url=endpoint,
+            data=json.dumps({"prompt": request_data}).encode(),
+            headers=headers,
+        )
+        return response["query_url"]
+
+    def _poll_for_completion(self, query_url):
+        start_time = time.time()
+        headers = self._prepare_headers()
+
+        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
+                time.sleep(self.POLLING_INTERVAL)
+            except (json.JSONDecodeError, KeyError) as e:
+                if BIZYAIR_DEBUG:
+                    print(
+                        f"Response parsing error: {e} | Raw response: {response.text}"
+                    )
+                time.sleep(self.POLLING_INTERVAL)
+
+        raise TimeoutError("Task processing timeout")
+
+    def _process_result(self, result_data):
+        video_url = result_data["data"]["payload"]
+        return (download_url_to_video_output(video_url),)
+
+    def generate_video(
+        self,
+        model_id: str,
+        prompt: str,
+        negative_prompt: str = "",
+        seed: int = 0,
+        image=None,
+        lora_config=[],
+        use_teacache="enable",
+        **kwargs,
+    ):
+        if download_url_to_video_output is None or tensor_to_base64_string is None:
+            raise ImportError(ERROR_MSG)
+
+        req_dict = {}
+        req_dict["guidance_scale"] = kwargs.pop("cfg", 6.0)
+        req_dict["num_inference_steps"] = kwargs.pop("steps", 30)
+        req_dict["prompt"] = prompt
+        req_dict["negative_prompt"] = negative_prompt
+        req_dict["seed"] = seed
+        if lora_config:
+            if len(lora_config) > 1:
+                raise NotImplementedError(f"TODO, tmp only support one lora")
+            req_dict["lora_name_list"] = [x[0] for x in lora_config]
+            req_dict["lora_weight_list"] = [x[1] for x in lora_config]
+        else:
+            req_dict["lora_name_list"] = []
+            req_dict["lora_weight_list"] = []
+
+        if use_teacache == "enable":
+            req_dict["teacache"] = 0.3
+        else:
+            req_dict["teacache"] = 0
+
+        req_dict["image"] = self._encode_image(image_tensor=image)
+        endpoint = self.MODEL_ENDPOINTS[model_id]
+        query_url = self._send_initial_request(endpoint, request_data=req_dict)
+        result = self._poll_for_completion(query_url)
+        return self._process_result(result)

bizyengine/version.txt

@@ -1 +1 @@
-1.2.5
+1.2.7

{bizyengine-1.2.5.dist-info → bizyengine-1.2.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bizyengine
-Version: 1.2.5
+Version: 1.2.7
 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.5.dist-info → bizyengine-1.2.7.dist-info}/RECORD

@@ -1,7 +1,7 @@
 bizyengine/__init__.py,sha256=GP9V-JM07fz7uv_qTB43QEA2rKdrVJxi5I7LRnn_3ZQ,914
-bizyengine/version.txt,sha256=_it8I3tabkKBiqPk9_xYIPScMiicMsliYO-JqMj209c,6
+bizyengine/version.txt,sha256=2ejoQpxqfQuWXk-sJqXv_3ED8FX6qYzOtb1HMf0kYDo,6
 bizyengine/bizy_server/__init__.py,sha256=SP9oSblnPo4KQyh7yOGD26YCskFAcQHAZy04nQBNRIw,200
-bizyengine/bizy_server/api_client.py,sha256=QFXzq-s2xURE9k4wLz5Hm_dxbyMH5fXhurwSYqJEZyU,39804
+bizyengine/bizy_server/api_client.py,sha256=urdV6GBXRcm-i3yNzjLKQoiN_5CC66Fd1QzXCGSa9CE,40496
 bizyengine/bizy_server/errno.py,sha256=nEr_A6ARwgIwlr1PFP8eg-HNAzz9r7l00fTKaq-ipxM,15826
 bizyengine/bizy_server/error_handler.py,sha256=MGrfO1AEqbfEgMWPL8B6Ypew_zHiQAdYGlhN9bZohrY,167
 bizyengine/bizy_server/execution.py,sha256=ayaEf6eGJKQsVZV-1_UlGlvwwmlH7FEek31Uq-MbUjA,1644
@@ -10,7 +10,7 @@ bizyengine/bizy_server/resp.py,sha256=iOFT5Ud7VJBP2uqkojJIgc3y2ifMjjEXoj0ewneL9l
 bizyengine/bizy_server/server.py,sha256=V67smRHnVRy_vS4o3D1_uSO7to5_3ie8gVEvcIlj6QU,46439
 bizyengine/bizy_server/stream_response.py,sha256=H2XHqlVRtQMhgdztAuG7l8-iV_Pm42u2x6WJ0gNVIW0,9654
 bizyengine/bizy_server/utils.py,sha256=C5tnMhvdtrrvwuCex3oERIGWrTWVb5dkXD1Txb5sJaE,2568
-bizyengine/bizyair_extras/__init__.py,sha256=6Su2AZMMPlBcOdxgy-BWq89m9MJl7voLSWeDcdyagIo,922
+bizyengine/bizyair_extras/__init__.py,sha256=oaKC7QzFbgJ_BWuc3o05gmIq-8ZhUB7VEzYXnlWjGv8,951
 bizyengine/bizyair_extras/nodes_advanced_refluxcontrol.py,sha256=cecfjrtnjJAty9aNkhz8BlmHUC1NImkFlUDiA0COEa4,2242
 bizyengine/bizyair_extras/nodes_cogview4.py,sha256=Ni0TDOycczyDhYPvSR68TxGV_wE2uhaxd8MIj-J4-3o,2031
 bizyengine/bizyair_extras/nodes_comfyui_detail_daemon.py,sha256=i71it24tiGvZ3h-XFWISr4CpZszUtPuz3UrZARYluLk,6169
@@ -34,6 +34,7 @@ bizyengine/bizyair_extras/nodes_testing_utils.py,sha256=qeJHlSq-c21Vx8H0oZw878Q0
 bizyengine/bizyair_extras/nodes_trellis.py,sha256=QU2dQsN_zKA91vTVtR4Af4kJXDJA8CDSf1PyIu_ywZc,7421
 bizyengine/bizyair_extras/nodes_ultimatesdupscale.py,sha256=6IZCDZ_9PhrKwJzCZGvKNxcRYwwkPRMNovF17nwzFqw,4130
 bizyengine/bizyair_extras/nodes_upscale_model.py,sha256=lrzA1BFI2w5aEPCmNPMh07s-WDzG-xTT49uU6WCnlP8,1151
+bizyengine/bizyair_extras/nodes_wan_i2v.py,sha256=mQaZsgjapXMyfmce7_sKI_nZH6A7-cwokdWTEmQVCa0,7345
 bizyengine/bizyair_extras/nodes_wan_video.py,sha256=d1mCcW9jCj-5Oymmymy0Vz-nwWv36FMGE5Gn-E7Rul4,1632
 bizyengine/bizyair_extras/route_bizyair_tools.py,sha256=QdgfiKeF6c4-5Bx5Pmz9RKlaAHreypy9SIg_krmLk-o,2079
 bizyengine/bizyair_extras/nodes_ipadapter_plus/__init__.py,sha256=ECKATm_EKi_4G47-FJI4-3rHO3iiF9FVakfSTE-pooE,36
@@ -74,7 +75,7 @@ bizyengine/misc/route_sam.py,sha256=-bMIR2QalfnszipGxSxvDAHGJa5gPSrjkYPb5baaRg4,
 bizyengine/misc/segment_anything.py,sha256=RRm8FOfDY9VxdVrLjcdzJRh2pSM-kmNcCySuYnx9l7w,8677
 bizyengine/misc/supernode.py,sha256=MPoJN6H_oCV00lmv1LWtGdQwTlyQPI6gXdMDXgkFd7g,4197
 bizyengine/misc/utils.py,sha256=_aO1lHtfDf7Bv0K4xvnZ1Fu5Y9MfMkuhkiS_g8687ng,6461
-bizyengine-1.2.5.dist-info/METADATA,sha256=_TjK2n7NeWTuZBuhivHRScZUbXPJXzPZa6-T8xYwnkA,574
-bizyengine-1.2.5.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
-bizyengine-1.2.5.dist-info/top_level.txt,sha256=2zapzqxX-we5cRyJkGf9bd5JinRtXp3-_uDI-xCAnc0,11
-bizyengine-1.2.5.dist-info/RECORD,,
+bizyengine-1.2.7.dist-info/METADATA,sha256=qo-32z-1Ony-F42-qUOCU0bbmPIfsbBqIrH0-0Cbu10,574
+bizyengine-1.2.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+bizyengine-1.2.7.dist-info/top_level.txt,sha256=2zapzqxX-we5cRyJkGf9bd5JinRtXp3-_uDI-xCAnc0,11
+bizyengine-1.2.7.dist-info/RECORD,,

{bizyengine-1.2.5.dist-info → bizyengine-1.2.7.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.4.0)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: true
 Tag: py3-none-any