improve MongoDB client management and storage init

2025-02-19 04:30:52 +08:00
parent 7a970451b9
commit 6d8e627f85
7 changed files with 92 additions and 97 deletions
--- a/lightrag/lightrag.py
+++ b/lightrag/lightrag.py
@@ -609,9 +609,9 @@ class LightRAG:
                    tasks.append(storage.finalize())

            await asyncio.gather(*tasks)
-            logger.debug("Finalized Storages")

            self.storages_status = StoragesStatus.FINALIZED
+            logger.debug("Finalized Storages")

    async def get_graph_labels(self):
        text = await self.chunk_entity_relation_graph.get_all_labels()