diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py index 8bcb6b51..13202992 100644 --- a/lightrag/lightrag.py +++ b/lightrag/lightrag.py @@ -1159,7 +1159,7 @@ class LightRAG: """ if param.mode in ["local", "global", "hybrid"]: response = await kg_query( - query, + query.strip(), self.chunk_entity_relation_graph, self.entities_vdb, self.relationships_vdb, @@ -1180,7 +1180,7 @@ class LightRAG: ) elif param.mode == "naive": response = await naive_query( - query, + query.strip(), self.chunks_vdb, self.text_chunks, param, @@ -1199,7 +1199,7 @@ class LightRAG: ) elif param.mode == "mix": response = await mix_kg_vector_query( - query, + query.strip(), self.chunk_entity_relation_graph, self.entities_vdb, self.relationships_vdb,