gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix linting

Browse Source
This commit is contained in:
yangdx
2025-04-10 20:32:40 +08:00
parent 339bc99259
commit 7d69449c67
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
lightrag/operate.py
View File

@@ -615,10 +615,14 @@ async def extract_entities(
# Merge results - only add entities and edges with new names # Merge results - only add entities and edges with new names
for entity_name, entities in glean_nodes.items(): for entity_name, entities in glean_nodes.items():
if entity_name not in maybe_nodes: # Only accetp entities with new name in gleaning stage if (
entity_name not in maybe_nodes
): # Only accetp entities with new name in gleaning stage
maybe_nodes[entity_name].extend(entities) maybe_nodes[entity_name].extend(entities)
for edge_key, edges in glean_edges.items(): for edge_key, edges in glean_edges.items():
if edge_key not in maybe_edges: # Only accetp edges with new name in gleaning stage if (
edge_key not in maybe_edges
): # Only accetp edges with new name in gleaning stage
maybe_edges[edge_key].extend(edges) maybe_edges[edge_key].extend(edges)
if now_glean_index == entity_extract_max_gleaning - 1: if now_glean_index == entity_extract_max_gleaning - 1: