Update agent.py

agent.py

@@ -10,7 +10,7 @@ from llama_index.llms.openai import OpenAI
 from llama_index.core.agent import FunctionCallingAgent
 from llama_index.core.tools import FunctionTool
 
-# --- TOOL FUNCTIONS --- #
+# === TOOL FUNCTIONS ===
 
 def reverse_sentence(sentence: str) -> str:
     """Reverse a sentence character by character."""
@@ -33,7 +33,11 @@ def convert_table_if_detected(question: str, file_context: str) -> str:
     """If question contains a table about * on set S, try parsing non-commutative set."""
     if "* on the set" in question and file_context:
         try:
-            table_lines = [line.strip() for line in file_context.splitlines() if '|' in line and '*' not in line[:2]]
+            table_lines = [
+                line.strip()
+                for line in file_context.splitlines()
+                if '|' in line and '*' not in line[:2]
+            ]
             headers = re.split(r'\|+', table_lines[0])[1:-1]
             data_rows = [re.split(r'\|+', row)[1:-1] for row in table_lines[1:]]
             index = [row[0] for row in data_rows]
@@ -68,10 +72,10 @@ def extract_excel_total_food_sales(file_path: str) -> str:
             total += float(amount)
     return f"${total:.2f}"
 
-# --- LLM SETUP --- #
+# === LLM SETUP ===
 llm = OpenAI(model="gpt-4o")
 
-# --- TOOL WRAPPING --- #
+# === TOOLS ===
 tools = [
     FunctionTool.from_defaults(fn=reverse_sentence),
     FunctionTool.from_defaults(fn=extract_vegetables_from_list),
@@ -82,16 +86,15 @@ agent = FunctionCallingAgent.from_tools(
     tools=tools,
     llm=llm,
     system_prompt=(
-        "You are an expert assistant solving GAIA benchmark tasks. "
-        "You are expected to respond with precise, short answers that match the expected format. "
-        "Use tools when available to analyze tables, audio, videos, and reasoning-based logic. "
-        "Never guess — always base your answers on facts from content or reliable lookup. "
-        "Respond with the answer only, without explanation."
+        "You are a strict and factual research agent solving GAIA benchmark questions. "
+        "You must answer precisely, based only on available information. "
+        "Never hallucinate, and always return concise, well-formatted answers. "
+        "Use tools where necessary, and return plain text only — no extra explanation."
     ),
-    verbose=True,
+    verbose=True
 )
 
-# --- RUNNER FUNCTION --- #
+# === MAIN AGENT CALL ===
 def answer_question(question: str, task_id: str = None, file_content: str = "") -> str:
     file_context = file_content or ""
     file_context = convert_table_if_detected(question, file_context)