diff --git a/web/app/components/app/configuration/dataset-config/settings-modal/index.tsx b/web/app/components/app/configuration/dataset-config/settings-modal/index.tsx index cc2fb061b8..5b01c36d40 100644 --- a/web/app/components/app/configuration/dataset-config/settings-modal/index.tsx +++ b/web/app/components/app/configuration/dataset-config/settings-modal/index.tsx @@ -31,6 +31,7 @@ import { import { ModelTypeEnum } from '@/app/components/header/account-setting/model-provider-page/declarations' import { fetchMembers } from '@/service/common' import type { Member } from '@/models/common' +import { IndexingType } from '@/app/components/datasets/create/step-two' type SettingsModalProps = { currentDataset: DataSet @@ -54,8 +55,6 @@ const SettingsModal: FC = ({ const { data: embeddingsModelList } = useModelList(ModelTypeEnum.textEmbedding) const { modelList: rerankModelList, - defaultModel: rerankDefaultModel, - currentModel: isRerankDefaultModelValid, } = useModelListAndDefaultModelAndCurrentProviderAndModel(ModelTypeEnum.rerank) const { t } = useTranslation() const { notify } = useToastContext() @@ -73,6 +72,7 @@ const SettingsModal: FC = ({ const [indexMethod, setIndexMethod] = useState(currentDataset.indexing_technique) const [retrievalConfig, setRetrievalConfig] = useState(localeCurrentDataset?.retrieval_model_dict as RetrievalConfig) + const [keywordNumber, setKeywordNumber] = useState(currentDataset.keyword_number ?? 10) const handleValueChange = (type: string, value: string) => { setLocaleCurrentDataset({ ...localeCurrentDataset, [type]: value }) @@ -124,6 +124,7 @@ const SettingsModal: FC = ({ description, permission, indexing_technique: indexMethod, + keyword_number: keywordNumber, retrieval_model: { ...retrievalConfig, score_threshold: retrievalConfig.score_threshold_enabled ? retrievalConfig.score_threshold : 0, @@ -248,13 +249,15 @@ const SettingsModal: FC = ({ setIndexMethod(v!)} + onChange={setIndexMethod} currentValue={currentDataset.indexing_technique} + keywordNumber={keywordNumber} + onKeywordNumberChange={setKeywordNumber} /> )} - {indexMethod === 'high_quality' && ( + {indexMethod === IndexingType.QUALIFIED && (
{t('datasetSettings.form.embeddingModel')}
@@ -333,7 +336,7 @@ const SettingsModal: FC = ({
- {indexMethod === 'high_quality' + {indexMethod === IndexingType.QUALIFIED ? (