src/agent.py

• Query · class · L11-L23 — class Query
• str · method · L19-L20 — def str(self)
• repr · method · L22-L23 — def repr(self)
• AgentId · class · L27-L35 — class AgentId
• str · method · L31-L32 — def str(self) -> str
• repr · method · L34-L35 — def repr(self) -> str
• RetrievedMemory · class · L39-L50 — class RetrievedMemory
• str · method · L46-L47 — def str(self)
• repr · method · L49-L50 — def repr(self)
• Agent · class · L53-L427 — class Agent
• init · method · L61-L100 — def init( self, agent_id: AgentId, task_type: str, if_use_memory: bool = False, memory_db: Optional[MemoryVectorDB] = None, n_retrieval: int = 3, if_confidence_weight: bool = False, retrieval_policy: str = "debate_state_aware", seed: Optional[int] = None, ref_lambda: Optional[float] = None, retrieve_first_round: bool = False, memory_agent_id: Optional[str] = None, ) -> None
• initialize_conversation · method · L106-L109 — def initialize_conversation(self, query: Query) -> None
• clear_conversation · method · L111-L113 — def clear_conversation(self, query: Query) -> None
• clear_all_conversations · method · L115-L117 — def clear_all_conversations(self) -> None
• _build_prompt · method · L123-L164 — def _build_prompt( self, query: Query, prev_responses: Optional[Dict[AgentId, Tuple[str, Optional[np.ndarray]]]], retrieved_memories: Optional[List[RetrievedMemory]] = None, other_confidences: Optional[Dict[AgentId, float]] = None, ) -> str
• build_messages · method · L166-L211 — def build_messages( self, query: Query, prev_responses: Optional[Dict[AgentId, Tuple[str, Optional[np.ndarray]]]] = None, prev_round_summary: Optional[str] = None, prev_consensus_ratio: Optional[float] = None, prefetched: Optional[Tuple[List[RetrievedMemory], Optional[Dict[AgentId, float]]]] = None, ) -> List[Dict[str, str]]
• record_response · method · L213-L216 — def record_response(self, query: Query, response: str) -> None
• batch_retrieve_all · method · L222-L322 — def batch_retrieve_all( self, queries: List[Query], prev_responses_map: Dict[int, Dict[AgentId, Tuple[str, Optional[np.ndarray]]]], prev_round_summaries: Dict[int, Optional[str]], prev_consensus_ratios: Dict[int, Optional[float]], ) -> Dict[int, Tuple[List[RetrievedMemory], Optional[Dict[AgentId, float]]]]
• _estimate_confidences_cached · method · L324-L381 — def _estimate_confidences_cached( self, prev_responses: Dict[AgentId, Tuple[str, Optional[np.ndarray]]], candidate_memories: List[Tuple[Dict, str, float, Optional[np.ndarray]]], resp_emb_cache: Dict[Tuple, Optional[np.ndarray]], ) -> Dict[AgentId, float]
• _build_query_text · method · L383-L410 — def _build_query_text( self, query: Query, prev_agent_response: Optional[str] = None, prev_round_summary: Optional[str] = None, ) -> str
• _to_retrieved_memories · method · L412-L427 — def _to_retrieved_memories( self, query_id: int, retrieved: List[Tuple[Dict, str, float, Optional[np.ndarray]]], ) -> List[RetrievedMemory]