gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Improve missing-vector warning logic in vector similarity

Browse source 
- Check for any missing vectors
- Separate no-vector vs partial-vector warnings
- Ensure early return on empty vectors
This commit is contained in:
yangdx 2025-08-14 14:24:15 +08:00
parent 5c7ae8721b
commit a11e8d77eb
1 changed files with 9 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
lightrag/utils.py
View file

@ -1713,10 +1713,15 @@ async def vector_similarity_sorting(
f"Vector similarity chunk selection: {len(chunk_vectors)} chunk vectors Retrieved" f"Vector similarity chunk selection: {len(chunk_vectors)} chunk vectors Retrieved"
) )
if not chunk_vectors: if not chunk_vectors or len(chunk_vectors) != len(all_chunk_ids):
logger.warning( if not chunk_vectors:
"Vector similarity chunk selection: no vectors retrieved from chunks_vdb" 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 [] return []
# Calculate cosine similarities # Calculate cosine similarities