nidZddlmZddlZddlZddlmZddlmZddl m Z ddl m Z m Z mZer ddlmZdd lmZd ZddZ dddZdS)z?PM Agent router - routes discussion between participant Agents.) annotationsN) TYPE_CHECKING)ValidationError)format_scratchpad_summary) AgentConfig MeetingState PMDecision)ContextManager) LLMClientavailable_agentsdict[str, AgentConfig]returnstrcpdd|D}d|dS)zBuild the PM Agent's system prompt with available agents listed. Args: available_agents: Mapping of agent name to AgentConfig. Returns: A fully-formed system prompt string for the PM LLM call.  c36K|]\}}d|d|jVdS)z- z: N)role).0nameconfigs -/root/projects/multi-agents-meeting/src/pm.py z)build_pm_system_prompt..sJ'3tV"T""V[""zYou are a meeting facilitator (PM). Your job is to: 1. Analyze the current whiteboard state 2. Decide which Agent should speak next, or if the meeting should end 3. Output ONLY valid JSON matching the schema below Available Agents: a You MUST output ONLY a JSON object with these fields: - "analysis": brief analysis of current whiteboard state - "next_action": either "CALL_AGENT" or "FINISH" - "target_agent": name of the agent to call (required if CALL_AGENT) - "prompt_for_agent": specific question for that agent (required if CALL_AGENT) - "final_report": complete Markdown meeting report (required if FINISH) Rules: - DO NOT wrap up prematurely. Explore the topic deeply before concluding. - Ask follow-up questions when an agent gives a surface-level answer. - Call the same agent multiple times if their area needs deeper exploration. - Challenge assumptions: ask agents to justify their reasoning with specifics. - Cross-pollinate: share one agent's perspective with another for counterpoints. - Ensure at least 2-3 rounds of back-and-forth before considering FINISH. - Call each agent at least once before finishing. - When finishing, produce a comprehensive report that reflects the depth of discussion. - The target_agent MUST be one of the available agents listed above. - Output ONLY the JSON object, no other text. - If you see entries from [HUMAN], treat them as high-priority stakeholder feedback. Address their concerns by calling relevant agents to respond.)joinitems)r agent_lists rbuild_pm_system_promptrsb7G7M7M7O7OJ C   C C CCrstaterclientr context_managerContextManager | Noner czt|}||||}nDt|j}d|jd}|jr|d|jdz }|d|jd|z }d|dg}d }ttD]} |r| dd |d d| || } | } | d rUd| vr| dddn| } | d r | dd} | } | ds6tjd| tj} | r| } | drdd} d}d}t'| D]N\}}|rd} |dkr|rd}|dkr| }|r"|dkr| dz } .|dkr| dz} | dkr| d|dz} nOt)j| }t-di|}|jdkr=|j|vr4d|jdd|}|cS#t(jt8f$r}t;|}Yd}~d}~wwxYwt=dtd|) aRun PM Agent to get the next routing decision. Calls the LLM with the current whiteboard state. Retries up to _MAX_RETRIES times when the response is invalid JSON, fails Pydantic validation, or references a non-existent agent. Each retry appends an error-feedback message so the LLM can self-correct. Args: state: Current MeetingState containing topic, scratchpad, and round. available_agents: Mapping of agent name to AgentConfig. client: Any object satisfying the LLMClient Protocol. context_manager: Optional ContextManager for whiteboard compression. Returns: A validated PMDecision instance. Raises: RuntimeError: When all retry attempts are exhausted without a valid decision. Nz## Meeting Topic z z## Background Context z## Current Whiteboard (Round z) user)rcontentz$Your previous response was invalid: z . Please output ONLY valid JSON.)systemmessagesz```r{z\{.*\}rF\T"} CALL_AGENTztarget_agent 'z!' is not available. Choose from: z, z&Failed to get valid PM decision after z attempts. Last error: )rbuild_pm_contextr scratchpadtopiccontext current_roundrange _MAX_RETRIESappendchatstrip startswithsplitendswithresearchDOTALLgroup enumeratejsonloadsr next_action target_agentrkeysJSONDecodeErrorrr RuntimeError)rr r r! system_prompt user_contentscratchpad_textr( last_errorattempt raw_responsecleanedmatchdepth in_string escape_nextichdatadecisionexcs rrun_pmrZ@s2++;<.C.C.E.E$F$FIIOOO$o6   SJ HHHH   $ $ $! $ $  sF(I0,I00J JJ )r rrr)N) rrr rr r r!r"rr )__doc__ __future__rrCr>typingrpydanticr src.agentr src.modelsrrr src.context_managerr src.llm_clientr r7rrZr0rrrcsEE"""""" $$$$$$//////<<<<<<<<<<)222222(((((( (C(C(C(C^.2 ooooooor