refactor: extract useVariableList; properties add output variable
This commit is contained in:
@@ -1967,6 +1967,7 @@ Memory Bear: After the rebellion, regional warlordism intensified for several re
|
||||
value: 'Value',
|
||||
addCase: 'Add Condition',
|
||||
addVariable: 'Add Variables',
|
||||
output: 'Output Variable'
|
||||
},
|
||||
|
||||
clear: 'Clear',
|
||||
|
||||
@@ -2061,6 +2061,7 @@ export const zh = {
|
||||
value: '值',
|
||||
addCase: '添加条件',
|
||||
addVariable: '添加变量',
|
||||
output: '输出变量'
|
||||
},
|
||||
|
||||
clear: '清空',
|
||||
|
||||
@@ -33,7 +33,8 @@ const InitialValuePlugin: React.FC<InitialValuePluginProps> = ({ value, options
|
||||
|
||||
useEffect(() => {
|
||||
if (value !== prevValueRef.current && !isUserInputRef.current) {
|
||||
editor.update(() => {
|
||||
queueMicrotask(() => {
|
||||
editor.update(() => {
|
||||
const root = $getRoot();
|
||||
root.clear();
|
||||
|
||||
@@ -98,7 +99,8 @@ const InitialValuePlugin: React.FC<InitialValuePluginProps> = ({ value, options
|
||||
});
|
||||
root.append(paragraph);
|
||||
}
|
||||
}, { discrete: true });
|
||||
}, { discrete: true });
|
||||
});
|
||||
}
|
||||
|
||||
prevValueRef.current = value;
|
||||
|
||||
@@ -17,7 +17,7 @@ const GroupVariableList: FC<GroupVariableListProps> = ({
|
||||
name,
|
||||
options = [],
|
||||
isCanAdd = false,
|
||||
size = "middle"
|
||||
size = "small"
|
||||
}) => {
|
||||
const { t } = useTranslation();
|
||||
const form = Form.useFormInstance();
|
||||
@@ -37,16 +37,10 @@ const GroupVariableList: FC<GroupVariableListProps> = ({
|
||||
}
|
||||
|
||||
return (
|
||||
<div className="rb:mb-4">
|
||||
<Row gutter={12} className="rb:mb-2!">
|
||||
<Col span={12}>
|
||||
<Form.Item
|
||||
noStyle
|
||||
>
|
||||
{t('workflow.config.var-aggregator.variable')}
|
||||
</Form.Item>
|
||||
</Col>
|
||||
</Row>
|
||||
<div>
|
||||
<div className="rb:font-medium rb:text-[12px] rb:mb-1">
|
||||
{t('workflow.config.var-aggregator.variable')}
|
||||
</div>
|
||||
|
||||
<Form.Item
|
||||
name={name}
|
||||
|
||||
@@ -34,9 +34,9 @@ const VariableSelect: FC<VariableSelectProps> = ({
|
||||
if (filterOption) {
|
||||
return (
|
||||
<span
|
||||
className={clsx("rb:w-full rb:wrap-break-word rb:line-clamp-1 rb:border rb:border-[#DFE4ED] rb:rounded-md rb:bg-white rb:text-[12px] rb:inline-flex rb:items-center rb:px-1.5 rb:cursor-pointer", {
|
||||
className={clsx("rb:max-w-full rb:wrap-break-word rb:line-clamp-1 rb:border rb:border-[#DFE4ED] rb:rounded-md rb:bg-white rb:text-[12px] rb:inline-flex rb:items-center rb:px-1.5 rb:cursor-pointer", {
|
||||
'rb:leading-5.5!': size !== 'small',
|
||||
'rb:leading-4!': size === 'small'
|
||||
'rb:leading-4! rb:text-[10px]!': size === 'small'
|
||||
})}
|
||||
contentEditable={false}
|
||||
>
|
||||
|
||||
@@ -0,0 +1,209 @@
|
||||
import { useMemo, useEffect, useState } from 'react';
|
||||
import { Graph, Node } from '@antv/x6';
|
||||
import type { Suggestion } from '../../Editor/plugin/AutocompletePlugin';
|
||||
import type { ChatVariable } from '../../../types';
|
||||
|
||||
const NODE_VARIABLES = {
|
||||
llm: [{ label: 'output', dataType: 'string', field: 'output' }],
|
||||
'jinja-render': [{ label: 'output', dataType: 'string', field: 'output' }],
|
||||
tool: [{ label: 'data', dataType: 'string', field: 'data' }],
|
||||
'knowledge-retrieval': [{ label: 'output', dataType: 'array[object]', field: 'output' }],
|
||||
'parameter-extractor': [
|
||||
{ label: '__is_success', dataType: 'number', field: '__is_success' },
|
||||
{ label: '__reason', dataType: 'string', field: '__reason' }
|
||||
],
|
||||
'http-request': [
|
||||
{ label: 'body', dataType: 'string', field: 'body' },
|
||||
{ label: 'status_code', dataType: 'number', field: 'status_code' }
|
||||
],
|
||||
'question-classifier': [{ label: 'class_name', dataType: 'string', field: 'class_name' }],
|
||||
'memory-read': [
|
||||
{ label: 'answer', dataType: 'string', field: 'answer' },
|
||||
{ label: 'intermediate_outputs', dataType: 'array[object]', field: 'intermediate_outputs' }
|
||||
]
|
||||
} as const;
|
||||
|
||||
const addVariable = (
|
||||
list: Suggestion[],
|
||||
keys: Set<string>,
|
||||
key: string,
|
||||
label: string,
|
||||
dataType: string,
|
||||
value: string,
|
||||
nodeData: any,
|
||||
extra?: Partial<Suggestion>
|
||||
) => {
|
||||
if (!keys.has(key)) {
|
||||
keys.add(key);
|
||||
list.push({ key, label, type: 'variable', dataType, value, nodeData, ...extra });
|
||||
}
|
||||
};
|
||||
|
||||
const processNodeVariables = (
|
||||
nodeData: any,
|
||||
dataNodeId: string,
|
||||
variableList: Suggestion[],
|
||||
addedKeys: Set<string>
|
||||
) => {
|
||||
const { type, config } = nodeData;
|
||||
|
||||
if (type in NODE_VARIABLES) {
|
||||
NODE_VARIABLES[type as keyof typeof NODE_VARIABLES].forEach(({ label, dataType, field }) => {
|
||||
addVariable(variableList, addedKeys, `${dataNodeId}_${label}`, label, dataType, `${dataNodeId}.${field}`, nodeData);
|
||||
});
|
||||
}
|
||||
|
||||
switch (type) {
|
||||
case 'start':
|
||||
[...(config?.variables?.defaultValue ?? []), ...(config?.variables?.value ?? [])].forEach((v: any) => {
|
||||
if (v?.name) addVariable(variableList, addedKeys, `${dataNodeId}_${v.name}`, v.name, v.type, `${dataNodeId}.${v.name}`, nodeData);
|
||||
});
|
||||
config?.variables?.sys?.forEach((v: any) => {
|
||||
if (v?.name) addVariable(variableList, addedKeys, `${dataNodeId}_sys_${v.name}`, `sys.${v.name}`, v.type, `sys.${v.name}`, nodeData);
|
||||
});
|
||||
break;
|
||||
|
||||
case 'parameter-extractor':
|
||||
(config?.params?.defaultValue || []).forEach((p: any) => {
|
||||
if (p?.name) addVariable(variableList, addedKeys, `${dataNodeId}_${p.name}`, p.name, p.type || 'string', `${dataNodeId}.${p.name}`, nodeData);
|
||||
});
|
||||
break;
|
||||
|
||||
case 'var-aggregator':
|
||||
if (config.group.defaultValue) {
|
||||
(config.group_variables.defaultValue || []).forEach((gv: any) => {
|
||||
if (gv?.key) {
|
||||
let dt = 'string';
|
||||
if (gv.value?.[0]) {
|
||||
const fv = variableList.find(v => `{{${v.value}}}` === gv.value[0]);
|
||||
if (fv) dt = fv.dataType;
|
||||
}
|
||||
addVariable(variableList, addedKeys, `${dataNodeId}_${gv.key}`, gv.key, dt, `${dataNodeId}.${gv.key}`, nodeData);
|
||||
}
|
||||
});
|
||||
} else {
|
||||
const fv = (config.group_variables.defaultValue || [])[0];
|
||||
let dt = 'any';
|
||||
if (fv) {
|
||||
const found = variableList.find(v => `{{${v.value}}}` === fv);
|
||||
if (found) dt = found.dataType;
|
||||
}
|
||||
addVariable(variableList, addedKeys, `${dataNodeId}_output`, 'output', dt, `${dataNodeId}.output`, nodeData);
|
||||
}
|
||||
break;
|
||||
|
||||
case 'iteration':
|
||||
let dt = 'string';
|
||||
if (nodeData.output) {
|
||||
const sv = variableList.find(v => v.value === nodeData.output);
|
||||
if (sv) dt = sv.dataType;
|
||||
}
|
||||
addVariable(variableList, addedKeys, `${dataNodeId}_output`, 'output', `array[${dt}]`, `${dataNodeId}.output`, nodeData);
|
||||
break;
|
||||
|
||||
case 'loop':
|
||||
(config.cycle_vars.defaultValue || []).forEach((cv: any) => {
|
||||
if (cv.name?.trim()) addVariable(variableList, addedKeys, `${dataNodeId}_cycle_${cv.name}`, cv.name, cv.type || 'string', `${dataNodeId}.${cv.name}`, nodeData);
|
||||
});
|
||||
break;
|
||||
}
|
||||
};
|
||||
|
||||
const hasOutputNodeTypes = [
|
||||
'llm',
|
||||
'knowledge-retrieval',
|
||||
'memory-read',
|
||||
'question-classifier',
|
||||
'var-aggregator',
|
||||
'http-request',
|
||||
'tool',
|
||||
'jinja-render'
|
||||
]
|
||||
export const getCurrentNodeVariables = (nodeData: any, values: any): Suggestion[] => {
|
||||
if (!nodeData || !hasOutputNodeTypes.includes(nodeData.type)) return [];
|
||||
const list: Suggestion[] = [];
|
||||
const keys = new Set<string>();
|
||||
const dataNodeId = nodeData.id;
|
||||
|
||||
processNodeVariables({
|
||||
...nodeData,
|
||||
config: {
|
||||
...nodeData.config,
|
||||
...values
|
||||
}
|
||||
}, dataNodeId, list, keys);
|
||||
return nodeData.type === 'var-aggregator' && !nodeData.config.group.defaultValue ? [] : list;
|
||||
};
|
||||
|
||||
export const useVariableList = (
|
||||
selectedNode: Node | null | undefined,
|
||||
graphRef: React.MutableRefObject<Graph | undefined>,
|
||||
chatVariables: ChatVariable[]
|
||||
) => {
|
||||
const [trigger, setTrigger] = useState(0);
|
||||
|
||||
const variableList = useMemo(() => {
|
||||
if (!selectedNode || !graphRef?.current) return [];
|
||||
|
||||
const list: Suggestion[] = [];
|
||||
const graph = graphRef.current;
|
||||
const edges = graph.getEdges();
|
||||
const nodes = graph.getNodes();
|
||||
const keys = new Set<string>();
|
||||
|
||||
const getPreviousNodes = (nodeId: string, visited = new Set<string>()): string[] => {
|
||||
if (visited.has(nodeId)) return [];
|
||||
visited.add(nodeId);
|
||||
const prev = edges.filter(e => e.getTargetCellId() === nodeId).map(e => e.getSourceCellId());
|
||||
return [...prev, ...prev.flatMap(id => getPreviousNodes(id, visited))];
|
||||
};
|
||||
|
||||
const getParentLoop = (nodeId: string): Node | null => {
|
||||
const node = nodes.find(n => n.id === nodeId);
|
||||
const cycle = node?.getData()?.cycle;
|
||||
if (cycle) {
|
||||
const parent = nodes.find(n => n.getData().id === cycle);
|
||||
if (parent?.getData()?.type === 'loop' || parent?.getData()?.type === 'iteration') return parent;
|
||||
}
|
||||
return null;
|
||||
};
|
||||
|
||||
const childIds = nodes.filter(n => n.getData()?.cycle === selectedNode.id).map(n => n.id);
|
||||
const parentLoop = getParentLoop(selectedNode.id);
|
||||
const relevantIds = [...getPreviousNodes(selectedNode.id), ...childIds, ...(parentLoop ? getPreviousNodes(parentLoop.id) : [])];
|
||||
|
||||
chatVariables?.forEach(v => addVariable(list, keys, `CONVERSATION_${v.name}`, v.name, v.type, `conv.${v.name}`, { type: 'CONVERSATION', name: 'CONVERSATION', icon: '' }, { group: 'CONVERSATION' }));
|
||||
|
||||
relevantIds.forEach(id => {
|
||||
const node = nodes.find(n => n.id === id);
|
||||
if (node) processNodeVariables(node.getData(), node.getData().id, list, keys);
|
||||
});
|
||||
|
||||
if (parentLoop) {
|
||||
const pd = parentLoop.getData();
|
||||
const pid = pd.id;
|
||||
if (pd.type === 'loop') {
|
||||
(pd.cycle_vars || []).forEach((cv: any) => addVariable(list, keys, `${pid}_cycle_${cv.name}`, cv.name, cv.type || 'String', `${pid}.${cv.name}`, pd));
|
||||
} else if (pd.type === 'iteration' && pd.config.input.defaultValue) {
|
||||
let itemType = 'object';
|
||||
const iv = list.find(v => `{{${v.value}}}` === pd.config.input.defaultValue);
|
||||
if (iv?.dataType.startsWith('array[')) itemType = iv.dataType.replace(/^array\[(.+)\]$/, '$1');
|
||||
addVariable(list, keys, `${pid}_item`, 'item', itemType, `${pid}.item`, pd);
|
||||
addVariable(list, keys, `${pid}_index`, 'index', 'number', `${pid}.index`, pd);
|
||||
}
|
||||
}
|
||||
|
||||
return list;
|
||||
}, [selectedNode, graphRef, trigger, chatVariables]);
|
||||
|
||||
useEffect(() => {
|
||||
if (!graphRef?.current) return;
|
||||
const graph = graphRef.current;
|
||||
const handler = () => setTrigger(p => p + 1);
|
||||
const events = ['edge:added', 'edge:removed', 'edge:changed', 'edge:connected', 'node:added', 'node:removed', 'node:change:data'];
|
||||
events.forEach(e => graph.on(e, handler));
|
||||
return () => events.forEach(e => graph.off(e, handler));
|
||||
}, [graphRef]);
|
||||
|
||||
return variableList;
|
||||
};
|
||||
File diff suppressed because it is too large
Load Diff
@@ -275,11 +275,6 @@ export const useWorkflowGraph = ({
|
||||
}, 100)
|
||||
}
|
||||
if (edges.length) {
|
||||
// 计算loop和iteration类型节点的数量
|
||||
const loopIterationCount = nodes.filter(node =>
|
||||
node.type === 'loop' || node.type === 'iteration'
|
||||
).length;
|
||||
|
||||
// 去重处理:对于if-else和question-classifier节点,不同连接桩允许连接到相同节点
|
||||
const uniqueEdges = edges.filter((edge, index, arr) => {
|
||||
return arr.findIndex(e => {
|
||||
|
||||
Reference in New Issue
Block a user