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/opensearch/opensearch_vector.py
+++ b/api/core/rag/datasource/vdb/opensearch/opensearch_vector.py
@@ -170,7 +170,7 @@ class OpenSearchVector(BaseVector):
                metadata = {}

            metadata["score"] = hit["_score"]
-            score_threshold = kwargs.get("score_threshold", 0.0)
+            score_threshold = float(kwargs.get("score_threshold") or 0.0)
            if hit["_score"] > score_threshold:
                doc = Document(page_content=hit["_source"].get(Field.CONTENT_KEY.value), metadata=metadata)
                docs.append(doc)