Feat: Add hatPromptEngine component #3221 (#4881)

### 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:
balibabu
2025-02-11 19:04:10 +08:00
committed by GitHub
parent 521d25d4e6
commit d197f33646
9 changed files with 398 additions and 142 deletions

View File

@ -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>
)}
/>
)}
</>
);
}