This commit is contained in:
jyong
2025-05-28 17:56:04 +08:00
parent 5fc2bc58a9
commit 7f59ffe7af
32 changed files with 680 additions and 202 deletions

View File

@ -173,7 +173,7 @@ class GraphEngine:
)
return
elif isinstance(item, NodeRunSucceededEvent):
if item.node_type == NodeType.END:
if item.node_type in (NodeType.END, NodeType.KNOWLEDGE_INDEX):
self.graph_runtime_state.outputs = (
dict(item.route_node_state.node_run_result.outputs)
if item.route_node_state.node_run_result
@ -319,7 +319,7 @@ class GraphEngine:
# It may not be necessary, but it is necessary. :)
if (
self.graph.node_id_config_mapping[next_node_id].get("data", {}).get("type", "").lower()
== NodeType.END.value
in [NodeType.END.value, NodeType.KNOWLEDGE_INDEX.value]
):
break