From 162faee4f2efc4d9261e4a76630f02c5aa97a05e Mon Sep 17 00:00:00 2001 From: Qin Liu Date: Tue, 27 Aug 2024 09:47:16 +0800 Subject: [PATCH] fix: set score_threshold to zero if it is None for MyScale vectordb (#7640) Co-authored-by: crazywoola <100913391+crazywoola@users.noreply.github.com> --- api/core/rag/datasource/vdb/myscale/myscale_vector.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api/core/rag/datasource/vdb/myscale/myscale_vector.py b/api/core/rag/datasource/vdb/myscale/myscale_vector.py index 4ae1a3395..05e75effe 100644 --- a/api/core/rag/datasource/vdb/myscale/myscale_vector.py +++ b/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"""