agent.py

from agent_config import create_agent

class BasicAgent:
    def __init__(self):
        print("BasicAgent initialized.")
        self.agent = create_agent()

    def __call__(self, question: str) -> str:
        try:
            return self.agent.run(question)
        except Exception as e:
            return f"Error: {e}"