Trying to add memory retention
This commit is contained in:
parent
8c93e45c2f
commit
814102e921
@ -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
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user