From 1b4aac394d1a5095224a735a83e3034d447231c5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=88=98=E6=A3=92=E6=A3=92?= Date: Mon, 11 Dec 2023 19:28:57 +0800 Subject: [PATCH] chore: update DEFAULT_SYSTEM_MSG and self.workspace. --- metagpt/actions/make_tools.py | 10 ++++++---- metagpt/tools/functions/libs/udf/__init__.py | 0 2 files changed, 6 insertions(+), 4 deletions(-) create mode 100644 metagpt/tools/functions/libs/udf/__init__.py diff --git a/metagpt/actions/make_tools.py b/metagpt/actions/make_tools.py index aa2ebe501..7cad8ef7b 100644 --- a/metagpt/actions/make_tools.py +++ b/metagpt/actions/make_tools.py @@ -12,9 +12,10 @@ from metagpt.actions.write_analysis_code import WriteCodeByGenerate class MakeTools(WriteCodeByGenerate): DEFAULT_SYSTEM_MSG = """Please Create a very General Function Code startswith `def` from any codes you got.\n - **Notice:1. The import statement must be written after `def`, it is very important for you. - 2. Reflect on whether it meets the requirements of a general function. - 3. Refactor your code to get the most efficient implementation for large input data in the shortest amount of time. + **Notice: + 1. Reflect on whether it meets the requirements of a general function. + 2. Refactor your code to get the most efficient implementation for large input data in the shortest amount of time. + 3. Use Google style for function annotations. 4. Write example code by using old varibales in old code, and make sure it could be execute in the user's machine.** """ @@ -26,7 +27,7 @@ class MakeTools(WriteCodeByGenerate): :param str workspace: tools code saved file path dir, defaults to None """ super().__init__(name, context, llm) - self.workspace = workspace or "." + self.workspace = workspace or str(Path(__file__).parents[1].joinpath("./tools/functions/libs/udf")) self.file_suffix: str = '.py' def parse_function_name(self, function_code: str) -> str: @@ -47,6 +48,7 @@ class MakeTools(WriteCodeByGenerate): saved_path = Path(self.workspace).joinpath(func_name+self.file_suffix) logger.info(f"Saved tool_code {func_name} in {str(saved_path)}.") saved_path.write_text(tool_code, encoding='utf-8') + # TODO: 保存到udf中，供WriteCodeWithMakeTools使用 @retry(stop=stop_after_attempt(3), wait=wait_fixed(1)) async def run(self, code_message: List[Message | Dict], **kwargs) -> str: diff --git a/metagpt/tools/functions/libs/udf/__init__.py b/metagpt/tools/functions/libs/udf/__init__.py new file mode 100644 index 000000000..e69de29bb