Merge pull request #885 from YanSte/remove-unsused

Cleanup of code

lightrag/api/lightrag_server.py

@@ -1682,11 +1682,6 @@ def create_app(args):
             trace_exception(e)
             raise HTTPException(status_code=500, detail=str(e))
 
-    # query all graph labels
-    @app.get("/graph/label/list")
-    async def get_graph_labels():
-        return await rag.get_graph_labels()
-
     # query all graph
     @app.get("/graphs")
     async def get_knowledge_graph(label: str):

lightrag/base.py

@@ -198,10 +198,6 @@ class BaseGraphStorage(StorageNameSpace, ABC):
     ) -> tuple[np.ndarray[Any, Any], list[str]]:
         """Get all labels in the graph."""
 
-    @abstractmethod
-    async def get_all_labels(self) -> list[str]:
-        """Get a knowledge graph of a node."""
-
     @abstractmethod
     async def get_knowledge_graph(
         self, node_label: str, max_depth: int = 5

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:

lightrag/lightrag.py

@@ -615,10 +615,6 @@ 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:

lightrag/tools/lightrag_visualizer/graph_visualizer.py

@@ -1,6 +1,6 @@
 from typing import Optional, Tuple, Dict, List
 import numpy as np
-import networkx as nx
+
 import pipmaster as pm
 
 # Added automatic libraries install using pipmaster
@@ -12,7 +12,10 @@ if not pm.is_installed("pyglm"):
     pm.install("pyglm")
 if not pm.is_installed("python-louvain"):
     pm.install("python-louvain")
+if not pm.is_installed("networkx"):
+    pm.install("networkx")
 
+import networkx as nx
 import moderngl
 from imgui_bundle import imgui, immapp, hello_imgui
 import community