adai/MemoryBear
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(web): node execution status reset

Browse Source
This commit is contained in:
zhaoying
2026-04-27 13:47:49 +08:00
parent 610ae27cf9
commit d5d81f0c4f
1 changed files with 7 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
web/src/views/Workflow/hooks/useWorkflowGraph.ts
View File

@@ -2,7 +2,7 @@
* @Author: ZhaoYing * @Author: ZhaoYing
* @Date: 2026-02-03 15:17:48 * @Date: 2026-02-03 15:17:48
* @Last Modified by: ZhaoYing * @Last Modified by: ZhaoYing
* @Last Modified time: 2026-04-24 17:21:09 * @Last Modified time: 2026-04-27 13:47:02
*/ */
import { Clipboard, Graph, Keyboard, MiniMap, Node, Snapline, History, type Edge } from '@antv/x6'; import { Clipboard, Graph, Keyboard, MiniMap, Node, Snapline, History, type Edge } from '@antv/x6';
import type { HistoryCommand as Command } from '@antv/x6/lib/plugin/history/type'; import type { HistoryCommand as Command } from '@antv/x6/lib/plugin/history/type';
@@ -1488,20 +1488,16 @@ export const useWorkflowGraph = ({
if (!graphRef.current) return; if (!graphRef.current) return;
const nodes = graphRef.current.getNodes(); const nodes = graphRef.current.getNodes();
const lastWithSub = [...chatHistory].reverse().find(item => item.subContent?.length); // Reset all node execution status on every chatHistory change
// Reset all node execution status first
nodes.forEach(node => { nodes.forEach(node => {
const data = node.getData(); const data = node.getData();
if (typeof data.executionStatus === 'string') { node.setData({ ...data, executionStatus: '' });
node.setData({ ...data, executionStatus: undefined });
}
}); });
if (!lastWithSub?.subContent) return;
// Build a nodeId -> status map first const lastAssistant = [...chatHistory].reverse().find(item => item.role === 'assistant');
const statusMap: Record<string, string> = {}; if (!lastAssistant?.subContent?.length) return;
lastWithSub.subContent.forEach(sub => { lastAssistant.subContent.forEach(sub => {
if (typeof sub.status === 'string') { if (typeof sub.status === 'string') {
statusMap[sub.node_id] = sub.status;
const node = nodes.find(n => n.getData()?.id === sub.node_id); const node = nodes.find(n => n.getData()?.id === sub.node_id);
if (node) { if (node) {
node.setData({ ...node.getData(), executionStatus: sub.status }); node.setData({ ...node.getData(), executionStatus: sub.status });