Procházet zdrojové kódy

fix: set score_threshold to zero if it is None for MyScale vectordb (#7640)

Co-authored-by: crazywoola <100913391+crazywoola@users.noreply.github.com>
Qin Liu před 7 měsíci
rodič
revize
162faee4f2

+ 1 - 1
api/core/rag/datasource/vdb/myscale/myscale_vector.py

@@ -122,7 +122,7 @@ class MyScaleVector(BaseVector):
 
     def _search(self, dist: str, order: SortOrder, **kwargs: Any) -> list[Document]:
         top_k = kwargs.get("top_k", 5)
-        score_threshold = kwargs.get("score_threshold", 0.0)
+        score_threshold = kwargs.get('score_threshold') or 0.0
         where_str = f"WHERE dist < {1 - score_threshold}" if \
             self._metric.upper() == "COSINE" and order == SortOrder.ASC and score_threshold > 0.0 else ""
         sql = f"""