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

Log all merges at appropriate level

Browse source
This commit is contained in:
yangdx 2025-09-10 18:37:13 +08:00
parent a4bfdb7ddf
commit db6bba80c9
1 changed files with 16 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
lightrag/operate.py
View file

@ -1340,17 +1340,20 @@ async def _merge_nodes_then_upsert(
) )
# Log based on actual LLM usage # Log based on actual LLM usage
if already_fragment > 0 or llm_was_used: if llm_was_used:
if llm_was_used: status_message = f"LLMmrg: `{entity_name}` | {already_fragment}+{num_fragment - already_fragment}{dd_message}"
status_message = f"LLMmrg: `{entity_name}` | {already_fragment}+{num_fragment - already_fragment}{dd_message}" else:
else: status_message = f"Merged: `{entity_name}` | {already_fragment}+{num_fragment - already_fragment}{dd_message}"
status_message = f"Merged: `{entity_name}` | {already_fragment}+{num_fragment - already_fragment}{dd_message}"
if already_fragment > 0 or llm_was_used:
logger.info(status_message) logger.info(status_message)
if pipeline_status is not None and pipeline_status_lock is not None: if pipeline_status is not None and pipeline_status_lock is not None:
async with pipeline_status_lock: async with pipeline_status_lock:
pipeline_status["latest_message"] = status_message pipeline_status["latest_message"] = status_message
pipeline_status["history_messages"].append(status_message) pipeline_status["history_messages"].append(status_message)
else:
logger.debug(status_message)
else: else:
logger.error(f"Entity {entity_name} has no description") logger.error(f"Entity {entity_name} has no description")
description = "(no description)" description = "(no description)"
@ -1458,17 +1461,20 @@ async def _merge_edges_then_upsert(
) )
# Log based on actual LLM usage # Log based on actual LLM usage
if already_fragment > 0 or llm_was_used: if llm_was_used:
if llm_was_used: status_message = f"LLMmrg: `{src_id}`~`{tgt_id}` | {already_fragment}+{num_fragment - already_fragment}{dd_message}"
status_message = f"LLMmrg: `{src_id}`~`{tgt_id}` | {already_fragment}+{num_fragment - already_fragment}{dd_message}" else:
else: status_message = f"Merged: `{src_id}`~`{tgt_id}` | {already_fragment}+{num_fragment - already_fragment}{dd_message}"
status_message = f"Merged: `{src_id}`~`{tgt_id}` | {already_fragment}+{num_fragment - already_fragment}{dd_message}"
if already_fragment > 0 or llm_was_used:
logger.info(status_message) logger.info(status_message)
if pipeline_status is not None and pipeline_status_lock is not None: if pipeline_status is not None and pipeline_status_lock is not None:
async with pipeline_status_lock: async with pipeline_status_lock:
pipeline_status["latest_message"] = status_message pipeline_status["latest_message"] = status_message
pipeline_status["history_messages"].append(status_message) pipeline_status["history_messages"].append(status_message)
else:
logger.debug(status_message)
else: else:
logger.error(f"Edge {src_id} - {tgt_id} has no description") logger.error(f"Edge {src_id} - {tgt_id} has no description")
description = "(no description)" description = "(no description)"