From 263595b980e9bc34c225b762a84f2b968f1a91d1 Mon Sep 17 00:00:00 2001 From: lidanyang Date: Wed, 27 Dec 2023 11:03:39 +0800 Subject: [PATCH] support load tools from file or file list --- metagpt/actions/write_analysis_code.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/metagpt/actions/write_analysis_code.py b/metagpt/actions/write_analysis_code.py index 9691f888f..2d9110e91 100644 --- a/metagpt/actions/write_analysis_code.py +++ b/metagpt/actions/write_analysis_code.py @@ -118,7 +118,13 @@ class WriteCodeWithTools(BaseWriteAnalysisCode): schema_module = schema_module or 'udf' self.available_tools.update({schema_module: schema_path}) else: - yml_files = schema_path.glob("*.yml") + if isinstance(schema_path, list): + yml_files = schema_path + elif isinstance(schema_path, Path) and schema_path.is_file(): + yml_files = [schema_path] + else: + yml_files = schema_path.glob("*.yml") + for yml_file in yml_files: module = yml_file.stem with open(yml_file, "r", encoding="utf-8") as f: