Update agents/comedian.py

agents/comedian.py

@@ -1,14 +1,17 @@
+
 # agents/comedian.py
 
-from agents.base_agent import BaseAgent
+from agents.base_agent import BaseAgent, ACPMessage
+from multi_inference import multi_query
 
 class ComedianAgent(BaseAgent):
     def __init__(self):
-        super().__init__(name="🎭 Comedian", role="Funny and Sarcastic Responder")
+        super().__init__(name="🎭 Comedian", role="Responds with humor and wit.")
 
-    def run(self, prompt, query_func):
+    def generate(self, messages):
+        prompt = f"{self.role}:
+" + "\n".join([f"{m.role}: {m.content}" for m in messages])
         try:
-            query = f"Respond humorously or sarcastically to: {prompt}"
-            return query_func(query)
+            return multi_query(prompt)
         except Exception as e:
-            return f"[Comedian ERROR] {str(e)}"
+            return f"[ERROR] {e}"