fix: run_python_file

agent_tools.py

@@ -301,38 +301,55 @@ Execute Python code from a file identified by task_id and file_name.
 Returns the numeric result if defined, otherwise stdout.
 """)
 def run_python_file(task_id: str, file_name: str) -> str:
+    file_path = file_name
+    buffer = io.StringIO()
+    old_stdout = sys.stdout
+    ns = {"__builtins__": __builtins__, "__name__": "__main__"}
     try:
-        # Завантажуємо файл
         file_url = f"{DEFAULT_API_URL}/files/{task_id}"
         r = requests.get(file_url, timeout=15, allow_redirects=True)
         if r.status_code != 200:
             return f"❌ Failed to download file: {r.status_code}"
 
-        # Зберігаємо файл з наданим file_name
-        with open(file_name, "wb") as f:
+        with open(file_path, "wb") as f:
             f.write(r.content)
 
-        # Читаємо код
-        with open(file_name, "r") as f:
+        with open(file_path, "r", encoding="utf-8", errors="replace") as f:
             code = f.read()
 
-        # Виконуємо код і зберігаємо stdout
-        buffer = io.StringIO()
         sys.stdout = buffer
-        local_vars = {}
-        exec(code, {"__builtins__": {}}, local_vars)
-        sys.stdout = sys.__stdout__
-
-        # Повертаємо змінну 'result', якщо вона визначена, інакше stdout
-        if "result" in local_vars:
-            return str(local_vars["result"])
+        try:
+            compiled = compile(code, file_path, "exec")
+            exec(compiled, ns, ns)
+        finally:
+            sys.stdout = old_stdout
+
+        if "result" in ns:
+            return str(ns["result"])
         else:
             output = buffer.getvalue().strip()
             return output or "No output produced."
 
     except Exception as e:
-        sys.stdout = sys.__stdout__
+        # Prefer returning a computed result or any partial stdout if available
+        try:
+            sys.stdout = old_stdout
+        except Exception:
+            pass
+        if "result" in ns:
+            return str(ns["result"])
+        output = buffer.getvalue().strip()
+        if output:
+            return output
         return f"❌ Error executing Python file: {e}"
+    finally:
+        # Ensure the downloaded code file is removed after execution
+        try:
+            if os.path.exists(file_path):
+                os.remove(file_path)
+        except Exception:
+            pass
+
 
 
 TOOLS: List[Callable[..., Any]] = [

test.py

@@ -107,7 +107,7 @@ questions = [
 # Test
 if __name__ == "__main__":
     agent = CUSTOM_AGENT()
-    q = questions[18]
+    q = questions[11]
     answer = agent.run(q)
     print("Answer:", answer)
 # audio 9 13