Feat: Add wencai operator #3221 (#9116)

### What problem does this PR solve?

Feat: Add wencai operator #3221

### Type of change


- [x] New Feature (non-breaking change which adds functionality)
This commit is contained in:
balibabu
2025-07-30 14:34:06 +08:00
committed by GitHub
parent b638d3f773
commit 391c5586dd
7 changed files with 117 additions and 28 deletions

View File

@ -1,3 +1,4 @@
import { FormContainer } from '@/components/form-container';
import { TopNFormField } from '@/components/top-n-item';
import {
Form,
@ -8,14 +9,34 @@ import {
FormMessage,
} from '@/components/ui/form';
import { RAGFlowSelect } from '@/components/ui/select';
import { useMemo } from 'react';
import { zodResolver } from '@hookform/resolvers/zod';
import { memo, useMemo } from 'react';
import { useForm, useFormContext } from 'react-hook-form';
import { useTranslation } from 'react-i18next';
import { z } from 'zod';
import { initialWenCaiValues } from '../../constant';
import { useFormValues } from '../../hooks/use-form-values';
import { useWatchFormChange } from '../../hooks/use-watch-form-change';
import { INextOperatorForm } from '../../interface';
import { WenCaiQueryTypeOptions } from '../../options';
import { DynamicInputVariable } from '../components/next-dynamic-input-variable';
import { buildOutputList } from '../../utils/build-output-list';
import { FormWrapper } from '../components/form-wrapper';
import { Output } from '../components/output';
import { QueryVariable } from '../components/query-variable';
const WenCaiForm = ({ form, node }: INextOperatorForm) => {
export const WenCaiPartialSchema = {
top_n: z.number(),
query_type: z.string(),
};
export const FormSchema = z.object({
...WenCaiPartialSchema,
query: z.string(),
});
export function WenCaiFormWidgets() {
const { t } = useTranslation();
const form = useFormContext();
const wenCaiQueryTypeOptions = useMemo(() => {
return WenCaiQueryTypeOptions.map((x) => ({
@ -24,32 +45,53 @@ const WenCaiForm = ({ form, node }: INextOperatorForm) => {
}));
}, [t]);
return (
<>
<TopNFormField max={99}></TopNFormField>
<FormField
control={form.control}
name="query_type"
render={({ field }) => (
<FormItem>
<FormLabel>{t('flow.queryType')}</FormLabel>
<FormControl>
<RAGFlowSelect {...field} options={wenCaiQueryTypeOptions} />
</FormControl>
<FormMessage />
</FormItem>
)}
/>
</>
);
}
const outputList = buildOutputList(initialWenCaiValues.outputs);
function WenCaiForm({ node }: INextOperatorForm) {
const defaultValues = useFormValues(initialWenCaiValues, node);
const form = useForm<z.infer<typeof FormSchema>>({
defaultValues,
resolver: zodResolver(FormSchema),
});
useWatchFormChange(node?.id, form);
return (
<Form {...form}>
<form
className="space-y-6"
onSubmit={(e) => {
e.preventDefault();
}}
>
<DynamicInputVariable node={node}></DynamicInputVariable>
<TopNFormField max={99}></TopNFormField>
<FormField
control={form.control}
name="query_type"
render={({ field }) => (
<FormItem>
<FormLabel>{t('flow.queryType')}</FormLabel>
<FormControl>
<RAGFlowSelect {...field} options={wenCaiQueryTypeOptions} />
</FormControl>
<FormMessage />
</FormItem>
)}
/>
</form>
<FormWrapper>
<FormContainer>
<QueryVariable></QueryVariable>
</FormContainer>
<FormContainer>
<WenCaiFormWidgets></WenCaiFormWidgets>
</FormContainer>
</FormWrapper>
<div className="p-5">
<Output list={outputList}></Output>
</div>
</Form>
);
};
}
export default WenCaiForm;
export default memo(WenCaiForm);