From 89290183c6ff4c41bfb7a47c2f3197d461031fd3 Mon Sep 17 00:00:00 2001
From: jyong <718720800@qq.com>
Date: Sun, 29 Sep 2024 15:36:59 +0800
Subject: [PATCH] add score threshold enabled

---
 api/services/external_knowledge_service.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/api/services/external_knowledge_service.py b/api/services/external_knowledge_service.py
index dc144cc097..8d77b1dc14 100644
--- a/api/services/external_knowledge_service.py
+++ b/api/services/external_knowledge_service.py
@@ -319,11 +319,12 @@ class ExternalDatasetService:
         headers = {"Content-Type": "application/json"}
         if settings.get("api_key"):
             headers["Authorization"] = f"Bearer {settings.get('api_key')}"
-
+        score_threshold_enabled = external_retrieval_parameters.get("score_threshold_enabled") or False
+        score_threshold = external_retrieval_parameters.get("score_threshold", 0.0) if score_threshold_enabled else 0.0
         request_params = {
             "retrieval_setting": {
                 "top_k": external_retrieval_parameters.get("top_k"),
-                "score_threshold": external_retrieval_parameters.get("score_threshold"),
+                "score_threshold": score_threshold,
             },
             "query": query,
             "knowledge_id": external_knowledge_binding.external_knowledge_id,