Update codex/sandbox.py

codex/sandbox.py

@@ -1,9 +1,21 @@
 import ast
 
 SAFE_NODES = (
-    ast.Module, ast.Assign, ast.Expr, ast.Call, ast.Name, ast.Load,
-    ast.Constant, ast.BinOp, ast.Add, ast.Sub, ast.Mult, ast.Div,
-    ast.If, ast.Compare, ast.Return, ast.FunctionDef, ast.arguments
+    ast.Module,
+    ast.Assign,
+    ast.Expr,
+    ast.Call,
+    ast.Name,
+    ast.Load,
+    ast.Constant,
+    ast.BinOp,
+    ast.Add,
+    ast.Sub,
+    ast.Mult,
+    ast.Div,
+    ast.Return,
+    ast.FunctionDef,
+    ast.arguments
 )
 
 def safe_execute(code):
@@ -11,8 +23,9 @@ def safe_execute(code):
 
     for node in ast.walk(tree):
         if not isinstance(node, SAFE_NODES):
-            raise ValueError("Unsafe code detected")
+            raise ValueError(f"Unsafe code detected: {type(node).__name__}")
 
     local_env = {}
     exec(compile(tree, "<sandbox>", "exec"), {"__builtins__": {}}, local_env)
+
     return local_env