diff --git a/docs/scripts/coverage.sh b/docs/scripts/coverage.sh index 648d9b412..a56571399 100755 --- a/docs/scripts/coverage.sh +++ b/docs/scripts/coverage.sh @@ -1 +1 @@ -coverage run --source ./metagpt -m pytest --durations=0 && coverage report -m && coverage html && open htmlcov/index.html +coverage run --source ./metagpt -m pytest --durations=0 --timeout=100 && coverage report -m && coverage html && open htmlcov/index.html diff --git a/examples/example.faiss b/examples/example.faiss index a5a539dc4..580946190 100644 Binary files a/examples/example.faiss and b/examples/example.faiss differ diff --git a/examples/example.pkl b/examples/example.pkl index a0e839763..f706fd803 100644 Binary files a/examples/example.pkl and b/examples/example.pkl differ diff --git a/setup.py b/setup.py index 29c44d3c1..a81be6115 100644 --- a/setup.py +++ b/setup.py @@ -39,6 +39,7 @@ extras_require["test"] = [ "pytest-mock", "pytest-html", "pytest-xdist", + "pytest-timeout", "connexion[uvicorn]~=3.0.5", "azure-cognitiveservices-speech~=1.31.0", "aioboto3~=11.3.0", diff --git a/tests/metagpt/serialize_deserialize/test_action.py b/tests/metagpt/serialize_deserialize/test_action.py index 677988e2f..81879e34e 100644 --- a/tests/metagpt/serialize_deserialize/test_action.py +++ b/tests/metagpt/serialize_deserialize/test_action.py @@ -27,6 +27,6 @@ async def test_action_deserialize(): new_action = Action(**serialized_data) - assert new_action.name == "" + assert new_action.name == "Action" assert isinstance(new_action.llm, type(LLM())) assert len(await new_action._aask("who are you")) > 0 diff --git a/tests/metagpt/serialize_deserialize/test_write_design.py b/tests/metagpt/serialize_deserialize/test_write_design.py index a2fce8047..7bcba3fc8 100644 --- a/tests/metagpt/serialize_deserialize/test_write_design.py +++ b/tests/metagpt/serialize_deserialize/test_write_design.py @@ -26,7 +26,7 @@ async def test_write_design_deserialize(): action = WriteDesign() serialized_data = action.model_dump() new_action = WriteDesign(**serialized_data) - assert new_action.name == "" + assert new_action.name == "WriteDesign" await new_action.run(with_messages="write a cli snake game") @@ -35,5 +35,5 @@ async def test_write_task_deserialize(): action = WriteTasks() serialized_data = action.model_dump() new_action = WriteTasks(**serialized_data) - assert new_action.name == "CreateTasks" + assert new_action.name == "WriteTasks" await new_action.run(with_messages="write a cli snake game") diff --git a/tests/metagpt/serialize_deserialize/test_write_docstring.py b/tests/metagpt/serialize_deserialize/test_write_docstring.py index 89ef6796b..e4116ab30 100644 --- a/tests/metagpt/serialize_deserialize/test_write_docstring.py +++ b/tests/metagpt/serialize_deserialize/test_write_docstring.py @@ -38,7 +38,7 @@ async def test_action_deserialize(style: str, part: str): new_action = WriteDocstring(**serialized_data) - assert not new_action.name + assert new_action.name == "WriteDocstring" assert new_action.desc == "Write docstring for code." ret = await new_action.run(code, style=style) assert part in ret