apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-08 15:05:17 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: Add tags to enable custom message classification

Browse source
This commit is contained in:
莘权 马 2023-08-22 11:10:08 +08:00
parent 2adcefc298
commit bc97b709bb
1 changed files with 10 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

10
metagpt/schema.py
View file

@ -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):