Update app.py

app.py

@@ -60,9 +60,8 @@ class GaiaSmolAgent:
             timeout=120.0, # Add a timeout to prevent hanging
         )
 
-        # A more sophisticated system prompt to guide the agent's reasoning.
-        # This improves its ability to handle complex GAIA questions.
-        system_prompt = """
+        # Store the sophisticated system prompt as an instance variable.
+        self.system_prompt = """
         You are an expert-level research assistant AI. Your sole purpose is to answer the user's question by breaking it down into logical steps and using the provided tools.
 
         **Available Tools:**
@@ -82,11 +81,10 @@ class GaiaSmolAgent:
         - Do not ask for clarification. Directly proceed to solve the problem.
         """
 
-        # Initialize a single, powerful agent instead of a planner/executor pair.
+        # Initialize the agent without the 'system_prompt' argument to prevent the TypeError.
         self.agent = CodeAgent(
             model=model,
             tools=[file_reader, DuckDuckGoSearchTool()],
-            system_prompt=system_prompt,
             add_base_tools=True,  # Provides the python interpreter and the final_answer function
         )
         print("Optimized GaiaSmolAgent initialized successfully.")
@@ -98,8 +96,10 @@ class GaiaSmolAgent:
         """
         print(f"Optimized Agent received question: {question[:100]}...")
         try:
+            # Combine the system prompt with the actual question to give the agent full context.
+            full_prompt = f"{self.system_prompt}\n\nUser Question: \"{question}\""
             # The agent now internally handles the reasoning, code generation, and execution in one step.
-            final_answer = self.agent.run(question)
+            final_answer = self.agent.run(full_prompt)
         except Exception as e:
             print(f"FATAL AGENT ERROR: An exception occurred during agent execution: {e}")
             print(traceback.format_exc()) # Print full traceback for easier debugging