Merge pull request #885 from YanSte/remove-unsused

Cleanup of code

lightrag/kg/age_impl.py

@@ -60,10 +60,6 @@ class AGEQueryException(Exception):
 @final
 @dataclass
 class AGEStorage(BaseGraphStorage):
-    @staticmethod
-    def load_nx_graph(file_name):
-        print("no preloading of graph with AGE in production")
-
     def __init__(self, namespace, global_config, embedding_func):
         super().__init__(
             namespace=namespace,
@@ -620,9 +616,6 @@ class AGEStorage(BaseGraphStorage):
     ) -> tuple[np.ndarray[Any, Any], list[str]]:
         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:

lightrag/kg/gremlin_impl.py

@@ -403,9 +403,6 @@ class GremlinStorage(BaseGraphStorage):
     ) -> tuple[np.ndarray[Any, Any], list[str]]:
         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:

lightrag/kg/mongo_impl.py

@@ -601,24 +601,6 @@ class MongoGraphStorage(BaseGraphStorage):
     # -------------------------------------------------------------------------
     #
 
-    async def get_all_labels(self) -> list[str]:
-        """
-        Get all existing node _id in the database
-        Returns:
-            [id1, id2, ...]  # Alphabetically sorted id list
-        """
-        # Use MongoDB's distinct and aggregation to get all unique labels
-        pipeline = [
-            {"$group": {"_id": "$_id"}},  # Group by _id
-            {"$sort": {"_id": 1}},  # Sort alphabetically
-        ]
-
-        cursor = self.collection.aggregate(pipeline)
-        labels = []
-        async for doc in cursor:
-            labels.append(doc["_id"])
-        return labels
-
     async def get_knowledge_graph(
         self, node_label: str, max_depth: int = 5
     ) -> KnowledgeGraph:

lightrag/kg/neo4j_impl.py

@@ -628,31 +628,6 @@ class Neo4JStorage(BaseGraphStorage):
         await traverse(label, 0)
         return result
 
-    async def get_all_labels(self) -> list[str]:
-        """
-        Get all existing node labels in the database
-        Returns:
-            ["Person", "Company", ...]  # Alphabetically sorted label list
-        """
-        async with self._driver.session(database=self._DATABASE) as session:
-            # Method 1: Direct metadata query (Available for Neo4j 4.3+)
-            # query = "CALL db.labels() YIELD label RETURN label"
-
-            # Method 2: Query compatible with older versions
-            query = """
-                MATCH (n)
-                WITH DISTINCT labels(n) AS node_labels
-                UNWIND node_labels AS label
-                RETURN DISTINCT label
-                ORDER BY label
-            """
-
-            result = await session.run(query)
-            labels = []
-            async for record in result:
-                labels.append(record["label"])
-            return labels
-
     async def delete_node(self, node_id: str) -> None:
         raise NotImplementedError
 

lightrag/kg/networkx_impl.py

@@ -167,9 +167,6 @@ class NetworkXStorage(BaseGraphStorage):
             if self._graph.has_edge(source, target):
                 self._graph.remove_edge(source, target)
 
-    async def get_all_labels(self) -> list[str]:
-        raise NotImplementedError
-
     async def get_knowledge_graph(
         self, node_label: str, max_depth: int = 5
     ) -> KnowledgeGraph:

lightrag/kg/oracle_impl.py

@@ -670,9 +670,6 @@ class OracleGraphStorage(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:

lightrag/kg/postgres_impl.py

@@ -180,12 +180,10 @@ class PostgreSQLDB:
             asyncpg.exceptions.UniqueViolationError,
             asyncpg.exceptions.DuplicateTableError,
         ) as e:
-            if upsert:
-                print("Key value duplicate, but upsert succeeded.")
-            else:
-                logger.error(f"Upsert error: {e}")
+            if not upsert:
+                logger.error(f"PostgreSQL, upsert error: {e}")
         except Exception as e:
-            logger.error(f"PostgreSQL database,\nsql:{sql},\ndata:{data},\nerror:{e}")
+            logger.error(f"PostgreSQL database, sql:{sql}, data:{data}, error:{e}")
             raise
 
 
@@ -1089,9 +1087,6 @@ class PGGraphStorage(BaseGraphStorage):
     ) -> tuple[np.ndarray[Any, Any], list[str]]:
         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:

lightrag/kg/tidb_impl.py

@@ -560,9 +560,6 @@ 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: