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

fix bug

Browse source
This commit is contained in:
geekan 2024-01-10 18:32:03 +08:00 • committed by 莘权 马
parent 0d742654d4
commit ae0a91c025
4 changed files with 17 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/metagpt/actions/test_write_code.py
View file

@ -19,8 +19,8 @@ from metagpt.const import (
TEST_OUTPUTS_FILE_REPO,
)
from metagpt.context import CONTEXT
from metagpt.llm import LLM
from metagpt.logs import logger
from metagpt.provider.openai_api import OpenAILLM as LLM
from metagpt.schema import CodingContext, Document
from metagpt.utils.common import aread
from tests.metagpt.actions.mock_markdown import TASKS_2, WRITE_CODE_PROMPT_SAMPLE
@ -32,7 +32,7 @@ async def test_write_code():
filename="task_filename.py", design_doc=Document(content="设计一个名为'add'的函数，该函数接受两个整数作为输入，并返回它们的和。")
)
doc = Document(content=ccontext.model_dump_json())
write_code = WriteCode(context=doc)
write_code = WriteCode(i_context=doc)
code = await write_code.run()
logger.info(code.model_dump_json())
@ -86,7 +86,7 @@ async def test_write_code_deps():
)
coding_doc = Document(root_path="snake1", filename="game.py", content=ccontext.json())
action = WriteCode(context=coding_doc)
action = WriteCode(i_context=coding_doc)
rsp = await action.run()
assert rsp
assert rsp.code_doc.content