bizyengine 1.2.43__py3-none-any.whl → 1.2.45__py3-none-any.whl

bizyengine/bizy_server/api_client.py

@@ -258,6 +258,7 @@ class APIClient:
         model_types: list[str] = None,
         base_models: list[str] = None,
         sort: str = None,
+        has_draft: bool = None,
         request_api_key: str = None,
     ) -> tuple[dict | None, ErrorNo | None]:
         server_url = f"{BIZYAIR_X_SERVER}/bizy_models/community"
@@ -270,6 +271,8 @@ class APIClient:
             params["base_models"] = base_models
         if sort:
             params["sort"] = sort
+        if has_draft is not None:
+            params["has_draft"] = has_draft
 
         headers, err = self.auth_header(api_key=request_api_key)
         if err is not None:

bizyengine/bizy_server/server.py

@@ -4,6 +4,7 @@ import json
 import logging
 import os
 import random
+import re
 import shutil
 import threading
 import time
@@ -175,6 +176,10 @@ class BizyAirServer:
             model_types = json_data.get("model_types", [])
             base_models = json_data.get("base_models", [])
             sort = json_data.get("sort", "")
+            has_draft = json_data.get("has_draft", None)
+            if has_draft is not None:
+                has_draft = bool(has_draft)
+
             resp, err = None, None
 
             request_api_key, err = _get_request_api_key(request.headers)
@@ -201,6 +206,7 @@ class BizyAirServer:
                     model_types=model_types,
                     base_models=base_models,
                     sort=sort,
+                    has_draft=has_draft,
                     request_api_key=request_api_key,
                 )
             elif mode == "official":
@@ -760,7 +766,13 @@ class BizyAirServer:
             all_models = await self.api_client.fetch_all_llm_models(
                 request_api_key=request_api_key
             )
-            llm_models = [model for model in all_models if "vl" not in model.lower()]
+            llm_models = [
+                model
+                for model in all_models
+                if not (
+                    re.search(r"\d+(\.\d+)?v", model.lower()) or "vl" in model.lower()
+                )
+            ]
             llm_models.append("No LLM Enhancement")
             return aiohttp.web.json_response(llm_models)
 
@@ -773,7 +785,11 @@ class BizyAirServer:
             all_models = await self.api_client.fetch_all_llm_models(
                 request_api_key=request_api_key
             )
-            vlm_models = [model for model in all_models if "vl" in model.lower()]
+            vlm_models = [
+                model
+                for model in all_models
+                if re.search(r"\d+(\.\d+)?v", model.lower()) or "vl" in model.lower()
+            ]
             vlm_models.append("No VLM Enhancement")
             return aiohttp.web.json_response(vlm_models)
 

bizyengine/misc/utils.py

@@ -185,7 +185,7 @@ def get_vlm_response(
 
     messages = [
         {
-            "role": "user",
+            "role": "system",
             "content": [{"type": "text", "text": system_prompt}],
         },  # 此方法皆适用于两种 VL 模型
         # {

bizyengine/version.txt

@@ -1 +1 @@
-1.2.43
+1.2.45

{bizyengine-1.2.43.dist-info → bizyengine-1.2.45.dist-info}/METADATA

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

@@ -1,13 +1,13 @@
 bizyengine/__init__.py,sha256=GP9V-JM07fz7uv_qTB43QEA2rKdrVJxi5I7LRnn_3ZQ,914
-bizyengine/version.txt,sha256=BtoigMhexC9rsdu6n-cE61VfK18rb-lvtD7WF4nXsdo,7
+bizyengine/version.txt,sha256=8kXgez0b1dAhIv4WEL8cE4ACDp44DIXKSzPEvWD0oAg,7
 bizyengine/bizy_server/__init__.py,sha256=SP9oSblnPo4KQyh7yOGD26YCskFAcQHAZy04nQBNRIw,200
-bizyengine/bizy_server/api_client.py,sha256=t6cwob7hs993oy5WdFcjKtIzfi3S_eUhODdDVv_Hedo,43822
+bizyengine/bizy_server/api_client.py,sha256=Z7G5IjaEqSJkF6nLLw2R3bpgBAOi5ClQiUbel6NMXmE,43932
 bizyengine/bizy_server/errno.py,sha256=1UiFmE2U7r7hCHgsw4-p_YL0VCmTJc9NyYDEbhkanaY,16336
 bizyengine/bizy_server/error_handler.py,sha256=MGrfO1AEqbfEgMWPL8B6Ypew_zHiQAdYGlhN9bZohrY,167
 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=M9P0oungSiUFGRxev3jJxhUTvjm8ixREQ-DCoyzU0gA,58617
+bizyengine/bizy_server/server.py,sha256=4b50TnW4ZSp9rlqhopO6pbqr5Nr0RvLfv3TQmPGMWqA,59062
 bizyengine/bizy_server/stream_response.py,sha256=H2XHqlVRtQMhgdztAuG7l8-iV_Pm42u2x6WJ0gNVIW0,9654
 bizyengine/bizy_server/utils.py,sha256=Kkn-AATZcdaDhg8Rg_EJW6aKqkyiSE2EYmuyOhUwXso,3863
 bizyengine/bizyair_extras/__init__.py,sha256=zzqOeOnd6uyyopeNFPa7asfG-sI4r3SSrKJkHGF4iGI,1064
@@ -78,8 +78,8 @@ bizyengine/misc/nodes_controlnet_union_sdxl.py,sha256=fYyu_XMY7mcX1Ad9x30q1tYB8m
 bizyengine/misc/route_sam.py,sha256=-bMIR2QalfnszipGxSxvDAHGJa5gPSrjkYPb5baaRg4,1561
 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.43.dist-info/METADATA,sha256=HkYWO2J_X38zk2PQjoDVRlDnoHa_lEQaNOP0YwvYBXk,675
-bizyengine-1.2.43.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-bizyengine-1.2.43.dist-info/top_level.txt,sha256=2zapzqxX-we5cRyJkGf9bd5JinRtXp3-_uDI-xCAnc0,11
-bizyengine-1.2.43.dist-info/RECORD,,
+bizyengine/misc/utils.py,sha256=CKduySGSMNGlJMImHyZmN-giABY5VUaB88f6Kq-HAV0,6831
+bizyengine-1.2.45.dist-info/METADATA,sha256=UodDBNwC4vmRZvkTbFi9aBevOnjfUZx1WzxaYnBimG0,675
+bizyengine-1.2.45.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+bizyengine-1.2.45.dist-info/top_level.txt,sha256=2zapzqxX-we5cRyJkGf9bd5JinRtXp3-_uDI-xCAnc0,11
+bizyengine-1.2.45.dist-info/RECORD,,