Parcourir la source

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 il y a 7 mois
Parent
commit
162faee4f2
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  1. 1 1
      api/core/rag/datasource/vdb/myscale/myscale_vector.py

+ 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"""