src/retrieval_policy.py

• _doc_embedding_matrix · function · L37-L49 — def _doc_embedding_matrix(candidates: List[Candidate], embedding_model=None) -> np.ndarray
• mmr_select · function · L52-L100 — def mmr_select( candidates: List[Candidate], n: int, lam: float, embedding_model=None, ) -> List[Candidate]
• diversity_select · function · L103-L131 — def diversity_select( candidates: List[Candidate], n: int, embedding_model=None, seed: Optional[int] = None, ) -> List[Candidate]
• RetrievalPolicy · class · L138-L179 — class RetrievalPolicy
• init · method · L143-L145 — def init(self, seed: Optional[int] = None): # Stateful RNG seeded once from the base seed.
• batch_retrieve · method · L147-L168 — def batch_retrieve( self, memory_db, agent_id: str, queries: List[Tuple[int, str]], n_retrieval: int, prev_consensus_ratios: Dict[int, Optional[float]], ) -> Dict[int, Tuple[List[Candidate], List[Candidate]]]
• _next_seed · method · L170-L172 — def _next_seed(self) -> int
• _filter · method · L175-L179 — def _filter(agent_id: str, correct_only: bool = False) -> Dict
• DebateStateAwarePolicy · class · L182-L218 — class DebateStateAwarePolicy(RetrievalPolicy)
• init · method · L192-L197 — def init(self, gamma: float = 0.9, ref_lambda: float = 0.7, adaptive_lambda: bool = True, seed: Optional[int] = None)
• _lambda_for · method · L199-L202 — def _lambda_for(self, consensus_ratio: Optional[float]) -> float
• batch_retrieve · method · L204-L218 — def batch_retrieve(self, memory_db, agent_id, queries, n_retrieval, prev_consensus_ratios)
• FixedLambdaPolicy · class · L221-L226 — class FixedLambdaPolicy(DebateStateAwarePolicy)
• init · method · L225-L226 — def init(self, ref_lambda: float = 0.7, seed: Optional[int] = None)
• SimilarityPolicy · class · L229-L243 — class SimilarityPolicy(RetrievalPolicy)
• batch_retrieve · method · L233-L243 — def batch_retrieve(self, memory_db, agent_id, queries, n_retrieval, prev_consensus_ratios)
• DiversityPolicy · class · L246-L263 — class DiversityPolicy(RetrievalPolicy)
• batch_retrieve · method · L250-L263 — def batch_retrieve(self, memory_db, agent_id, queries, n_retrieval, prev_consensus_ratios)
• RandomPolicy · class · L266-L281 — class RandomPolicy(RetrievalPolicy)
• batch_retrieve · method · L270-L281 — def batch_retrieve(self, memory_db, agent_id, queries, n_retrieval, prev_consensus_ratios)
• PositiveOnlyPolicy · class · L284-L298 — class PositiveOnlyPolicy(RetrievalPolicy)
• batch_retrieve · method · L288-L298 — def batch_retrieve(self, memory_db, agent_id, queries, n_retrieval, prev_consensus_ratios)
• build_retrieval_policy · function · L311-L328 — def build_retrieval_policy(name: str, seed: Optional[int] = None, **policy_kwargs) -> RetrievalPolicy