From fea01c5a251544b7c4c1887f4d13933190672231 Mon Sep 17 00:00:00 2001
From: Ibrahim Haroon <99413953+Ibrahim-Haroon@users.noreply.github.com>
Date: Tue, 3 Jun 2025 19:20:31 -0400
Subject: [PATCH] fix: handle case where distance is 0 by setting score to
 infinity

---
 llama_stack/providers/inline/vector_io/faiss/faiss.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/llama_stack/providers/inline/vector_io/faiss/faiss.py b/llama_stack/providers/inline/vector_io/faiss/faiss.py
index 47256d88d..0602e48b7 100644
--- a/llama_stack/providers/inline/vector_io/faiss/faiss.py
+++ b/llama_stack/providers/inline/vector_io/faiss/faiss.py
@@ -112,7 +112,7 @@ class FaissIndex(EmbeddingIndex):
             if i < 0:
                 continue
             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)