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

Fix MongoDB vector query embedding format compatibility

Browse source 
* Convert numpy arrays to lists
* Ensure MongoDB compatibility
This commit is contained in:
yangdx 2025-08-29 18:51:53 +08:00
parent 03d0fa3014
commit f3989548b9
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
lightrag/kg/mongo_impl.py
View file

@ -1814,7 +1814,11 @@ class MongoVectorDBStorage(BaseVectorStorage):
) -> list[dict[str, Any]]: ) -> list[dict[str, Any]]:
"""Queries the vector database using Atlas Vector Search.""" """Queries the vector database using Atlas Vector Search."""
if query_embedding is not None: if query_embedding is not None:
query_vector = query_embedding # Convert numpy array to list if needed for MongoDB compatibility
if hasattr(query_embedding, "tolist"):
query_vector = query_embedding.tolist()
else:
query_vector = list(query_embedding)
else: else:
# Generate the embedding # Generate the embedding
embedding = await self.embedding_func( embedding = await self.embedding_func(