瀏覽代碼

fix: knowledge retrieval score threshold setting (#5658)

zxhlyh 10 月之前
父節點
當前提交
2119d59da8

+ 4 - 1
web/app/components/app/configuration/dataset-config/settings-modal/index.tsx

@@ -99,7 +99,10 @@ const SettingsModal: FC<SettingsModalProps> = ({
           description,
           permission,
           indexing_technique: indexMethod,
-          retrieval_model: postRetrievalConfig,
+          retrieval_model: {
+            ...postRetrievalConfig,
+            score_threshold: postRetrievalConfig.score_threshold_enabled ? postRetrievalConfig.score_threshold : 0,
+          },
           embedding_model: localeCurrentDataset.embedding_model,
           embedding_model_provider: localeCurrentDataset.embedding_model_provider,
         },

+ 5 - 1
web/app/components/datasets/settings/form/index.tsx

@@ -25,6 +25,7 @@ import {
   useModelListAndDefaultModelAndCurrentProviderAndModel,
 } from '@/app/components/header/account-setting/model-provider-page/hooks'
 import { ModelTypeEnum } from '@/app/components/header/account-setting/model-provider-page/declarations'
+import type { DefaultModel } from '@/app/components/header/account-setting/model-provider-page/declarations'
 
 const rowClass = `
   flex justify-between py-4 flex-wrap gap-y-2
@@ -110,7 +111,10 @@ const Form = () => {
           description,
           permission,
           indexing_technique: indexMethod,
-          retrieval_model: postRetrievalConfig,
+          retrieval_model: {
+            ...postRetrievalConfig,
+            score_threshold: postRetrievalConfig.score_threshold_enabled ? postRetrievalConfig.score_threshold : 0,
+          },
           embedding_model: embeddingModel.model,
           embedding_model_provider: embeddingModel.provider,
         },