agents/base_agent.py

# agents/base_agent.py

class BaseAgent:
    def __init__(self, name="Agent", role="Generic Assistant"):
        self.name = name
        self.role = role

    def run(self, prompt, query_func):
        """
        Should be overridden by child classes.
        """
        raise NotImplementedError("Each agent must implement its own `run` method.")