Update agent.py

agent.py

@@ -77,7 +77,7 @@ 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."),
+    FunctionTool.from_defaults(search_wikipedia, return_direct=True, 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."),
@@ -88,6 +88,7 @@ TOOLS = [
 llm = OpenAI(model="gpt-4")
 
 agent = FunctionAgent(
+    max_iterations=5,
     tools=TOOLS,
     llm=llm,
     verbose=True,
@@ -101,6 +102,24 @@ You have access to the following tools:
 - Audio transcriber (Whisper, MP3)
 - Excel food sales analyzer
 
+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\".
+4. Follow formats strictly: comma-separated lists, numeric values, chess notation, names only, etc.
+5. Avoid all explanation unless requested.
+"""
+
+    system_prompt="""
+You are a highly capable AI agent taking the GAIA benchmark test.
+
+You have access to the following tools:
+- Wikipedia search for factual lookups
+- Python runner for math, logic, or text analysis
+- YouTube transcript fetcher (via URL)
+- Audio transcriber (Whisper, MP3)
+- Excel food sales analyzer
+
 Rules:
 1. Always try to use a tool if relevant.
 2. Return ONLY the final answer in the requested format.
@@ -112,4 +131,4 @@ Rules:
     tools=TOOLS,
     llm=llm,
     verbose=True,
-    
+