From 35e8a501c54762bd95bddebc1b3c8367a8993238 Mon Sep 17 00:00:00 2001
From: lidanyang <lidanyang@fuzhi.ai>
Date: Fri, 1 Dec 2023 11:59:28 +0800
Subject: [PATCH] add log print

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

diff --git a/metagpt/actions/write_analysis_code.py b/metagpt/actions/write_analysis_code.py
index 6fff1c66f..e81228109 100644
--- a/metagpt/actions/write_analysis_code.py
+++ b/metagpt/actions/write_analysis_code.py
@@ -4,10 +4,10 @@
 @Author  :   orange-crow
 @File    :   write_code_v2.py
 """
-import json
 from typing import Dict, List, Union, Tuple
 
 from metagpt.actions import Action
+from metagpt.logs import logger
 from metagpt.prompts.ml_engineer import (
     TOOL_RECOMMENDATION_PROMPT,
     SELECT_FUNCTION_TOOLS,
@@ -174,6 +174,7 @@ class WriteCodeWithTools(BaseWriteAnalysisCode):
             task, task_guide, available_tools
         )
         recommend_tools, tool_catalog = self._parse_recommend_tools(task_type, recommend_tools)
+        logger.info(f"Recommended tools for every steps: {recommend_tools}")
 
         special_prompt = ML_SPECIFIC_PROMPT.get(task_type, "")
         module_name = ML_MODULE_MAP[task_type]