From 945c24ae157cd116516cd267af3c1541dde03ee9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=8E=98=E6=9D=83=20=E9=A9=AC?= Date: Wed, 7 Aug 2024 14:53:37 +0800 Subject: [PATCH] fixbug: unit test --- metagpt/actions/design_api_an.py | 4 ++-- metagpt/utils/make_sk_kernel.py | 4 ++-- tests/metagpt/serialize_deserialize/test_role.py | 4 ++-- tests/metagpt/serialize_deserialize/test_serdeser_base.py | 6 +++--- tests/metagpt/serialize_deserialize/test_sk_agent.py | 4 ++++ .../metagpt/serialize_deserialize/test_write_code_review.py | 4 ++++ 6 files changed, 17 insertions(+), 9 deletions(-) diff --git a/metagpt/actions/design_api_an.py b/metagpt/actions/design_api_an.py index cffb09c764..ca7aea95a9 100644 --- a/metagpt/actions/design_api_an.py +++ b/metagpt/actions/design_api_an.py @@ -5,7 +5,7 @@ @Author : alexanderwu @File : design_api_an.py """ -from typing import List,Optional +from typing import List, Optional from metagpt.actions.action_node import ActionNode from metagpt.utils.mermaid import MMC1, MMC2 @@ -45,7 +45,7 @@ example=["main.py", "game.py", "new_feature.py"], ) -#optional,because low success reproduction of class diagram in non py project. +# optional,because low success reproduction of class diagram in non py project. DATA_STRUCTURES_AND_INTERFACES = ActionNode( key="Data structures and interfaces", expected_type=Optional[str], diff --git a/metagpt/utils/make_sk_kernel.py b/metagpt/utils/make_sk_kernel.py index 283a682d64..67d865b374 100644 --- a/metagpt/utils/make_sk_kernel.py +++ b/metagpt/utils/make_sk_kernel.py @@ -21,12 +21,12 @@ def make_sk_kernel(): if llm := config.get_azure_llm(): kernel.add_chat_service( "chat_completion", - AzureChatCompletion(llm.model, llm.base_url, llm.api_key), + AzureChatCompletion(deployment_name=llm.model, base_url=llm.base_url, api_key=llm.api_key), ) elif llm := config.get_openai_llm(): kernel.add_chat_service( "chat_completion", - OpenAIChatCompletion(llm.model, llm.api_key), + OpenAIChatCompletion(ai_model_id=llm.model, api_key=llm.api_key), ) return kernel diff --git a/tests/metagpt/serialize_deserialize/test_role.py b/tests/metagpt/serialize_deserialize/test_role.py index aaf7c19357..807849751a 100644 --- a/tests/metagpt/serialize_deserialize/test_role.py +++ b/tests/metagpt/serialize_deserialize/test_role.py @@ -28,9 +28,9 @@ def test_roles(context): role_a = RoleA() - assert len(role_a.rc.watch) == 1 + assert len(role_a.rc.watch) == 2 role_b = RoleB() - assert len(role_a.rc.watch) == 1 + assert len(role_a.rc.watch) == 2 assert len(role_b.rc.watch) == 1 role_d = RoleD(actions=[ActionOK()]) diff --git a/tests/metagpt/serialize_deserialize/test_serdeser_base.py b/tests/metagpt/serialize_deserialize/test_serdeser_base.py index ccc582f652..84058925eb 100644 --- a/tests/metagpt/serialize_deserialize/test_serdeser_base.py +++ b/tests/metagpt/serialize_deserialize/test_serdeser_base.py @@ -8,7 +8,7 @@ from pydantic import BaseModel, Field -from metagpt.actions import Action, ActionOutput +from metagpt.actions import Action, ActionOutput, UserRequirement from metagpt.actions.action_node import ActionNode from metagpt.actions.fix_bug import FixBug from metagpt.roles.role import Role, RoleReactMode @@ -68,7 +68,7 @@ class RoleA(Role): def __init__(self, **kwargs): super(RoleA, self).__init__(**kwargs) self.set_actions([ActionPass]) - self._watch([FixBug]) + self._watch([FixBug, UserRequirement]) class RoleB(Role): @@ -93,7 +93,7 @@ class RoleC(Role): def __init__(self, **kwargs): super(RoleC, self).__init__(**kwargs) self.set_actions([ActionOK, ActionRaise]) - self._watch([FixBug]) + self._watch([FixBug, UserRequirement]) self.rc.react_mode = RoleReactMode.BY_ORDER self.rc.memory.ignore_id = True diff --git a/tests/metagpt/serialize_deserialize/test_sk_agent.py b/tests/metagpt/serialize_deserialize/test_sk_agent.py index 97c0ade991..a4bf4ec6d6 100644 --- a/tests/metagpt/serialize_deserialize/test_sk_agent.py +++ b/tests/metagpt/serialize_deserialize/test_sk_agent.py @@ -15,3 +15,7 @@ async def test_sk_agent_serdeser(): new_role = SkAgent(**ser_role_dict) assert new_role.name == "Sunshine" assert len(new_role.actions) == 1 + + +if __name__ == "__main__": + pytest.main([__file__, "-s"]) diff --git a/tests/metagpt/serialize_deserialize/test_write_code_review.py b/tests/metagpt/serialize_deserialize/test_write_code_review.py index 32a017a974..4ced53ce84 100644 --- a/tests/metagpt/serialize_deserialize/test_write_code_review.py +++ b/tests/metagpt/serialize_deserialize/test_write_code_review.py @@ -29,3 +29,7 @@ def div(a: int, b: int = 0): assert new_action.name == "WriteCodeReview" await new_action.run() + + +if __name__ == "__main__": + pytest.main([__file__, "-s"])