From 7d9ab8af8ba77090bb141b0632578aea334eaaf3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=96=B9=E5=8D=9A?= <1016318004@qq.com> Date: Mon, 24 Feb 2025 09:32:51 +0800 Subject: [PATCH] =?UTF-8?q?=E8=A1=A5=E5=85=85=E7=BC=BA=E5=A4=B1=E5=8F=82?= =?UTF-8?q?=E6=95=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/scheduler/pool/loader/flow.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/apps/scheduler/pool/loader/flow.py b/apps/scheduler/pool/loader/flow.py index c455a030a..c0d39df34 100644 --- a/apps/scheduler/pool/loader/flow.py +++ b/apps/scheduler/pool/loader/flow.py @@ -29,6 +29,19 @@ async def search_step_type(node_id: str) -> str: return "" return call_id +async def search_step_name(node_id: str) -> str: + node_collection = MongoDB.get_collection("node") + # 查询 Node 集合获取对应的 call_id + node_doc = await node_collection.find_one({"_id": node_id}) + if not node_doc: + LOGGER.error(f"Node {node_id} not found") + return "" + call_id = node_doc.get("name") + if not call_id: + LOGGER.error(f"Node {node_id} has no associated call_id") + return "" + return call_id + class FlowLoader: """工作流加载器""" @@ -63,8 +76,10 @@ class FlowLoader: for step in flow_yaml["steps"]: if step["node"] in ["start", "end"]: step["type"] = step["node"] + step["name"] = step["node"] else: step["type"] = await search_step_type(step["node"]) + step["name"] = await search_step_name(step["node"]) try: # 检查Flow格式，并转换为Flow对象 -- Gitee