|
|
|
@@ -59,10 +59,10 @@ const SettingsModal: FC<SettingsModalProps> = ({ |
|
|
|
const { t } = useTranslation() |
|
|
|
const { notify } = useToastContext() |
|
|
|
const ref = useRef(null) |
|
|
|
const isExternal = currentDataset.provider === 'external' |
|
|
|
const [topK, setTopK] = useState(currentDataset?.external_retrieval_model.top_k ?? 2) |
|
|
|
const [scoreThreshold, setScoreThreshold] = useState(currentDataset?.external_retrieval_model.score_threshold ?? 0.5) |
|
|
|
const [scoreThresholdEnabled, setScoreThresholdEnabled] = useState(currentDataset?.external_retrieval_model.score_threshold_enabled ?? false) |
|
|
|
|
|
|
|
const { setShowAccountSettingModal } = useModalContext() |
|
|
|
const [loading, setLoading] = useState(false) |
|
|
|
const { isCurrentWorkspaceDatasetOperator } = useAppContext() |
|
|
|
@@ -122,19 +122,21 @@ const SettingsModal: FC<SettingsModalProps> = ({ |
|
|
|
description, |
|
|
|
permission, |
|
|
|
indexing_technique: indexMethod, |
|
|
|
external_retrieval_model: { |
|
|
|
top_k: topK, |
|
|
|
score_threshold: scoreThreshold, |
|
|
|
score_threshold_enabled: scoreThresholdEnabled, |
|
|
|
}, |
|
|
|
retrieval_model: { |
|
|
|
...postRetrievalConfig, |
|
|
|
score_threshold: postRetrievalConfig.score_threshold_enabled ? postRetrievalConfig.score_threshold : 0, |
|
|
|
}, |
|
|
|
external_knowledge_id: currentDataset!.external_knowledge_info.external_knowledge_id, |
|
|
|
external_knowledge_api_id: currentDataset!.external_knowledge_info.external_knowledge_api_id, |
|
|
|
embedding_model: localeCurrentDataset.embedding_model, |
|
|
|
embedding_model_provider: localeCurrentDataset.embedding_model_provider, |
|
|
|
...(isExternal && { |
|
|
|
external_knowledge_id: currentDataset!.external_knowledge_info.external_knowledge_id, |
|
|
|
external_knowledge_api_id: currentDataset!.external_knowledge_info.external_knowledge_api_id, |
|
|
|
external_retrieval_model: { |
|
|
|
top_k: topK, |
|
|
|
score_threshold: scoreThreshold, |
|
|
|
score_threshold_enabled: scoreThresholdEnabled, |
|
|
|
}, |
|
|
|
}), |
|
|
|
}, |
|
|
|
} as any |
|
|
|
if (permission === 'partial_members') { |