run pre-commit to find potential issues and fix them

tests/metagpt/serialize_deserialize/test_environment.py

@@ -11,7 +11,11 @@ from metagpt.environment import Environment
 from metagpt.roles.project_manager import ProjectManager
 from metagpt.schema import Message
 from metagpt.utils.common import any_to_str
-from tests.metagpt.serialize_deserialize.test_serdeser_base import RoleC, ActionOK, serdeser_path
+from tests.metagpt.serialize_deserialize.test_serdeser_base import (
+    ActionOK,
+    RoleC,
+    serdeser_path,
+)
 
 
 def test_env_serialize():
@@ -35,10 +39,7 @@ def test_environment_serdeser():
     ic_obj = ActionNode.create_model_class("prd", out_mapping)
 
     message = Message(
-        content="prd",
-        instruct_content=ic_obj(**out_data),
-        role="product manager",
-        cause_by=any_to_str(UserRequirement)
+        content="prd", instruct_content=ic_obj(**out_data), role="product manager", cause_by=any_to_str(UserRequirement)
     )
 
     environment = Environment()

tests/metagpt/serialize_deserialize/test_memory.py

@@ -14,17 +14,14 @@ from tests.metagpt.serialize_deserialize.test_serdeser_base import serdeser_path
 
 
 def test_memory_serdeser():
-    msg1 = Message(role="Boss",
-                   content="write a snake game",
-                   cause_by=UserRequirement)
+    msg1 = Message(role="Boss", content="write a snake game", cause_by=UserRequirement)
 
     out_mapping = {"field2": (list[str], ...)}
     out_data = {"field2": ["field2 value1", "field2 value2"]}
     ic_obj = ActionNode.create_model_class("system_design", out_mapping)
-    msg2 = Message(role="Architect",
-                   instruct_content=ic_obj(**out_data),
-                   content="system design content",
-                   cause_by=WriteDesign)
+    msg2 = Message(
+        role="Architect", instruct_content=ic_obj(**out_data), content="system design content", cause_by=WriteDesign
+    )
 
     memory = Memory()
     memory.add_batch([msg1, msg2])
@@ -40,17 +37,14 @@ def test_memory_serdeser():
 
 
 def test_memory_serdeser_save():
-    msg1 = Message(role="User",
-                   content="write a 2048 game",
-                   cause_by=UserRequirement)
+    msg1 = Message(role="User", content="write a 2048 game", cause_by=UserRequirement)
 
     out_mapping = {"field1": (list[str], ...)}
     out_data = {"field1": ["field1 value1", "field1 value2"]}
     ic_obj = ActionNode.create_model_class("system_design", out_mapping)
-    msg2 = Message(role="Architect",
-                   instruct_content=ic_obj(**out_data),
-                   content="system design content",
-                   cause_by=WriteDesign)
+    msg2 = Message(
+        role="Architect", instruct_content=ic_obj(**out_data), content="system design content", cause_by=WriteDesign
+    )
 
     memory = Memory()
     memory.add_batch([msg1, msg2])

tests/metagpt/serialize_deserialize/test_role.py

@@ -16,7 +16,12 @@ from metagpt.roles.product_manager import ProductManager
 from metagpt.roles.role import Role
 from metagpt.schema import Message
 from metagpt.utils.common import format_trackback_info
-from tests.metagpt.serialize_deserialize.test_serdeser_base import RoleA, RoleB, RoleC, serdeser_path
+from tests.metagpt.serialize_deserialize.test_serdeser_base import (
+    RoleA,
+    RoleB,
+    RoleC,
+    serdeser_path,
+)
 
 
 def test_roles():
@@ -75,12 +80,10 @@ async def test_role_serdeser_interrupt():
     role_c = RoleC()
     shutil.rmtree(SERDESER_PATH.joinpath("team"), ignore_errors=True)
 
-    stg_path = SERDESER_PATH.joinpath(f"team", "environment", "roles", "{role_c.__class__.__name__}_{role_c.name}")
+    stg_path = SERDESER_PATH.joinpath("team", "environment", "roles", f"{role_c.__class__.__name__}_{role_c.name}")
     try:
-        await role_c.run(
-            with_message=Message(content="demo", cause_by=UserRequirement)
-        )
-    except Exception as exp:
+        await role_c.run(with_message=Message(content="demo", cause_by=UserRequirement))
+    except Exception:
         logger.error(f"Exception in `role_a.run`, detail: {format_trackback_info()}")
         role_c.serialize(stg_path)
 
@@ -90,6 +93,4 @@ async def test_role_serdeser_interrupt():
     assert new_role_a._rc.state == 1
 
     with pytest.raises(Exception):
-        await role_c.run(
-            with_message=Message(content="demo", cause_by=UserRequirement)
-        )
+        await role_c.run(with_message=Message(content="demo", cause_by=UserRequirement))

tests/metagpt/serialize_deserialize/test_schema.py

@@ -14,12 +14,7 @@ def test_message_serdeser():
     out_data = {"field3": "field3 value3", "field4": ["field4 value1", "field4 value2"]}
     ic_obj = ActionNode.create_model_class("code", out_mapping)
 
-    message = Message(
-        content="code",
-        instruct_content=ic_obj(**out_data),
-        role="engineer",
-        cause_by=WriteCode
-    )
+    message = Message(content="code", instruct_content=ic_obj(**out_data), role="engineer", cause_by=WriteCode)
     ser_data = message.dict()
     assert ser_data["cause_by"] == "metagpt.actions.write_code.WriteCode"
     assert ser_data["instruct_content"]["class"] == "code"
@@ -31,14 +26,11 @@ def test_message_serdeser():
 
 
 def test_message_without_postprocess():
-    """ to explain `instruct_content` should be postprocessed """
+    """to explain `instruct_content` should be postprocessed"""
     out_mapping = {"field1": (list[str], ...)}
     out_data = {"field1": ["field1 value1", "field1 value2"]}
     ic_obj = ActionNode.create_model_class("code", out_mapping)
-    message = MockMessage(
-        content="code",
-        instruct_content=ic_obj(**out_data)
-    )
+    message = MockMessage(content="code", instruct_content=ic_obj(**out_data))
     ser_data = message.dict()
     assert ser_data["instruct_content"] == {"field1": ["field1 value1", "field1 value2"]}
 

tests/metagpt/serialize_deserialize/test_serdeser_base.py

@@ -16,7 +16,8 @@ serdeser_path = Path(__file__).absolute().parent.joinpath("..", "..", "data", "s
 
 
 class MockMessage(BaseModel):
-    """ to test normal dict without postprocess """
+    """to test normal dict without postprocess"""
+
     content: str = ""
     instruct_content: BaseModel = Field(default=None)
 
@@ -26,9 +27,7 @@ class ActionPass(Action):
 
     async def run(self, messages: list["Message"]) -> ActionOutput:
         await asyncio.sleep(5)  # sleep to make other roles can watch the executed Message
-        output_mapping = {
-            "result": (str, ...)
-        }
+        output_mapping = {"result": (str, ...)}
         pass_class = ActionNode.create_model_class("pass", output_mapping)
         pass_output = ActionOutput("ActionPass run passed", pass_class(**{"result": "pass result"}))
 

tests/metagpt/serialize_deserialize/test_team.py

@@ -8,10 +8,16 @@ import shutil
 import pytest
 
 from metagpt.const import SERDESER_PATH
-from metagpt.roles import ProjectManager, ProductManager, Architect
-from metagpt.team import Team
 from metagpt.logs import logger
-from tests.metagpt.serialize_deserialize.test_serdeser_base import RoleA, RoleB, RoleC, serdeser_path, ActionOK
+from metagpt.roles import Architect, ProductManager, ProjectManager
+from metagpt.team import Team
+from tests.metagpt.serialize_deserialize.test_serdeser_base import (
+    ActionOK,
+    RoleA,
+    RoleB,
+    RoleC,
+    serdeser_path,
+)
 
 
 def test_team_deserialize():
@@ -110,10 +116,8 @@ async def test_team_recover_multi_roles_save():
     role_a = RoleA()
     role_b = RoleB()
 
-    assert role_a.subscription == {"tests.metagpt.serialize_deserialize.test_serdeser_base.RoleA",
-                                   "RoleA"}
-    assert role_b.subscription == {"tests.metagpt.serialize_deserialize.test_serdeser_base.RoleB",
-                                   "RoleB"}
+    assert role_a.subscription == {"tests.metagpt.serialize_deserialize.test_serdeser_base.RoleA", "RoleA"}
+    assert role_b.subscription == {"tests.metagpt.serialize_deserialize.test_serdeser_base.RoleB", "RoleB"}
     assert role_b._rc.watch == {"tests.metagpt.serialize_deserialize.test_serdeser_base.ActionPass"}
 
     company = Team()

tests/metagpt/serialize_deserialize/test_write_code.py

@@ -19,8 +19,9 @@ def test_write_design_serialize():
 
 @pytest.mark.asyncio
 async def test_write_code_deserialize():
-    context = CodingContext(filename="test_code.py",
-                            design_doc=Document(content="write add function to calculate two numbers"))
+    context = CodingContext(
+        filename="test_code.py", design_doc=Document(content="write add function to calculate two numbers")
+    )
     doc = Document(content=context.json())
     action = WriteCode(context=doc)
     serialized_data = action.dict()

tests/metagpt/serialize_deserialize/test_write_code_review.py

@@ -18,7 +18,7 @@ def div(a: int, b: int = 0):
     context = CodingContext(
         filename="test_op.py",
         design_doc=Document(content="divide two numbers"),
-        code_doc=Document(content=code_content)
+        code_doc=Document(content=code_content),
     )
 
     action = WriteCodeReview(context=context)

tests/metagpt/test_environment.py

@@ -6,9 +6,10 @@
 @File    : test_environment.py
 """
 
-import pytest
 from pathlib import Path
 
+import pytest
+
 from metagpt.actions import UserRequirement
 from metagpt.environment import Environment
 from metagpt.logs import logger
@@ -16,7 +17,6 @@ from metagpt.manager import Manager
 from metagpt.roles import Architect, ProductManager, Role
 from metagpt.schema import Message
 
-
 serdeser_path = Path(__file__).absolute().parent.joinpath("../data/serdeser_storage")
 
 
@@ -26,23 +26,16 @@ def env():
 
 
 def test_add_role(env: Environment):
-    role = ProductManager(name="Alice",
-                          profile="product manager",
-                          goal="create a new product",
-                          constraints="limited resources")
+    role = ProductManager(
+        name="Alice", profile="product manager", goal="create a new product", constraints="limited resources"
+    )
     env.add_role(role)
     assert env.get_role(role.profile) == role
 
 
 def test_get_roles(env: Environment):
-    role1 = Role(name="Alice",
-                 profile="product manager",
-                 goal="create a new product",
-                 constraints="limited resources")
-    role2 = Role(name="Bob",
-                 profile="engineer",
-                 goal="develop the new product",
-                 constraints="short deadline")
+    role1 = Role(name="Alice", profile="product manager", goal="create a new product", constraints="limited resources")
+    role2 = Role(name="Bob", profile="engineer", goal="develop the new product", constraints="short deadline")
     env.add_role(role1)
     env.add_role(role2)
     roles = env.get_roles()
@@ -51,14 +44,10 @@ def test_get_roles(env: Environment):
 
 @pytest.mark.asyncio
 async def test_publish_and_process_message(env: Environment):
-    product_manager = ProductManager(name="Alice",
-                                     profile="Product Manager",
-                                     goal="做AI Native产品",
-                                     constraints="资源有限")
-    architect = Architect(name="Bob",
-                          profile="Architect",
-                          goal="设计一个可用、高效、较低成本的系统，包括数据结构与接口",
-                          constraints="资源有限，需要节省成本")
+    product_manager = ProductManager(name="Alice", profile="Product Manager", goal="做AI Native产品", constraints="资源有限")
+    architect = Architect(
+        name="Bob", profile="Architect", goal="设计一个可用、高效、较低成本的系统，包括数据结构与接口", constraints="资源有限，需要节省成本"
+    )
 
     env.add_roles([product_manager, architect])
 

tests/metagpt/test_schema.py

@@ -9,12 +9,13 @@
 """
 
 import json
+
 import pytest
 
 from metagpt.actions import Action
-from metagpt.schema import AIMessage, Message, SystemMessage, UserMessage
 from metagpt.actions.action_node import ActionNode
 from metagpt.actions.write_code import WriteCode
+from metagpt.schema import AIMessage, Message, SystemMessage, UserMessage
 from metagpt.utils.common import any_to_str
 
 
@@ -77,24 +78,13 @@ def test_message_serdeser():
     out_data = {"field3": "field3 value3", "field4": ["field4 value1", "field4 value2"]}
     ic_obj = ActionNode.create_model_class("code", out_mapping)
 
-    message = Message(
-        content="code",
-        instruct_content=ic_obj(**out_data),
-        role="engineer",
-        cause_by=WriteCode
-    )
+    message = Message(content="code", instruct_content=ic_obj(**out_data), role="engineer", cause_by=WriteCode)
     message_dict = message.dict()
     assert message_dict["cause_by"] == "metagpt.actions.write_code.WriteCode"
     assert message_dict["instruct_content"] == {
         "class": "code",
-        "mapping": {
-            "field3": "(<class 'str'>, Ellipsis)",
-            "field4": "(list[str], Ellipsis)"
-        },
-        "value": {
-            "field3": "field3 value3",
-            "field4": ["field4 value1", "field4 value2"]
-        }
+        "mapping": {"field3": "(<class 'str'>, Ellipsis)", "field4": "(list[str], Ellipsis)"},
+        "value": {"field3": "field3 value3", "field4": ["field4 value1", "field4 value2"]},
     }
 
     new_message = Message(**message_dict)

tests/metagpt/test_team.py

@@ -2,8 +2,8 @@
 # -*- coding: utf-8 -*-
 # @Desc   : unittest of team
 
-from metagpt.team import Team
 from metagpt.roles.project_manager import ProjectManager
+from metagpt.team import Team
 
 
 def test_team():