Update agent.py

agent.py

@@ -1,4 +1,4 @@
-# agent.py — final version using FunctionCallingAgent
+# agent.py — zaktualizowany z obsługą błędów, braków plików i bezpiecznym Pythonem
 
 import os
 import asyncio
@@ -24,16 +24,19 @@ wiki_api = WikipediaAPIWrapper(top_k_results=1, doc_content_chars_max=1000)
 
 def search_wikipedia(query: str) -> str:
     """Search Wikipedia for a given query and return relevant summary."""
-    return wiki_api.run(query)
+    try:
+        return wiki_api.run(query)
+    except Exception as e:
+        return f"Tool not available. ({e})"
 
-# Tool 2 — Python with output
+# Tool 2 — Python safe runner
 python_tool = PythonREPLTool()
 
 def run_python_code(code: str) -> str:
-    """Run Python code and return printed result."""
+    """Run Python code safely."""
     try:
-        if "print(" not in code:
-            code = f"print({code})"
+        if 'print' not in code:
+            code = f"print({repr(code)})"
         return python_tool.run(code)
     except Exception as e:
         return f"[PYTHON ERROR] {e}"
@@ -47,23 +50,25 @@ def get_youtube_transcript(url: str) -> str:
         docs = loader.load()
         return " ".join(d.page_content for d in docs)
     except Exception as e:
-        return f"[YOUTUBE ERROR] {e}"
+        return f"Tool not available. ({e})"
 
 # Tool 4 — Whisper transcription
 
 def transcribe_audio(file_path: str) -> str:
-    """Transcribe an MP3 file to text using Whisper."""
+    if not os.path.exists(file_path):
+        return "Tool not available. (Missing file)"
     try:
         model = whisper.load_model("base")
         res = model.transcribe(file_path)
         return res["text"]
     except Exception as e:
-        return f"[AUDIO ERROR] {e}"
+        return f"Tool not available. ({e})"
 
-# Tool 5 — Excel parser
+# Tool 5 — Excel food sum
 
 def extract_excel_total_food_sales(file_path: str) -> str:
-    """Sum sales from Excel where category is 'food'."""
+    if not os.path.exists(file_path):
+        return "Tool not available. (Missing file)"
     try:
         wb = openpyxl.load_workbook(file_path)
         sheet = wb.active
@@ -73,9 +78,9 @@ def extract_excel_total_food_sales(file_path: str) -> str:
                 total += float(amount or 0)
         return f"${total:.2f}"
     except Exception as e:
-        return f"[EXCEL ERROR] {e}"
+        return f"Tool not available. ({e})"
 
-# Assemble tools with return_direct
+# Assemble tools
 TOOLS = [
     FunctionTool.from_defaults(search_wikipedia, return_direct=True, name="search_wikipedia"),
     FunctionTool.from_defaults(run_python_code, return_direct=True, name="run_python"),
@@ -84,32 +89,25 @@ TOOLS = [
     FunctionTool.from_defaults(extract_excel_total_food_sales, return_direct=True, name="extract_excel_total_food_sales")
 ]
 
-# Create agent with FunctionCallingAgent
+# Build agent
 llm = OpenAI(model="gpt-4")
 
 agent = FunctionCallingAgent.from_tools(
     tools=TOOLS,
     llm=llm,
     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.
-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.
+You are a highly capable AI agent completing the GAIA benchmark.
+
+You MUST:
+- Always use tools when available.
+- If a tool returns 'None' or fails, try another or say 'Tool not available'.
+- Return only final answer in strict format (comma-separated, numbers, names, no explanations).
+- Do not guess. If unsure, state 'Tool not available'.
+- NEVER return raw tool errors, only result or fallback.
 """
 )
 
-# Agent answer wrapper
+# Agent call wrapper
 async def answer_question(question: str) -> str:
     try:
         response = await agent.achat(question)