Update app.py

app.py

@@ -27,6 +27,22 @@ SYSTEM_HINT = (
 )
 
 
+import os
+from smolagents import (
+    CodeAgent,
+    DuckDuckGoSearchTool,
+    InferenceClientModel,
+    WikipediaSearchTool,
+)
+
+GAIA_SYSTEM_PROMPT = """You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template:
+FINAL ANSWER: [YOUR FINAL ANSWER].
+YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings.
+If you are asked for a number, don't use commas to write your number, and don't use units such as $ or percent sign unless specified otherwise.
+If you are asked for a string, don't use articles, don't use abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise.
+If you are asked for a comma separated list, apply the above rules depending on whether the element to be put in the list is a number or a string."""
+
+
 class BasicAgent:
     def __init__(self):
         self.agent = CodeAgent(
@@ -36,30 +52,20 @@ class BasicAgent:
                 token=os.getenv("HF_TOKEN"),
             ),
             max_steps=12,
-            additional_authorized_imports=["pandas", "numpy"],
+            additional_authorized_imports=["pandas", "numpy", "datetime", "re", "json", "math"],
         )
-GAIA_SYSTEM_PROMPT = """You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template:
-FINAL ANSWER: [YOUR FINAL ANSWER].
-YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings.
-If you are asked for a number, don't use commas to write your number, and don't use units such as $ or percent sign unless specified otherwise.
-If you are asked for a string, don't use articles, don't use abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise.
-If you are asked for a comma separated list, apply the above rules depending on whether the element to be put in the list is a number or a string."""
-
-
-def __call__(self, question: str) -> str:
-    try:
-        raw = str(self.agent.run(f"{GAIA_SYSTEM_PROMPT}\n\nQuestion: {question}"))
-        # Извлекаем только то, что после "FINAL ANSWER:"
-        if "FINAL ANSWER:" in raw:
-            answer = raw.split("FINAL ANSWER:")[-1].strip()
-            # Убираем точку в конце, если есть
-            answer = answer.rstrip(".").strip()
-            return answer
-        return raw.strip()
-    except Exception as e:
-        print(f"Agent failed: {e}")
-        return "unable to answer"
 
+    def __call__(self, question: str) -> str:
+        try:
+            raw = str(self.agent.run(f"{GAIA_SYSTEM_PROMPT}\n\nQuestion: {question}"))
+            if "FINAL ANSWER:" in raw:
+                answer = raw.split("FINAL ANSWER:")[-1].strip()
+                answer = answer.rstrip(".").strip()
+                return answer
+            return raw.strip()
+        except Exception as e:
+            print(f"Agent failed: {e}")
+            return "unable to answer"