feat: optimize graph visualize

Author: earayu

aperag/graph/lightrag/lightrag.py

@@ -375,7 +375,23 @@ async def get_knowledge_graph(
             KnowledgeGraph: Knowledge graph containing nodes and edges
         """
 
-        return await self.chunk_entity_relation_graph.get_knowledge_graph(node_label, max_depth, max_nodes)
+        kg = await self.chunk_entity_relation_graph.get_knowledge_graph(node_label, max_depth, max_nodes)
+
+        # Clean up descriptions in nodes by replacing GRAPH_FIELD_SEP with double newlines
+        if kg.nodes and len(kg.nodes) > 0:
+            for node in kg.nodes:
+                if node.properties.get("description"):
+                    # Replace <SEP> with double newlines for better readability
+                    node.properties["description"] = node.properties["description"].replace(GRAPH_FIELD_SEP, "\n\n")
+
+        # Clean up descriptions in edges by replacing GRAPH_FIELD_SEP with double newlines
+        if kg.edges and len(kg.edges) > 0:
+            for edge in kg.edges:
+                if edge.properties.get("description"):
+                    # Replace <SEP> with double newlines for better readability
+                    edge.properties["description"] = edge.properties["description"].replace(GRAPH_FIELD_SEP, "\n\n")
+
+        return kg
 
     def _get_storage_class(self, storage_name: str) -> Callable[..., Any]:
         # Direct class lookup from registry instead of dynamic import

aperag/service/graph_service.py

@@ -20,6 +20,7 @@
 from aperag.db.ops import async_db_ops
 from aperag.exceptions import CollectionNotFoundException
 from aperag.graph import lightrag_manager
+from aperag.graph.lightrag.types import KnowledgeGraph
 from aperag.schema import view_models
 from aperag.utils.utils import utc_now
 
@@ -92,7 +93,7 @@ async def get_knowledge_graph(
                 mode_description = f"subgraph from '{label}'"
 
             # Get knowledge graph
-            kg = await rag.get_knowledge_graph(
+            kg: KnowledgeGraph = await rag.get_knowledge_graph(
                 node_label=node_label,
                 max_depth=max_depth,
                 max_nodes=query_max_nodes,