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

add eda prompt

Browse source
This commit is contained in:
lidanyang 2024-02-07 12:02:46 +08:00
parent 7fff68d564
commit f51df40ebb
2 changed files with 13 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
metagpt/tools/tool_type.py
View file

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