phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-16 18:08:09 +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-03 19:20:31 -04:00
parent ef885d2147
commit fea01c5a25
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/inline/vector_io/faiss/faiss.py
View file

@ -112,7 +112,7 @@ class FaissIndex(EmbeddingIndex):
if i < 0: if i < 0:
continue continue
chunks.append(self.chunk_by_index[int(i)]) chunks.append(self.chunk_by_index[int(i)])
scores.append(1.0 / float(d)) scores.append(1.0 / float(d) if d != 0 else float("inf"))
return QueryChunksResponse(chunks=chunks, scores=scores) return QueryChunksResponse(chunks=chunks, scores=scores)