Explorar o código

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

Rhys hai 1 mes
pai
achega
548f6ef2b6
Modificáronse 1 ficheiros con 3 adicións e 2 borrados
  1. 3 2
      api/core/rag/datasource/vdb/chroma/chroma_vector.py

+ 3 - 2
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,