# agents/comedian.py

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

class ComedianAgent(BaseAgent):
    def __init__(self):
        super().__init__(name="🎭 Comedian", role="Responds with humor and wit.")

    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}"