From 6f8312ddd0c3bc2f55ea18c3af2fbf8f6f4b141c Mon Sep 17 00:00:00 2001 From: Ibrahim Haroon Date: Fri, 6 Jun 2025 15:31:25 -0400 Subject: [PATCH] fix: handle case where distance is 0 by setting score to infinity --- llama_stack/providers/remote/vector_io/pgvector/pgvector.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llama_stack/providers/remote/vector_io/pgvector/pgvector.py b/llama_stack/providers/remote/vector_io/pgvector/pgvector.py index ea918c552..7d58a49f3 100644 --- a/llama_stack/providers/remote/vector_io/pgvector/pgvector.py +++ b/llama_stack/providers/remote/vector_io/pgvector/pgvector.py @@ -116,7 +116,7 @@ class PGVectorIndex(EmbeddingIndex): scores = [] for doc, dist in results: 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)