Kanium-PUBLIC/KaniumCogs
5
0
Fork 0
Code Issues 6 Pull Requests Packages Projects 1 Releases Wiki Activity

Trying to trim memory properly

Browse Source
This commit is contained in:
AllfatherHatt 2025-02-21 00:38:08 +01:00
parent 814102e921
commit 2724f7cee6
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
reginaldCog/reginald.py
View File

@ -94,9 +94,6 @@ class ReginaldCog(commands.Cog):
formatted_messages.append({"role": "user", "content": f"{user_name}: {prompt}"}) formatted_messages.append({"role": "user", "content": f"{user_name}: {prompt}"})
response_text = await self.generate_response(api_key, formatted_messages) response_text = await self.generate_response(api_key, formatted_messages)
memory.append({"user": user_name, "content": prompt})
memory.append({"user": "Reginald", "content": response_text})
if len(memory) > self.short_term_memory_limit: if len(memory) > self.short_term_memory_limit:
summary = await self.summarize_memory(memory) summary = await self.summarize_memory(memory)
@ -118,7 +115,9 @@ class ReginaldCog(commands.Cog):
keep_count = max(1, int(len(memory) * retention_ratio)) # Keep at least 1 message keep_count = max(1, int(len(memory) * retention_ratio)) # Keep at least 1 message
memory = memory[-keep_count:] # Remove oldest 75%, keep recent memory = memory[-keep_count:] # Remove oldest 75%, keep recent
memory.append({"user": user_name, "content": prompt})
memory.append({"user": "Reginald", "content": response_text})
short_memory[channel_id] = memory short_memory[channel_id] = memory
await ctx.send(response_text[:2000]) await ctx.send(response_text[:2000])