agents/base_agent.py

# agents/base_agent.py

class BaseAgent:
    def __init__(self, name, role_description):
        self.name = name
        self.role = role_description

    def generate_response(self, prompt, inference_fn):
        system_prompt = f"You are {self.name}, a {self.role}. Respond uniquely and in your voice."
        full_prompt = f"{system_prompt}\nUser: {prompt}\n{self.name}:"
        return inference_fn(full_prompt)