bizyengine 1.2.66__py3-none-any.whl → 1.2.67__py3-none-any.whl

bizyengine/bizyair_extras/nodes_gemini.py

@@ -463,3 +463,143 @@ class NanoBananaPro(BizyAirBaseNode):
         except Exception as e:
             logging.error(f"Gemini API error: {e}")
             raise e
+
+
+class NanoBananaProOfficial(BizyAirBaseNode):
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "prompt": (
+                    "STRING",
+                    {
+                        "multiline": True,
+                        "default": "",
+                    },
+                ),
+                "temperature": (
+                    "FLOAT",
+                    {"default": 1.0, "min": 0.0, "max": 1.0, "step": 0.05},
+                ),
+                "top_p": (
+                    "FLOAT",
+                    {"default": 1.0, "min": 0.0, "max": 1.0, "step": 0.05},
+                ),
+                "seed": ("INT", {"default": 0, "min": 0, "max": 2147483647}),
+                "max_tokens": ("INT", {"default": 32768, "min": 1, "max": 32768}),
+                "aspect_ratio": (
+                    [
+                        "1:1",
+                        "2:3",
+                        "3:2",
+                        "3:4",
+                        "4:3",
+                        "4:5",
+                        "5:4",
+                        "9:16",
+                        "16:9",
+                        "21:9",
+                        "auto",
+                    ],
+                    {"default": "auto", "tooltip": "Output image aspect ratio"},
+                ),
+                "resolution": (["1K", "2K", "4K", "auto"], {"default": "auto"}),
+            },
+            "optional": {
+                "images": ("IMAGE",),
+            },
+        }
+
+    RETURN_TYPES = ("IMAGE", "STRING")
+    FUNCTION = "execute"
+    OUTPUT_NODE = False
+    CATEGORY = "☁️BizyAir/External APIs/Gemini"
+    NODE_DISPLAY_NAME = "NanoBananaPro (Official Parameters)"
+
+    def execute(
+        self,
+        prompt,
+        temperature,
+        top_p,
+        seed,
+        max_tokens,
+        aspect_ratio,
+        resolution,
+        images=None,
+        **kwargs,
+    ):
+        try:
+            url = f"{BIZYAIR_SERVER_ADDRESS}/proxy_inference/VertexAI/gemini-3-pro-image-preview"
+            extra_data = pop_api_key_and_prompt_id(kwargs)
+
+            if images is not None and len(images) > 14:
+                raise ValueError("Maximum number of images is 14")
+            parts = [{"text": prompt}]
+            for batch_number, image in enumerate(images if images is not None else []):
+                if image is not None:
+                    parts.append(
+                        {
+                            "inline_data": {
+                                "mime_type": "image/png",
+                                "data": tensor_to_base64_string(image),
+                            }
+                        }
+                    )
+
+            image_config = {}
+            if resolution != "auto":
+                image_config["imageSize"] = resolution
+            if aspect_ratio != "auto":
+                image_config["aspectRatio"] = aspect_ratio
+            generation_config = {
+                "seed": seed,
+                "responseModalities": ["TEXT", "IMAGE"],
+                "temperature": temperature,
+                "topP": top_p,
+                "maxOutputTokens": max_tokens,
+            }
+            if image_config != {}:
+                generation_config["imageConfig"] = image_config
+            logging.info(f"Generation config: {generation_config}")
+            data = {
+                "contents": {
+                    "parts": parts,
+                    "role": "user",
+                },
+                "generationConfig": generation_config,
+            }
+            json_payload = json.dumps(data).encode("utf-8")
+            headers = client.headers(api_key=extra_data["api_key"])
+            headers["X-BIZYAIR-PROMPT-ID"] = extra_data[
+                "prompt_id"
+            ]  # 额外参数vertexai会拒绝，所以用请求头传
+            resp = client.send_request(
+                url=url,
+                data=json_payload,
+                headers=headers,
+            )
+            # 解析潜在错误
+            prompt_feedback = resp.get("promptFeedback", None)
+            if prompt_feedback:
+                logging.error(f"Response: {resp}")
+                raise ValueError(f"Prompt blocked: {prompt_feedback}")
+            if len(resp.get("candidates", [])) == 0:
+                logging.error(f"Response: {resp}")
+                raise ValueError("No candidates found in response")
+            if resp["candidates"][0]["finishReason"] != "STOP":
+                logging.error(f"Response: {resp}")
+                raise ValueError(
+                    f"Erroneous finish reason: {resp['candidates'][0]['finishReason']}"
+                )
+
+            # 解析文本
+            text = get_text_from_response(resp)
+
+            # 解析base64图片
+            image = get_image_from_response(resp)
+
+            return (image, text)
+
+        except Exception as e:
+            logging.error(f"Gemini API error: {e}")
+            raise e

bizyengine/version.txt

@@ -1 +1 @@
-1.2.66
+1.2.67

{bizyengine-1.2.66.dist-info → bizyengine-1.2.67.dist-info}/METADATA

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

@@ -1,5 +1,5 @@
 bizyengine/__init__.py,sha256=GP9V-JM07fz7uv_qTB43QEA2rKdrVJxi5I7LRnn_3ZQ,914
-bizyengine/version.txt,sha256=JfKoIMXW0UfH6cucqqNqD89qwq9gR_SqQ1Ky-4wtr3c,7
+bizyengine/version.txt,sha256=HO17D2oKn4i7o0xBaxcgdF_1EGXo3mJ-AFhW9WyJaYo,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
@@ -22,7 +22,7 @@ bizyengine/bizyair_extras/nodes_custom_sampler.py,sha256=NK-7sdcp8oxJisjTEFfBskk
 bizyengine/bizyair_extras/nodes_dataset.py,sha256=htF0YZb_FHncLhLDEbJfNCVqJ6rvlo1ZLk7iY42Rylc,3440
 bizyengine/bizyair_extras/nodes_differential_diffusion.py,sha256=nSrbD-w0XtrwktwzME5M0Vmi1sI7Z08AqwgymTdThqo,370
 bizyengine/bizyair_extras/nodes_flux.py,sha256=ls94kGBuBNgW5c6uhG36iZLk1TTM2TIoTTcpERgEE5E,2683
-bizyengine/bizyair_extras/nodes_gemini.py,sha256=yMngTTCmgOV3lB4Wnqi1A-qSk4MFGGTBSbe0hK-o9QE,16097
+bizyengine/bizyair_extras/nodes_gemini.py,sha256=i6_YqG4n7aVdqVtDhMbNeP0SZqp21JYe1MBqISe5LPQ,21054
 bizyengine/bizyair_extras/nodes_hunyuan3d.py,sha256=dWHLeqX68N7zKnfDMzm9nutmCNtFT6-wwt7P5cPDu7Q,2058
 bizyengine/bizyair_extras/nodes_image_utils.py,sha256=BldF_CKD2M01K8-SnG-QV86u3HZqFz_GP5GrCQ5CFDQ,2875
 bizyengine/bizyair_extras/nodes_ip2p.py,sha256=GSEFJvrs4f2tv0xwYkWqc8uhsXrzAJVPvvwcw0gTjR0,619
@@ -97,7 +97,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.66.dist-info/METADATA,sha256=SV507-RvZb4rbAXDX8mWZ9Lp3n-6XK86OYaIqIEF524,735
-bizyengine-1.2.66.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-bizyengine-1.2.66.dist-info/top_level.txt,sha256=2zapzqxX-we5cRyJkGf9bd5JinRtXp3-_uDI-xCAnc0,11
-bizyengine-1.2.66.dist-info/RECORD,,
+bizyengine-1.2.67.dist-info/METADATA,sha256=ws1b3lePVEyV62m-FLAO6M3irolR1ErxM6pR2Tarz88,735
+bizyengine-1.2.67.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+bizyengine-1.2.67.dist-info/top_level.txt,sha256=2zapzqxX-we5cRyJkGf9bd5JinRtXp3-_uDI-xCAnc0,11
+bizyengine-1.2.67.dist-info/RECORD,,