piZ#dZddlmZddlZddlZddlmZddlmZddl m Z m Z erddl m Z ejeZdZd Zd Zd Zd ZddZGddZdS)zIntelligent context compression for multi-agent meetings. Prevents whiteboard explosion by compressing scratchpad content while preserving recent entries in full and keeping full transcripts. ) annotationsN) TYPE_CHECKING)format_scratchpad_summary) MeetingStateScratchpadEntry) LLMClienti zSummarize the following meeting discussion into a structured briefing. Preserve: key arguments, disagreements, decisions, open questions. Remove: rhetorical flourishes, examples used for illustration, repetition. Output in bullet points, max 800 tokens.zSummarize the following background material into a structured briefing. Preserve: key facts, requirements, constraints, goals. Remove: verbose explanations, examples, repetition. Output in bullet points, max 1500 tokens.textstrreturnintc|sdSttjd|}t||z }|dz }|dz }t||zS)aEstimate token count for mixed Chinese/English text. Uses a heuristic: Chinese characters ~1.5 chars/token, English/ASCII ~4 chars/token. No external dependencies. Args: text: Input text to estimate. Returns: Approximate token count. rz[\u4e00-\u9fff\u3400-\u4dbf]g?g@)lenrefindallr)r chinese_chars other_charschinese_tokens other_tokenss :/root/projects/multi-agents-meeting/src/context_manager.pyestimate_tokensr)sb q #BDIIJJMd))m+K"S(N$L ~ , - --c<eZdZdZdddZdd ZddZddZddZdS)ContextManagera]Manage meeting context to prevent whiteboard explosion. Uses a three-level compression strategy: - Level 0: Full content (under budget) - Level 1: Truncate early entries, keep recent in full - Level 2: LLM-summarize early entries, keep recent in full Args: budget_tokens: Maximum tokens for the scratchpad portion. p budget_tokensrr Nonec||_dS)N)r)selfrs r__init__zContextManager.__init__Ms*rentrieslist[ScratchpadEntry]clientrr cl|sdSt|}t|}||jkrtd||St |t kr6t |dkr tdt |dz nd}nt }|dkr |d| n|}|dkr || dng}|rt|nd}t|} td|j| z } g} |D]S} | j } t | tkr| dtdz} | d| j d | Td | }t|}|| kr4| dkr.| r,td || z|}|r|d |zz }|Std |r0t|}|td |dg}n%|td |dg}d|S|rd|d|Sd|S)uCompress scratchpad entries to fit within budget. Strategy (progressive disclosure): 1. If raw scratchpad < budget → return full text (Level 0) 2. If exceeds → truncate early entries, keep recent 2 full (Level 1) 3. If still exceeds → LLM-summarize early entries (Level 2) Args: entries: All scratchpad entries. client: LLM client for Level 2 summarization. Returns: Compressed scratchpad text fitting within budget. zNo discussion yet.z$Level 0: scratchpad fits (%d tokens)rNz...[z]:  z,Level 1: truncated early entries (%d tokens)zLevel 2: LLM summarizationuserrolecontentsystemmessagesz[Discussion Summary] z[Earlier Discussion Summary] z [Recent Entries] )rrrloggerdebugr_RECENT_ENTRIES_TO_KEEPminmaxr-_TRUNCATE_CHARSappend agent_namejoinchat_SUMMARIZE_SCRATCHPAD_PROMPT)r r"r$ full_text full_tokens recent_count early_entriesrecent_entries recent_text recent_tokens early_budgettruncated_linesentryr-truncated_texttruncated_tokenscombinedearly_full_textsummarys rcompact_scratchpadz!ContextManager.compact_scratchpadPs& (''-g66 %i00  $, , , LL? M M M  w<<2 2 27:7||a7G7G3q#g,,"2333QLL2L3?!3C3C,// 4@14D4D,00"CQY/???WY ' 44 1d0=@AA &(" G GEmG7||o--!"2?"23e;  " "#Eu'7#E#EG#E#E F F F F?33*>:: | + + q0@0@_0@ LLGIY\iIi j j j%H /D;..O  1222  67 FFOkk3#)oFFG"GG kk3#)i@@A"G6G55 5  a`G``S^`` `1111rstatercfd|jdg}|jrO|jdks|js|d|jdn|d|jd||j|}|d|jd|d|S)aRBuild compressed context for the PM agent. Round 0 (first round): includes full context. Later rounds: uses context_summary if available. Args: state: Current meeting state. client: LLM client for compression. Returns: Formatted string for PM's user message. z## Meeting Topic r)rz## Background Context z!## Background Context (Summary) z## Current Whiteboard (Round z) )topiccontext current_roundcontext_summaryr7rK scratchpadr9)r rLr$partsscratchpad_texts rbuild_pm_contextzContextManager.build_pm_contexts"C5;BBBC = ^"a''u/D' JJJJKKKK \5CX\\\]]]11%2BFKK `U5H````aaayyr target_agent pm_promptcZg}|jr|d|jdn%|jr|d|jd||j|}|||r|d|d|S)aBuild compressed context for a specific agent. Agents get context_summary (not full context) and compressed scratchpad. Args: state: Current meeting state. target_agent: Name of the agent receiving this context. pm_prompt: The PM's specific question for this agent. client: LLM client for compression. Returns: Formatted string for the agent's scratchpad_summary parameter. z [Background] r)z [PM's Question] )rQr7rOrKrRr9)r rLrVrWr$rSrTs rbuild_agent_contextz"ContextManager.build_agent_contexts(   = LLC%*?CCC D D D D ] = LL;%-;;; < < <11%2BFKK _%%%  < LL:y:: ; ; ;yyrrOc|sdSt|}|tkr|S|td|dg}|S)a/Generate a summary of long background context. If context is short enough, returns it as-is. Args: context: Original background context. client: LLM client for summarization. Returns: Original context if short, LLM summary if long. r'r*r+r.)r_CONTEXT_SUMMARY_THRESHOLDr:_SUMMARIZE_CONTEXT_PROMPT)r rOr$tokensrJs rsummarize_contextz ContextManager.summarize_contexts_  2 )) / / /N++,%'::;  rN)r)rrr r)r"r#r$rr r )rLrr$rr r ) rLrrVr rWr r$rr r )rOr r$rr r ) __name__ __module__ __qualname____doc__r!rKrUrYr^rrrrAs  +++++S2S2S2S2j    @$ $ $ $ Lrr)r r r r)rb __future__rloggingrtypingr src.agentr src.modelsrrsrc.llm_clientr getLoggerr_r1r[r3r6r;r\rrrcrrrks #""""" //////44444444)((((((  8 $ $"/0....0EEEEEEEEEEr