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

chapter6/agent-with-event-trigger/tests/test_code_interpreter.py

@@ -0,0 +1,73 @@
+"""Regression tests for the code interpreter's execution namespace."""
+
+import importlib.util
+import sys
+import types
+import unittest
+from pathlib import Path
+from unittest.mock import patch
+
+
+def _load_agent_module():
+    openai = types.ModuleType("openai")
+    openai.OpenAI = type("OpenAI", (), {})
+
+    mcp = types.ModuleType("mcp")
+    mcp.__path__ = []
+    mcp.ClientSession = type("ClientSession", (), {})
+    mcp.StdioServerParameters = type("StdioServerParameters", (), {})
+
+    mcp_client = types.ModuleType("mcp.client")
+    mcp_client.__path__ = []
+    mcp_stdio = types.ModuleType("mcp.client.stdio")
+    mcp_stdio.stdio_client = lambda *args, **kwargs: None
+    mcp_types = types.ModuleType("mcp.types")
+    mcp_types.TextContent = type("TextContent", (), {})
+
+    stubs = {
+        "openai": openai,
+        "mcp": mcp,
+        "mcp.client": mcp_client,
+        "mcp.client.stdio": mcp_stdio,
+        "mcp.types": mcp_types,
+    }
+    module_path = Path(__file__).resolve().parents[1] / "agent.py"
+    module_name = "_event_trigger_agent_under_test"
+    spec = importlib.util.spec_from_file_location(module_name, module_path)
+    module = importlib.util.module_from_spec(spec)
+
+    with patch.dict(sys.modules, stubs):
+        sys.modules[module_name] = module
+        sys.path.insert(0, str(module_path.parent))
+        try:
+            spec.loader.exec_module(module)
+        finally:
+            sys.path.pop(0)
+
+    return module
+
+
+class CodeInterpreterNamespaceTests(unittest.TestCase):
+    def test_code_interpreter_shares_names_with_defined_functions(self):
+        agent_module = _load_agent_module()
+        agent = agent_module.EventTriggeredAgent.__new__(agent_module.EventTriggeredAgent)
+
+        result = agent._tool_code_interpreter(
+            "value = 5\n"
+            "def double():\n"
+            "    return value * 2\n"
+            "print(double())"
+        )
+
+        self.assertEqual(
+            result,
+            {
+                "success": True,
+                "stdout": "10\n",
+                "stderr": "",
+            },
+        )
+
+
+if __name__ == "__main__":
+    unittest.main()