Remove summary length check for entity relations

- Summary now determined by num_fragment
This commit is contained in:
yangdx
2025-04-20 12:36:32 +08:00
parent 48e49fbe34
commit 4ae5246a7e

View File

@@ -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(