Revert "Cleanup of code"

2025-02-20 15:09:43 +01:00
parent c431cd584a
commit 678e0f9aea
12 changed files with 84 additions and 7 deletions
--- a/lightrag/lightrag.py
+++ b/lightrag/lightrag.py
@@ -458,6 +458,10 @@ class LightRAG:
            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()
+        return text
+
    async def get_knowledge_graph(
        self, nodel_label: str, max_depth: int
    ) -> KnowledgeGraph: