From 77eacec1b85fdf6e33684dd3c4c5a3d3555dec17 Mon Sep 17 00:00:00 2001
From: yzlin <yzlin@fuzhi.ai>
Date: Wed, 13 Mar 2024 16:29:28 +0800
Subject: [PATCH] recover legacy task type

---
 metagpt/strategy/task_type.py         | 14 ++++++++++++++
 metagpt/tools/libs/email_login.py     |  2 +-
 metagpt/tools/libs/gpt_v_generator.py |  2 +-
 3 files changed, 16 insertions(+), 2 deletions(-)

diff --git a/metagpt/strategy/task_type.py b/metagpt/strategy/task_type.py
index 9eeeb79ce..7c88817cc 100644
--- a/metagpt/strategy/task_type.py
+++ b/metagpt/strategy/task_type.py
@@ -54,6 +54,20 @@ class TaskType(Enum):
     )
     OTHER = TaskTypeDef(name="other", desc="Any tasks not in the defined categories")
 
+    # Legacy TaskType to support tool recommendation using type match. You don't need to define task types if you have no human priors to inject.
+    TEXT2IMAGE = TaskTypeDef(
+        name="text2image",
+        desc="Related to text2image, image2image using stable diffusion model.",
+    )
+    WEBSCRAPING = TaskTypeDef(
+        name="web scraping",
+        desc="For scraping data from web pages.",
+    )
+    EMAIL_LOGIN = TaskTypeDef(
+        name="email login",
+        desc="For logging to an email.",
+    )
+
     @property
     def type_name(self):
         return self.value.name
diff --git a/metagpt/tools/libs/email_login.py b/metagpt/tools/libs/email_login.py
index 757ac2b87..32626ac55 100644
--- a/metagpt/tools/libs/email_login.py
+++ b/metagpt/tools/libs/email_login.py
@@ -23,7 +23,7 @@ IMAP_SERVERS = {
 }
 
 
-@register_tool()
+@register_tool(tags=["email login"])
 def email_login_imap(email_address, email_password):
     """
     Use imap_tools package to log in to your email (the email that supports IMAP protocol) to verify and return the account object.
diff --git a/metagpt/tools/libs/gpt_v_generator.py b/metagpt/tools/libs/gpt_v_generator.py
index 0e9f34770..4eba3d5ee 100644
--- a/metagpt/tools/libs/gpt_v_generator.py
+++ b/metagpt/tools/libs/gpt_v_generator.py
@@ -28,7 +28,7 @@ As the design pays tribute to large companies, sometimes it is normal for some c
 Now, please generate the corresponding webpage code including HTML, CSS and JavaScript:"""
 
 
-@register_tool(include_functions=["__init__", "generate_webpages", "save_webpages"])
+@register_tool(tags=["image2webpage"], include_functions=["__init__", "generate_webpages", "save_webpages"])
 class GPTvGenerator:
     """Class for generating webpage code from a given webpage screenshot.