bizyengine 1.2.28__py3-none-any.whl → 1.2.30__py3-none-any.whl

bizyengine/bizy_server/server.py

@@ -3,6 +3,7 @@ import configparser
 import json
 import logging
 import os
+import random
 import shutil
 import threading
 import time
@@ -14,7 +15,8 @@ import execution
 import openai
 from server import PromptServer
 
-from bizyengine.core.common.env_var import BIZYAIR_SERVER_MODE
+from bizyengine.core.common.env_var import BIZYAIR_SERVER_ADDRESS, BIZYAIR_SERVER_MODE
+from bizyengine.core.configs.conf import ModelRule, config_manager
 
 from .api_client import APIClient
 from .errno import ErrorNo, errnos
@@ -47,6 +49,16 @@ def _get_request_api_key(request_headers):
     return None, None
 
 
+def _get_api_key_from_cookie(cookie_str):
+    if not cookie_str:
+        return None
+    cookies = cookie_str.split("; ")
+    for cookie in cookies:
+        if cookie.startswith("apiKey="):
+            return cookie[7:]
+    return None
+
+
 class BizyAirServer:
     def __init__(self):
         BizyAirServer.instance = self
@@ -64,6 +76,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())
@@ -836,6 +870,60 @@ class BizyAirServer:
 
                 return ErrResponse(errnos.MODEL_API_ERROR)
 
+        @self.prompt_server.routes.post(f"/{MODEL_API}/image-edit")
+        async def image_edit(request):
+            try:
+                request_api_key, err = _get_request_api_key(request.headers)
+                if err:
+                    return ErrResponse(err)
+                auth_header = request.headers.get("Authorization", "")
+                cookie_str = request.headers.get("Cookie", "")
+                cookie_api_key = _get_api_key_from_cookie(cookie_str)
+                request_data = await request.json()
+                target_url = f"{BIZYAIR_SERVER_ADDRESS}{config_manager.get_rules('comfyagent-flux1-kontext')[0].route}"
+                forward_payload = {
+                    "prompt": request_data.get("prompt", "请编辑这张图片"),
+                    "image": request_data.get("image", ""),
+                    "seed": request_data.get("seed", random.randint(1, 9999999999)),
+                    "num_inference_steps": request_data.get("num_inference_steps", 20),
+                    "guidance_scale": request_data.get("guidance_scale", 2.5),
+                }
+                async with aiohttp.ClientSession() as session:
+                    headers = {
+                        "Content-Type": "application/json",
+                        "User-Agent": "BizyAir Server",
+                        "x-bizyair-client-version": "1.2.21",
+                    }
+                    if request_api_key:
+                        headers["Authorization"] = f"Bearer {request_api_key}"
+                    elif cookie_api_key:
+                        headers["Authorization"] = f"Bearer {cookie_api_key}"
+                    elif auth_header:
+                        if not auth_header.startswith("Bearer "):
+                            headers["Authorization"] = f"Bearer {auth_header}"
+                        else:
+                            headers["Authorization"] = auth_header
+                    else:
+                        from bizyengine.core.common.env_var import load_api_key
+
+                        has_key, file_api_key = load_api_key()
+                        if has_key and file_api_key:
+                            headers["Authorization"] = f"Bearer {file_api_key}"
+                            logging.debug("Using API key from api_key.ini file")
+                    async with session.post(
+                        target_url, headers=headers, json=forward_payload, timeout=120
+                    ) as response:
+                        response_data = await response.json()
+                        logging.debug(response_data)
+
+                        if response.status == 200:
+                            return OKResponse(response_data)
+                        else:
+                            return ErrResponse(errnos.MODEL_API_ERROR)
+
+            except Exception:
+                return ErrResponse(errnos.MODEL_API_ERROR)
+
         @self.prompt_server.routes.post(f"/{MODEL_API}/images")
         async def image_generations(request):
             try:

bizyengine/core/configs/models.json

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

bizyengine/core/configs/models.yaml

@@ -51,6 +51,25 @@ model_rules:
           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-kontext
+    describe: flux1-kontext for bizybot only
+    score: 0
+    route: /supernode/comfyagent-flux1-kontext
+    nodes:
+      - class_type: comfyagent-flux1-kontext
+
   - mode_type: unet
     base_model: FLUX
     describe: flux1-schnell

bizyengine/version.txt

@@ -1 +1 @@
-1.2.28
+1.2.30

{bizyengine-1.2.28.dist-info → bizyengine-1.2.30.dist-info}/METADATA

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

@@ -1,5 +1,5 @@
 bizyengine/__init__.py,sha256=GP9V-JM07fz7uv_qTB43QEA2rKdrVJxi5I7LRnn_3ZQ,914
-bizyengine/version.txt,sha256=5L1cydx4tADQd48y4rHkNWBcq8rAyN4XJZ3nQNbzL1Q,6
+bizyengine/version.txt,sha256=JlhCJgvOLCf0knGqzSmKCYF_ZcqtpjyEa5v066aASRM,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=Eyg1wqyvr7n2kQGLIKvPKyhUIPoG92refahTQjkTweo,57815
 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
@@ -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=vQH4xVvybUYuoSWoT8kHLVBapB-ItVzsd0HFjSb56Hc,2705
-bizyengine/core/configs/models.yaml,sha256=i7Om9jajsysqLQxrVt_8D2NbryyaVwhoQXdse7Sf9Ss,9364
+bizyengine/core/configs/models.json,sha256=NqYVmp-ICpr1vFgafWjJMu5dyuRt9ffVUJmov3_-xRY,2746
+bizyengine/core/configs/models.yaml,sha256=K6b6eNMZCM_bHWBME8DlF2FyonD_9qXLBHcxKiP6WYg,9859
 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.28.dist-info/METADATA,sha256=cmzaHXGfelwLTQvVs-T9dETqQ8N5Ji-Z034Bqwe7WAA,675
-bizyengine-1.2.28.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-bizyengine-1.2.28.dist-info/top_level.txt,sha256=2zapzqxX-we5cRyJkGf9bd5JinRtXp3-_uDI-xCAnc0,11
-bizyengine-1.2.28.dist-info/RECORD,,
+bizyengine-1.2.30.dist-info/METADATA,sha256=Sk1xkJ14fUPjoOi-V_qMnI07h-uQGMVp5gKarw8uVTg,675
+bizyengine-1.2.30.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+bizyengine-1.2.30.dist-info/top_level.txt,sha256=2zapzqxX-we5cRyJkGf9bd5JinRtXp3-_uDI-xCAnc0,11
+bizyengine-1.2.30.dist-info/RECORD,,