AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-15 21:45:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Rhys 2025-02-25 08:40:22 +07:00 committed by GitHub
parent b15ff4eb8c
commit 548f6ef2b6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
api/core/rag/datasource/vdb/chroma/chroma_vector.py
View File

@ -111,8 +111,9 @@ class ChromaVector(BaseVector):
for index in range(len(ids)): for index in range(len(ids)):
distance = distances[index] distance = distances[index]
metadata = dict(metadatas[index]) metadata = dict(metadatas[index])
if distance >= score_threshold: score = 1 - distance
metadata["score"] = distance if score > score_threshold:
metadata["score"] = score
doc = Document( doc = Document(
page_content=documents[index], page_content=documents[index],
metadata=metadata, metadata=metadata,