pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-08 20:42:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Feat: Construct the to field of the classification operator when saving data #3221 (#8610)

Browse Source 
### What problem does this PR solve?

Feat: Construct the to field of the classification operator when saving
data #3221

### Type of change


- [x] New Feature (non-breaking change which adds functionality)
This commit is contained in:
balibabu
2025-07-02 09:49:42 +08:00
committed by GitHub
parent 0b40eb3e90
commit 212d5ce7ff
3 changed files with 44 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
web/src/pages/agent/utils.ts
View File

@ -1,5 +1,6 @@
import {
IAgentForm,
ICategorizeForm,
ICategorizeItem,
ICategorizeItemResult,
} from '@/interfaces/database/agent';
@ -158,6 +159,33 @@ function buildAgentTools(edges: Edge[], nodes: Node[], nodeId: string) {
return params;
}
function filterTargetsBySourceHandleId(edges: Edge[], handleId: string) {
return edges.filter((x) => x.sourceHandle === handleId).map((x) => x.target);
}
function buildCategorizeTos(edges: Edge[], nodes: Node[], nodeId: string) {
const node = nodes.find((x) => x.id === nodeId);
const params = { ...(node?.data.form ?? {}) } as ICategorizeForm;
if (node && node.data.label === Operator.Categorize) {
const subEdges = edges.filter((x) => x.source === nodeId);
const categoryDescription = params.category_description || {};
const nextCategoryDescription = Object.entries(categoryDescription).reduce<
ICategorizeForm['category_description']
>((pre, [key, val]) => {
pre[key] = {
...val,
to: filterTargetsBySourceHandleId(subEdges, key),
};
return pre;
}, {});
params.category_description = nextCategoryDescription;
}
return params;
}
const buildOperatorParams = (operatorName: string) =>
pipe(
removeUselessDataInTheOperator(operatorName),
@ -190,8 +218,20 @@ export const buildDslComponentsByGraph = (
.forEach((x) => {
const id = x.id;
const operatorName = x.data.label;
let params = x?.data.form ?? {};
switch (operatorName) {
case Operator.Agent:
params = buildAgentTools(edges, nodes, id);
break;
case Operator.Categorize:
params = buildCategorizeTos(edges, nodes, id);
break;
default:
break;
}
const params = buildAgentTools(edges, nodes, id);
components[id] = {
obj: {
...(oldDslComponents[id]?.obj ?? {}),