diff --git a/metagpt/prompts/tool_types.py b/metagpt/prompts/tool_types.py index f27fbea99..8728d22f3 100644 --- a/metagpt/prompts/tool_types.py +++ b/metagpt/prompts/tool_types.py @@ -1,3 +1,10 @@ +# Prompt for using tools of "eda" type +EDA_PROMPT = """ +The current task is about exploratory data analysis, please note the following: +- Distinguish column types with `select_dtypes` for tailored analysis and visualization, such as correlation. +- Remember to `import numpy as np` before using Numpy functions. +""" + # Prompt for using tools of "data_preprocess" type DATA_PREPROCESS_PROMPT = """ The current task is about data preprocessing, please note the following: diff --git a/metagpt/tools/tool_type.py b/metagpt/tools/tool_type.py index 6fa971c56..7f3f132a6 100644 --- a/metagpt/tools/tool_type.py +++ b/metagpt/tools/tool_type.py @@ -1,6 +1,7 @@ from enum import Enum from metagpt.prompts.tool_types import ( + EDA_PROMPT, DATA_PREPROCESS_PROMPT, FEATURE_ENGINEERING_PROMPT, IMAGE2WEBPAGE_PROMPT, @@ -11,7 +12,11 @@ from metagpt.tools.tool_data_type import ToolTypeDef class ToolType(Enum): - EDA = ToolTypeDef(name="eda", desc="For performing exploratory data analysis") + EDA = ToolTypeDef( + name="eda", + desc="For performing exploratory data analysis", + usage_prompt=EDA_PROMPT, + ) DATA_PREPROCESS = ToolTypeDef( name="data_preprocess", desc="Only for changing value inplace.",