fix: incorrect score in the chroma vector (#14273)

2025-02-25 08:40:22 +07:00 · 2025-02-25 08:40:22 +07:00 · 548f6ef2b6
parent b15ff4eb8c
commit 548f6ef2b6
1 changed files with 3 additions and 2 deletions
--- a/api/core/rag/datasource/vdb/chroma/chroma_vector.py
+++ b/api/core/rag/datasource/vdb/chroma/chroma_vector.py
@ -111,8 +111,9 @@ class ChromaVector(BaseVector):
        for index in range(len(ids)):
            distance = distances[index]
            metadata = dict(metadatas[index])
-            if distance >= score_threshold:
-                metadata["score"] = distance
+            score = 1 - distance
+            if score > score_threshold:
+                metadata["score"] = score
                doc = Document(
                    page_content=documents[index],
                    metadata=metadata,