From bc97b709bb17e7d25cc48f49632648ff5cb32624 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=8E=98=E6=9D=83=20=E9=A9=AC?= <mashenquan@fuzhi.ai>
Date: Tue, 22 Aug 2023 11:10:08 +0800
Subject: [PATCH] feat: Add tags to enable custom message classification

---
 metagpt/schema.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/metagpt/schema.py b/metagpt/schema.py
index 2e4a6c62f..140f207c8 100644
--- a/metagpt/schema.py
+++ b/metagpt/schema.py
@@ -45,6 +45,16 @@ class Message:
             "content": self.content
         }
 
+    def add_tag(self, tag):
+        if self.tags is None:
+            self.tags = set()
+        self.tags.add(tag)
+
+    def remove_tag(self, tag):
+        if self.tags is None:
+            return
+        self.tags.remove(tag)
+
 
 @dataclass
 class UserMessage(Message):