From 9651cdd735bf82928f3ada3d299d0c442edbfd73 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=88=98=E6=A3=92=E6=A3=92?= Date: Tue, 12 Dec 2023 14:45:06 +0800 Subject: [PATCH] update function_schema. --- metagpt/tools/functions/libs/udf/__init__.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/metagpt/tools/functions/libs/udf/__init__.py b/metagpt/tools/functions/libs/udf/__init__.py index e44e97c41..c9c818a96 100644 --- a/metagpt/tools/functions/libs/udf/__init__.py +++ b/metagpt/tools/functions/libs/udf/__init__.py @@ -25,8 +25,9 @@ def extract_function_signatures(file_path): module_name = Path(file_path).parts[-1][:-len(Path(file_path).suffix)] module = importlib.import_module(f"metagpt.tools.functions.libs.udf.{module_name}") # 获取函数注释和函数路径 - function_schema = {'name': function_signature, 'doc': inspect.getdoc(getattr(module, function_name)), - 'path': f'from metagpt.tools.functions.libs.udf.{module_name} import {function_name}'} + function_schema = {'udf_name': function_signature, + 'udf_path': f'from metagpt.tools.functions.libs.udf.{module_name} import {function_name}', + 'udf_doc': inspect.getdoc(getattr(module, function_name))} function_signatures.append(function_schema) return function_signatures @@ -48,4 +49,4 @@ folder_path = str(Path(__file__).parent.absolute()) function_signatures = get_function_signatures_in_folder(folder_path) UDFS = [func for func in function_signatures - if not func['name'].startswith(('extract_function_signatures', 'get_function_signatures_in_folder'))] + if not func['udf_name'].startswith(('extract_function_signatures', 'get_function_signatures_in_folder'))]