bizyengine 1.2.76__py3-none-any.whl → 1.2.78__py3-none-any.whl

bizyengine/bizy_server/api_client.py

@@ -9,6 +9,7 @@ from openai import OpenAI
 import bizyengine.core as core
 from bizyengine.core.common import get_api_key
 from bizyengine.core.common.env_var import (
+    BIZYAIR_DEBUG,
     BIZYAIR_PRODUCTION_TEST,
     BIZYAIR_SERVER_MODE,
     BIZYAIR_X_SERVER,
@@ -62,88 +63,55 @@ class APIClient:
             errnos.INVALID_API_KEY.message = error_message
             return None, errnos.INVALID_API_KEY
 
+    async def _handle_response(self, response, method, url):
+        try:
+            resp_text = await response.text()
+        except Exception:
+            resp_text = ""
+
+        try:
+            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}"
+            )
+            isJson = isinstance(resp_json, dict)
+            return None, ErrorNo(
+                response.status,
+                resp_json.get("code", response.status) if isJson else resp_json,
+                None,
+                {
+                    user_profile.getLang(): (
+                        resp_json.get("message", resp_text) if isJson else resp_json
+                    )
+                },
+            )
+        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, 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) if isJson else resp_json,
-                        None,
-                        {
-                            user_profile.getLang(): (
-                                resp_json.get("message", await response.text())
-                                if isJson
-                                else resp_json
-                            )
-                        },
-                    )
-                return resp_json, None
+                return await self._handle_response(response, "GET", url)
 
     async def do_post(self, url, data=None, headers=None):
         async with await self.get_session() as session:
             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) if isJson else resp_json,
-                        None,
-                        {
-                            user_profile.getLang(): (
-                                resp_json.get("message", await response.text())
-                                if isJson
-                                else resp_json
-                            )
-                        },
-                    )
-                return resp_json, None
+                return await self._handle_response(response, "POST", url)
 
     async def do_put(self, url, data=None, headers=None):
         async with await self.get_session() as session:
             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) if isJson else resp_json,
-                        None,
-                        {
-                            user_profile.getLang(): (
-                                resp_json.get("message", await response.text())
-                                if isJson
-                                else resp_json
-                            )
-                        },
-                    )
-                return resp_json, None
+                return await self._handle_response(response, "PUT", url)
 
     async def do_delete(self, url, data=None, headers=None):
         async with await self.get_session() as session:
             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) if isJson else resp_json,
-                        None,
-                        {
-                            user_profile.getLang(): (
-                                resp_json.get("message", await response.text())
-                                if isJson
-                                else resp_json
-                            )
-                        },
-                    )
-                return resp_json, None
+                return await self._handle_response(response, "DELETE", url)
 
     async def user_info(self) -> tuple[dict | None, ErrorNo | None]:
         headers, err = self.auth_header()
@@ -1216,13 +1184,13 @@ class APIClient:
         try:
             data, err = await self.do_get(url, params=params, headers=headers)
             if err is not None:
-                print(f"fetch_all_models: error fetching models: {err}")
+                print(f"fetch_all_models: error fetching models: {str(err)}")
                 return []
 
             all_models = [model["id"] for model in data["data"]]
             return all_models
         except aiohttp.ClientError as e:
-            print(f"Error fetching models: {e}")
+            print(f"Error fetching models: {str(e)}")
             return []
         except asyncio.exceptions.TimeoutError:
             print("Request to fetch models timed out")
@@ -1243,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

@@ -22,6 +22,9 @@ class ErrorNo:
     def set_message(self, msg, lang="en"):
         self.messages[lang] = msg
 
+    def __str__(self) -> str:
+        return f"ErrorNo(http_status_code={self.http_status_code}, code={self.code}, messages={self.messages}, data={self.data})"
+
     message = property(fget=get_message, fset=set_message)
 
 
@@ -507,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
@@ -951,6 +952,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], "")