Merge pull request #31 from iorisa/fixbug/last_history_id

fixbug: last history id
This commit is contained in:
Justin-ZL 2023-09-04 21:50:06 +08:00 committed by GitHub
commit 847a79d822
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -115,8 +115,9 @@ class BrainMemory(pydantic.BaseModel):
def add_history(self, msg: Message):
if msg.id:
if int(msg.id) < int(self.last_history_id):
if self.to_int(msg.id, 0) < self.to_int(self.last_history_id, -1):
return
self.last_history_id = str(self.to_int(msg.id, 0))
self.history.append(msg.dict())
self.is_dirty = True
@ -125,3 +126,10 @@ class BrainMemory(pydantic.BaseModel):
if m.get("content") == text:
return True
return False
@staticmethod
def to_int(v, default_value):
try:
return int(v)
except:
return default_value