apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-04-27 01:36:29 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge branch 'code_intepreter' of https://gitlab.deepwisdomai.com/agents/data_agents_opt into code_intepreter

Browse source
This commit is contained in:
yzlin 2024-02-02 18:52:45 +08:00
commit 3894334b52
4 changed files with 38 additions and 83 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tests/metagpt/actions/test_execute_nb_code.py
View file

@ -8,8 +8,6 @@ async def test_code_running():
executor = ExecuteNbCode()
output, is_success = await executor.run("print('hello world!')")
assert is_success
output, is_success = await executor.run({"code": "print('hello world!')", "language": "python"})
assert is_success
@pytest.mark.asyncio

10
tests/metagpt/actions/test_write_analysis_code.py
View file

@ -15,16 +15,20 @@ async def test_write_code_by_list_plan():
write_code = WriteCodeByGenerate()
execute_code = ExecuteNbCode()
messages = []
plan = ["随机生成一个pandas DataFrame时间序列", "绘制这个时间序列的直方图", "求均值"]
plan = ["随机生成一个pandas DataFrame时间序列", "绘制这个时间序列的直方图", "回顾已完成的任务", "求均值", "总结"]
for task in plan:
print(f"\n任务: {task}\n\n")
messages.append(Message(task, role="assistant"))
code = await write_code.run(messages)
if task.startswith(("回顾", "总结")):
assert code["language"] == "markdown"
else:
assert code["language"] == "python"
messages.append(Message(code["code"], role="assistant"))
assert len(code) > 0
output = await execute_code.run(code["code"])
output, _ = await execute_code.run(**code)
print(f"\n[Output]: 任务{task}的执行结果是: \n{output}\n")
messages.append(output[0])
messages.append(output)
@pytest.mark.asyncio