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

2024-09-13 14:24:35 +08:00
parent a45ac6ab98
commit 08c486452f
14 changed files with 17 additions and 17 deletions
--- a/api/core/rag/datasource/vdb/qdrant/qdrant_vector.py
+++ b/api/core/rag/datasource/vdb/qdrant/qdrant_vector.py
@@ -333,13 +333,13 @@ class QdrantVector(BaseVector):
            limit=kwargs.get("top_k", 4),
            with_payload=True,
            with_vectors=True,
-            score_threshold=kwargs.get("score_threshold", 0.0),
+            score_threshold=float(kwargs.get("score_threshold") or 0.0),
        )
        docs = []
        for result in results:
            metadata = result.payload.get(Field.METADATA_KEY.value) or {}
            # duplicate check score threshold
-            score_threshold = kwargs.get("score_threshold", 0.0)
+            score_threshold = float(kwargs.get("score_threshold") or 0.0)
            if result.score > score_threshold:
                metadata["score"] = result.score
                doc = Document(