mirror of
https://github.com/infiniflow/ragflow.git
synced 2025-12-22 06:06:40 +08:00
### What problem does this PR solve? Feat: Add hatPromptEngine component #3221 ### Type of change - [x] New Feature (non-breaking change which adds functionality)
This commit is contained in:
@ -1,7 +1,25 @@
|
||||
import { LlmModelType } from '@/constants/knowledge';
|
||||
import { useTranslate } from '@/hooks/common-hooks';
|
||||
import { useSelectLlmOptionsByModelType } from '@/hooks/llm-hooks';
|
||||
import { Form, Select, Slider } from 'antd';
|
||||
import { Select as AntSelect, Form, Slider } from 'antd';
|
||||
import { useFormContext } from 'react-hook-form';
|
||||
import {
|
||||
FormControl,
|
||||
FormField,
|
||||
FormItem,
|
||||
FormLabel,
|
||||
FormMessage,
|
||||
} from './ui/form';
|
||||
import {
|
||||
Select,
|
||||
SelectContent,
|
||||
SelectGroup,
|
||||
SelectItem,
|
||||
SelectLabel,
|
||||
SelectTrigger,
|
||||
SelectValue,
|
||||
} from './ui/select';
|
||||
import { FormSlider } from './ui/slider';
|
||||
|
||||
type FieldType = {
|
||||
rerank_id?: string;
|
||||
@ -18,7 +36,7 @@ export const RerankItem = () => {
|
||||
name={'rerank_id'}
|
||||
tooltip={t('rerankTip')}
|
||||
>
|
||||
<Select
|
||||
<AntSelect
|
||||
options={allOptions[LlmModelType.Rerank]}
|
||||
allowClear
|
||||
placeholder={t('rerankPlaceholder')}
|
||||
@ -55,3 +73,81 @@ const Rerank = () => {
|
||||
};
|
||||
|
||||
export default Rerank;
|
||||
|
||||
const RerankId = 'rerank_id';
|
||||
|
||||
function RerankFormField() {
|
||||
const form = useFormContext();
|
||||
const { t } = useTranslate('knowledgeDetails');
|
||||
const allOptions = useSelectLlmOptionsByModelType();
|
||||
const options = allOptions[LlmModelType.Rerank];
|
||||
|
||||
return (
|
||||
<FormField
|
||||
control={form.control}
|
||||
name={RerankId}
|
||||
render={({ field }) => (
|
||||
<FormItem>
|
||||
<FormLabel>{t('rerankModel')}</FormLabel>
|
||||
<FormControl>
|
||||
<Select onValueChange={field.onChange} {...field}>
|
||||
<SelectTrigger
|
||||
className="w-[280px]"
|
||||
value={field.value}
|
||||
onReset={() => {
|
||||
form.resetField(RerankId);
|
||||
}}
|
||||
>
|
||||
<SelectValue placeholder={t('rerankPlaceholder')} />
|
||||
</SelectTrigger>
|
||||
<SelectContent>
|
||||
{options.map((x) => (
|
||||
<SelectGroup key={x.label}>
|
||||
<SelectLabel>{x.label}</SelectLabel>
|
||||
{x.options.map((y) => (
|
||||
<SelectItem
|
||||
value={y.value}
|
||||
key={y.value}
|
||||
disabled={y.disabled}
|
||||
>
|
||||
{y.label}
|
||||
</SelectItem>
|
||||
))}
|
||||
</SelectGroup>
|
||||
))}
|
||||
</SelectContent>
|
||||
</Select>
|
||||
</FormControl>
|
||||
<FormMessage />
|
||||
</FormItem>
|
||||
)}
|
||||
/>
|
||||
);
|
||||
}
|
||||
|
||||
export function RerankFormFields() {
|
||||
const { control, watch } = useFormContext();
|
||||
const { t } = useTranslate('knowledgeDetails');
|
||||
const rerankId = watch(RerankId);
|
||||
|
||||
return (
|
||||
<>
|
||||
<RerankFormField></RerankFormField>
|
||||
{rerankId && (
|
||||
<FormField
|
||||
control={control}
|
||||
name={'top_k'}
|
||||
render={({ field }) => (
|
||||
<FormItem>
|
||||
<FormLabel>{t('topK')}</FormLabel>
|
||||
<FormControl>
|
||||
<FormSlider {...field} max={2048} min={1}></FormSlider>
|
||||
</FormControl>
|
||||
<FormMessage />
|
||||
</FormItem>
|
||||
)}
|
||||
/>
|
||||
)}
|
||||
</>
|
||||
);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user