diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py index ac38dc4c..2ab9f89a 100644 --- a/lightrag/lightrag.py +++ b/lightrag/lightrag.py @@ -1799,26 +1799,12 @@ class LightRAG: ) ) - # Update pipeline status after getting affected_nodes - async with pipeline_status_lock: - log_message = f"Found {len(affected_nodes)} affected entities" - logger.info(log_message) - pipeline_status["latest_message"] = log_message - pipeline_status["history_messages"].append(log_message) - affected_edges = ( await self.chunk_entity_relation_graph.get_edges_by_chunk_ids( list(chunk_ids) ) ) - # Update pipeline status after getting affected_edges - async with pipeline_status_lock: - log_message = f"Found {len(affected_edges)} affected relations" - logger.info(log_message) - pipeline_status["latest_message"] = log_message - pipeline_status["history_messages"].append(log_message) - except Exception as e: logger.error(f"Failed to analyze affected graph elements: {e}") raise Exception(f"Failed to analyze graph dependencies: {e}") from e @@ -1836,6 +1822,14 @@ class LightRAG: elif remaining_sources != sources: entities_to_rebuild[node_label] = remaining_sources + async with pipeline_status_lock: + log_message = ( + f"Found {len(entities_to_rebuild)} affected entities" + ) + logger.info(log_message) + pipeline_status["latest_message"] = log_message + pipeline_status["history_messages"].append(log_message) + # Process relationships for edge_data in affected_edges: src = edge_data.get("source") @@ -1857,6 +1851,14 @@ class LightRAG: elif remaining_sources != sources: relationships_to_rebuild[edge_tuple] = remaining_sources + async with pipeline_status_lock: + log_message = ( + f"Found {len(relationships_to_rebuild)} affected relations" + ) + logger.info(log_message) + pipeline_status["latest_message"] = log_message + pipeline_status["history_messages"].append(log_message) + except Exception as e: logger.error(f"Failed to process graph analysis results: {e}") raise Exception(f"Failed to process graph dependencies: {e}") from e