fix: score_threshold handling in vector search methods (#8356)

2024-09-13 14:24:35 +08:00 · 2024-09-13 14:24:35 +08:00 · 08c486452f
commit 08c486452f
parent a45ac6ab98
14 changed files with 17 additions and 17 deletions
--- a/api/core/rag/datasource/vdb/tencent/tencent_vector.py
+++ b/api/core/rag/datasource/vdb/tencent/tencent_vector.py
@ -153,7 +153,7 @@ class TencentVector(BaseVector):
            limit=kwargs.get("top_k", 4),
            timeout=self._client_config.timeout,
        )
-        score_threshold = kwargs.get("score_threshold", 0.0)
+        score_threshold = float(kwargs.get("score_threshold") or 0.0)
        return self._get_search_res(res, score_threshold)

    def search_by_full_text(self, query: str, **kwargs: Any) -> list[Document]: