bizydraft 0.1.26__py3-none-any.whl → 0.1.28__py3-none-any.whl

bizydraft/server.py

@@ -1,6 +1,8 @@
 import os
+from typing import Union
 
 from loguru import logger
+from .workflow_io import parse_workflow_io
 
 try:
     from server import PromptServer
@@ -10,8 +12,7 @@ except ImportError:
     )
     exit(1)
 
-
-from .resp import OKResponse, ErrResponse
+from .resp import OKResponse, ErrResponse, JsonResponse
 
 _API_PREFIX = "bizyair"
 _SERVER_MODE_HC_FLAG = True
@@ -32,7 +33,7 @@ class BizyDraftServer:
 
     def setup_routes(self):
         @self.prompt_server.routes.get(f"/{_API_PREFIX}/are_you_alive")
-        async def are_you_alive(request):
+        async def are_you_alive(request) -> Union[OKResponse, ErrResponse]:
             if _SERVER_MODE_HC_FLAG:
                 return OKResponse()
             return ErrResponse(500)
@@ -40,7 +41,22 @@ class BizyDraftServer:
         @self.prompt_server.routes.post(
             f"/{_API_PREFIX}/are_you_alive_{BIZYAIR_MAGIC_STRING}"
         )
-        async def toggle_are_you_alive(request):
+        async def toggle_are_you_alive(request) -> OKResponse:
             global _SERVER_MODE_HC_FLAG
             _SERVER_MODE_HC_FLAG = not _SERVER_MODE_HC_FLAG
             return OKResponse()
+
+        @self.prompt_server.routes.post(f"/{_API_PREFIX}/workflow_io")
+        async def workflow_io(request) -> Union[JsonResponse, ErrResponse]:
+            try:
+                data = await request.json()
+            except Exception as e:
+                logger.error(f"解析 request.json() 失败: {e}")
+                return ErrResponse(400)
+
+            try:
+                response = parse_workflow_io(data)
+                return JsonResponse(200, response)
+            except Exception as e:
+                logger.error(f"parse_workflow_io 处理失败: {e}")
+                return ErrResponse(500)

bizydraft/workflow_io.py

@@ -0,0 +1,101 @@
+from typing import Any, Dict, List, Set
+
+BASIC_TYPES = (int, float, str, bool)
+
+
+def is_basic_type(val: Any) -> bool:
+    return isinstance(val, BASIC_TYPES)
+
+
+def get_input_params(prompt: Dict[str, Any]) -> List[Dict[str, Any]]:
+    """
+    遍历所有节点inputs，保留基础类型参数。
+    返回格式：[{node_id, class_type, param_name, param_value}]
+    """
+    input_params = []
+    for node_id, node in prompt.items():
+        class_type = node.get("class_type")
+        for k, v in node.get("inputs", {}).items():
+            if is_basic_type(v):
+                input_params.append(
+                    {
+                        "node_id": node_id,
+                        "class_type": class_type,
+                        "param_name": k,
+                        "param_value": v,
+                    }
+                )
+    return input_params
+
+
+def get_leaf_nodes_from_prompt(prompt: Dict[str, Any]) -> Set[str]:
+    """
+    仅通过prompt参数推断叶子节点（即没有被其他节点inputs引用的节点）。
+    返回节点id集合（字符串）。
+    """
+    referenced = set()
+    for node_id, node in prompt.items():
+        for v in node.get("inputs", {}).values():
+            # 如果是引用（如["4", 1]），则v为list且第一个元素为节点id
+            if isinstance(v, list) and len(v) > 0 and isinstance(v[0], str):
+                referenced.add(v[0])
+    all_nodes = set(prompt.keys())
+    leaf_nodes = all_nodes - referenced
+    return leaf_nodes
+
+
+def summarize_params(
+    input_params: List[Dict[str, Any]], exclude_node_ids: Set[str]
+) -> List[Dict[str, Any]]:
+    """
+    按节点类型和参数名整理参数信息，便于输出。
+    exclude_node_ids: 需要从inputs中排除的节点id集合
+    """
+    summary = {}
+    for param in input_params:
+        node_id = param["node_id"]
+        if node_id in exclude_node_ids:
+            continue
+        key = (param["class_type"], node_id)
+        if key not in summary:
+            summary[key] = {
+                "name": param["class_type"],
+                "displayName": param["class_type"],
+                "nodeId": node_id,
+                "params": [],
+            }
+        summary[key]["params"].append(
+            {
+                "name": param["param_name"],
+                "displayName": param["param_name"],
+                "type": type(param["param_value"]).__name__.upper(),
+                "defaultValue": param["param_value"],
+            }
+        )
+    return list(summary.values())
+
+
+def parse_workflow_io(request_data) -> Dict[str, Any]:
+    prompt = request_data.get("prompt", {})
+
+    # 1. 输出参数（推断叶子节点）
+    leaf_nodes = get_leaf_nodes_from_prompt(prompt)
+    output_nodes = []
+    for node_id in leaf_nodes:
+        node = prompt[node_id]
+        class_type = node.get("class_type")
+        output_nodes.append(
+            {
+                "nodeId": node_id,
+                "name": class_type,
+                "displayName": class_type,
+                "params": [],
+            }
+        )
+
+    # 2. 输入参数（排除已作为输出的节点）
+    input_params = get_input_params(prompt)
+    input_summary = summarize_params(input_params, exclude_node_ids=leaf_nodes)
+
+    # 3. 构造response_body.json格式
+    return {"data": {"inputs": input_summary, "outputs": output_nodes}}

{bizydraft-0.1.26.dist-info → bizydraft-0.1.28.dist-info}/METADATA

@@ -1,4 +1,4 @@
 Metadata-Version: 2.4
 Name: bizydraft
-Version: 0.1.26
+Version: 0.1.28
 Summary: bizydraft

{bizydraft-0.1.26.dist-info → bizydraft-0.1.28.dist-info}/RECORD

@@ -4,7 +4,8 @@ bizydraft/hijack_routes.py,sha256=MlietlmTU-Lj7evZo5yZE-ghQJZ9_1vdpyHXAXYfoFw,74
 bizydraft/postload.py,sha256=YfiFjJ7MLN6WNar19lLdAL_3IPr5GyI3G4rbZIiAgXU,562
 bizydraft/prestartup_patch.py,sha256=Rh_D-rEUmPaojSrl8CEBMAhSgwtLSm6gH2Mzmk5NCuQ,316
 bizydraft/resp.py,sha256=8INvKOe5Dgai3peKfqKjrhUoYeuXWXn358w30-_cY-A,369
-bizydraft/server.py,sha256=ES1fBYr_Ufv_YWsynLck1ln9bXTGjK3KXas1CK9WBqM,1358
+bizydraft/server.py,sha256=0Um4lA_UZN-SRD_nW0lo0pekI6hPNuDgQjKB21ZsXcw,2102
+bizydraft/workflow_io.py,sha256=lqLWhqNs5uJZ0IlTGyoEfBk6WjsWm0Btl6bQY-nldLo,3364
 bizydraft/static/js/handleStyle.js,sha256=P2DNMiKHr--ekn3CGNOAHU46OJCCsifejxw97kLmb7Q,1676
 bizydraft/static/js/hookLoadImage.js,sha256=GklOwfIkXhSfCFrHBz4pPqaSnBiKjegbpDknrzvXYOc,5767
 bizydraft/static/js/main.js,sha256=cZ-7wR9T8aNLzIrTjc-g9xVZf7z6TXWl1zhP_wXSSVo,150
@@ -12,7 +13,7 @@ bizydraft/static/js/postEvent.js,sha256=lXo35DUiBmzGFMtOL2ULcjbCscNb5eAZ98_eX7en
 bizydraft/static/js/socket.js,sha256=vewGQLgdJSrjvqptEDxLirZRXZUjWTJDmDdjs4mbf4k,2436
 bizydraft/static/js/tool.js,sha256=Ejyb5GiFhuoCp5_dBvu2suagbxGHCkR2Lg_4rMKN2ls,756
 bizydraft/static/js/uploadFile.js,sha256=5OjLLaCHZkobLltNs27K5GgZBFHZJowBwS-oakLoNW4,4985
-bizydraft-0.1.26.dist-info/METADATA,sha256=rQW4tgpR5Qtc6eHJy8KRVfyAPyFSwZY01Ohug0DbIN4,73
-bizydraft-0.1.26.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-bizydraft-0.1.26.dist-info/top_level.txt,sha256=XtoBq6hjZhXIM7aas4GtPDtAiKo8FdLzMABXW8qqQ8M,10
-bizydraft-0.1.26.dist-info/RECORD,,
+bizydraft-0.1.28.dist-info/METADATA,sha256=gj7ghFoGT71u0GqpFG-lMtXaRGE7spfbvug-GuLW2yY,73
+bizydraft-0.1.28.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+bizydraft-0.1.28.dist-info/top_level.txt,sha256=XtoBq6hjZhXIM7aas4GtPDtAiKo8FdLzMABXW8qqQ8M,10
+bizydraft-0.1.28.dist-info/RECORD,,