mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-09-06 11:31:12 +00:00
Native data AI application framework based on AWEL+AGENT (#1152)
Co-authored-by: Fangyin Cheng <staneyffer@gmail.com> Co-authored-by: lcx01800250 <lcx01800250@alibaba-inc.com> Co-authored-by: licunxing <864255598@qq.com> Co-authored-by: Aralhi <xiaoping0501@gmail.com> Co-authored-by: xuyuan23 <643854343@qq.com> Co-authored-by: aries_ckt <916701291@qq.com> Co-authored-by: hzh97 <2976151305@qq.com>
This commit is contained in:
100
web/utils/flow.ts
Normal file
100
web/utils/flow.ts
Normal file
@@ -0,0 +1,100 @@
|
||||
import { IFlowData, IFlowDataNode, IFlowNode } from '@/types/flow';
|
||||
import { Node } from 'reactflow';
|
||||
|
||||
export const getUniqueNodeId = (nodeData: IFlowNode, nodes: Node[]) => {
|
||||
let count = 0;
|
||||
nodes.forEach((node) => {
|
||||
if (node.data.name === nodeData.name) {
|
||||
count++;
|
||||
}
|
||||
});
|
||||
return `${nodeData.id}_${count}`;
|
||||
};
|
||||
|
||||
// 驼峰转下划线,接口协议字段命名规范
|
||||
export const mapHumpToUnderline = (flowData: IFlowData) => {
|
||||
/**
|
||||
* sourceHandle -> source_handle,
|
||||
* targetHandle -> target_handle,
|
||||
* positionAbsolute -> position_absolute
|
||||
*/
|
||||
const { nodes, edges, ...rest } = flowData;
|
||||
const newNodes = nodes.map((node) => {
|
||||
const { positionAbsolute, ...rest } = node;
|
||||
return {
|
||||
position_absolute: positionAbsolute,
|
||||
...rest,
|
||||
};
|
||||
});
|
||||
const newEdges = edges.map((edge) => {
|
||||
const { sourceHandle, targetHandle, ...rest } = edge;
|
||||
return {
|
||||
source_handle: sourceHandle,
|
||||
target_handle: targetHandle,
|
||||
...rest,
|
||||
};
|
||||
});
|
||||
return {
|
||||
nodes: newNodes,
|
||||
edges: newEdges,
|
||||
...rest,
|
||||
};
|
||||
};
|
||||
|
||||
export const mapUnderlineToHump = (flowData: IFlowData) => {
|
||||
/**
|
||||
* source_handle -> sourceHandle,
|
||||
* target_handle -> targetHandle,
|
||||
* position_absolute -> positionAbsolute
|
||||
*/
|
||||
const { nodes, edges, ...rest } = flowData;
|
||||
const newNodes = nodes.map((node) => {
|
||||
const { position_absolute, ...rest } = node;
|
||||
return {
|
||||
positionAbsolute: position_absolute,
|
||||
...rest,
|
||||
};
|
||||
});
|
||||
const newEdges = edges.map((edge) => {
|
||||
const { source_handle, target_handle, ...rest } = edge;
|
||||
return {
|
||||
sourceHandle: source_handle,
|
||||
targetHandle: target_handle,
|
||||
...rest,
|
||||
};
|
||||
});
|
||||
return {
|
||||
nodes: newNodes,
|
||||
edges: newEdges,
|
||||
...rest,
|
||||
};
|
||||
};
|
||||
|
||||
export const checkFlowDataRequied = (flowData: IFlowData) => {
|
||||
const { nodes, edges } = flowData;
|
||||
// check the input, parameters that are required
|
||||
let result: [boolean, IFlowDataNode, string] = [true, nodes[0], ''];
|
||||
outerLoop: for (let i = 0; i < nodes.length; i++) {
|
||||
const node = nodes[i].data;
|
||||
const { inputs = [], parameters = [] } = node;
|
||||
// check inputs
|
||||
for (let j = 0; j < inputs.length; j++) {
|
||||
if (!edges.some((edge) => edge.targetHandle === `${nodes[i].id}|inputs|${j}`)) {
|
||||
result = [false, nodes[i], `The input ${inputs[j].type_name} of node ${node.label} is required`];
|
||||
break outerLoop;
|
||||
}
|
||||
}
|
||||
// check parameters
|
||||
for (let k = 0; k < parameters.length; k++) {
|
||||
const parameter = parameters[k];
|
||||
if (!parameter.optional && parameter.category === 'resource' && !edges.some((edge) => edge.targetHandle === `${nodes[i].id}|parameters|${k}`)) {
|
||||
result = [false, nodes[i], `The parameter ${parameter.type_name} of node ${node.label} is required`];
|
||||
break outerLoop;
|
||||
} else if (!parameter.optional && parameter.category === 'common' && (parameter.value === undefined || parameter.value === null)) {
|
||||
result = [false, nodes[i], `The parameter ${parameter.type_name} of node ${node.label} is required`];
|
||||
break outerLoop;
|
||||
}
|
||||
}
|
||||
}
|
||||
return result;
|
||||
};
|
Reference in New Issue
Block a user