phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-28 10:54:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: handle case where distance is 0 by setting score to infinity

Browse source
This commit is contained in:
Ibrahim Haroon 2025-06-06 15:31:25 -04:00 • committed by Ibrahim Haroon
parent 33ecefd284
commit 6f8312ddd0
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/providers/remote/vector_io/pgvector/pgvector.py
View file

@ -116,7 +116,7 @@ class PGVectorIndex(EmbeddingIndex):
scores = [] scores = []
for doc, dist in results: for doc, dist in results:
chunks.append(Chunk(**doc)) chunks.append(Chunk(**doc))
scores.append(1.0 / float(dist)) scores.append(1.0 / float(dist) if dist != 0 else float("inf"))
return QueryChunksResponse(chunks=chunks, scores=scores) return QueryChunksResponse(chunks=chunks, scores=scores)