apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-20 15:38:09 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge branch 'fix_write_code_pipeline' into dev

Browse source
This commit is contained in:
yzlin 2023-11-24 14:23:12 +08:00
commit a9b46579b4
4 changed files with 676 additions and 45 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -2,25 +2,24 @@ import pytest
from metagpt.actions.write_analysis_code import WriteCodeByGenerate
from metagpt.actions.execute_code import ExecutePyCode
from metagpt.schema import Message
@pytest.mark.asyncio
async def test_write_code():
write_code = WriteCodeByGenerate()
code = await write_code.run("Write a hello world code.")
assert "language" in code.content
assert "code" in code.content
print(code)
# @pytest.mark.asyncio
# async def test_write_code():
# write_code = WriteCodeFunction()
# code = await write_code.run("Write a hello world code.")
# assert len(code) > 0
# print(code)
@pytest.mark.asyncio
async def test_write_code_by_list_prompt():
write_code = WriteCodeByGenerate()
msg = ["a=[1,2,5,10,-10]", "写出求a中最大值的代码python"]
code = await write_code.run(msg)
assert "language" in code.content
assert "code" in code.content
print(code)
# @pytest.mark.asyncio
# async def test_write_code_by_list_prompt():
# write_code = WriteCodeFunction()
# msg = ["a=[1,2,5,10,-10]", "写出求a中最大值的代码python"]
# code = await write_code.run(msg)
# assert len(code) > 0
# print(code)
@pytest.mark.asyncio
@ -31,11 +30,10 @@ async def test_write_code_by_list_plan():
plan = ["随机生成一个pandas DataFrame时间序列", "绘制这个时间序列的直方图", "求均值"]
for task in plan:
print(f"\n任务: {task}\n\n")
messages.append(task)
messages.append(Message(task, role='assistant'))
code = await write_code.run(messages)
messages.append(code)
assert "language" in code.content
assert "code" in code.content
messages.append(Message(code, role='assistant'))
assert len(code) > 0
output = await execute_code.run(code)
print(f"\n[Output]: 任务{task}的执行结果是: \n{output}\n")
messages.append(output)
messages.append(output[0])