fix bug

app.py

@@ -207,14 +207,22 @@ class GAIAAgent:
         self.setup_model(api_key)
         self.setup_tools()
         
+        # Create custom prompt template based on our system prompt
+        self.custom_prompt = self.create_system_prompt()
+        
         # Create code execution agent (based on smolagents)
         self.code_agent = CodeAgent(
             model=self.model,
             tools=self.tools,
-            system_prompt=self.create_system_prompt(),
             verbosity_level=1  # 0=quiet, 1=normal, 2=verbose
         )
         
+        # Modify the agent's prompt templates to include our custom prompt
+        # This is how smolagents handles custom system prompts
+        if hasattr(self.code_agent, 'prompt_templates') and 'system_prompt' in self.code_agent.prompt_templates:
+            original_prompt = self.code_agent.prompt_templates['system_prompt']
+            self.code_agent.prompt_templates['system_prompt'] = original_prompt + "\n\n" + self.custom_prompt
+        
         # Set up state machine workflow (inspired by langgraph)
         self.setup_workflow()
         
@@ -289,7 +297,7 @@ class GAIAAgent:
                     5. For numerical answers, return only the number (no units or explanation).
                     6. For text answers, ensure exact matching of the correct words.
                     IMPORTANT: The final answer must be simple and direct, without extra explanation. For example, if the question is "What is 2+2?", the answer should simply be "4", not "2+2 equals 4".
-                """
+                    """
     
     def setup_workflow(self):
         """Set up the agent's state workflow (inspired by langgraph)"""