From 548f6ef2b6bd07f4560c8a4474e206bfa2f3d498 Mon Sep 17 00:00:00 2001 From: Rhys Date: Tue, 25 Feb 2025 08:40:22 +0700 Subject: [PATCH] fix: incorrect score in the chroma vector (#14273) --- api/core/rag/datasource/vdb/chroma/chroma_vector.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/api/core/rag/datasource/vdb/chroma/chroma_vector.py b/api/core/rag/datasource/vdb/chroma/chroma_vector.py index 907c4d2285..81a1207d1d 100644 --- 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,