bizyengine 1.2.77__py3-none-any.whl → 1.2.79__py3-none-any.whl

bizyengine/bizy_server/api_client.py

@@ -1,6 +1,7 @@
 import asyncio
 import json
 import os
+import urllib
 
 import aiohttp
 from openai import OpenAI
@@ -72,7 +73,6 @@ class APIClient:
             resp_json = json.loads(resp_text)
         except Exception:
             resp_json = resp_text
-
         if response.status != 200:
             print(
                 f"\033[31m[BizyAir]\033[0m API Error: {method} {url} {response.status} Body: {resp_text}"
@@ -91,8 +91,11 @@ class APIClient:
         return resp_json, None
 
     async def do_get(self, url, params=None, headers=None):
+        if params:
+            query_string = urllib.parse.urlencode(params, doseq=True)
+            url = f"{url}?{query_string}"
         async with await self.get_session() as session:
-            async with session.get(url, params=params, headers=headers) as response:
+            async with session.get(url, headers=headers) as response:
                 return await self._handle_response(response, "GET", url)
 
     async def do_post(self, url, data=None, headers=None):
@@ -1208,3 +1211,22 @@ class APIClient:
         except aiohttp.ClientError as e:
             print(f"get_plugin_tmp_token error: {e}")
             return None, errnos.GET_PLUGIN_TMP_TOKEN
+
+    async def trd_api_pricing(self, model: str, data, request_api_key: str = None):
+        server_url = f"{BIZYAIR_Y_SERVER}/trd_api/node/{model}/price"
+
+        headers, err = self.auth_header(api_key=request_api_key)
+        if err is not None:
+            return None, err
+
+        try:
+            ret, err = await self.do_post(server_url, headers=headers, data=data)
+            if err is not None:
+                return None, err
+
+            return ret["data"], None
+        except Exception as e:
+            print(
+                f"\033[31m[BizyAir]\033[0m Fail to get trd api node {model} pricing: {str(e)}"
+            )
+            return None, errnos.TRD_API_PRICING

bizyengine/bizy_server/errno.py

@@ -510,3 +510,10 @@ class errnos:
         None,
         {"en": "Failed to get tmp token", "zh": "获取临时令牌失败"},
     )
+
+    TRD_API_PRICING = ErrorNo(
+        500,
+        500162,
+        None,
+        {"en": "Failed to get trd api node pricing", "zh": "获取第三方API节点定价失败"},
+    )

bizyengine/bizy_server/server.py

@@ -1,4 +1,5 @@
 import asyncio
+import errno
 import json
 import logging
 import re
@@ -50,6 +51,10 @@ if not BIZYAIR_DEBUG:
 def _get_request_api_key(request_headers):
     if BIZYAIR_SERVER_MODE:
         encrypted_api_key = request_headers.get("Authorization")
+        if not encrypted_api_key:
+            # 尝试从cookie中读取auth_token
+            cookie_str = request_headers.get("Cookie")
+            encrypted_api_key = _get_auth_token_from_cookie(cookie_str)
         return decrypt_apikey(encrypted_api_key)
     return None, None
 
@@ -64,6 +69,17 @@ def _get_api_key_from_cookie(cookie_str):
     return None
 
 
+def _get_auth_token_from_cookie(cookie_str):
+    if not cookie_str:
+        return None
+    cookies = cookie_str.split(";")
+    for cookie in cookies:
+        cookie = cookie.strip()
+        if cookie.startswith("auth_token="):
+            return cookie[11:]
+    return None
+
+
 class BizyAirServer:
     def __init__(self):
         BizyAirServer.instance = self
@@ -951,6 +967,24 @@ class BizyAirServer:
                 print(f"\033[31m[BizyAir]\033[0m Chat request failed: {str(e)}")
                 return ErrResponse(errnos.MODEL_API_ERROR)
 
+        @self.prompt_server.routes.post(f"/{API_PREFIX}/trd_api_pricing")
+        async def trd_api_pricing(request):
+            request_api_key, err = _get_request_api_key(request.headers)
+            if err:
+                return ErrResponse(err)
+            model = request.rel_url.query.get("model", "")
+            if not model:
+                return ErrResponse(errnos.INVALID_MODEL_ID)
+            data = {}
+            if request.body_exists:
+                data = await request.json()
+            pricing, err = await self.api_client.trd_api_pricing(
+                model=model, data=data, request_api_key=request_api_key
+            )
+            if err:
+                return ErrResponse(err)
+            return OKResponse(pricing)
+
         # 服务器模式独占
         if BIZYAIR_SERVER_MODE:
             return

bizyengine/bizyair_extras/third_party_api/__init__.py

@@ -8,6 +8,7 @@ modules = [
     ".nodes_sora",
     ".nodes_veo3",
     ".nodes_wan_api",
+    ".nodes_vidu",
 ]
 from bizyengine.core.common.utils import safe_star_import
 

bizyengine/bizyair_extras/third_party_api/nodes_doubao.py

@@ -4,8 +4,11 @@ from .trd_nodes_base import BizyAirTrdApiBaseNode
 
 
 class Seedream4(BizyAirTrdApiBaseNode):
-    RETURN_TYPES = ("IMAGE",)
-    RETURN_NAMES = ("images",)
+    RETURN_TYPES = (
+        "IMAGE",
+        """{"doubao-seedream-4-0-250828": "doubao-seedream-4-0-250828"}""",
+    )
+    RETURN_NAMES = ("images", "bizyair_model_name")
     CATEGORY = "☁️BizyAir/External APIs/Doubao"
     NODE_DISPLAY_NAME = "Seedream4"
 
@@ -149,12 +152,15 @@ class Seedream4(BizyAirTrdApiBaseNode):
 
     def handle_outputs(self, outputs):
         images = self.combine_images(outputs[1])
-        return (images,)
+        return (images, "")
 
 
 class Seedream4_5(BizyAirTrdApiBaseNode):
-    RETURN_TYPES = ("IMAGE",)
-    RETURN_NAMES = ("images",)
+    RETURN_TYPES = (
+        "IMAGE",
+        """{"doubao-seedream-4-5-251128": "doubao-seedream-4-5-251128"}""",
+    )
+    RETURN_NAMES = ("images", "bizyair_model_name")
     CATEGORY = "☁️BizyAir/External APIs/Doubao"
     NODE_DISPLAY_NAME = "Seedream 4.5"
     INPUT_IS_LIST = True
@@ -251,7 +257,8 @@ class Seedream4_5(BizyAirTrdApiBaseNode):
         # 多图的情况可以认为图片输入都是List[Image Batch]
         total_input_images = 0
         for _, img_batch in enumerate(images if images is not None else []):
-            total_input_images += img_batch.shape[0]
+            if img_batch is not None:
+                total_input_images += img_batch.shape[0]
         extra_images, total_extra_images = self.get_extra_images(**kwargs)
         total_input_images += total_extra_images
         if total_input_images > 14:
@@ -316,12 +323,15 @@ class Seedream4_5(BizyAirTrdApiBaseNode):
 
     def handle_outputs(self, outputs):
         images = self.combine_images(outputs[1])
-        return (images,)
+        return (images, "")
 
 
 class Seededit3(BizyAirTrdApiBaseNode):
-    RETURN_TYPES = ("IMAGE",)
-    RETURN_NAMES = ("image",)
+    RETURN_TYPES = (
+        "IMAGE",
+        """{"doubao-seededit-3-0-i2i-250628": "doubao-seededit-3-0-i2i-250628"}""",
+    )
+    RETURN_NAMES = ("image", "bizyair_model_name")
     CATEGORY = "☁️BizyAir/External APIs/Doubao"
     NODE_DISPLAY_NAME = "Seededit3"
 
@@ -374,12 +384,15 @@ class Seededit3(BizyAirTrdApiBaseNode):
 
     def handle_outputs(self, outputs):
         images = self.combine_images(outputs[1])
-        return (images,)
+        return (images, "")
 
 
 class Seedance_1_0_T2V_API(BizyAirTrdApiBaseNode):
-    RETURN_TYPES = ("VIDEO",)
-    RETURN_NAMES = ("video",)
+    RETURN_TYPES = (
+        "VIDEO",
+        """{"doubao-seedance-1-0-pro-250528": "doubao-seedance-1-0","doubao-seedance-1-0-pro-fast-251015": "doubao-seedance-1-0"}""",
+    )
+    RETURN_NAMES = ("video", "bizyair_model_name")
     CATEGORY = "☁️BizyAir/External APIs/Doubao"
     NODE_DISPLAY_NAME = "Seedance 1.0 Pro Text To Video"
 
@@ -458,12 +471,15 @@ class Seedance_1_0_T2V_API(BizyAirTrdApiBaseNode):
         return data, "doubao-seedance-1-0"
 
     def handle_outputs(self, outputs):
-        return (outputs[0][0],)
+        return (outputs[0][0], "")
 
 
 class Seedance_1_0_I2V_API(BizyAirTrdApiBaseNode):
-    RETURN_TYPES = ("VIDEO",)
-    RETURN_NAMES = ("video",)
+    RETURN_TYPES = (
+        "VIDEO",
+        """{"doubao-seedance-1-0-pro-250528": "doubao-seedance-1-0","doubao-seedance-1-0-pro-fast-251015": "doubao-seedance-1-0"}""",
+    )
+    RETURN_NAMES = ("video", "bizyair_model_name")
     CATEGORY = "☁️BizyAir/External APIs/Doubao"
     NODE_DISPLAY_NAME = "Seedance 1.0 Pro Image To Video"
 
@@ -568,4 +584,4 @@ class Seedance_1_0_I2V_API(BizyAirTrdApiBaseNode):
         return data, "doubao-seedance-1-0"
 
     def handle_outputs(self, outputs):
-        return (outputs[0][0],)
+        return (outputs[0][0], "")

bizyengine/bizyair_extras/third_party_api/nodes_flux.py

@@ -5,8 +5,11 @@ from .trd_nodes_base import BizyAirTrdApiBaseNode
 
 class Flux_Kontext_API(BizyAirTrdApiBaseNode):
     NODE_DISPLAY_NAME = "Flux Kontext API"
-    RETURN_TYPES = ("IMAGE",)
-    RETURN_NAMES = ("image",)
+    RETURN_TYPES = (
+        "IMAGE",
+        """{"flux-kontext-pro": "flux-kontext","flux-kontext-max": "flux-kontext"}""",
+    )
+    RETURN_NAMES = ("image", "bizyair_model_name")
     CATEGORY = "☁️BizyAir/External APIs/Flux"
 
     @classmethod
@@ -58,13 +61,13 @@ class Flux_Kontext_API(BizyAirTrdApiBaseNode):
 
     def handle_outputs(self, outputs):
         images = self.combine_images(outputs[1])
-        return (images,)
+        return (images, "")
 
 
 class Flux_2_API(BizyAirTrdApiBaseNode):
     NODE_DISPLAY_NAME = "Flux 2 API"
-    RETURN_TYPES = ("IMAGE",)
-    RETURN_NAMES = ("image",)
+    RETURN_TYPES = ("IMAGE", """{"flux-2-pro": "flux-2","flux-2-flex": "flux-2"}""")
+    RETURN_NAMES = ("image", "bizyair_model_name")
     CATEGORY = "☁️BizyAir/External APIs/Flux"
 
     @classmethod
@@ -170,4 +173,4 @@ class Flux_2_API(BizyAirTrdApiBaseNode):
 
     def handle_outputs(self, outputs):
         images = self.combine_images(outputs[1])
-        return (images,)
+        return (images, "")

bizyengine/bizyair_extras/third_party_api/nodes_gemini.py

@@ -60,7 +60,12 @@ def build_prompt_for_operation(
 
 class NanoBanana(BizyAirTrdApiBaseNode):
     NODE_DISPLAY_NAME = "NanoBanana"
-    RETURN_TYPES = ("IMAGE", "STRING")
+    RETURN_TYPES = (
+        "IMAGE",
+        "STRING",
+        """{"gemini-2.5-flash-image": "gemini-2.5-flash-image"}""",
+    )
+    RETURN_NAMES = ("image", "string", "bizyair_model_name")
     CATEGORY = "☁️BizyAir/External APIs/Gemini"
 
     @classmethod
@@ -114,6 +119,7 @@ class NanoBanana(BizyAirTrdApiBaseNode):
                     },
                 ),
             },
+            "hidden": {"bizyair_model_name": {"default": "gemini-2.5-flash-image"}},
         }
 
     def handle_inputs(self, headers, prompt_id, **kwargs):
@@ -176,13 +182,17 @@ class NanoBanana(BizyAirTrdApiBaseNode):
             raise ValueError("No image found in response")
         if len(outputs[2]) > 0:
             text = outputs[2][0]
-        return (image, text)
+        return (image, text, "")
 
 
 class NanoBananaPro(BizyAirTrdApiBaseNode):
     NODE_DISPLAY_NAME = "NanoBananaPro"
-    RETURN_TYPES = ("IMAGE", "STRING")
-    RETURN_NAMES = ("image", "string")
+    RETURN_TYPES = (
+        "IMAGE",
+        "STRING",
+        """{"gemini-3-pro-image-preview": "gemini-3-pro-image-preview"}""",
+    )
+    RETURN_NAMES = ("image", "string", "bizyair_model_name")
     CATEGORY = "☁️BizyAir/External APIs/Gemini"
     INPUT_IS_LIST = True
 
@@ -273,7 +283,8 @@ class NanoBananaPro(BizyAirTrdApiBaseNode):
         # 多图的情况可以认为图片输入都是List[Image Batch]
         total_input_images = 0
         for _, img_batch in enumerate(images if images is not None else []):
-            total_input_images += img_batch.shape[0]
+            if img_batch is not None:
+                total_input_images += img_batch.shape[0]
         extra_images, total_extra_images = self.get_extra_images(**kwargs)
         total_input_images += total_extra_images
         if total_input_images > 14:
@@ -333,7 +344,7 @@ class NanoBananaPro(BizyAirTrdApiBaseNode):
             raise ValueError("No image found in response")
         if len(outputs[2]) > 0:
             text = outputs[2][0]
-        return (image, text)
+        return (image, text, "")
 
 
 class NanoBananaProOfficial(BizyAirTrdApiBaseNode):
@@ -390,8 +401,12 @@ class NanoBananaProOfficial(BizyAirTrdApiBaseNode):
             },
         }
 
-    RETURN_TYPES = ("IMAGE", "STRING")
-    RETURN_NAMES = ("image", "string")
+    RETURN_TYPES = (
+        "IMAGE",
+        "STRING",
+        """{"gemini-3-pro-image-preview": "gemini-3-pro-image-preview"}""",
+    )
+    RETURN_NAMES = ("image", "string", "bizyair_model_name")
     CATEGORY = "☁️BizyAir/External APIs/Gemini"
     NODE_DISPLAY_NAME = "NanoBananaPro (Official Parameters)"
     INPUT_IS_LIST = True
@@ -409,7 +424,8 @@ class NanoBananaProOfficial(BizyAirTrdApiBaseNode):
         # 多图的情况可以认为图片输入都是List[Image Batch]
         total_input_images = 0
         for _, img_batch in enumerate(images if images is not None else []):
-            total_input_images += img_batch.shape[0]
+            if img_batch is not None:
+                total_input_images += img_batch.shape[0]
         total_input_images += total_extra_images
         if total_input_images > 14:
             raise ValueError("Maximum number of images is 14")
@@ -457,4 +473,4 @@ class NanoBananaProOfficial(BizyAirTrdApiBaseNode):
             raise ValueError("No image found in response")
         if len(outputs[2]) > 0:
             text = outputs[2][0]
-        return (image, text)
+        return (image, text, "")

bizyengine/bizyair_extras/third_party_api/nodes_gpt.py

@@ -5,8 +5,8 @@ from .trd_nodes_base import BizyAirTrdApiBaseNode
 
 class GPT_IMAGE_1_T2I_API(BizyAirTrdApiBaseNode):
     NODE_DISPLAY_NAME = "GPT Image 1 Text To Image"
-    RETURN_TYPES = ("IMAGE",)
-    RETURN_NAMES = ("images",)
+    RETURN_TYPES = ("IMAGE", """{"gpt-image-1": "gpt-image-1"}""")
+    RETURN_NAMES = ("images", "bizyair_model_name")
     CATEGORY = "☁️BizyAir/External APIs/OpenAI"
 
     @classmethod
@@ -40,13 +40,13 @@ class GPT_IMAGE_1_T2I_API(BizyAirTrdApiBaseNode):
 
     def handle_outputs(self, outputs):
         images = self.combine_images(outputs[1])
-        return (images,)
+        return (images, "")
 
 
 class GPT_IMAGE_EDIT_API(BizyAirTrdApiBaseNode):
     NODE_DISPLAY_NAME = "GPT Image Edit"
-    RETURN_TYPES = ("IMAGE",)
-    RETURN_NAMES = ("images",)
+    RETURN_TYPES = ("IMAGE", """{"gpt-image-1": "gpt-image-1"}""")
+    RETURN_NAMES = ("images", "bizyair_model_name")
     CATEGORY = "☁️BizyAir/External APIs/OpenAI"
 
     @classmethod
@@ -98,4 +98,4 @@ class GPT_IMAGE_EDIT_API(BizyAirTrdApiBaseNode):
 
     def handle_outputs(self, outputs):
         images = self.combine_images(outputs[1])
-        return (images,)
+        return (images, "")

bizyengine/bizyair_extras/third_party_api/nodes_hailuo.py

@@ -28,8 +28,8 @@ class Hailuo2_3_T2V(BizyAirTrdApiBaseNode):
         }
 
     NODE_DISPLAY_NAME = "Hailuo2.3 Text To Video"
-    RETURN_TYPES = ("VIDEO",)
-    RETURN_NAMES = ("video",)
+    RETURN_TYPES = ("VIDEO", """{"MiniMax-Hailuo-2.3-t2v": "MiniMax-Hailuo-2.3-t2v"}""")
+    RETURN_NAMES = ("video", "bizyair_model_name")
     CATEGORY = "☁️BizyAir/External APIs/Hailuo"
 
     def handle_inputs(self, headers, prompt_id, **kwargs):
@@ -50,7 +50,7 @@ class Hailuo2_3_T2V(BizyAirTrdApiBaseNode):
         return data, "MiniMax-Hailuo-2.3-t2v"
 
     def handle_outputs(self, outputs):
-        return (outputs[0][0],)
+        return (outputs[0][0], "")
 
 
 class Hailuo2_3_I2V(BizyAirTrdApiBaseNode):
@@ -79,8 +79,8 @@ class Hailuo2_3_I2V(BizyAirTrdApiBaseNode):
         }
 
     NODE_DISPLAY_NAME = "Hailuo2.3 Image To Video"
-    RETURN_TYPES = ("VIDEO",)
-    RETURN_NAMES = ("video",)
+    RETURN_TYPES = ("VIDEO", """{"MiniMax-Hailuo-2.3-i2v": "MiniMax-Hailuo-2.3-i2v"}""")
+    RETURN_NAMES = ("video", "bizyair_model_name")
     CATEGORY = "☁️BizyAir/External APIs/Hailuo"
 
     def handle_inputs(self, headers, prompt_id, **kwargs):
@@ -112,4 +112,4 @@ class Hailuo2_3_I2V(BizyAirTrdApiBaseNode):
         return data, "MiniMax-Hailuo-2.3-i2v"
 
     def handle_outputs(self, outputs):
-        return (outputs[0][0],)
+        return (outputs[0][0], "")