Update agent.py

agent.py

@@ -77,9 +77,9 @@ def extract_excel_total_food_sales(file_path: str) -> str:
 
 # Assemble tools with proper descriptions
 TOOLS = [
-    FunctionTool.from_defaults(search_wikipedia, name="search_wikipedia", description="Search Wikipedia for facts and lists.", return_direct=True),
-    FunctionTool.from_defaults(run_python_code, name="run_python", description="Run Python code for logic, math, or set processing.", return_direct=True),
-    FunctionTool.from_defaults(get_youtube_transcript, name="get_youtube_transcript", description="Fetch transcript from YouTube video by URL.", return_direct=True),
+    FunctionTool.from_defaults(search_wikipedia, name="search_wikipedia", description="Search Wikipedia for facts and lists."),
+    FunctionTool.from_defaults(run_python_code, name="run_python", description="Run Python code for logic, math, or set processing."),
+    FunctionTool.from_defaults(get_youtube_transcript, name="get_youtube_transcript", description="Fetch transcript from YouTube video by URL."),
     FunctionTool.from_defaults(transcribe_audio, name="transcribe_audio", description="Transcribe MP3 audio file using Whisper."),
     FunctionTool.from_defaults(extract_excel_total_food_sales, name="extract_excel_total_food_sales", description="Sum total sales from Excel where category is 'food'.")
 ]
@@ -90,8 +90,6 @@ llm = OpenAI(model="gpt-4")
 agent = FunctionAgent(
     tools=TOOLS,
     llm=llm,
-    system_prompt="...",
-    max_iterations=5,
     verbose=True,
     system_prompt="""
 You are a highly capable AI agent taking the GAIA benchmark test.
@@ -106,17 +104,12 @@ You have access to the following tools:
 Rules:
 1. Always try to use a tool if relevant.
 2. Return ONLY the final answer in the requested format.
-3. Do not guess. If a tool fails, say "Tool not available".
+3. Do not guess. If a tool fails, say \"Tool not available\".
 4. Follow formats strictly: comma-separated lists, numeric values, chess notation, names only, etc.
 5. Avoid all explanation unless requested.
-"""
-)
+""",
 
-# Stable async answer handler
-async def answer_question(question: str) -> str:
-    try:
-        result = await agent.run(question)
-        return str(result).strip()
-    except Exception as e:
-        print("❌ Agent error:", e)
-        return "[ERROR] " + str(e)
+    tools=TOOLS,
+    llm=llm,
+    verbose=True,
+