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/chroma/chroma_vector.py
+++ b/api/core/rag/datasource/vdb/chroma/chroma_vector.py
@@ -92,7 +92,7 @@ class ChromaVector(BaseVector):
    def search_by_vector(self, query_vector: list[float], **kwargs: Any) -> list[Document]:
        collection = self._client.get_or_create_collection(self._collection_name)
        results: QueryResult = collection.query(query_embeddings=query_vector, n_results=kwargs.get("top_k", 4))
-        score_threshold = kwargs.get("score_threshold", 0.0)
+        score_threshold = float(kwargs.get("score_threshold") or 0.0)

        ids: list[str] = results["ids"][0]
        documents: list[str] = results["documents"][0]