05bf01b058
### What problem does this PR solve? Feat: Displays the output variable type selected by the loop operator #3221 ### Type of change - [x] New Feature (non-breaking change which adds functionality)
27 lines
750 B
TypeScript
27 lines
750 B
TypeScript
import { RAGFlowNodeType } from '@/interfaces/database/flow';
|
|
import { isEmpty } from 'lodash';
|
|
import { useMemo } from 'react';
|
|
import { initialIterationValues } from '../../constant';
|
|
import { OutputObject } from './interface';
|
|
|
|
function convertToArray(outputObject: OutputObject) {
|
|
return Object.entries(outputObject).map(([key, value]) => ({
|
|
name: key,
|
|
ref: value.ref,
|
|
type: value.type,
|
|
}));
|
|
}
|
|
|
|
export function useValues(node?: RAGFlowNodeType) {
|
|
const values = useMemo(() => {
|
|
const formData = node?.data?.form;
|
|
|
|
if (isEmpty(formData)) {
|
|
return { ...initialIterationValues, outputs: [] };
|
|
}
|
|
|
|
return { ...formData, outputs: convertToArray(formData.outputs) };
|
|
}, [node?.data?.form]);
|
|
|
|
return values;
|
|
}
|