From 4ae5246a7e557926424ec01173f1a4f5438fd890 Mon Sep 17 00:00:00 2001
From: yangdx <yangdx@znipower.com>
Date: Sun, 20 Apr 2025 12:36:32 +0800
Subject: [PATCH] Remove summary length check for entity relations

- Summary now determined by num_fragment
---
 lightrag/operate.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/lightrag/operate.py b/lightrag/operate.py
index 8eb7cf24..466eca50 100644
--- a/lightrag/operate.py
+++ b/lightrag/operate.py
@@ -121,8 +121,11 @@ async def _handle_entity_relation_summary(
     )
 
     tokens = tokenizer.encode(description)
-    if len(tokens) < summary_max_tokens:  # No need for summary
-        return description
+
+    ### summarize is not determined here anymore (It's determined by num_fragment now)
+    # if len(tokens) < summary_max_tokens:  # No need for summary
+    #     return description
+
     prompt_template = PROMPTS["summarize_entity_descriptions"]
     use_description = tokenizer.decode(tokens[:llm_max_tokens])
     context_base = dict(