Pārlūkot izejas kodu

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

tags/1.0.0
Rhys pirms 8 mēnešiem
vecāks
revīzija
548f6ef2b6
Revīzijas autora e-pasta adrese nav piesaistīta nevienam kontam
1 mainītis faili ar 3 papildinājumiem un 2 dzēšanām
  1. 3
    2
      api/core/rag/datasource/vdb/chroma/chroma_vector.py

+ 3
- 2
api/core/rag/datasource/vdb/chroma/chroma_vector.py Parādīt failu

@@ -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,

Notiek ielāde…
Atcelt
Saglabāt