ai-agent-book 精选快照（<2MB 代码与文档，来自 github.com/bojieli/ai-agent-book）

chapter5/code-for-logic/sandbox.py

@@ -0,0 +1,52 @@
+"""
+极简 Code Interpreter 沙箱：在子进程中执行模型生成的 Python 代码。
+
+- 用独立子进程运行，避免污染主进程、并可强制超时。
+- 子进程使用与主程序相同的解释器(sys.executable)，因此已预装 python-constraint。
+- 捕获 stdout / stderr 一并返回给模型，让它能看到求解结果或报错信息。
+"""
+import subprocess
+import sys
+import tempfile
+import os
+
+
+def run_python(code: str, timeout: int = 20) -> str:
+    """在子进程沙箱中执行 code，返回合并后的 stdout/stderr 文本。"""
+    with tempfile.NamedTemporaryFile("w", suffix=".py", delete=False,
+                                     encoding="utf-8") as f:
+        f.write(code)
+        path = f.name
+    try:
+        proc = subprocess.run(
+            [sys.executable, path],
+            capture_output=True, text=True, timeout=timeout,
+        )
+        out = proc.stdout
+        if proc.stderr.strip():
+            out += "\n[stderr]\n" + proc.stderr
+        if not out.strip():
+            out = "(代码已执行，但没有任何输出。记得用 print() 打印结果。)"
+        return out.strip()
+    except subprocess.TimeoutExpired:
+        return f"[错误] 代码执行超时（超过 {timeout} 秒）。"
+    finally:
+        os.unlink(path)
+
+
+if __name__ == "__main__":
+    # 自测：用 python-constraint 求解一个最简单的 K&K 谜题
+    demo = """
+from constraint import Problem
+p = Problem()
+# True=骑士(说真话), False=无赖(说假话)
+p.addVariable('A', [True, False])
+p.addVariable('B', [True, False])
+# A 说"B 是无赖"：A 的真值 == (B 是无赖) 即 A == (not B)
+p.addConstraint(lambda a, b: a == (not b), ['A', 'B'])
+# B 说"我们都不是骑士"：B == (not A and not B)
+p.addConstraint(lambda a, b: b == ((not a) and (not b)), ['A', 'B'])
+for s in p.getSolutions():
+    print({k: 'knight' if v else 'knave' for k, v in s.items()})
+"""
+    print(run_python(demo))