f7b6c4ca99
### What problem does this PR solve? Feat: Add StringTransform operator #3221 ### Type of change - [x] New Feature (non-breaking change which adds functionality)
61 lines
1.5 KiB
TypeScript
61 lines
1.5 KiB
TypeScript
import { SelectWithSearch } from '@/components/originui/select-with-search';
|
|
import {
|
|
FormControl,
|
|
FormField,
|
|
FormItem,
|
|
FormLabel,
|
|
FormMessage,
|
|
} from '@/components/ui/form';
|
|
import { ReactNode, useMemo } from 'react';
|
|
import { useFormContext } from 'react-hook-form';
|
|
import { useTranslation } from 'react-i18next';
|
|
import { VariableType } from '../../constant';
|
|
import { useBuildQueryVariableOptions } from '../../hooks/use-get-begin-query';
|
|
|
|
type QueryVariableProps = {
|
|
name?: string;
|
|
type?: VariableType;
|
|
label?: ReactNode;
|
|
};
|
|
|
|
export function QueryVariable({
|
|
name = 'query',
|
|
type,
|
|
label,
|
|
}: QueryVariableProps) {
|
|
const { t } = useTranslation();
|
|
const form = useFormContext();
|
|
|
|
const nextOptions = useBuildQueryVariableOptions();
|
|
|
|
const finalOptions = useMemo(() => {
|
|
return type
|
|
? nextOptions.map((x) => {
|
|
return { ...x, options: x.options.filter((y) => y.type === type) };
|
|
})
|
|
: nextOptions;
|
|
}, [nextOptions, type]);
|
|
|
|
return (
|
|
<FormField
|
|
control={form.control}
|
|
name={name}
|
|
render={({ field }) => (
|
|
<FormItem>
|
|
{label || (
|
|
<FormLabel tooltip={t('chat.modelTip')}>
|
|
{t('flow.query')}
|
|
</FormLabel>
|
|
)}
|
|
<FormControl>
|
|
<SelectWithSearch
|
|
options={finalOptions}
|
|
{...field}
|
|
></SelectWithSearch>
|
|
</FormControl>
|
|
<FormMessage />
|
|
</FormItem>
|
|
)}
|
|
/>
|
|
);
|
|
}
|