mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-08-26 01:52:16 +08:00
### What problem does this PR solve? Feat: Delete the agent and tool nodes downstream of the agent node #3221 ### Type of change - [x] New Feature (non-breaking change which adds functionality)
This commit is contained in:
34
web/src/pages/agent/utils/delete-node.ts
Normal file
34
web/src/pages/agent/utils/delete-node.ts
Normal file
@@ -0,0 +1,34 @@
|
||||
import { Edge } from '@xyflow/react';
|
||||
import { filterAllDownstreamAgentAndToolNodeIds } from './filter-downstream-nodes';
|
||||
|
||||
// Delete all downstream agent and tool operators of the current agent operator
|
||||
export function deleteAllDownstreamAgentsAndTool(
|
||||
nodeId: string,
|
||||
edges: Edge[],
|
||||
) {
|
||||
const downstreamAgentAndToolNodeIds = filterAllDownstreamAgentAndToolNodeIds(
|
||||
edges,
|
||||
[nodeId],
|
||||
);
|
||||
|
||||
const downstreamAgentAndToolEdges = downstreamAgentAndToolNodeIds.reduce<
|
||||
Edge[]
|
||||
>((pre, cur) => {
|
||||
const relatedEdges = edges.filter(
|
||||
(x) => x.source === cur || x.target === cur,
|
||||
);
|
||||
|
||||
relatedEdges.forEach((x) => {
|
||||
if (!pre.some((y) => y.id !== x.id)) {
|
||||
pre.push(x);
|
||||
}
|
||||
});
|
||||
|
||||
return pre;
|
||||
}, []);
|
||||
|
||||
return {
|
||||
downstreamAgentAndToolNodeIds,
|
||||
downstreamAgentAndToolEdges,
|
||||
};
|
||||
}
|
||||
Reference in New Issue
Block a user