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/kg/tidb_impl.py
+++ b/lightrag/kg/tidb_impl.py
@@ -560,6 +560,9 @@ class TiDBGraphStorage(BaseGraphStorage):
    async def delete_node(self, node_id: str) -> None:
        raise NotImplementedError

+    async def get_all_labels(self) -> list[str]:
+        raise NotImplementedError
+
    async def get_knowledge_graph(
        self, node_label: str, max_depth: int = 5
    ) -> KnowledgeGraph: