From f51df40ebb4cd5af6263c613e2ffa898a3ae1ccd Mon Sep 17 00:00:00 2001
From: lidanyang <lidanyang@fuzhi.ai>
Date: Wed, 7 Feb 2024 12:02:46 +0800
Subject: [PATCH] add eda prompt

---
 metagpt/prompts/tool_types.py | 7 +++++++
 metagpt/tools/tool_type.py    | 7 ++++++-
 2 files changed, 13 insertions(+), 1 deletion(-)

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.",