Create codex/sandbox.py

codex/sandbox.py

@@ -0,0 +1,18 @@
+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
+)
+
+def safe_execute(code):
+    tree = ast.parse(code)
+
+    for node in ast.walk(tree):
+        if not isinstance(node, SAFE_NODES):
+            raise ValueError("Unsafe code detected")
+
+    local_env = {}
+    exec(compile(tree, "<sandbox>", "exec"), {"__builtins__": {}}, local_env)
+    return local_env