From 97a2ee4ef10b1b41e7c808b0118f33b3e9e5b22e Mon Sep 17 00:00:00 2001 From: yangdx Date: Sat, 25 Oct 2025 11:17:43 +0800 Subject: [PATCH] Rename rebuild function name and improve relationship logging format --- lightrag/lightrag.py | 4 ++-- lightrag/operate.py | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py index 24ea0209..cc1dfcd7 100644 --- a/lightrag/lightrag.py +++ b/lightrag/lightrag.py @@ -87,7 +87,7 @@ from lightrag.operate import ( merge_nodes_and_edges, kg_query, naive_query, - _rebuild_knowledge_from_chunks, + rebuild_knowledge_from_chunks, ) from lightrag.constants import GRAPH_FIELD_SEP from lightrag.utils import ( @@ -3347,7 +3347,7 @@ class LightRAG: # 8. Rebuild entities and relationships from remaining chunks if entities_to_rebuild or relationships_to_rebuild: try: - await _rebuild_knowledge_from_chunks( + await rebuild_knowledge_from_chunks( entities_to_rebuild=entities_to_rebuild, relationships_to_rebuild=relationships_to_rebuild, knowledge_graph_inst=self.chunk_entity_relation_graph, diff --git a/lightrag/operate.py b/lightrag/operate.py index 36c8251d..cca9db15 100644 --- a/lightrag/operate.py +++ b/lightrag/operate.py @@ -502,7 +502,7 @@ async def _handle_single_relationship_extraction( return None -async def _rebuild_knowledge_from_chunks( +async def rebuild_knowledge_from_chunks( entities_to_rebuild: dict[str, list[str]], relationships_to_rebuild: dict[tuple[str, str], list[str]], knowledge_graph_inst: BaseGraphStorage, @@ -723,7 +723,7 @@ async def _rebuild_knowledge_from_chunks( rebuilt_relationships_count += 1 except Exception as e: failed_relationships_count += 1 - status_message = f"Failed to rebuild `{src} - {tgt}`: {e}" + status_message = f"Failed to rebuild `{src}`~`{tgt}`: {e}" logger.info(status_message) # Per requirement, change to info if pipeline_status is not None and pipeline_status_lock is not None: async with pipeline_status_lock: @@ -1487,7 +1487,7 @@ async def _rebuild_single_relationship( raise # Re-raise exception # Log rebuild completion with truncation info - status_message = f"Rebuild `{src} - {tgt}` from {len(chunk_ids)} chunks" + status_message = f"Rebuild `{src}`~`{tgt}` from {len(chunk_ids)} chunks" if truncation_info: status_message += f" ({truncation_info})" # Add truncation info from apply_source_ids_limit if truncation occurred