awesome-bookmarks/shell_gpt
3
1
Fork 0
mirror of https://github.com/TheR1D/shell_gpt.git synced 2026-06-02 06:14:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

Message cache length fix

Browse Source
This commit is contained in:
Farkhod Sadykov
2025-04-08 12:19:56 +02:00
parent 859c97915a
commit f66f65a68c
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
sgpt/handlers/chat_handler.py
+7 -2
View File
@@ -71,8 +71,13 @@ class ChatSession:
def _write(self, messages: List[Dict[str, str]], chat_id: str) -> None:
file_path = self.storage_path / chat_id
message_role_item = messages.pop(0)
json.dump([message_role_item] + messages[-self.length :], file_path.open("w"))
if len(messages) > self.length:
message_role_item = messages[0]
json.dump(
[message_role_item] + messages[-self.length :], file_path.open("w")
)
return
json.dump(messages[-self.length :], file_path.open("w"))
def invalidate(self, chat_id: str) -> None:
file_path = self.storage_path / chat_id