fixbug: brain memory serialize

This commit is contained in:
莘权 马 2023-08-27 15:11:28 +08:00
parent 903e89cec3
commit 3e9151e52e
4 changed files with 90 additions and 16 deletions

View file

@ -70,6 +70,22 @@ class Message:
def is_contain(self, tag):
return self.is_contain_tags([tag])
def dict(self):
"""pydantic-like `dict` function"""
full = {
"instruct_content": self.instruct_content,
"cause_by": self.cause_by,
"sent_from": self.sent_from,
"send_to": self.send_to,
"tags": self.tags
}
m = {"content": self.content}
for k, v in full.items():
if v:
m[k] = v
return m
@dataclass
class UserMessage(Message):
@ -101,7 +117,6 @@ class AIMessage(Message):
super().__init__(content, 'assistant')
if __name__ == '__main__':
test_content = 'test_message'
msgs = [