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/relyt/relyt_vector.py
+++ b/api/core/rag/datasource/vdb/relyt/relyt_vector.py
@ -230,7 +230,7 @@ class RelytVector(BaseVector):
        # Organize results.
        docs = []
        for document, score in results:
-            score_threshold = kwargs.get("score_threshold", 0.0)
+            score_threshold = float(kwargs.get("score_threshold") or 0.0)
            if 1 - score > score_threshold:
                docs.append(document)
        return docs