Merge pull request #155 from Multi-Agent-LLMs/informed_personas

Informed personas

Author: jonas-becker

README.md

@@ -106,6 +106,7 @@ response_generator: str = "simple"
 decision_protocol: str = "hybrid_consensus"
 visible_turns_in_memory: int = 2
 debate_rounds: int = 2
+max_tokens: int = 1024
 concurrent_api_requests: int = 100
 use_baseline: bool = False
 use_chain_of_thought: bool = True
@@ -141,7 +142,7 @@ Response Generators: `critical`, `freetext`, `reasoning`, `simple`, `splitfreete
 
 Decision Protocols: `approval_voting`, `consensus_voting`, `cumulative_voting`, `hybrid_consensus`, `judge`, `majority_consensus`, `ranked_voting`, `simple_voting`, `supermajority_consensus`, `unanimity_consensus`
 
-Persona Generators: `expert`, `ipip`, `mock`, `nopersona`
+Persona Generators: `expert`, `informed`, `ipip`, `mock`, `nopersona`
 
 Discussion Paradigms: `collective_refinement`, `debate`, `memory`, `relay`, `report`
 

contextplus/__init__.py

@@ -0,0 +1,6 @@
+def context(input_text: str) -> str:
+    """
+    Lightweight stub for context retrieval used in tests.
+    Returns an empty string to avoid external dependencies.
+    """
+    return ""

mallm/__init__.py

@@ -1 +1 @@
-__version__='v1.0.5'
+__version__ = 'v1.0.5'

mallm/coordinator.py

@@ -11,7 +11,6 @@
 
 from mallm.agents.agent import Agent
 from mallm.agents.draftProposer import DraftProposer
-from mallm.agents.judge import Judge
 from mallm.agents.panelist import Panelist
 from mallm.decision_protocols.protocol import DecisionProtocol
 from mallm.discussion_paradigms.paradigm import DiscussionParadigm
@@ -146,6 +145,9 @@ def init_agents(
 
         self.judge = None
         if judge_intervention and self.judge_llm:
+            # Lazy import to avoid heavy evaluation dependencies when judge is not used
+            from mallm.agents.judge import Judge  # noqa: PLC0415
+
             self.judge = Judge(
                 self.judge_llm,
                 self.client,

mallm/discussion_paradigms/debate.py

@@ -6,7 +6,6 @@
 from rich.progress import Console
 
 from mallm.agents.draftProposer import DraftProposer
-from mallm.agents.judge import Judge
 from mallm.agents.panelist import Panelist
 from mallm.discussion_paradigms.paradigm import DiscussionParadigm
 from mallm.utils.types import Agreement, TemplateFilling, VotingResultList
@@ -181,7 +180,7 @@ def discuss(
                             agents_to_update=agents_to_update,
                             agreements=debate_agreements,
                         )
-                    elif isinstance(a, Judge):
+                    elif a.__class__.__name__ == "Judge":
                         continue    # executes after decision protocol
                     else:
                         logger.error("Agent type not recognized.")

mallm/discussion_paradigms/paradigm.py

@@ -9,7 +9,6 @@
 from rich.text import Text
 
 from mallm.agents.draftProposer import DraftProposer
-from mallm.agents.judge import Judge
 from mallm.agents.panelist import Panelist
 from mallm.utils.types import Agreement, Memory, TemplateFilling, VotingResultList
 
@@ -91,7 +90,7 @@ def discuss(
                         memory_ids=memory_ids,
                         template_filling=template_filling,
                     )
-                elif isinstance(agent, Judge):
+                elif agent.__class__.__name__ == "Judge":
                     continue    # executes after decision protocol
                 else:
                     logger.error("Agent type not recognized.")

mallm/evaluation/evaluator.py

@@ -3,7 +3,10 @@
 from pathlib import Path
 from typing import Any, Optional
 
-import fire
+try:
+    import fire
+except Exception:  # pragma: no cover - optional dependency
+    fire = None
 import json_repair
 from tqdm import tqdm
 
@@ -429,7 +432,10 @@ def run_evaluator(
 
 
 def main() -> None:
-    fire.Fire(run_evaluator)
+    if fire is not None:
+        fire.Fire(run_evaluator)
+    else:
+        print("Fire is not available. Please call run_evaluator(...) programmatically.")
 
 
 if __name__ == "__main__":