liqiang
b119135836
Build latest book artifacts / build (push) Canceled after 0s
dependency resolution / resolve (3.11) (push) Canceled after 0s
dependency resolution / resolve (3.13) (push) Canceled after 0s
deploy-pages / build (push) Canceled after 0s
deploy-pages / deploy (push) Canceled after 0s
i18n consistency check / check (push) Canceled after 0s
provider adoption tests / test (chapter2/context-compression) (push) Canceled after 0s
provider adoption tests / test (chapter2/prompt-injection) (push) Canceled after 0s
provider adoption tests / test (chapter2/system-hint) (push) Canceled after 0s
provider adoption tests / test (chapter3/log-sanitization) (push) Canceled after 0s
web-search-agent tests / test (push) Canceled after 0s
web-search-agent tests / agentbook (push) Canceled after 0s
74 lines
2.1 KiB
Python
74 lines
2.1 KiB
Python
"""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()
|