run pre-commit to find potential issues and fix them

metagpt/actions/action.py

@@ -8,13 +8,18 @@
 
 from __future__ import annotations
 
-from typing import Optional, Any
+from typing import Any, Optional
 
 from pydantic import BaseModel, Field
 
 from metagpt.llm import LLM
 from metagpt.provider.base_gpt_api import BaseGPTAPI
-from metagpt.schema import CodingContext, CodeSummarizeContext, TestingContext, RunCodeContext
+from metagpt.schema import (
+    CodeSummarizeContext,
+    CodingContext,
+    RunCodeContext,
+    TestingContext,
+)
 
 action_subclass_registry = {}
 

metagpt/actions/debug_error.py

@@ -17,7 +17,7 @@ from metagpt.config import CONFIG
 from metagpt.const import TEST_CODES_FILE_REPO, TEST_OUTPUTS_FILE_REPO
 from metagpt.llm import LLM, BaseGPTAPI
 from metagpt.logs import logger
-from metagpt.schema import RunCodeResult, RunCodeContext
+from metagpt.schema import RunCodeContext, RunCodeResult
 from metagpt.utils.common import CodeParser
 from metagpt.utils.file_repository import FileRepository
 

metagpt/actions/design_api.py

@@ -45,9 +45,11 @@ class WriteDesign(Action):
     name: str = ""
     context: Optional[str] = None
     llm: BaseGPTAPI = Field(default_factory=LLM)
-    desc: str = "Based on the PRD, think about the system design, and design the corresponding APIs, " \
-                "data structures, library tables, processes, and paths. Please provide your design, feedback " \
-                "clearly and in detail."
+    desc: str = (
+        "Based on the PRD, think about the system design, and design the corresponding APIs, "
+        "data structures, library tables, processes, and paths. Please provide your design, feedback "
+        "clearly and in detail."
+    )
 
     async def run(self, with_messages: Message, schema: str = CONFIG.prompt_schema):
         # Use `git diff` to identify which PRD documents have been modified in the `docs/prds` directory.

metagpt/actions/fix_bug.py

@@ -9,6 +9,7 @@ from metagpt.actions import Action
 
 class FixBug(Action):
     """Fix bug action without any implementation details"""
+
     name: str = "FixBug"
 
     async def run(self, *args, **kwargs):

metagpt/actions/prepare_documents.py

@@ -25,6 +25,7 @@ from metagpt.utils.git_repository import GitRepository
 
 class PrepareDocuments(Action):
     """PrepareDocuments Action: initialize project folder and add new requirements to docs/requirements.txt."""
+
     name: str = "PrepareDocuments"
     context: Optional[str] = None
     llm: BaseGPTAPI = Field(default_factory=LLM)

metagpt/actions/run_code.py

@@ -24,7 +24,7 @@ from metagpt.actions.action import Action
 from metagpt.config import CONFIG
 from metagpt.llm import LLM, BaseGPTAPI
 from metagpt.logs import logger
-from metagpt.schema import RunCodeResult, RunCodeContext
+from metagpt.schema import RunCodeContext, RunCodeResult
 from metagpt.utils.exceptions import handle_exception
 
 PROMPT_TEMPLATE = """

metagpt/actions/search_and_summarize.py

@@ -5,18 +5,18 @@
 @Author  : alexanderwu
 @File    : search_google.py
 """
+from typing import Optional
+
 import pydantic
-from typing import Optional, Any
-from pydantic import BaseModel, Field
+from pydantic import Field, root_validator
 
 from metagpt.actions import Action
+from metagpt.config import CONFIG, Config
 from metagpt.llm import LLM
-from metagpt.provider.base_gpt_api import BaseGPTAPI
-from metagpt.config import Config, CONFIG
 from metagpt.logs import logger
+from metagpt.provider.base_gpt_api import BaseGPTAPI
 from metagpt.schema import Message
 from metagpt.tools.search_engine import SearchEngine
-from pydantic import root_validator
 
 SEARCH_AND_SUMMARIZE_SYSTEM = """### Requirements
 1. Please summarize the latest dialogue based on the reference information (secondary) and dialogue history (primary). Do not include text that is irrelevant to the conversation.
@@ -120,7 +120,7 @@ class SearchAndSummarize(Action):
         engine = values.get("engine")
         search_func = values.get("search_func")
         config = Config()
-        
+
         if engine is None:
             engine = config.search_engine
         try:
@@ -135,7 +135,7 @@ class SearchAndSummarize(Action):
         if self.search_engine is None:
             logger.warning("Configure one of SERPAPI_API_KEY, SERPER_API_KEY, GOOGLE_API_KEY to unlock full feature")
             return ""
-        
+
         query = context[-1].content
         # logger.debug(query)
         rsp = await self.search_engine.run(query)
@@ -144,9 +144,9 @@ class SearchAndSummarize(Action):
             logger.error("empty rsp...")
             return ""
         # logger.info(rsp)
-        
+
         system_prompt = [system_text]
-        
+
         prompt = SEARCH_AND_SUMMARIZE_PROMPT.format(
             ROLE=self.prefix,
             CONTEXT=rsp,

metagpt/actions/write_code_review.py

@@ -142,15 +142,9 @@ class WriteCodeReview(Action):
         iterative_code = self.context.code_doc.content
         k = CONFIG.code_review_k_times or 1
         for i in range(k):
-            format_example = FORMAT_EXAMPLE.format(
-                filename=self.context.code_doc.filename
-            )
-            task_content = (
-                self.context.task_doc.content if self.context.task_doc else ""
-            )
-            code_context = await WriteCode.get_codes(
-                self.context.task_doc, exclude=self.context.filename
-            )
+            format_example = FORMAT_EXAMPLE.format(filename=self.context.code_doc.filename)
+            task_content = self.context.task_doc.content if self.context.task_doc else ""
+            code_context = await WriteCode.get_codes(self.context.task_doc, exclude=self.context.filename)
             context = "\n".join(
                 [
                     "## System Design\n" + str(self.context.design_doc) + "\n",

metagpt/actions/write_prd.py

@@ -143,8 +143,9 @@ class WritePRD(Action):
 
     async def _update_prd(self, requirement_doc, prd_doc, prds_file_repo, *args, **kwargs) -> Document | None:
         if not prd_doc:
-            prd = await self._run_new_requirement(requirements=[requirement_doc.content if requirement_doc else ""],
-                                                  *args, **kwargs)
+            prd = await self._run_new_requirement(
+                requirements=[requirement_doc.content if requirement_doc else ""], *args, **kwargs
+            )
             new_prd_doc = Document(
                 root_path=PRDS_FILE_REPO,
                 filename=FileRepository.new_filename() + ".json",

metagpt/actions/write_test.py

@@ -9,14 +9,15 @@
 """
 
 from typing import Optional
+
 from pydantic import Field
 
-from metagpt.llm import LLM
-from metagpt.provider.base_gpt_api import BaseGPTAPI
 from metagpt.actions.action import Action
 from metagpt.config import CONFIG
 from metagpt.const import TEST_CODES_FILE_REPO
+from metagpt.llm import LLM
 from metagpt.logs import logger
+from metagpt.provider.base_gpt_api import BaseGPTAPI
 from metagpt.schema import Document, TestingContext
 from metagpt.utils.common import CodeParser
 

metagpt/environment.py

@@ -56,12 +56,14 @@ class Environment(BaseModel):
         roles_path = stg_path.joinpath("roles.json")
         roles_info = []
         for role_key, role in self.roles.items():
-            roles_info.append({
-                "role_class": role.__class__.__name__,
-                "module_name": role.__module__,
-                "role_name": role.name,
-                "role_sub_tags": list(self.members.get(role))
-            })
+            roles_info.append(
+                {
+                    "role_class": role.__class__.__name__,
+                    "module_name": role.__module__,
+                    "role_name": role.name,
+                    "role_sub_tags": list(self.members.get(role)),
+                }
+            )
             role.serialize(stg_path=stg_path.joinpath(f"roles/{role.__class__.__name__}_{role.name}"))
         write_json_file(roles_path, roles_info)
 
@@ -70,7 +72,7 @@ class Environment(BaseModel):
 
     @classmethod
     def deserialize(cls, stg_path: Path) -> "Environment":
-        """ stg_path: ./storage/team/environment/ """
+        """stg_path: ./storage/team/environment/"""
         roles_path = stg_path.joinpath("roles.json")
         roles_info = read_json_file(roles_path)
         roles = []
@@ -83,9 +85,7 @@ class Environment(BaseModel):
         history = read_json_file(stg_path.joinpath("history.json"))
         history = history.get("content")
 
-        environment = Environment(**{
-            "history": history
-        })
+        environment = Environment(**{"history": history})
         environment.add_roles(roles)
 
         return environment

metagpt/memory/longterm_memory.py

@@ -5,9 +5,7 @@
 """
 
 from typing import Optional
-from pydantic import Field
 
-from typing import Optional
 from pydantic import Field
 
 from metagpt.logs import logger
@@ -22,6 +20,7 @@ class LongTermMemory(Memory):
     - recover memory when it staruped
     - update memory when it changed
     """
+
     memory_storage: MemoryStorage = Field(default_factory=MemoryStorage)
     rc: Optional["RoleContext"] = None
     msg_from_recover: bool = False

metagpt/memory/memory.py

@@ -6,7 +6,6 @@
 @File    : memory.py
 @Modified By: mashenquan, 2023-11-1. According to RFC 116: Updated the type of index key.
 """
-import copy
 from collections import defaultdict
 from pathlib import Path
 from typing import Iterable, Set
@@ -14,11 +13,17 @@ from typing import Iterable, Set
 from pydantic import BaseModel, Field
 
 from metagpt.schema import Message
-from metagpt.utils.common import any_to_str, any_to_str_set, read_json_file, write_json_file
+from metagpt.utils.common import (
+    any_to_str,
+    any_to_str_set,
+    read_json_file,
+    write_json_file,
+)
 
 
 class Memory(BaseModel):
     """The most basic memory: super-memory"""
+
     storage: list[Message] = []
     index: dict[str, list[Message]] = Field(default_factory=defaultdict(list))
 
@@ -32,14 +37,14 @@ class Memory(BaseModel):
         self.index = new_index
 
     def serialize(self, stg_path: Path):
-        """ stg_path = ./storage/team/environment/ or ./storage/team/environment/roles/{role_class}_{role_name}/ """
+        """stg_path = ./storage/team/environment/ or ./storage/team/environment/roles/{role_class}_{role_name}/"""
         memory_path = stg_path.joinpath("memory.json")
         storage = self.dict()
         write_json_file(memory_path, storage)
 
     @classmethod
     def deserialize(cls, stg_path: Path) -> "Memory":
-        """ stg_path = ./storage/team/environment/ or ./storage/team/environment/roles/{role_class}_{role_name}/"""
+        """stg_path = ./storage/team/environment/ or ./storage/team/environment/roles/{role_class}_{role_name}/"""
         memory_path = stg_path.joinpath("memory.json")
 
         memory_dict = read_json_file(memory_path)
@@ -68,7 +73,7 @@ class Memory(BaseModel):
         return [message for message in self.storage if content in message.content]
 
     def delete_newest(self) -> "Message":
-        """ delete the newest message from the storage"""
+        """delete the newest message from the storage"""
         if len(self.storage) > 0:
             newest_msg = self.storage.pop()
             if newest_msg.cause_by and newest_msg in self.index[newest_msg.cause_by]:

metagpt/provider/postprecess/base_postprecess_plugin.py

@@ -4,7 +4,6 @@
 
 from typing import Union
 
-from metagpt.logs import logger
 from metagpt.utils.repair_llm_raw_output import (
     RepairType,
     extract_content_from_output,

metagpt/roles/architect.py

@@ -5,7 +5,6 @@
 @Author  : alexanderwu
 @File    : architect.py
 """
-from pydantic import Field
 
 from metagpt.actions import WritePRD
 from metagpt.actions.design_api import WriteDesign
@@ -22,11 +21,14 @@ class Architect(Role):
         goal (str): Primary goal or responsibility of the architect.
         constraints (str): Constraints or guidelines for the architect.
     """
+
     name: str = "Bob"
     profile: str = "Architect"
     goal: str = "design a concise, usable, complete software system"
-    constraints: str = "make sure the architecture is simple enough and use  appropriate open source " \
-                       "libraries. Use same language as user requirement"
+    constraints: str = (
+        "make sure the architecture is simple enough and use  appropriate open source "
+        "libraries. Use same language as user requirement"
+    )
 
     def __init__(self, **kwargs) -> None:
         super().__init__(**kwargs)

metagpt/roles/customer_service.py

@@ -6,7 +6,6 @@
 @File    : sales.py
 """
 from typing import Optional
-from pydantic import Field
 
 from metagpt.roles import Sales
 
@@ -27,14 +26,11 @@ DESC = """
 
 
 class CustomerService(Sales):
-
     name: str = "Xiaomei"
     profile: str = "Human customer service"
     desc: str = DESC
 
     store: Optional[str] = None
 
-    def __init__(
-            self,
-            **kwargs):
+    def __init__(self, **kwargs):
         super().__init__(**kwargs)

metagpt/roles/engineer.py

@@ -24,8 +24,6 @@ from collections import defaultdict
 from pathlib import Path
 from typing import Set
 
-from pydantic import Field
-
 from metagpt.actions import Action, WriteCode, WriteCodeReview, WriteTasks
 from metagpt.actions.fix_bug import FixBug
 from metagpt.actions.summarize_code import SummarizeCode
@@ -69,11 +67,14 @@ class Engineer(Role):
         n_borg (int): Number of borgs.
         use_code_review (bool): Whether to use code review.
     """
+
     name: str = "Alex"
     profile: str = "Engineer"
     goal: str = "write elegant, readable, extensible, efficient code"
-    constraints: str = "the code should conform to standards like google-style and be modular and maintainable. " \
-                       "Use same language as user requirement"
+    constraints: str = (
+        "the code should conform to standards like google-style and be modular and maintainable. "
+        "Use same language as user requirement"
+    )
     n_borg: int = 1
     use_code_review: bool = False
     code_todos: list = []
@@ -212,7 +213,7 @@ class Engineer(Role):
 
     @staticmethod
     async def _new_coding_context(
-            filename, src_file_repo, task_file_repo, design_file_repo, dependency
+        filename, src_file_repo, task_file_repo, design_file_repo, dependency
     ) -> CodingContext:
         old_code_doc = await src_file_repo.get(filename)
         if not old_code_doc:

metagpt/roles/product_manager.py

@@ -7,7 +7,6 @@
 @Modified By: mashenquan, 2023/11/27. Add `PrepareDocuments` action according to Section 2.2.3.5.1 of RFC 135.
 """
 
-from pydantic import Field
 
 from metagpt.actions import UserRequirement, WritePRD
 from metagpt.actions.prepare_documents import PrepareDocuments
@@ -25,6 +24,7 @@ class ProductManager(Role):
         goal (str): Goal of the product manager.
         constraints (str): Constraints or limitations for the product manager.
     """
+
     name: str = "Alice"
     profile: str = "Product Manager"
     goal: str = "efficiently create a successful product that meets market demands and user expectations"

metagpt/roles/project_manager.py

@@ -5,7 +5,6 @@
 @Author  : alexanderwu
 @File    : project_manager.py
 """
-from pydantic import Field
 
 from metagpt.actions import WriteTasks
 from metagpt.actions.design_api import WriteDesign
@@ -22,10 +21,13 @@ class ProjectManager(Role):
         goal (str): Goal of the project manager.
         constraints (str): Constraints or limitations for the project manager.
     """
+
     name: str = "Eve"
     profile: str = "Project Manager"
-    goal: str = "break down tasks according to PRD/technical design, generate a task list, and analyze task " \
-                "dependencies to start with the prerequisite modules"
+    goal: str = (
+        "break down tasks according to PRD/technical design, generate a task list, and analyze task "
+        "dependencies to start with the prerequisite modules"
+    )
     constraints: str = "use same language as user requirement"
 
     def __init__(self, **kwargs) -> None:

metagpt/roles/qa_engineer.py

@@ -15,13 +15,8 @@
     of SummarizeCode.
 """
 
-from pydantic import Field
 
-from metagpt.actions import (
-    DebugError,
-    RunCode,
-    WriteTest,
-)
+from metagpt.actions import DebugError, RunCode, WriteTest
 from metagpt.actions.summarize_code import SummarizeCode
 from metagpt.config import CONFIG
 from metagpt.const import (
@@ -40,8 +35,9 @@ class QaEngineer(Role):
     name: str = "Edward"
     profile: str = "QaEngineer"
     goal: str = "Write comprehensive and robust tests to ensure codes will work as expected without bugs"
-    constraints: str = "The test code you write should conform to code standard like PEP8, be modular, " \
-                       "easy to read and maintain"
+    constraints: str = (
+        "The test code you write should conform to code standard like PEP8, be modular, " "easy to read and maintain"
+    )
     test_round_allowed: int = 5
 
     def __init__(self, **kwargs):
@@ -118,7 +114,8 @@ class QaEngineer(Role):
         )
         run_code_context.code = None
         run_code_context.test_code = None
-        recipient = parse_recipient(result.summary)  # the recipient might be Engineer or myself
+        # the recipient might be Engineer or myself
+        recipient = parse_recipient(result.summary)
         mappings = {"Engineer": "Alex", "QaEngineer": "Edward"}
         self.publish_message(
             Message(

metagpt/roles/role.py

@@ -23,7 +23,7 @@ from __future__ import annotations
 
 from enum import Enum
 from pathlib import Path
-from typing import Iterable, Set, Type, Any
+from typing import Any, Iterable, Set, Type
 
 from pydantic import BaseModel, Field
 
@@ -37,7 +37,13 @@ from metagpt.logs import logger
 from metagpt.memory import Memory
 from metagpt.provider.base_gpt_api import BaseGPTAPI
 from metagpt.schema import Message, MessageQueue
-from metagpt.utils.common import any_to_str, read_json_file, write_json_file, import_class, role_raise_decorator
+from metagpt.utils.common import (
+    any_to_str,
+    import_class,
+    read_json_file,
+    role_raise_decorator,
+    write_json_file,
+)
 from metagpt.utils.repair_llm_raw_output import extract_state_value_from_output
 
 PREFIX_TEMPLATE = """You are a {profile}, named {name}, your goal is {goal}, and the constraint is {constraints}. """
@@ -82,18 +88,22 @@ class RoleReactMode(str, Enum):
 
 class RoleContext(BaseModel):
     """Role Runtime Context"""
+
     # # env exclude=True to avoid `RecursionError: maximum recursion depth exceeded in comparison`
     env: "Environment" = Field(default=None, exclude=True)
     # TODO judge if ser&deser
-    msg_buffer: MessageQueue = Field(default_factory=MessageQueue,
-                                     exclude=True)  # Message Buffer with Asynchronous Updates
+    msg_buffer: MessageQueue = Field(
+        default_factory=MessageQueue, exclude=True
+    )  # Message Buffer with Asynchronous Updates
     memory: Memory = Field(default_factory=Memory)
     # long_term_memory: LongTermMemory = Field(default_factory=LongTermMemory)
     state: int = Field(default=-1)  # -1 indicates initial or termination state where todo is None
     todo: Action = Field(default=None, exclude=True)
     watch: set[str] = Field(default_factory=set)
     news: list[Type[Message]] = Field(default=[], exclude=True)  # TODO not used
-    react_mode: RoleReactMode = RoleReactMode.REACT  # see `Role._set_react_mode` for definitions of the following two attributes
+    react_mode: RoleReactMode = (
+        RoleReactMode.REACT
+    )  # see `Role._set_react_mode` for definitions of the following two attributes
     max_react_loop: int = 1
 
     class Config:
@@ -120,6 +130,7 @@ role_subclass_registry = {}
 
 class Role(BaseModel):
     """Role/Agent"""
+
     name: str = ""
     profile: str = ""
     goal: str = ""
@@ -145,7 +156,7 @@ class Role(BaseModel):
         "_states": [],
         "_actions": [],
         "_rc": RoleContext(),
-        "_subscription": set()
+        "_subscription": set(),
     }
 
     __hash__ = object.__hash__  # support Role as hashable type in `Environment.members`
@@ -206,14 +217,14 @@ class Role(BaseModel):
         return f"{self.name}({self.profile})"
 
     def serialize(self, stg_path: Path = None):
-        stg_path = SERDESER_PATH.joinpath(f"team/environment/roles/{self.__class__.__name__}_{self.name}") \
-            if stg_path is None else stg_path
+        stg_path = (
+            SERDESER_PATH.joinpath(f"team/environment/roles/{self.__class__.__name__}_{self.name}")
+            if stg_path is None
+            else stg_path
+        )
 
         role_info = self.dict(exclude={"_rc": {"memory": True, "msg_buffer": True}, "_llm": True})
-        role_info.update({
-            "role_class": self.__class__.__name__,
-            "module_name": self.__module__
-        })
+        role_info.update({"role_class": self.__class__.__name__, "module_name": self.__module__})
         role_info_path = stg_path.joinpath("role_info.json")
         write_json_file(role_info_path, role_info)
 
@@ -221,7 +232,7 @@ class Role(BaseModel):
 
     @classmethod
     def deserialize(cls, stg_path: Path) -> "Role":
-        """ stg_path = ./storage/team/environment/roles/{role_class}_{role_name}"""
+        """stg_path = ./storage/team/environment/roles/{role_class}_{role_name}"""
         role_info_path = stg_path.joinpath("role_info.json")
         role_info = read_json_file(role_info_path)
 
@@ -328,12 +339,9 @@ class Role(BaseModel):
         """Get the role prefix"""
         if self.desc:
             return self.desc
-        return PREFIX_TEMPLATE.format(**{
-            "profile": self.profile,
-            "name": self.name,
-            "goal": self.goal,
-            "constraints": self.constraints
-        })
+        return PREFIX_TEMPLATE.format(
+            **{"profile": self.profile, "name": self.name, "goal": self.goal, "constraints": self.constraints}
+        )
 
     async def _think(self) -> None:
         """Think about what to do and decide on the next action"""

metagpt/roles/sales.py

@@ -7,7 +7,6 @@
 """
 
 from typing import Optional
-from pydantic import Field
 
 from metagpt.actions import SearchAndSummarize
 from metagpt.roles import Role
@@ -15,7 +14,6 @@ from metagpt.tools import SearchEngineType
 
 
 class Sales(Role):
-
     name: str = "Xiaomei"
     profile: str = "Retail sales guide"
     desc: str = "I am a sales guide in retail. My name is Xiaomei. I will answer some customer questions next, and I "

metagpt/roles/searcher.py

@@ -35,7 +35,7 @@ class Searcher(Role):
     goal: str = "Provide search services for users"
     constraints: str = "Answer is rich and complete"
     engine: SearchEngineType = SearchEngineType.SERPAPI_GOOGLE
-    
+
     def __init__(self, **kwargs) -> None:
         """
         Initializes the Searcher role with given attributes.

metagpt/schema.py

@@ -23,7 +23,7 @@ from abc import ABC
 from asyncio import Queue, QueueEmpty, wait_for
 from json import JSONDecodeError
 from pathlib import Path
-from typing import Dict, List, Optional, Set, Type, TypedDict, TypeVar, Any
+from typing import Any, Dict, List, Optional, Set, Type, TypedDict, TypeVar
 
 from pydantic import BaseModel, Field
 
@@ -38,9 +38,12 @@ from metagpt.const import (
 )
 from metagpt.logs import logger
 from metagpt.utils.common import any_to_str, any_to_str_set, import_class
-from metagpt.utils.serialize import actionoutout_schema_to_mapping, actionoutput_mapping_to_str, \
-    actionoutput_str_to_mapping
 from metagpt.utils.exceptions import handle_exception
+from metagpt.utils.serialize import (
+    actionoutout_schema_to_mapping,
+    actionoutput_mapping_to_str,
+    actionoutput_str_to_mapping,
+)
 
 
 class RawMessage(TypedDict):
@@ -119,8 +122,9 @@ class Message(BaseModel):
             kwargs["instruct_content"] = ic_new
 
         kwargs["id"] = kwargs.get("id", uuid.uuid4().hex)
-        kwargs["cause_by"] = any_to_str(kwargs.get("cause_by",
-                                                   import_class("UserRequirement", "metagpt.actions.add_requirement")))
+        kwargs["cause_by"] = any_to_str(
+            kwargs.get("cause_by", import_class("UserRequirement", "metagpt.actions.add_requirement"))
+        )
         kwargs["sent_from"] = any_to_str(kwargs.get("sent_from", ""))
         kwargs["send_to"] = any_to_str_set(kwargs.get("send_to", {MESSAGE_ROUTE_TO_ALL}))
         super(Message, self).__init__(**kwargs)
@@ -138,7 +142,7 @@ class Message(BaseModel):
         super().__setattr__(key, new_val)
 
     def dict(self, *args, **kwargs) -> "DictStrAny":
-        """ overwrite the `dict` to dump dynamic pydantic model"""
+        """overwrite the `dict` to dump dynamic pydantic model"""
         obj_dict = super(Message, self).dict(*args, **kwargs)
         ic = self.instruct_content
         if ic:
@@ -208,9 +212,7 @@ class MessageQueue(BaseModel):
 
     _queue: Queue = Field(default_factory=Queue)
 
-    _private_attributes = {
-        "_queue": Queue()
-    }
+    _private_attributes = {"_queue": Queue()}
 
     class Config:
         arbitrary_types_allowed = True

metagpt/startup.py

@@ -1,9 +1,9 @@
 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
 import asyncio
+from pathlib import Path
 
 import typer
-from pathlib import Path
 
 from metagpt.config import CONFIG
 
@@ -32,7 +32,7 @@ def startup(
         help="The maximum number of times the 'SummarizeCode' action is automatically invoked, with -1 indicating "
         "unlimited. This parameter is used for debugging the workflow.",
     ),
-    recover_path: str = typer.Option(default=None, help="recover the project from existing serialized storage")
+    recover_path: str = typer.Option(default=None, help="recover the project from existing serialized storage"),
 ):
     """Run a startup. Be a boss."""
     from metagpt.roles import (

metagpt/team.py

@@ -8,20 +8,24 @@
         Section 2.2.3.3 of RFC 135.
 """
 
-from pathlib import Path
 import warnings
+from pathlib import Path
 
 from pydantic import BaseModel, Field
 
 from metagpt.actions import UserRequirement
 from metagpt.config import CONFIG
-from metagpt.const import MESSAGE_ROUTE_TO_ALL
-from metagpt.const import SERDESER_PATH
+from metagpt.const import MESSAGE_ROUTE_TO_ALL, SERDESER_PATH
 from metagpt.environment import Environment
 from metagpt.logs import logger
 from metagpt.roles import Role
 from metagpt.schema import Message
-from metagpt.utils.common import NoMoneyException, read_json_file, write_json_file, serialize_decorator
+from metagpt.utils.common import (
+    NoMoneyException,
+    read_json_file,
+    serialize_decorator,
+    write_json_file,
+)
 
 
 class Team(BaseModel):
@@ -51,12 +55,14 @@ class Team(BaseModel):
 
     @classmethod
     def deserialize(cls, stg_path: Path) -> "Team":
-        """ stg_path = ./storage/team """
+        """stg_path = ./storage/team"""
         # recover team_info
         team_info_path = stg_path.joinpath("team_info.json")
         if not team_info_path.exists():
-            raise FileNotFoundError("recover storage meta file `team_info.json` not exist, "
-                                    "not to recover and please start a new project.")
+            raise FileNotFoundError(
+                "recover storage meta file `team_info.json` not exist, "
+                "not to recover and please start a new project."
+            )
 
         team_info: dict = read_json_file(team_info_path)
 

metagpt/utils/common.py

@@ -22,8 +22,7 @@ import re
 import traceback
 import typing
 from pathlib import Path
-from typing import Any
-from typing import List, Tuple, Union, get_args, get_origin
+from typing import Any, List, Tuple, Union, get_args, get_origin
 
 import aiofiles
 import loguru
@@ -219,7 +218,7 @@ class OutputParser:
 
         if start_index != -1 and end_index != -1:
             # Extract the structure part
-            structure_text = text[start_index: end_index + 1]
+            structure_text = text[start_index : end_index + 1]
 
             try:
                 # Attempt to convert the text to a Python data type using ast.literal_eval
@@ -439,7 +438,7 @@ def read_json_file(json_file: str, encoding=None) -> list[Any]:
     with open(json_file, "r", encoding=encoding) as fin:
         try:
             data = json.load(fin)
-        except Exception as exp:
+        except Exception:
             raise ValueError(f"read json file: {json_file} failed")
     return data
 
@@ -474,9 +473,9 @@ def serialize_decorator(func):
         try:
             result = await func(self, *args, **kwargs)
             return result
-        except KeyboardInterrupt as kbi:
+        except KeyboardInterrupt:
             logger.error(f"KeyboardInterrupt occurs, start to serialize the project, exp:\n{format_trackback_info()}")
-        except Exception as exp:
+        except Exception:
             logger.error(f"Exception occurs, start to serialize the project, exp:\n{format_trackback_info()}")
         self.serialize()  # Team.serialize
 
@@ -491,14 +490,18 @@ def role_raise_decorator(func):
             logger.error(f"KeyboardInterrupt: {kbi} occurs, start to serialize the project")
             if self.latest_observed_msg:
                 self._rc.memory.delete(self.latest_observed_msg)
-            raise Exception(format_trackback_info(limit=None))  # raise again to make it captured outside
-        except Exception as exp:
+            # raise again to make it captured outside
+            raise Exception(format_trackback_info(limit=None))
+        except Exception:
             if self.latest_observed_msg:
-                logger.warning("There is a exception in role's execution, in order to resume, "
-                               "we delete the newest role communication message in the role's memory.")
+                logger.warning(
+                    "There is a exception in role's execution, in order to resume, "
+                    "we delete the newest role communication message in the role's memory."
+                )
                 # remove role newest observed msg to make it observed again
                 self._rc.memory.delete(self.latest_observed_msg)
-            raise Exception(format_trackback_info(limit=None))  # raise again to make it captured outside
+            # raise again to make it captured outside
+            raise Exception(format_trackback_info(limit=None))
 
     return wrapper