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

Refactor: move force_llm_summary_on_merge to global_config access

Browse source 
- Remove parameter from function signature
- Access from global_config instead
- Improve code consistency
This commit is contained in:
yangdx 2025-08-26 18:02:39 +08:00
parent 01a2c79f29
commit 79e0226b2b
1 changed files with 5 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
lightrag/operate.py
View file

@ -117,7 +117,6 @@ async def _handle_entity_relation_summary(
description_type: str,
entity_or_relation_name: str,
description_list: list[str],
force_llm_summary_on_merge: int,
seperator: str,
global_config: dict,
llm_response_cache: BaseKVStorage | None = None,
@ -152,6 +151,7 @@ async def _handle_entity_relation_summary(
tokenizer: Tokenizer = global_config["tokenizer"]
summary_context_size = global_config["summary_context_size"]
summary_max_tokens = global_config["summary_max_tokens"]
force_llm_summary_on_merge = global_config["force_llm_summary_on_merge"]
current_list = description_list[:] # Copy the list to avoid modifying original
llm_was_used = False # Track whether LLM was used during the entire process
@ -880,14 +880,12 @@ async def _rebuild_single_entity(
# deduplicate descriptions
description_list = list(dict.fromkeys(relationship_descriptions))
# Generate description from relationships or fallback to current
# Generate final description from relationships or fallback to current
if description_list:
force_llm_summary_on_merge = global_config["force_llm_summary_on_merge"]
final_description, _ = await _handle_entity_relation_summary(
"Entity",
entity_name,
description_list,
force_llm_summary_on_merge,
GRAPH_FIELD_SEP,
global_config,
llm_response_cache=llm_response_cache,
@ -923,14 +921,12 @@ async def _rebuild_single_entity(
else current_entity.get("entity_type", "UNKNOWN")
)
# Generate final description and update storage
force_llm_summary_on_merge = global_config["force_llm_summary_on_merge"]
# Generate final description from entities or fallback to current
if description_list:
final_description, _ = await _handle_entity_relation_summary(
"Entity",
entity_name,
description_list,
force_llm_summary_on_merge,
GRAPH_FIELD_SEP,
global_config,
llm_response_cache=llm_response_cache,
@ -1005,14 +1001,12 @@ async def _rebuild_single_relationship(
weight = sum(weights) if weights else current_relationship.get("weight", 1.0)
# Use summary if description has too many fragments
force_llm_summary_on_merge = global_config["force_llm_summary_on_merge"]
# Generate final description from relations or fallback to current
if description_list:
final_description, _ = await _handle_entity_relation_summary(
"Relation",
f"{src}-{tgt}",
description_list,
force_llm_summary_on_merge,
GRAPH_FIELD_SEP,
global_config,
llm_response_cache=llm_response_cache,
@ -1096,6 +1090,7 @@ async def _merge_nodes_then_upsert(
reverse=True,
)[0][0] # Get the entity type with the highest count
# merge and deduplicate description
description_list = list(
dict.fromkeys(
already_description
@ -1103,7 +1098,6 @@ async def _merge_nodes_then_upsert(
)
)
force_llm_summary_on_merge = global_config["force_llm_summary_on_merge"]
num_fragment = len(description_list)
already_fragment = len(already_description)
deduplicated_num = already_fragment + len(nodes_data) - num_fragment
@ -1117,7 +1111,6 @@ async def _merge_nodes_then_upsert(
"Entity",
entity_name,
description_list,
force_llm_summary_on_merge,
GRAPH_FIELD_SEP,
global_config,
llm_response_cache,
@ -1222,7 +1215,6 @@ async def _merge_edges_then_upsert(
)
)
force_llm_summary_on_merge = global_config["force_llm_summary_on_merge"]
num_fragment = len(description_list)
already_fragment = len(already_description)
deduplicated_num = already_fragment + len(edges_data) - num_fragment
@ -1236,7 +1228,6 @@ async def _merge_edges_then_upsert(
"Relation",
f"({src_id}, {tgt_id})",
description_list,
force_llm_summary_on_merge,
GRAPH_FIELD_SEP,
global_config,
llm_response_cache,