src/multi_agent_debate.py

• remove_think_tags · function · L17-L18 — def remove_think_tags(text: str) -> str
• MultiAgentDebate · class · L21-L460 — class MultiAgentDebate
• init · method · L36-L62 — def init( self, agents: List[Agent], model: LanguageModel, task_type: str, max_round: int = 3, consensus_threshold: float = 0.8, if_summarize_state: bool = False, if_confidence_weight: bool = False, ) -> None
• run · method · L68-L117 — def run(self, queries: List[Query], is_training: bool = True, batch_idx: int = 0) -> List[Dict[str, Any]]
• _run_debate_round · method · L123-L273 — def _run_debate_round(self, queries: List[Query]) -> None
• _summarize_debate_state · method · L275-L326 — def _summarize_debate_state(self, queries: List[Query]) -> None
• _check_consensus · method · L328-L357 — def _check_consensus(self, query: Query, pred_answers: List[str]) -> tuple[bool, float, str]
• _initialize_debate · method · L363-L373 — def _initialize_debate(self, queries: List[Query]) -> None
• _clear_debate · method · L375-L382 — def _clear_debate(self) -> None
• _save_memory_data · method · L388-L452 — def _save_memory_data(self, queries: List[Query], batch_idx: int = 0) -> None
• save_debate_log · method · L454-L457 — def save_debate_log(self, filename: str) -> None
• get_token_usage_summary · method · L459-L460 — def get_token_usage_summary(self)