修正

metagpt/tools/__init__.py

@@ -6,6 +6,7 @@
 @File    : __init__.py
 """
 
+
 from enum import Enum
 
 

metagpt/tools/code_interpreter.py

@@ -1,16 +1,16 @@
-import inspect
 import re
-import textwrap
-from pathlib import Path
 from typing import List, Callable
+from pathlib import Path
 
 import wrapt
+import textwrap
+import inspect
 from interpreter.interpreter import Interpreter
 
-from metagpt.actions.clone_function import CloneFunction, run_function_code, run_function_script
-from metagpt.config import CONFIG
 from metagpt.logs import logger
+from metagpt.config import CONFIG
 from metagpt.utils.highlight import highlight
+from metagpt.actions.clone_function import CloneFunction, run_function_code, run_function_script
 
 
 def extract_python_code(code: str):
@@ -36,7 +36,6 @@ def extract_python_code(code: str):
 
 class OpenCodeInterpreter(object):
     """https://github.com/KillianLucas/open-interpreter"""
-
     def __init__(self, auto_run: bool = True) -> None:
         interpreter = Interpreter()
         interpreter.auto_run = auto_run
@@ -127,5 +126,4 @@ class OpenInterpreterDecorator(object):
             except Exception as e:
                 raise Exception("Could not evaluate Python code", e)
             return res
-
         return wrapper(wrapped)

metagpt/tools/prompt_writer.py

@@ -10,7 +10,6 @@ from typing import Union
 
 class GPTPromptGenerator:
     """Using LLM, given an output, request LLM to provide input (supporting instruction, chatbot, and query styles)"""
-
     def __init__(self):
         self._generators = {i: getattr(self, f"gen_{i}_style") for i in ['instruction', 'chatbot', 'query']}
 

metagpt/tools/sd_engine.py

@@ -10,8 +10,8 @@ import os
 from os.path import join
 from typing import List
 
-from PIL import Image, PngImagePlugin
 from aiohttp import ClientSession
+from PIL import Image, PngImagePlugin
 
 from metagpt.config import Config
 from metagpt.const import WORKSPACE_ROOT
@@ -64,12 +64,12 @@ class SDEngine:
         logger.info(self.sd_t2i_url)
 
     def construct_payload(
-            self,
-            prompt,
-            negtive_prompt=default_negative_prompt,
-            width=512,
-            height=512,
-            sd_model="galaxytimemachinesGTM_photoV20",
+        self,
+        prompt,
+        negtive_prompt=default_negative_prompt,
+        width=512,
+        height=512,
+        sd_model="galaxytimemachinesGTM_photoV20",
     ):
         # Configure the payload with provided inputs
         self.payload["prompt"] = prompt
@@ -120,13 +120,11 @@ def decode_base64_to_image(img, save_name):
     image.save(f"{save_name}.png", pnginfo=pnginfo)
     return pnginfo, image
 
-
 def batch_decode_base64_to_image(imgs, save_dir="", save_name=""):
     for idx, _img in enumerate(imgs):
         save_name = join(save_dir, save_name)
         decode_base64_to_image(_img, save_name=save_name)
 
-
 if __name__ == "__main__":
     engine = SDEngine()
     prompt = "pixel style, game design, a game interface should be minimalistic and intuitive with the score and high score displayed at the top. The snake and its food should be easily distinguishable. The game should have a simple color scheme, with a contrasting color for the snake and its food. Complete interface boundary"

metagpt/tools/search_engine.py

@@ -20,7 +20,7 @@ class SkSearchEngine:
 
     @sk_function(
         description="searches results from Google. Useful when you need to find short "
-                    "and succinct answers about a specific topic. Input should be a search query.",
+        "and succinct answers about a specific topic. Input should be a search query.",
         name="searchAsync",
         input_description="search",
     )
@@ -42,7 +42,7 @@ class SearchEngine:
     """
 
     def __init__(
-            self,
+        self,
             engine: Optional[SearchEngineType] = None,
             run_func: Callable[[str, int, bool], Coroutine[None, None, Union[str, list[str]]]] = None,
     ):
@@ -68,19 +68,19 @@ class SearchEngine:
 
     @overload
     def run(
-            self,
-            query: str,
-            max_results: int = 8,
-            as_string: Literal[True] = True,
+        self,
+        query: str,
+        max_results: int = 8,
+        as_string: Literal[True] = True,
     ) -> str:
         ...
 
     @overload
     def run(
-            self,
-            query: str,
-            max_results: int = 8,
-            as_string: Literal[False] = False,
+        self,
+        query: str,
+        max_results: int = 8,
+        as_string: Literal[False] = False,
     ) -> list[dict[str, str]]:
         ...
 

metagpt/tools/search_engine_ddg.py

@@ -25,10 +25,10 @@ class DDGAPIWrapper:
     """
 
     def __init__(
-            self,
-            *,
-            loop: asyncio.AbstractEventLoop | None = None,
-            executor: futures.Executor | None = None,
+        self,
+        *,
+        loop: asyncio.AbstractEventLoop | None = None,
+        executor: futures.Executor | None = None,
     ):
         kwargs = {}
         if CONFIG.global_proxy:
@@ -39,29 +39,29 @@ class DDGAPIWrapper:
 
     @overload
     def run(
-            self,
-            query: str,
-            max_results: int = 8,
-            as_string: Literal[True] = True,
-            focus: list[str] | None = None,
+        self,
+        query: str,
+        max_results: int = 8,
+        as_string: Literal[True] = True,
+        focus: list[str] | None = None,
     ) -> str:
         ...
 
     @overload
     def run(
-            self,
-            query: str,
-            max_results: int = 8,
-            as_string: Literal[False] = False,
-            focus: list[str] | None = None,
+        self,
+        query: str,
+        max_results: int = 8,
+        as_string: Literal[False] = False,
+        focus: list[str] | None = None,
     ) -> list[dict[str, str]]:
         ...
 
     async def run(
-            self,
-            query: str,
-            max_results: int = 8,
-            as_string: bool = True,
+        self,
+        query: str,
+        max_results: int = 8,
+        as_string: bool = True,
     ) -> str | list[dict]:
         """Return the results of a Google search using the official Google API
 

metagpt/tools/search_engine_googleapi.py

@@ -76,11 +76,11 @@ class GoogleAPIWrapper(BaseModel):
         return service.cse()
 
     async def run(
-            self,
-            query: str,
-            max_results: int = 8,
-            as_string: bool = True,
-            focus: list[str] | None = None,
+        self,
+        query: str,
+        max_results: int = 8,
+        as_string: bool = True,
+        focus: list[str] | None = None,
     ) -> str | list[dict]:
         """Return the results of a Google search using the official Google API.
 

metagpt/tools/translator.py

@@ -24,4 +24,4 @@ class Translator:
 
     @classmethod
     def translate_prompt(cls, original, lang='中文'):
-        return prompt.format(LANG=lang, ORIGINAL=original)
+        return prompt.format(LANG=lang, ORIGINAL=original)

metagpt/tools/web_browser_engine.py

@@ -12,9 +12,9 @@ from metagpt.utils.parse_html import WebPage
 
 class WebBrowserEngine:
     def __init__(
-            self,
-            engine: WebBrowserEngineType | None = None,
-            run_func: Callable[..., Coroutine[Any, Any, WebPage | list[WebPage]]] | None = None,
+        self,
+        engine: WebBrowserEngineType | None = None,
+        run_func: Callable[..., Coroutine[Any, Any, WebPage | list[WebPage]]] | None = None,
     ):
         engine = engine or CONFIG.web_browser_engine
 
@@ -46,9 +46,7 @@ class WebBrowserEngine:
 if __name__ == "__main__":
     import fire
 
-
     async def main(url: str, *urls: str, engine_type: Literal["playwright", "selenium"] = "playwright", **kwargs):
         return await WebBrowserEngine(WebBrowserEngineType(engine_type), **kwargs).run(url, *urls)
 
-
     fire.Fire(main)

metagpt/tools/web_browser_engine_playwright.py

@@ -23,10 +23,10 @@ class PlaywrightWrapper:
     """
 
     def __init__(
-            self,
-            browser_type: Literal["chromium", "firefox", "webkit"] | None = None,
-            launch_kwargs: dict | None = None,
-            **kwargs,
+        self,
+        browser_type: Literal["chromium", "firefox", "webkit"] | None = None,
+        launch_kwargs: dict | None = None,
+        **kwargs,
     ) -> None:
         if browser_type is None:
             browser_type = CONFIG.playwright_browser_type
@@ -139,12 +139,11 @@ async def _log_stream(sr, log_func):
 _install_lock: asyncio.Lock = None
 _install_cache = set()
 
+
 if __name__ == "__main__":
     import fire
 
-
     async def main(url: str, *urls: str, browser_type: str = "chromium", **kwargs):
         return await PlaywrightWrapper(browser_type, **kwargs).run(url, *urls)
 
-
     fire.Fire(main)

metagpt/tools/web_browser_engine_selenium.py

@@ -28,12 +28,12 @@ class SeleniumWrapper:
     """
 
     def __init__(
-            self,
-            browser_type: Literal["chrome", "firefox", "edge", "ie"] | None = None,
-            launch_kwargs: dict | None = None,
-            *,
-            loop: asyncio.AbstractEventLoop | None = None,
-            executor: futures.Executor | None = None,
+        self,
+        browser_type: Literal["chrome", "firefox", "edge", "ie"] | None = None,
+        launch_kwargs: dict | None = None,
+        *,
+        loop: asyncio.AbstractEventLoop | None = None,
+        executor: futures.Executor | None = None,
     ) -> None:
         if browser_type is None:
             browser_type = CONFIG.selenium_browser_type
@@ -117,9 +117,7 @@ def _gen_get_driver_func(browser_type, *args, executable_path=None):
 if __name__ == "__main__":
     import fire
 
-
     async def main(url: str, *urls: str, browser_type: str = "chrome", **kwargs):
         return await SeleniumWrapper(browser_type, **kwargs).run(url, *urls)
 
-
     fire.Fire(main)