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

feat: 修改output_path

Browse source
This commit is contained in:
莘权 马 2024-06-05 12:11:45 +08:00
parent 04e72c0597
commit 4359abd84a
8 changed files with 83 additions and 73 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -6,12 +6,12 @@
@File : test_design_api.py
@Modifiled By: mashenquan, 2023-12-6. According to RFC 135
"""
import json
from pathlib import Path
import pytest
from metagpt.actions.design_api import WriteDesign
from metagpt.const import METAGPT_ROOT
from metagpt.const import DEFAULT_WORKSPACE_ROOT, METAGPT_ROOT
from metagpt.logs import logger
from metagpt.schema import AIMessage, Message
from metagpt.utils.project_repo import ProjectRepo
@ -74,8 +74,7 @@ async def test_design_api(context, user_requirement, prd_filename, legacy_design
)
assert isinstance(result, AIMessage)
assert result.content
m = json.loads(result.content)
assert m
assert str(DEFAULT_WORKSPACE_ROOT) in result.content
@pytest.mark.parametrize(
@ -97,7 +96,7 @@ async def test_design_api_dir(context, user_requirement, prd_filename, legacy_de
user_requirement=user_requirement,
prd_filename=prd_filename,
legacy_design_filename=legacy_design_filename,
output_path=context.config.project_path,
output_pathname=str(Path(context.config.project_path) / "1.txt"),
)
assert isinstance(result, AIMessage)
assert result.content

18
tests/metagpt/actions/test_write_prd.py
View file

@ -6,12 +6,13 @@
@File : test_write_prd.py
@Modified By: mashenquan, 2023-11-1. According to Chapter 2.2.1 and 2.2.2 of RFC 116, replace `handle` with `run`.
"""
import json
import uuid
from pathlib import Path
import pytest
from metagpt.actions import UserRequirement, WritePRD
from metagpt.const import REQUIREMENT_FILENAME
from metagpt.const import DEFAULT_WORKSPACE_ROOT, REQUIREMENT_FILENAME
from metagpt.logs import logger
from metagpt.roles.product_manager import ProductManager
from metagpt.roles.role import RoleReactMode
@ -80,10 +81,12 @@ async def test_write_prd_api(context):
result = await action.run(user_requirement="write a snake game.")
assert isinstance(result, AIMessage)
assert result.content
m = json.loads(result.content)
assert m
assert str(DEFAULT_WORKSPACE_ROOT) in result.content
result = await action.run(user_requirement="write a snake game.", output_path=str(context.config.project_path))
result = await action.run(
user_requirement="write a snake game.",
output_pathname=str(Path(context.config.project_path) / f"{uuid.uuid4().hex}.json"),
)
assert isinstance(result, AIMessage)
assert result.content
assert result.instruct_content
@ -94,12 +97,11 @@ async def test_write_prd_api(context):
result = await action.run(user_requirement="Add moving enemy.", legacy_prd_filename=legacy_prd_filename)
assert isinstance(result, AIMessage)
assert result.content
m = json.loads(result.content)
assert m
assert str(DEFAULT_WORKSPACE_ROOT) in result.content
result = await action.run(
user_requirement="Add moving enemy.",
output_path=str(context.config.project_path),
output_pathname=str(Path(context.config.project_path) / f"{uuid.uuid4().hex}.json"),
legacy_prd_filename=legacy_prd_filename,
)
assert isinstance(result, AIMessage)