From 86e320be1187ef4738a8000e270cc69cdbf31030 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:57:22 +0800 Subject: [PATCH] update for no_udf_found. --- metagpt/actions/write_analysis_code.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/metagpt/actions/write_analysis_code.py b/metagpt/actions/write_analysis_code.py index 725c4aa2a..663f76b7b 100644 --- a/metagpt/actions/write_analysis_code.py +++ b/metagpt/actions/write_analysis_code.py @@ -211,13 +211,16 @@ class WriteCodeWithUDFs(WriteCodeByGenerate): from metagpt.tools.functions.libs.udf import UDFS DEFAULT_SYSTEM_MSG = f"""Please remember these functions, you will use these functions to write code:\n - {UDFS} + {UDFS}, **Notice: 1. if no right udf for user requirement, please send `No udf found`** """ async def aask_code_and_text(self, context: List[Dict], **kwargs) -> Tuple[str]: rsp = await self.llm.acompletion(context, **kwargs) rsp_content = self.llm.get_choice_text(rsp) code = CodeParser.parse_code(None, rsp_content) + if code.startswith('No udf found') or rsp_content.startswith('No udf found'): + rsp_content = 'No udf found' + code = 'No udf found' return code, rsp_content async def run(self, context: List[Message], plan: Plan = None, task_guide: str = "", **kwargs) -> str: