From f5d21081917c80136e75e45effa4e575adf7b371 Mon Sep 17 00:00:00 2001 From: Ibrahim Haroon Date: Fri, 6 Jun 2025 15:33:05 -0400 Subject: [PATCH] fix: handle case where distance is 0 by setting score to infinity --- llama_stack/providers/remote/vector_io/weaviate/weaviate.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llama_stack/providers/remote/vector_io/weaviate/weaviate.py b/llama_stack/providers/remote/vector_io/weaviate/weaviate.py index e6fe8ccd3..6f2027dad 100644 --- a/llama_stack/providers/remote/vector_io/weaviate/weaviate.py +++ b/llama_stack/providers/remote/vector_io/weaviate/weaviate.py @@ -76,7 +76,7 @@ class WeaviateIndex(EmbeddingIndex): continue chunks.append(chunk) - scores.append(1.0 / doc.metadata.distance) + scores.append(1.0 / doc.metadata.distance if doc.metadata.distance != 0 else float("inf")) return QueryChunksResponse(chunks=chunks, scores=scores)