TheLovinator/ANewDawn
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add conversation memory reset command

Browse source 
Introduces a new /reset command to allow authorized users to reset the conversation memory for a channel. Also adds a new_conversation option to the /ask command to start a fresh conversation, and implements the reset_memory function in misc.py.
This commit is contained in:
Joakim Hellsén 2025-09-26 01:06:18 +02:00
commit eec1ed4f59
2 changed files with 50 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

14
misc.py
View file

@ -57,6 +57,20 @@ agent: Agent[BotDependencies, str] = Agent(
)
def reset_memory(channel_id: str) -> None:
"""Reset the conversation memory for a specific channel.
Args:
channel_id (str): The ID of the channel to reset memory for.
"""
if channel_id in recent_messages:
del recent_messages[channel_id]
logger.info("Reset memory for channel %s", channel_id)
if channel_id in last_trigger_time:
del last_trigger_time[channel_id]
logger.info("Reset trigger times for channel %s", channel_id)
def _message_text_length(msg: ModelRequest | ModelResponse) -> int:
"""Compute the total text length of all text parts in a message.