From a11e8d77eb122f2d97528b08d8111e4058d45c60 Mon Sep 17 00:00:00 2001 From: yangdx Date: Thu, 14 Aug 2025 14:24:15 +0800 Subject: [PATCH] Improve missing-vector warning logic in vector similarity - Check for any missing vectors - Separate no-vector vs partial-vector warnings - Ensure early return on empty vectors --- lightrag/utils.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/lightrag/utils.py b/lightrag/utils.py index b0327842..249c1a38 100644 --- a/lightrag/utils.py +++ b/lightrag/utils.py @@ -1713,10 +1713,15 @@ async def vector_similarity_sorting( f"Vector similarity chunk selection: {len(chunk_vectors)} chunk vectors Retrieved" ) - if not chunk_vectors: - logger.warning( - "Vector similarity chunk selection: no vectors retrieved from chunks_vdb" - ) + if not chunk_vectors or len(chunk_vectors) != len(all_chunk_ids): + if not chunk_vectors: + logger.warning( + "Vector similarity chunk selection: no vectors retrieved from chunks_vdb" + ) + else: + logger.warning( + f"Vector similarity chunk selection: found {len(chunk_vectors)} but expecting {len(all_chunk_ids)}" + ) return [] # Calculate cosine similarities