jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: score_threshold handling in vector search methods (#8356)

Browse source
This commit is contained in:
-LAN- 2024-09-13 14:24:35 +08:00 • committed by GitHub
commit 08c486452f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
14 changed files with 17 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

4
api/core/rag/datasource/vdb/qdrant/qdrant_vector.py
View file

@ -333,13 +333,13 @@ class QdrantVector(BaseVector):
limit=kwargs.get("top_k", 4),
with_payload=True,
with_vectors=True,
score_threshold=kwargs.get("score_threshold", 0.0),
score_threshold=float(kwargs.get("score_threshold") or 0.0),
)
docs = []
for result in results:
metadata = result.payload.get(Field.METADATA_KEY.value) or {}
# duplicate check score threshold
score_threshold = kwargs.get("score_threshold", 0.0)
score_threshold = float(kwargs.get("score_threshold") or 0.0)
if result.score > score_threshold:
metadata["score"] = result.score
doc = Document(