Update agent.py

agent.py

@@ -1,4 +1,4 @@
-# agent.py — alternatywa z FunctionAgent (stabilna dla GAIA)
+# agent.py — final version using FunctionCallingAgent
 
 import os
 import asyncio
@@ -75,16 +75,16 @@ def extract_excel_total_food_sales(file_path: str) -> str:
     except Exception as e:
         return f"[EXCEL ERROR] {e}"
 
-# Assemble tools with proper descriptions
+# Assemble tools with return_direct
 TOOLS = [
-    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."),
-    FunctionTool.from_defaults(extract_excel_total_food_sales, name="extract_excel_total_food_sales", description="Sum total sales from Excel where category is 'food'.")
+    FunctionTool.from_defaults(search_wikipedia, return_direct=True, name="search_wikipedia"),
+    FunctionTool.from_defaults(run_python_code, return_direct=True, name="run_python"),
+    FunctionTool.from_defaults(get_youtube_transcript, return_direct=True, name="get_youtube_transcript"),
+    FunctionTool.from_defaults(transcribe_audio, return_direct=True, name="transcribe_audio"),
+    FunctionTool.from_defaults(extract_excel_total_food_sales, return_direct=True, name="extract_excel_total_food_sales")
 ]
 
-# Create agent with FunctionAgent (compatible with run)
+# Create agent with FunctionCallingAgent
 llm = OpenAI(model="gpt-4")
 
 agent = FunctionCallingAgent.from_tools(
@@ -108,4 +108,12 @@ Rules:
 5. Avoid all explanation unless requested.
 """
 )
-    
+
+# Agent answer wrapper
+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)