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."),
-    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(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(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,7 +90,9 @@ llm = OpenAI(model="gpt-4")
 agent = FunctionAgent(
     tools=TOOLS,
     llm=llm,
-    verbose=True,
+    system_prompt="...",
+    max_iterations=5,
+    verbose=True
     system_prompt="""
 You are a highly capable AI agent taking the GAIA benchmark test.