apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-05-15 11:02:36 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

add log print

Browse source
This commit is contained in:
lidanyang 2023-12-01 11:59:28 +08:00
parent aad201e06f
commit 35e8a501c5
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
metagpt/actions/write_analysis_code.py
View file

@ -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]