From 10d488c49a2aaf93f93e5ff43daf58811b5cd195 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 11:12:17 +0800 Subject: [PATCH] fix: path in function_signatures. --- metagpt/tools/functions/libs/udf/__init__.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/metagpt/tools/functions/libs/udf/__init__.py b/metagpt/tools/functions/libs/udf/__init__.py index c581dd992..e44e97c41 100644 --- a/metagpt/tools/functions/libs/udf/__init__.py +++ b/metagpt/tools/functions/libs/udf/__init__.py @@ -22,11 +22,11 @@ def extract_function_signatures(file_path): # 获取函数签名 function_signature = f"{function_name}({', '.join(args)})" # 导入函数 - module = Path(file_path).parts[-1][:-len(Path(file_path).suffix)] - module = importlib.import_module(f"metagpt.tools.functions.libs.udf.{module}") + 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} import function_name'} + 'path': f'from metagpt.tools.functions.libs.udf.{module_name} import {function_name}'} function_signatures.append(function_schema) return function_signatures