From 19b0120c15c3ad5cce82256f2cdb374df4507f72 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=88=98=E6=A3=92=E6=A3=92?= <liubangbang@fuzhi.ai>
Date: Wed, 20 Dec 2023 09:58:31 +0800
Subject: [PATCH] restore task_type value.

---
 metagpt/actions/write_analysis_code.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/metagpt/actions/write_analysis_code.py b/metagpt/actions/write_analysis_code.py
index bc069414f..88f22684d 100644
--- a/metagpt/actions/write_analysis_code.py
+++ b/metagpt/actions/write_analysis_code.py
@@ -178,7 +178,7 @@ class WriteCodeWithTools(BaseWriteAnalysisCode):
         column_info: str = "",
         **kwargs,
     ) -> Tuple[List[Message], str]:
-        task_type = plan.current_task.task_type or 'udf'
+        task_type = plan.current_task.task_type
         available_tools = self.available_tools.get(task_type, {})
         special_prompt = ML_SPECIFIC_PROMPT.get(task_type, "")
         code_steps = plan.current_task.code_steps