diff --git a/telegram/kb_retrieval.py b/telegram/kb_retrieval.py index a1e9241..87106e2 100644 --- a/telegram/kb_retrieval.py +++ b/telegram/kb_retrieval.py @@ -581,8 +581,8 @@ def format_context_for_prompt(ctx: KBContext) -> str: sections.append("## Matched Entities") for i, ent in enumerate(ctx.entities): sections.append(f"**{ent.name}** ({ent.entity_type}, {ent.domain})") - # Top entity gets full content (up to 2000 chars), rest get truncated - if i == 0: + # Top 3 entities get full content, rest get truncated + if i < 3: sections.append(ent.overview[:2000]) else: sections.append(ent.overview[:500])