feat: is dirty

This commit is contained in:
莘权 马 2023-09-04 15:39:11 +08:00
parent 0a494171fa
commit fb6bb4b692

View file

@ -35,14 +35,17 @@ class BrainMemory(pydantic.BaseModel):
historical_summary_fingerprint: List[str] = []
historical_summary: str = ""
last_history_id: str = ""
is_dirty: bool = False
def add_talk(self, msg: Message):
msg.add_tag(MessageType.Talk.value)
self.history.append(msg.dict())
self.is_dirty = True
def add_answer(self, msg: Message):
msg.add_tag(MessageType.Answer.value)
self.history.append(msg.dict())
self.is_dirty = True
def get_knowledge(self) -> str:
texts = [Message(**m).content for m in self.knowledge]