Update app.py

app.py

@@ -53,25 +53,28 @@ def calculate(expression: str) -> str:
 # --- Agent ---
 class GAIAAgent:
     def __init__(self):
-        model = OpenAIServerModel(model_id="gpt-3.5-turbo", api_key=os.getenv("OPENAI_API_KEY"))
-        self.agent = ToolCallingAgent(
-            name="GAIA_Agent",
-            description="AI assistant using web_search and calculate tools.",
-            tools=[web_search, calculate],
-            model=model,
-        )
+        self.system_prompt = """You are an AI assistant that answers questions using tools:
+- Use web_search for factual queries
+- Use calculate for math problems
+- Be concise and accurate."""
 
-    def __call__(self, question: str) -> str:
         try:
-            return self.agent.run(
-                question,
-                system_prompt=(
-                    "You are an AI assistant. Use web_search for factual queries "
-                    "and calculate for math problems. Be concise and accurate."
-                )
+            self.agent = ToolCallingAgent(
+                name="GAIA_Agent",
+                description=self.system_prompt,  # this sets the prompt
+                tools=[web_search, calculate],
+                model=client.chat.completions  # callable, no kwargs
             )
+            print("✅ Agent initialized successfully")
+        except Exception as e:
+            raise RuntimeError(f"Agent init failed: {str(e)}")
+
+    def __call__(self, question: str) -> str:
+        try:
+            response = self.agent.run(question)  # <-- no system_prompt here
+            return str(response) if response else "No answer generated"
         except Exception as e:
-            return f"Error: {e}"
+            return f"Error: {str(e)}"
 
 # --- Gradio UI + Submission ---
 def submit_answers(profile: gr.OAuthProfile | None):