bizyengine 1.2.27__py3-none-any.whl → 1.2.29__py3-none-any.whl

bizyengine/bizy_server/server.py

@@ -64,6 +64,28 @@ class BizyAirServer:
         async def get_server_mode(request):
             return OKResponse({"server_mode": BIZYAIR_SERVER_MODE})
 
+        @self.prompt_server.routes.get(f"/{API_PREFIX}/proxy_view")
+        async def proxy_view(request):
+            filename = request.rel_url.query.get("filename", "")
+            if not filename:
+                return aiohttp.web.Response(status=400, text="Missing filename")
+            host = request.url.host
+            port = request.url.port
+            view_url = (
+                f"http://{host}:{port}/view?filename={urllib.parse.quote(filename)}"
+            )
+            async with aiohttp.ClientSession() as session:
+                async with session.get(view_url) as resp:
+                    if resp.status != 200:
+                        return aiohttp.web.Response(
+                            status=resp.status, text="Failed to fetch image"
+                        )
+                    content = await resp.read()
+                    content_type = resp.headers.get(
+                        "Content-Type", "application/octet-stream"
+                    )
+                    return aiohttp.web.Response(body=content, content_type=content_type)
+
         @self.prompt_server.routes.get(f"/{COMMUNITY_API}/model_types")
         async def list_model_types(request):
             return OKResponse(types())

bizyengine/bizyair_extras/nodes_flux.py

@@ -38,6 +38,23 @@ class CLIPTextEncodeFlux(BizyAirBaseNode):
     #     return (new_clip,)
 
 
+class ReferenceLatent(BizyAirBaseNode):
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "conditioning": ("CONDITIONING",),
+            },
+            "optional": {
+                "latent": ("LATENT",),
+            },
+        }
+
+    RETURN_TYPES = ("CONDITIONING",)
+    CATEGORY = "advanced/conditioning/edit_models"
+    DESCRIPTION = "This node sets the guiding latent for an edit model. If the model supports it you can chain multiple to set multiple reference images."
+
+
 class FluxGuidance(BizyAirBaseNode):
     @classmethod
     def INPUT_TYPES(s):

bizyengine/core/configs/models.json

@@ -54,7 +54,8 @@
       "meijia_flux_lora_rank16_bf16.safetensors"
     ]
   },
-  "unet": ["shuttle-3.1-aesthetic.safetensors",{
+  "unet": ["shuttle-3.1-aesthetic.safetensors",
+    "flux1-dev-kontext_fp8_scaled.safetensors", "flux1-dev-kontext-onediff.safetensors", {
     "kolors": [
       "Kolors-Inpainting.safetensors",
       "Kolors.safetensors"

bizyengine/core/configs/models.yaml

@@ -40,6 +40,27 @@ model_rules:
         inputs:
           unet_name:
             - ^flux/flux1-dev.sft$
+  - mode_type: unet
+    base_model: FLUX-kontext
+    describe: flux1-kontext
+    score: 3
+    route: /supernode/bizyair-comfyui-flux1-kontext-fp8
+    nodes:
+      - class_type: UNETLoader
+        inputs:
+          unet_name:
+            - ^flux1-dev-kontext_fp8_scaled.safetensors$
+
+  - mode_type: unet
+    base_model: FLUX-kontext
+    describe: flux1-kontext
+    score: 3
+    route: /supernode/bizyair-comfyui-flux1-kontext-onediff
+    nodes:
+      - class_type: UNETLoader
+        inputs:
+          unet_name:
+            - ^flux1-dev-kontext-onediff.safetensors$
 
   - mode_type: unet
     base_model: FLUX

bizyengine/core/data_types.py

@@ -1,4 +1,7 @@
 # https://docs.comfy.org/essentials/custom_node_datatypes#model-datatypes
+from functools import singledispatch
+from typing import Any, Dict, List, Tuple, Union
+
 # Model datatypes
 MODEL = "BIZYAIR_MODEL"
 CLIP = "BIZYAIR_CLIP"
@@ -31,3 +34,24 @@ def is_model_datatype(datatype):
 # https://docs.comfy.org/essentials/custom_node_images_and_masks
 def is_send_request_datatype(datatype: str) -> bool:
     return datatype in {"IMAGE", "LATENT", "MASK", "STRING", "FLOAT", "INT"}
+
+
+@singledispatch
+def convert_to_custom_type(data: Any) -> Any:
+    return data
+
+
+@convert_to_custom_type.register(str)
+def _(data: str) -> Union[str, Any]:
+    return BIZYAIR_TYPE_MAP.get(data, data)
+
+
+@convert_to_custom_type.register(dict)
+def _(data: Dict) -> Dict:
+    return {k: convert_to_custom_type(v) for k, v in data.items()}
+
+
+@convert_to_custom_type.register(list)
+@convert_to_custom_type.register(tuple)
+def _(data: Union[List, Tuple]) -> Union[List, Tuple]:
+    return type(data)(convert_to_custom_type(item) for item in data)

bizyengine/core/nodes_base.py

@@ -10,7 +10,7 @@ from bizyengine.core.common.env_var import BIZYAIR_DEBUG, BIZYAIR_SERVER_MODE
 from bizyengine.core.configs.conf import config_manager
 
 from .common.client import get_api_key
-from .data_types import is_send_request_datatype
+from .data_types import convert_to_custom_type, is_send_request_datatype
 from .nodes_io import BizyAirNodeIO, create_node_data
 
 try:
@@ -134,6 +134,7 @@ def ensure_hidden_unique_id_and_prompt(org_input_types_func):
         nonlocal original_has_unique_id
 
         result = org_input_types_func()
+        result = convert_to_custom_type(result)
         if "hidden" not in result:
             result["hidden"] = {"unique_id": "UNIQUE_ID"}
         elif "unique_id" not in result["hidden"]:
@@ -156,6 +157,10 @@ class BizyAirBaseNode:
     def __init_subclass__(cls, **kwargs):
         if not cls.CATEGORY.startswith(f"{LOGO}{PREFIX}"):
             cls.CATEGORY = f"{LOGO}{PREFIX}/{cls.CATEGORY}"
+
+        if hasattr(cls, "RETURN_TYPES"):
+            setattr(cls, "RETURN_TYPES", convert_to_custom_type(cls.RETURN_TYPES))
+
         register_node(cls, PREFIX)
         cls.setup_input_types()
 

bizyengine/version.txt

@@ -1 +1 @@
-1.2.27
+1.2.29

{bizyengine-1.2.27.dist-info → bizyengine-1.2.29.dist-info}/METADATA

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

@@ -1,5 +1,5 @@
 bizyengine/__init__.py,sha256=GP9V-JM07fz7uv_qTB43QEA2rKdrVJxi5I7LRnn_3ZQ,914
-bizyengine/version.txt,sha256=hPsbIxO28ragdKYMcOjrjXZov_fwllVD_xQUQMCih60,6
+bizyengine/version.txt,sha256=8UbXkWb3cQmVLROEmDOpmLGeuURRw6434TkE3B5WEfI,6
 bizyengine/bizy_server/__init__.py,sha256=SP9oSblnPo4KQyh7yOGD26YCskFAcQHAZy04nQBNRIw,200
 bizyengine/bizy_server/api_client.py,sha256=t6cwob7hs993oy5WdFcjKtIzfi3S_eUhODdDVv_Hedo,43822
 bizyengine/bizy_server/errno.py,sha256=Q-U96XnZQCuPH_44Om8wnc2-Kh7qFqwLKtox27msU54,16095
@@ -7,7 +7,7 @@ bizyengine/bizy_server/error_handler.py,sha256=MGrfO1AEqbfEgMWPL8B6Ypew_zHiQAdYG
 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=9zCDdtK-0FwA6VYCBhrur99WQs2geNvA2jlfncMnSj0,53520
+bizyengine/bizy_server/server.py,sha256=bPJaLX8AoHPq4u-q5nuwf58jezXNQdsHO_wUibi10Ik,54594
 bizyengine/bizy_server/stream_response.py,sha256=H2XHqlVRtQMhgdztAuG7l8-iV_Pm42u2x6WJ0gNVIW0,9654
 bizyengine/bizy_server/utils.py,sha256=CIoRrCy8Wv4D9FDdr_QtNP03AjWw8M6GSm2QcrVJl6Y,3739
 bizyengine/bizyair_extras/__init__.py,sha256=EjyLEdLpPm5I4Z25YmnoK4PYQDmsEyXtCXQt9UDGmi0,1009
@@ -21,7 +21,7 @@ bizyengine/bizyair_extras/nodes_controlnet.py,sha256=79eM_Q-5iddWamIQo5T47HPchqQ
 bizyengine/bizyair_extras/nodes_custom_sampler.py,sha256=NK-7sdcp8oxJisjTEFfBskknQJF5I1fRwQkJbG3dKfc,3457
 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=jdi8KM_s8a_2mEksHyM1StHVW0Odm0d3JXNOztOOrxA,2157
+bizyengine/bizyair_extras/nodes_flux.py,sha256=ls94kGBuBNgW5c6uhG36iZLk1TTM2TIoTTcpERgEE5E,2683
 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
@@ -44,9 +44,9 @@ 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/core/__init__.py,sha256=EV9ZtTwOHC0S_eNvCu-tltIydfxfMsH59LbgVX4e_1c,359
-bizyengine/core/data_types.py,sha256=U1Ai149lvbVA-z59sfgniES9KSsyFIbDs_vcjpjlUK4,967
+bizyengine/core/data_types.py,sha256=2f7QqqZvhKmXw3kZV1AvXuPTda34b4wXQE9tyO8nUSM,1595
 bizyengine/core/image_utils.py,sha256=--DmQb3R9Ev21MfZG9rfgOGsU2zRywJ-hpiNNN0N8p8,8586
-bizyengine/core/nodes_base.py,sha256=n54RfLRnCb3lNd2euYj-IVQWlf7_CkVr33bzpPmVyEU,8821
+bizyengine/core/nodes_base.py,sha256=h2f_FWTWj6lpdKjwHRuOmoRifqwkV59ovM2ZxPK4h9c,9019
 bizyengine/core/nodes_io.py,sha256=VhwRwYkGp0g3Mh0hx1OSwNZbV06NEV13w2aODSiAm5M,2832
 bizyengine/core/commands/__init__.py,sha256=82yRdMT23RTiZPkFW_G3fVa-fj3-TUAXnj6cnGA3xRA,22
 bizyengine/core/commands/base.py,sha256=TYH9lhr033B2roBLPkWkxcvoz_fW3cdzx_bvP_FI7dg,635
@@ -61,8 +61,8 @@ bizyengine/core/common/client.py,sha256=1Ka8DIjbmD9Gme9c_Q1zwXXueSCP3_OSdEDyGYEo
 bizyengine/core/common/env_var.py,sha256=1EAW3gOXY2bKouCqrGa583vTJRdDasQ1IsFTnzDg7Dk,3450
 bizyengine/core/common/utils.py,sha256=bm-XmSPy83AyjD0v5EfWp6jiO6_5p7rkZ_HQAuVmgmo,3086
 bizyengine/core/configs/conf.py,sha256=D_UWG9SSJnK5EhbrfNFryJQ8hUwwdvhOGlq1TielwpI,3830
-bizyengine/core/configs/models.json,sha256=jCrqQgjVeHugLb191Xay5rg0m3duTVISPp_GxVGQ3HA,2656
-bizyengine/core/configs/models.yaml,sha256=x8TOo-VHMKtrxfWO5_OUtFg6wNqxFQbf_eOWWfRBjAk,9082
+bizyengine/core/configs/models.json,sha256=NqYVmp-ICpr1vFgafWjJMu5dyuRt9ffVUJmov3_-xRY,2746
+bizyengine/core/configs/models.yaml,sha256=ZsHC2vBS8jfhwgT7NId2G7Hf_iJX3Az7ykbPN3h-1EQ,9648
 bizyengine/core/path_utils/__init__.py,sha256=JVpqNHgaKiEtTI8_r47af8GtWHxrtOockQ6Qpzp9_MQ,260
 bizyengine/core/path_utils/path_manager.py,sha256=tRVAcpsYvfWD-tK7khLvNCZayB0wpU9L0tRTH4ZESzM,10549
 bizyengine/core/path_utils/utils.py,sha256=kQfPQjGU27qF9iyzRxLSRg5cMsd-VixuCUldART7cgY,2394
@@ -77,7 +77,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=deQjBgLAkxIr-NaOMm77TcgBT3ExAp0MFm5ehUJ3CGs,6829
-bizyengine-1.2.27.dist-info/METADATA,sha256=ok50MkdSb5QqGRuIQMF7dR44VipfqVZKqm1RzZcAWaU,675
-bizyengine-1.2.27.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-bizyengine-1.2.27.dist-info/top_level.txt,sha256=2zapzqxX-we5cRyJkGf9bd5JinRtXp3-_uDI-xCAnc0,11
-bizyengine-1.2.27.dist-info/RECORD,,
+bizyengine-1.2.29.dist-info/METADATA,sha256=7ephykrOGACrIgVCPXBXRq1SqHhkGJ7r-HmB7geWUT0,675
+bizyengine-1.2.29.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+bizyengine-1.2.29.dist-info/top_level.txt,sha256=2zapzqxX-we5cRyJkGf9bd5JinRtXp3-_uDI-xCAnc0,11
+bizyengine-1.2.29.dist-info/RECORD,,