fix ruff check error

tests/metagpt/actions/test_run_code.py

@@ -6,24 +6,23 @@
 @File    : test_run_code.py
 """
 import pytest
-import asyncio
+
 from metagpt.actions.run_code import RunCode
 
+
 @pytest.mark.asyncio
 async def test_run_text():
-    action = RunCode()
-    result, errs = await RunCode.run_text('result = 1 + 1')
+    result, errs = await RunCode.run_text("result = 1 + 1")
     assert result == 2
     assert errs == ""
 
-    result, errs = await RunCode.run_text('result = 1 / 0')
+    result, errs = await RunCode.run_text("result = 1 / 0")
     assert result == ""
     assert "ZeroDivisionError" in errs
 
+
 @pytest.mark.asyncio
 async def test_run_script():
-    action = RunCode()
-    
     # Successful command
     out, err = await RunCode.run_script(".", command=["echo", "Hello World"])
     assert out.strip() == "Hello World"
@@ -33,6 +32,7 @@ async def test_run_script():
     out, err = await RunCode.run_script(".", command=["python", "-c", "print(1/0)"])
     assert "ZeroDivisionError" in err
 
+
 @pytest.mark.asyncio
 async def test_run():
     action = RunCode()
@@ -47,10 +47,11 @@ async def test_run():
         test_file_name="",
         command=["echo", "Hello World"],
         working_directory=".",
-        additional_python_paths=[]
+        additional_python_paths=[],
     )
     assert "PASS" in result
 
+
 @pytest.mark.asyncio
 async def test_run_failure():
     action = RunCode()
@@ -65,6 +66,6 @@ async def test_run_failure():
         test_file_name="",
         command=["python", "-c", "print(1/0)"],
         working_directory=".",
-        additional_python_paths=[]
+        additional_python_paths=[],
     )
-    assert "FAIL" in result
+    assert "FAIL" in result

tests/metagpt/actions/test_write_code_review.py

@@ -8,8 +8,6 @@
 import pytest
 
 from metagpt.actions.write_code_review import WriteCodeReview
-from metagpt.logs import logger
-from tests.metagpt.actions.mock import SEARCH_CODE_SAMPLE
 
 
 @pytest.mark.asyncio
@@ -20,11 +18,7 @@ def add(a, b):
 """
     # write_code_review = WriteCodeReview("write_code_review")
 
-    code = await WriteCodeReview().run(
-        context="编写一个从a加b的函数，返回a+b",
-        code=code,
-        filename="math.py"
-    )
+    code = await WriteCodeReview().run(context="编写一个从a加b的函数，返回a+b", code=code, filename="math.py")
 
     # 我们不能精确地预测生成的代码评审，但我们可以检查返回的是否为字符串
     assert isinstance(code, str)
@@ -33,6 +27,7 @@ def add(a, b):
     captured = capfd.readouterr()
     print(f"输出内容: {captured.out}")
 
+
 # @pytest.mark.asyncio
 # async def test_write_code_review_directly():
 #     code = SEARCH_CODE_SAMPLE

tests/metagpt/roles/ui_role.py

@@ -2,22 +2,19 @@
 # @Date    : 2023/7/15 16:40
 # @Author  : stellahong (stellahong@fuzhi.ai)
 # @Desc    :
-import re
 import os
-from importlib import import_module
+import re
 from functools import wraps
+from importlib import import_module
 
-from metagpt.logs import logger
-from metagpt.actions import Action, ActionOutput
-from metagpt.roles import ProductManager, Role
-from metagpt.schema import Message
+from metagpt.actions import Action, ActionOutput, WritePRD
 from metagpt.const import WORKSPACE_ROOT
-
-from metagpt.actions import WritePRD
-from metagpt.software_company import SoftwareCompany
+from metagpt.logs import logger
+from metagpt.roles import Role
+from metagpt.schema import Message
 from metagpt.tools.sd_engine import SDEngine
 
-PROMPT_TEMPLATE = '''
+PROMPT_TEMPLATE = """
 # Context
 {context}
 
@@ -34,9 +31,9 @@ Attention: Use '##' to split sections, not '#', and '## <SECTION_NAME>' SHOULD W
 ## CSS Styles (styles.css):Provide as Plain text,use standard css code
 ## Anything UNCLEAR:Provide as Plain text. Make clear here.
 
-'''
+"""
 
-FORMAT_EXAMPLE = '''
+FORMAT_EXAMPLE = """
 
 ## UI Design Description
 ```Snake games are classic and addictive games with simple yet engaging elements. Here are the main elements commonly found in snake games ```
@@ -126,7 +123,7 @@ body {
 ## Anything UNCLEAR
 There are no unclear points.
 
-'''
+"""
 
 OUTPUT_MAPPING = {
     "UI Design Description": (str, ...),
@@ -139,25 +136,25 @@ OUTPUT_MAPPING = {
 
 def load_engine(func):
     """Decorator to load an engine by file name and engine name."""
-    
+
     @wraps(func)
     def wrapper(*args, **kwargs):
         file_name, engine_name = func(*args, **kwargs)
-        engine_file = import_module(file_name, package='metagpt')
+        engine_file = import_module(file_name, package="metagpt")
         ip_module_cls = getattr(engine_file, engine_name)
         try:
             engine = ip_module_cls()
         except:
             engine = None
-        
+
         return engine
-    
+
     return wrapper
 
 
 def parse(func):
     """Decorator to parse information using regex pattern."""
-    
+
     @wraps(func)
     def wrapper(*args, **kwargs):
         context, pattern = func(*args, **kwargs)
@@ -168,30 +165,30 @@ def parse(func):
         else:
             text_info = context
             logger.info("未找到匹配的内容")
-        
+
         return text_info
-    
+
     return wrapper
 
 
 class UIDesign(Action):
     """Class representing the UI Design action."""
-    
+
     def __init__(self, name, context=None, llm=None):
         super().__init__(name, context, llm)  # 需要调用LLM进一步丰富UI设计的prompt
-    
+
     @parse
     def parse_requirement(self, context: str):
         """Parse UI Design draft from the context using regex."""
         pattern = r"## UI Design draft.*?\n(.*?)## Anything UNCLEAR"
         return context, pattern
-    
+
     @parse
     def parse_ui_elements(self, context: str):
         """Parse Selected Elements from the context using regex."""
         pattern = r"## Selected Elements.*?\n(.*?)## HTML Layout"
         return context, pattern
-    
+
     @parse
     def parse_css_code(self, context: str):
         pattern = r"```css.*?\n(.*?)## Anything UNCLEAR"
@@ -201,7 +198,7 @@ class UIDesign(Action):
     def parse_html_code(self, context: str):
         pattern = r"```html.*?\n(.*?)```"
         return context, pattern
-    
+
     async def draw_icons(self, context, *args, **kwargs):
         """Draw icons using SDEngine."""
         engine = SDEngine()
@@ -215,20 +212,20 @@ class UIDesign(Action):
             prompts_batch.append(prompt)
         await engine.run_t2i(prompts_batch)
         logger.info("Finish icon design using StableDiffusion API")
-    
+
     async def _save(self, css_content, html_content):
-        save_dir = WORKSPACE_ROOT / "resources" / 'codes'
+        save_dir = WORKSPACE_ROOT / "resources" / "codes"
         if not os.path.exists(save_dir):
             os.makedirs(save_dir, exist_ok=True)
         # Save CSS and HTML content to files
-        css_file_path = save_dir / f"ui_design.css"
-        html_file_path = save_dir / f"ui_design.html"
-        
-        with open(css_file_path, 'w') as css_file:
+        css_file_path = save_dir / "ui_design.css"
+        html_file_path = save_dir / "ui_design.html"
+
+        with open(css_file_path, "w") as css_file:
             css_file.write(css_content)
-        with open(html_file_path, 'w') as html_file:
+        with open(html_file_path, "w") as html_file:
             html_file.write(html_content)
-    
+
     async def run(self, requirements: list[Message], *args, **kwargs) -> ActionOutput:
         """Run the UI Design action."""
         # fixme: update prompt (根据需求细化prompt）
@@ -249,23 +246,27 @@ class UIDesign(Action):
 
 class UI(Role):
     """Class representing the UI Role."""
-    
-    def __init__(self, name="Catherine", profile="UI Design",
-                 goal="Finish a workable and good User Interface design based on a product design",
-                 constraints="Give clear layout description and use standard icons to finish the design",
-                 skills=["SD"]):
+
+    def __init__(
+        self,
+        name="Catherine",
+        profile="UI Design",
+        goal="Finish a workable and good User Interface design based on a product design",
+        constraints="Give clear layout description and use standard icons to finish the design",
+        skills=["SD"],
+    ):
         super().__init__(name, profile, goal, constraints)
         self.load_skills(skills)
         self._init_actions([UIDesign])
         self._watch([WritePRD])
-    
+
     @load_engine
     def load_sd_engine(self):
         """Load the SDEngine."""
         file_name = ".tools.sd_engine"
         engine_name = "SDEngine"
         return file_name, engine_name
-    
+
     def load_skills(self, skills):
         """Load skills for the UI Role."""
         # todo: 添加其他出图engine
@@ -273,4 +274,3 @@ class UI(Role):
             if skill == "SD":
                 self.sd_engine = self.load_sd_engine()
                 logger.info(f"load skill engine {self.sd_engine}")
-    

tests/metagpt/tools/test_web_browser_engine.py

@@ -1,6 +1,6 @@
 import pytest
-from metagpt.config import Config
-from metagpt.tools import web_browser_engine, WebBrowserEngineType
+
+from metagpt.tools import WebBrowserEngineType, web_browser_engine
 
 
 @pytest.mark.asyncio

tests/metagpt/utils/test_serialize.py

@@ -3,94 +3,64 @@
 # @Desc   : the unittest of serialize
 
 from typing import List, Tuple
-import pytest
 
-from pydantic import create_model
-
-from metagpt.actions.action_output import ActionOutput
 from metagpt.actions import WritePRD
+from metagpt.actions.action_output import ActionOutput
 from metagpt.schema import Message
-from metagpt.utils.serialize import actionoutout_schema_to_mapping, serialize_message, deserialize_message
+from metagpt.utils.serialize import (
+    actionoutout_schema_to_mapping,
+    deserialize_message,
+    serialize_message,
+)
 
 
 def test_actionoutout_schema_to_mapping():
-    schema = {
-        'title': 'test',
-        'type': 'object',
-        'properties': {
-            'field': {
-                'title': 'field',
-                'type': 'string'
-            }
-        }
-    }
+    schema = {"title": "test", "type": "object", "properties": {"field": {"title": "field", "type": "string"}}}
     mapping = actionoutout_schema_to_mapping(schema)
-    assert mapping['field'] == (str, ...)
+    assert mapping["field"] == (str, ...)
 
     schema = {
-        'title': 'test',
-        'type': 'object',
-        'properties': {
-            'field': {
-                'title': 'field',
-                'type': 'array',
-                'items': {
-                    'type': 'string'
-                }
-            }
-        }
+        "title": "test",
+        "type": "object",
+        "properties": {"field": {"title": "field", "type": "array", "items": {"type": "string"}}},
     }
     mapping = actionoutout_schema_to_mapping(schema)
-    assert mapping['field'] == (List[str], ...)
+    assert mapping["field"] == (List[str], ...)
 
     schema = {
-        'title': 'test',
-        'type': 'object',
-        'properties': {
-            'field': {
-                'title': 'field',
-                'type': 'array',
-                'items': {
-                    'type': 'array',
-                    'minItems': 2,
-                    'maxItems': 2,
-                    'items': [
-                        {
-                            'type': 'string'
-                        },
-                        {
-                            'type': 'string'
-                        }
-                    ]
-                }
+        "title": "test",
+        "type": "object",
+        "properties": {
+            "field": {
+                "title": "field",
+                "type": "array",
+                "items": {
+                    "type": "array",
+                    "minItems": 2,
+                    "maxItems": 2,
+                    "items": [{"type": "string"}, {"type": "string"}],
+                },
             }
-        }
+        },
     }
     mapping = actionoutout_schema_to_mapping(schema)
-    assert mapping['field'] == (List[Tuple[str, str]], ...)
+    assert mapping["field"] == (List[Tuple[str, str]], ...)
 
     assert True, True
 
 
 def test_serialize_and_deserialize_message():
-    out_mapping = {
-        'field1': (str, ...),
-        'field2': (List[str], ...)
-    }
-    out_data = {
-        'field1': 'field1 value',
-        'field2': ['field2 value1', 'field2 value2']
-    }
-    ic_obj = ActionOutput.create_model_class('prd', out_mapping)
+    out_mapping = {"field1": (str, ...), "field2": (List[str], ...)}
+    out_data = {"field1": "field1 value", "field2": ["field2 value1", "field2 value2"]}
+    ic_obj = ActionOutput.create_model_class("prd", out_mapping)
 
-    message = Message(content='prd demand',
-                      instruct_content=ic_obj(**out_data),
-                      role='user',
-                      cause_by=WritePRD)  # WritePRD as test action
+    message = Message(
+        content="prd demand", instruct_content=ic_obj(**out_data), role="user", cause_by=WritePRD
+    )  # WritePRD as test action
 
     message_ser = serialize_message(message)
 
     new_message = deserialize_message(message_ser)
     assert new_message.content == message.content
     assert new_message.cause_by == message.cause_by
-    assert new_message.instruct_content.field1 == out_data['field1']
+    assert new_message.instruct_content.field1 == out_data["field1"]