apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-05 14:55:18 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

add timeout and retry when code execution

Browse source
This commit is contained in:
lidanyang 2024-01-09 16:14:44 +08:00
parent fe6e46d478
commit fd9f2416ff
2 changed files with 54 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

9
tests/metagpt/actions/test_execute_code.py
View file

@ -88,3 +88,12 @@ def test_truncate():
assert truncate(output) == output
output = "hello world"
assert truncate(output, 5) == "Truncated to show only the last 5 characters\nworld"
@pytest.mark.asyncio
async def test_run_with_timeout():
pi = ExecutePyCode(timeout=1)
code = "import time; time.sleep(2)"
message, success = await pi.run(code)
assert not success
assert message == "TimeoutError"