From f3989548b9f63ba96aaf6ecfcb87baeb9dcce72a Mon Sep 17 00:00:00 2001 From: yangdx Date: Fri, 29 Aug 2025 18:51:53 +0800 Subject: [PATCH] Fix MongoDB vector query embedding format compatibility * Convert numpy arrays to lists * Ensure MongoDB compatibility --- lightrag/kg/mongo_impl.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/lightrag/kg/mongo_impl.py b/lightrag/kg/mongo_impl.py index 8d52af64..9e4d7e67 100644 --- a/lightrag/kg/mongo_impl.py +++ b/lightrag/kg/mongo_impl.py @@ -1814,7 +1814,11 @@ class MongoVectorDBStorage(BaseVectorStorage): ) -> list[dict[str, Any]]: """Queries the vector database using Atlas Vector Search.""" 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: # Generate the embedding embedding = await self.embedding_func(