fixbug: last_talk

This commit is contained in:
莘权 马 2023-09-05 00:38:31 +08:00
parent 18a65470f0
commit 22dbe3b224
2 changed files with 3 additions and 3 deletions

View file

@ -34,7 +34,7 @@ class BrainMemory(pydantic.BaseModel):
historical_summary: str = ""
last_history_id: str = ""
is_dirty: bool = False
last_talk: str = ""
last_talk: str = None
def add_talk(self, msg: Message):
msg.add_tag(MessageType.Talk.value)
@ -127,5 +127,5 @@ class BrainMemory(pydantic.BaseModel):
def pop_last_talk(self):
v = self.last_talk
self.last_talk = ""
self.last_talk = None
return v

View file

@ -120,7 +120,7 @@ class Assistant(Role):
async def refine_memory(self) -> str:
history_text = self.memory.history_text
last_talk = self.memory.last_talk
last_talk = self.memory.pop_last_talk()
if last_talk is None: # No user feedback, unsure if past conversation is finished.
return None
if history_text == "":