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

mv tool_type def

Browse source
This commit is contained in:
yzlin 2024-02-05 00:05:16 +08:00
parent ca6749b5f1
commit a609946029
1 changed files with 35 additions and 60 deletions
Download patch file Download diff file Expand all files Collapse all files

95
metagpt/tools/tool_types.py
View file

@ -9,68 +9,43 @@ from metagpt.prompts.tool_types import (
)
from metagpt.tools.tool_data_type import ToolType
Eda = ToolType(name="eda", desc="For performing exploratory data analysis")
DataPreprocess = ToolType(
name="data_preprocess",
desc="Only for changing value inplace.",
usage_prompt=DATA_PREPROCESS_PROMPT,
)
FeatureEngineering = ToolType(
name="feature_engineering",
desc="Only for creating new columns for input data.",
usage_prompt=FEATURE_ENGINEERING_PROMPT,
)
ModelTrain = ToolType(
name="model_train",
desc="Only for training model.",
usage_prompt=MODEL_TRAIN_PROMPT,
)
ModelEvaluate = ToolType(
name="model_evaluate",
desc="Only for evaluating model.",
usage_prompt=MODEL_EVALUATE_PROMPT,
)
StableDiffusion = ToolType(
name="stable_diffusion",
desc="Related to text2image, image2image using stable diffusion model.",
)
Image2Webpage = ToolType(
name="image2webpage",
desc="For converting image into webpage code.",
usage_prompt=IMAGE2WEBPAGE_PROMPT,
)
WebScraping = ToolType(
name="web_scraping",
desc="For scraping data from web pages.",
)
Other = ToolType(name="other", desc="Any tools not in the defined categories")
class ToolTypes(Enum):
EDA = Eda
DATA_PREPROCESS = DataPreprocess
FEATURE_ENGINEERING = FeatureEngineering
MODEL_TRAIN = ModelTrain
MODEL_EVALUATE = ModelEvaluate
STABLE_DIFFUSION = StableDiffusion
IMAGE2WEBPAGE = Image2Webpage
WEBSCRAPING = WebScraping
OTHER = Other
EDA = ToolType(name="eda", desc="For performing exploratory data analysis")
DATA_PREPROCESS = ToolType(
name="data_preprocess",
desc="Only for changing value inplace.",
usage_prompt=DATA_PREPROCESS_PROMPT,
)
FEATURE_ENGINEERING = ToolType(
name="feature_engineering",
desc="Only for creating new columns for input data.",
usage_prompt=FEATURE_ENGINEERING_PROMPT,
)
MODEL_TRAIN = ToolType(
name="model_train",
desc="Only for training model.",
usage_prompt=MODEL_TRAIN_PROMPT,
)
MODEL_EVALUATE = ToolType(
name="model_evaluate",
desc="Only for evaluating model.",
usage_prompt=MODEL_EVALUATE_PROMPT,
)
STABLE_DIFFUSION = ToolType(
name="stable_diffusion",
desc="Related to text2image, image2image using stable diffusion model.",
)
IMAGE2WEBPAGE = ToolType(
name="image2webpage",
desc="For converting image into webpage code.",
usage_prompt=IMAGE2WEBPAGE_PROMPT,
)
WEBSCRAPING = ToolType(
name="web_scraping",
desc="For scraping data from web pages.",
)
OTHER = ToolType(name="other", desc="Any tools not in the defined categories")
def __missing__(self, key):
return self.OTHER