|
@@ -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') {
|