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

bizyengine/bizyair_extras/nodes_gemini.py

@@ -306,3 +306,160 @@ class NanoBanana(BizyAirBaseNode):
         except Exception as e:
             logging.error(f"Gemini API error: {e}")
             raise e
+
+
+class NanoBananaPro(BizyAirBaseNode):
+    def __init__(self):
+        pass
+
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "prompt": (
+                    "STRING",
+                    {
+                        "multiline": True,
+                        "default": "",
+                    },
+                ),
+                "operation": (
+                    ["generate", "edit", "style_transfer", "object_insertion"],
+                    {
+                        "default": "generate",
+                        "tooltip": "Choose the type of image operation",
+                    },
+                ),
+                "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", "16:9", "9:16", "4:3", "3:4"],
+                    {"default": "1:1", "tooltip": "Output image aspect ratio"},
+                ),
+                "resolution": (["1K", "2K", "4K"], {"default": "1K"}),
+            },
+            "optional": {
+                "images": ("IMAGE",),
+                "quality": (
+                    ["standard", "high"],
+                    {"default": "high", "tooltip": "Image generation quality"},
+                ),
+                "character_consistency": (
+                    "BOOLEAN",
+                    {
+                        "default": True,
+                        "tooltip": "Maintain character consistency across edits",
+                    },
+                ),
+            },
+        }
+
+    RETURN_TYPES = ("IMAGE", "STRING")
+    FUNCTION = "execute"
+    OUTPUT_NODE = False
+    CATEGORY = "☁️BizyAir/External APIs/Gemini"
+
+    def execute(
+        self,
+        prompt,
+        operation,
+        temperature,
+        top_p,
+        seed,
+        max_tokens,
+        aspect_ratio,
+        resolution,
+        images=None,
+        quality=None,
+        character_consistency=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 = []
+            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),
+                            }
+                        }
+                    )
+
+            prompt = build_prompt_for_operation(
+                prompt,
+                operation,
+                has_references=len(parts) > 0,
+                aspect_ratio=aspect_ratio,
+                character_consistency=character_consistency,
+            )
+            if quality == "high":
+                prompt += " Use the highest quality settings available."
+            parts.append({"text": prompt})
+
+            data = {
+                "contents": {
+                    "parts": parts,
+                    "role": "user",
+                },
+                "generationConfig": {
+                    "seed": seed,
+                    "responseModalities": ["TEXT", "IMAGE"],
+                    "temperature": temperature,
+                    "topP": top_p,
+                    "maxOutputTokens": max_tokens,
+                    "imageConfig": {
+                        "imageSize": resolution,
+                        "aspectRatio": aspect_ratio,
+                    },
+                },
+            }
+            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.64
+1.2.66

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

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

@@ -1,5 +1,5 @@
 bizyengine/__init__.py,sha256=GP9V-JM07fz7uv_qTB43QEA2rKdrVJxi5I7LRnn_3ZQ,914
-bizyengine/version.txt,sha256=YIUL7E8BvoLTU2x9ILvIB3uukzbDUbiLiPtnJTt77J8,7
+bizyengine/version.txt,sha256=JfKoIMXW0UfH6cucqqNqD89qwq9gR_SqQ1Ky-4wtr3c,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=XLRzYv9GRW95QAJKXF_djdg8_izW14hyfp7SCYGeD8I,10562
+bizyengine/bizyair_extras/nodes_gemini.py,sha256=yMngTTCmgOV3lB4Wnqi1A-qSk4MFGGTBSbe0hK-o9QE,16097
 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.64.dist-info/METADATA,sha256=tVL9_y_h7QUsyDmvoEuxYtR0dNRtsGLZblfPjOc8lvg,735
-bizyengine-1.2.64.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-bizyengine-1.2.64.dist-info/top_level.txt,sha256=2zapzqxX-we5cRyJkGf9bd5JinRtXp3-_uDI-xCAnc0,11
-bizyengine-1.2.64.dist-info/RECORD,,
+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,,