baseline/custom_agent.py

from smolagents import CodeAgent

class CustomCodeAgent(CodeAgent):

    def initialize_system_prompt(self):
        return self.system_prompt_template