gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add TODO: There is performance when iterating get_all_labels

Browse source
This commit is contained in:
yangdx 2025-06-24 11:32:28 +08:00
parent 8b6dcfb6eb
commit 2946bbdb71
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
lightrag/lightrag.py
View file

@ -1762,6 +1762,7 @@ class LightRAG:
graph_db_lock = get_graph_db_lock(enable_logging=False) graph_db_lock = get_graph_db_lock(enable_logging=False)
async with graph_db_lock: async with graph_db_lock:
# Process entities # Process entities
# TODO There is performance when iterating get_all_labels for PostgresSQL
all_labels = await self.chunk_entity_relation_graph.get_all_labels() all_labels = await self.chunk_entity_relation_graph.get_all_labels()
for node_label in all_labels: for node_label in all_labels:
node_data = await self.chunk_entity_relation_graph.get_node( node_data = await self.chunk_entity_relation_graph.get_node(
@ -1785,6 +1786,7 @@ class LightRAG:
) )
# Process relationships # Process relationships
# TODO There is performance when iterating get_all_labels for PostgresSQL
for node_label in all_labels: for node_label in all_labels:
node_edges = await self.chunk_entity_relation_graph.get_node_edges( node_edges = await self.chunk_entity_relation_graph.get_node_edges(
node_label node_label