agents/trader.py

# agents/trader.py

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

class TraderAgent(BaseAgent):
    def __init__(self):
        super().__init__(name="💹 Trader", role="Analyzes markets and gives financial insights.")

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