agents/lawyer.py

# agents/lawyer.py

from agents.base_agent import BaseAgent, ACPMessage
from multi_inference import multi_query

class LawyerAgent(BaseAgent):
    def __init__(self):
        super().__init__(name="⚖️ Lawyer", role="Analyzes from a legal and ethical perspective.")

    def generate(self, messages):
        prompt = f"{self.role}:\n" + "\n".join([f"{m.role}: {m.content}" for m in messages])
        try:
            return multi_query(prompt)
        except Exception as e:
            return f"[ERROR] {e}"