Update agent.py

agent.py

@@ -1,3 +1,5 @@
+import os
+
 from smolagents import (
     CodeAgent, 
     OpenAIServerModel, 
@@ -25,7 +27,7 @@ def run_agent(level, question, file_name, ground_truth):
             verbosity_level = 2
         )
 
-    agent.prompt_templates["system_prompt"] = agent.prompt_templates["system_prompt"] + '\nWhen answering, provide ONLY the precise answer requested. Do not include explanations, steps, reasoning, or additional text. Be direct and specific. GAIA benchmark requires exact matching answers. For example, if asked "What is the capital of France?", respond simply with "Paris".'
+    agent.prompt_templates["system_prompt"] = os.environ["SYSTEM_PROMPT"]
 
     print(agent.prompt_templates["system_prompt"])