Fjerkroa/discord_bot
1
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Save history on short patch.

Browse Source
This commit is contained in:
OK 2023-03-27 14:13:10 +02:00
parent 2cd66b55ee
commit 2ee2b092d6
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
fjerkroa_bot/ai_responder.py
View File

@ -101,7 +101,11 @@ class AIResponder(object):
user_ma = re.match(user_re, message.user) user_ma = re.match(user_re, message.user)
if chan_ma and user_ma: if chan_ma and user_ma:
self.history.append({"role": "user", "content": str(message)}) self.history.append({"role": "user", "content": str(message)})
self.history = self.history[-limit:] if len(self.history) > limit:
self.history = self.history[-limit:]
if self.history_file is not None:
with open(self.history_file, 'wb') as fd:
pickle.dump(self.history, fd)
return True return True
return False return False
@ -119,6 +123,7 @@ class AIResponder(object):
return answer, limit return answer, limit
except openai.error.InvalidRequestError as err: except openai.error.InvalidRequestError as err:
if 'maximum context length is' in str(err) and limit > 4: if 'maximum context length is' in str(err) and limit > 4:
logging.warning(f"context length exceeded, reduce the limit {limit}: {str(err)}")
limit -= 1 limit -= 1
return None, limit return None, limit
raise err raise err