earthjasonlin/ai-mail-bot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: also delete user's message when cleaning up reply/hint flow

Browse Source
This commit is contained in:
Zichao Lin
2026-07-02 20:56:30 +08:00
parent 08a1a32367
commit be412168bb
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/tg_bot.py
View File

@@ -320,6 +320,7 @@ def _handle_message(tg_cfg: TelegramConfig, cfg: Config, msg: dict):
if not conv: if not conv:
return return
user_msg_id = msg["message_id"]
chat_id_str = str(chat_id) chat_id_str = str(chat_id)
prompt_msg_id = conv.get("prompt_msg_id") prompt_msg_id = conv.get("prompt_msg_id")
@@ -327,10 +328,12 @@ def _handle_message(tg_cfg: TelegramConfig, cfg: Config, msg: dict):
ctx = _email_contexts.get(conv["summary_msg_id"]) ctx = _email_contexts.get(conv["summary_msg_id"])
if not ctx: if not ctx:
_conversations.pop(chat_id, None) _conversations.pop(chat_id, None)
delete_message(tg_cfg, chat_id_str, user_msg_id)
if prompt_msg_id: if prompt_msg_id:
delete_message(tg_cfg, chat_id_str, prompt_msg_id) delete_message(tg_cfg, chat_id_str, prompt_msg_id)
return return
delete_message(tg_cfg, chat_id_str, user_msg_id)
if prompt_msg_id: if prompt_msg_id:
delete_message(tg_cfg, chat_id_str, prompt_msg_id) delete_message(tg_cfg, chat_id_str, prompt_msg_id)
_do_send_reply(tg_cfg, cfg, chat_id, conv["summary_msg_id"], ctx, text) _do_send_reply(tg_cfg, cfg, chat_id, conv["summary_msg_id"], ctx, text)
@@ -341,10 +344,12 @@ def _handle_message(tg_cfg: TelegramConfig, cfg: Config, msg: dict):
ctx = _email_contexts.get(conv["summary_msg_id"]) ctx = _email_contexts.get(conv["summary_msg_id"])
if not ctx: if not ctx:
_conversations.pop(chat_id, None) _conversations.pop(chat_id, None)
delete_message(tg_cfg, chat_id_str, user_msg_id)
if prompt_msg_id: if prompt_msg_id:
delete_message(tg_cfg, chat_id_str, prompt_msg_id) delete_message(tg_cfg, chat_id_str, prompt_msg_id)
return return
delete_message(tg_cfg, chat_id_str, user_msg_id)
if prompt_msg_id: if prompt_msg_id:
delete_message(tg_cfg, chat_id_str, prompt_msg_id) delete_message(tg_cfg, chat_id_str, prompt_msg_id)
try: try: