Trying to add memory retention

This commit is contained in:
AllfatherHatt 2025-02-21 00:30:35 +01:00
parent 8c93e45c2f
commit 814102e921

View File

@ -97,6 +97,7 @@ class ReginaldCog(commands.Cog):
memory.append({"user": user_name, "content": prompt})
memory.append({"user": "Reginald", "content": response_text})
if len(memory) > self.short_term_memory_limit:
summary = await self.summarize_memory(memory)
@ -112,7 +113,11 @@ class ReginaldCog(commands.Cog):
if len(mid_memory[channel_id]) > 10: # Keep only last 10 summaries
mid_memory[channel_id].pop(0)
memory = memory[-100:]
# ✅ Only prune short-term memory if a new summary was made
retention_ratio = 0.25 # Keep 25% of messages for immediate continuity
keep_count = max(1, int(len(memory) * retention_ratio)) # Keep at least 1 message
memory = memory[-keep_count:] # Remove oldest 75%, keep recent
short_memory[channel_id] = memory