agents/base_agent.py

# base_agent.py

from abc import ABC, abstractmethod
from multi_inference import multi_query

class ACPMessage:
    def __init__(self, role, content):
        self.role = role
        self.content = content

class BaseAgent(ABC):
    def __init__(self, name, role):
        self.name = name
        self.role = role
        self.memory = []

    @abstractmethod
    def generate(self, messages):
        pass