d9b98cbb18
### What problem does this PR solve? Feat: Convert the prompt field of the agent operator to an array #3221 ### Type of change - [x] New Feature (non-breaking change which adds functionality)
30 lines
748 B
TypeScript
30 lines
748 B
TypeScript
import { useFetchModelId } from '@/hooks/logic-hooks';
|
|
import { RAGFlowNodeType } from '@/interfaces/database/flow';
|
|
import { get, isEmpty } from 'lodash';
|
|
import { useMemo } from 'react';
|
|
import { initialAgentValues } from '../../constant';
|
|
|
|
export function useValues(node?: RAGFlowNodeType) {
|
|
const llmId = useFetchModelId();
|
|
|
|
const defaultValues = useMemo(
|
|
() => ({
|
|
...initialAgentValues,
|
|
llm_id: llmId,
|
|
prompts: '',
|
|
}),
|
|
[llmId],
|
|
);
|
|
|
|
const values = useMemo(() => {
|
|
const formData = node?.data?.form;
|
|
|
|
if (isEmpty(formData)) {
|
|
return defaultValues;
|
|
}
|
|
|
|
return { ...formData, prompts: get(formData, 'prompts.0.content', '') };
|
|
}, [defaultValues, node?.data?.form]);
|
|
|
|
return values;
|
|
}
|