bizyengine 1.2.70__py3-none-any.whl → 1.2.72__py3-none-any.whl

bizyengine/bizy_server/api_client.py

@@ -1227,3 +1227,19 @@ class APIClient:
         except asyncio.exceptions.TimeoutError:
             print("Request to fetch models timed out")
             return []
+
+    async def get_plugin_tmp_token(self, api_key: str = None):
+        url = f"{BIZYAIR_X_SERVER}/auth/plugin_tmp_token"
+        headers, err = self.auth_header(api_key=api_key)
+        if err is not None:
+            print(f"get_plugin_tmp_token: error getting headers: {err}")
+            return None, err
+        try:
+            data, err = await self.do_post(url, headers=headers)
+            if err is not None:
+                print(f"get_plugin_tmp_token error: {err}")
+                return None, err
+            return data["data"], None
+        except aiohttp.ClientError as e:
+            print(f"get_plugin_tmp_token error: {e}")
+            return None, errnos.GET_PLUGIN_TMP_TOKEN

bizyengine/bizy_server/errno.py

@@ -500,3 +500,10 @@ class errnos:
         None,
         {"en": "Streaming connection closed unexpectedly", "zh": "流式连接异常关闭"},
     )
+
+    GET_PLUGIN_TMP_TOKEN = ErrorNo(
+        500,
+        500161,
+        None,
+        {"en": "Failed to get tmp token", "zh": "获取临时令牌失败"},
+    )

bizyengine/bizy_server/server.py

@@ -1342,6 +1342,16 @@ class BizyAirServer:
 
             return OKResponse(resp)
 
+        @self.prompt_server.routes.post(f"/{API_PREFIX}/auth/plugin_tmp_token")
+        async def get_plugin_tmp_token(request):
+            api_key = request.rel_url.query.get("api_key", "")
+
+            resp, err = await self.api_client.get_plugin_tmp_token(api_key=api_key)
+            if err is not None:
+                return ErrResponse(err)
+
+            return OKResponse(resp)
+
     async def send_json(self, event, data, sid=None):
         message = {"type": event, "data": data}
 

bizyengine/bizyair_extras/third_party_api/nodes_doubao.py

@@ -152,6 +152,137 @@ class Seedream4(BizyAirTrdApiBaseNode):
         return (images,)
 
 
+class Seedream4_5(BizyAirTrdApiBaseNode):
+    RETURN_TYPES = ("IMAGE",)
+    RETURN_NAMES = ("images",)
+    CATEGORY = "☁️BizyAir/External APIs/Doubao"
+    NODE_DISPLAY_NAME = "Seedream 4.5"
+
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "prompt": (
+                    "STRING",
+                    {
+                        "multiline": True,
+                        "default": "",
+                    },
+                ),
+                "model": (
+                    ["doubao-seedream-4-5-251128"],
+                    {"default": "doubao-seedream-4-5-251128"},
+                ),
+                "size": (
+                    [
+                        "2K",
+                        "4K",
+                        "Custom",
+                    ],
+                    {
+                        "default": "2K",
+                    },
+                ),
+                "custom_width": (
+                    "INT",
+                    {
+                        "default": 2048,
+                        "min": 480,
+                        "max": 4096,
+                        "tooltip": "总像素取值范围 [2560x1440=3686400, 4096x4096=16777216], 宽高比取值范围：[1/16, 16]",
+                    },
+                ),
+                "custom_height": (
+                    "INT",
+                    {
+                        "default": 2048,
+                        "min": 480,
+                        "max": 4096,
+                        "tooltip": "总像素取值范围 [2560x1440=3686400, 4096x4096=16777216], 宽高比取值范围：[1/16, 16]",
+                    },
+                ),
+                "max_images": (
+                    "INT",
+                    {
+                        "default": 1,
+                        "min": 1,
+                        "max": 15,
+                        "tooltip": "实际可生成的图片数量，除受到 max_images 影响外，还受到输入的参考图数量影响。输入的参考图数量+最终生成的图片数量≤15张。",
+                    },
+                ),
+                "optimize_prompt": (
+                    ["disabled", "standard"],
+                    {
+                        "default": "disabled",
+                        "tooltip": "提示词优化功能的配置。standard：标准模式，生成内容的质量更高，耗时较长。",
+                    },
+                ),
+            },
+            "optional": {
+                "images": (
+                    "IMAGE",
+                    {"tooltip": "参考图，数量不超过14张。会影响组图生成数量"},
+                ),
+            },
+        }
+
+    def handle_inputs(self, headers, prompt_id, **kwargs):
+        # up to 14 ref images
+        # 10mb each
+        # custom total pixels [3686400, 16777216]
+        # custom size ratio [1/16, 16]
+        model = kwargs.get("model", "doubao-seedream-4-0-250828")
+        prompt = kwargs.get("prompt", "")
+        size = kwargs.get("size", "2K")
+        images = kwargs.get("images", None)
+        optimize_prompt = kwargs.get("optimize_prompt", "disabled")
+        max_images = kwargs.get("max_images", 1)
+
+        if images is not None and len(images) > 14:
+            raise ValueError("Maximum number of images is 14")
+        if size == "Custom":
+            width = kwargs.get("custom_width", 2048)
+            height = kwargs.get("custom_height", 2048)
+            if width * height < 3686400 or width * height > 16777216:
+                raise ValueError("Total pixels must be between 3686400 and 16777216")
+            if width / height < 1 / 16 or width / height > 16:
+                raise ValueError("Width/height ratio must be between 1/16 and 16")
+            size = f"{width}x{height}"
+
+        parts = []
+        for batch_number, img in enumerate(images if images is not None else []):
+            if img is not None:
+                bio = tensor_to_bytesio(image=img, total_pixels=4096 * 4096)
+                length = bio.getbuffer().nbytes
+                if length > 10 * 1024 * 1024:
+                    raise ValueError(
+                        "Image size is too large, Seedream 4.5 only supports up to 10MB"
+                    )
+                url = self.upload_file(
+                    bio,
+                    f"{prompt_id}_{batch_number}.png",
+                    headers,
+                )
+                parts.append(url)
+
+        data = {
+            "prompt": prompt,
+            "size": size,
+            "model": model,
+            "max_images": max_images,
+        }
+        if len(parts) > 0:
+            data["image"] = parts
+        if optimize_prompt != "disabled":
+            data["optimize_prompt"] = optimize_prompt
+
+        return data, model
+
+    def handle_outputs(self, outputs):
+        images = self.combine_images(outputs[1])
+        return (images,)
+
+
 class Seededit3(BizyAirTrdApiBaseNode):
     RETURN_TYPES = ("IMAGE",)
     RETURN_NAMES = ("image",)

bizyengine/version.txt

@@ -1 +1 @@
-1.2.70
+1.2.72

{bizyengine-1.2.70.dist-info → bizyengine-1.2.72.dist-info}/METADATA

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

@@ -1,13 +1,13 @@
 bizyengine/__init__.py,sha256=GP9V-JM07fz7uv_qTB43QEA2rKdrVJxi5I7LRnn_3ZQ,914
-bizyengine/version.txt,sha256=uk2qWzu3m_QjuvcgyNyzSyiqDzc3zRKB23GUBdFjPNU,7
+bizyengine/version.txt,sha256=a7iR7IBhAjmgseNlnvFyQ7DcWKZVPC_fBtzuP4Q3RPY,7
 bizyengine/bizy_server/__init__.py,sha256=SP9oSblnPo4KQyh7yOGD26YCskFAcQHAZy04nQBNRIw,200
-bizyengine/bizy_server/api_client.py,sha256=Z7G5IjaEqSJkF6nLLw2R3bpgBAOi5ClQiUbel6NMXmE,43932
-bizyengine/bizy_server/errno.py,sha256=RIyvegX3lzpx_1L1q2XVvu3on0kvYgKiUQ8U3ZtyF68,16823
+bizyengine/bizy_server/api_client.py,sha256=fF2VApJjPcPWGQwYWSEaIN12c8WT95lj1_QHX83WbeQ,44630
+bizyengine/bizy_server/errno.py,sha256=ikb4Z3MRMTjosi9AC1epaOwE3kJsD6CcFcFp56AKF-Y,16986
 bizyengine/bizy_server/error_handler.py,sha256=MGrfO1AEqbfEgMWPL8B6Ypew_zHiQAdYGlhN9bZohrY,167
 bizyengine/bizy_server/execution.py,sha256=ayaEf6eGJKQsVZV-1_UlGlvwwmlH7FEek31Uq-MbUjA,1644
 bizyengine/bizy_server/profile.py,sha256=f4juAzJ73gCm0AhagYpt9WnG8HEI6xze_U96-omBLqU,3044
 bizyengine/bizy_server/resp.py,sha256=iOFT5Ud7VJBP2uqkojJIgc3y2ifMjjEXoj0ewneL9lc,710
-bizyengine/bizy_server/server.py,sha256=ZLWTDRtTGV601JGcJYJsp_NUgsJyKoDdR7KrStLzW5g,57490
+bizyengine/bizy_server/server.py,sha256=tdr_Um6aRed1YVMo7D9TpUyAu1ZQEyiHbQhyDfhtVoY,57877
 bizyengine/bizy_server/stream_response.py,sha256=H2XHqlVRtQMhgdztAuG7l8-iV_Pm42u2x6WJ0gNVIW0,9654
 bizyengine/bizy_server/utils.py,sha256=t3y3ZTDzFa8K4wXlzgLVaFNCizgylsKsd9K3rLL4sGw,3986
 bizyengine/bizyair_extras/__init__.py,sha256=9iPmEyR7F1IXbUaBNS90ivW9ul18GcuWFxRfFv2ieAw,1011
@@ -46,7 +46,7 @@ bizyengine/bizyair_extras/nodes_ipadapter_plus/nodes_ipadapter_plus.py,sha256=lO
 bizyengine/bizyair_extras/nodes_kolors_mz/__init__.py,sha256=HsCCCphW8q0SrWEiFlZKK_W2lQr1T0UJIJL7gEn37ME,3729
 bizyengine/bizyair_extras/oauth_callback/main.py,sha256=KQOZWor3kyNx8xvUNHYNMoHfCF9g_ht13_iPk4K_5YM,3633
 bizyengine/bizyair_extras/third_party_api/__init__.py,sha256=etiPBCIxOBD6hbrVhdivaRVOPrAp6z79YDWyJgqr_b4,320
-bizyengine/bizyair_extras/third_party_api/nodes_doubao.py,sha256=w3U9ZGtEplrp414bPlQOW8az_TbDiWydrrKbCGMIjVo,14348
+bizyengine/bizyair_extras/third_party_api/nodes_doubao.py,sha256=FtpAnxOuAND_xUgrdovw6hz5wHUm4VC92pd-fynpJpM,19239
 bizyengine/bizyair_extras/third_party_api/nodes_flux.py,sha256=9o7imVDn9qXnUokkAjTH9yNdpXwcuQ8p3rsv5Y5eywQ,6135
 bizyengine/bizyair_extras/third_party_api/nodes_gemini.py,sha256=-P6PO0-qwwUbOYaqbS6SSg3w6_yAkLpBtL3mNq0PNSc,14587
 bizyengine/bizyair_extras/third_party_api/nodes_gpt.py,sha256=pvIlwjjHnk_XCa4eJERBcsWonaBd24xP0jFVQLJXdQc,3098
@@ -103,7 +103,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=nXXTPkj4WBvds4EWjI9c-ydeWwmXl8Vwrdu-4Fh62g8,12914
-bizyengine-1.2.70.dist-info/METADATA,sha256=Um3qFw_Q7YFwuzu1E7ChgasGkFVBw5fkw-N7qsYsZ0c,735
-bizyengine-1.2.70.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-bizyengine-1.2.70.dist-info/top_level.txt,sha256=2zapzqxX-we5cRyJkGf9bd5JinRtXp3-_uDI-xCAnc0,11
-bizyengine-1.2.70.dist-info/RECORD,,
+bizyengine-1.2.72.dist-info/METADATA,sha256=avhbzY9k0yg3bnxWYcckNHBkPYIghct_pVI1FfR_OiE,735
+bizyengine-1.2.72.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+bizyengine-1.2.72.dist-info/top_level.txt,sha256=2zapzqxX-we5cRyJkGf9bd5JinRtXp3-_uDI-xCAnc0,11
+bizyengine-1.2.72.dist-info/RECORD,,