Add env FORCE_LLM_SUMMARY_ON_MERGE

This commit is contained in:
yangdx
2025-04-10 17:29:07 +08:00
parent 35431644ad
commit 3007dff153
5 changed files with 74 additions and 50 deletions

View File

@@ -103,8 +103,10 @@ class LightRAG:
entity_extract_max_gleaning: int = field(default=1)
"""Maximum number of entity extraction attempts for ambiguous content."""
entity_summary_to_max_tokens: int = field(
default=int(os.getenv("MAX_TOKEN_SUMMARY", 500))
summary_to_max_tokens: int = field(default=int(os.getenv("MAX_TOKEN_SUMMARY", 500)))
force_llm_summary_on_merge: int = field(
default=int(os.getenv("FORCE_LLM_SUMMARY_ON_MERGE", 6))
)
# Text chunking