feat: two-stage status messages: 获取邮件中 -> AI思考中

This commit is contained in:
2026-07-13 08:15:25 +08:00
parent 85b02651b9
commit 02b0941ed0
2 changed files with 12 additions and 10 deletions

View File

@@ -23,15 +23,18 @@ def poll_accounts(cfg: Config, skip_uids: set[bytes] | None = None) -> Generator
def ai_process(cfg: Config, acct_idx: int, mail: Email) -> Optional[dict]:
logger.info(f" 正在摘要: {mail.subject}")
# 先发思考中消息
thinking_msg_id = send_thinking(cfg.telegram, cfg.telegram.chat_id)
logger.info(f" 思考中消息已发送: msg_id={thinking_msg_id}")
# 阶段1显示获取邮件中
msg_id = send_thinking(cfg.telegram, cfg.telegram.chat_id, "📥 获取邮件中...")
logger.info(f" 获取邮件中消息已发送: msg_id={msg_id}")
# 阶段2AI 处理,更新为思考中
edit_message(cfg.telegram, cfg.telegram.chat_id, msg_id, "🤖 AI思考中...")
summary = summarize_email(cfg.ai, mail.recipient, mail.subject, mail.sender, mail.body, mail.account_email)
summary["recipient"] = mail.recipient
if "@" not in summary.get("sender", ""):
summary["sender"] = mail.sender
text = format_summary(summary, mail.account_email)
# 阶段3编辑为最终结果
return {
"text": text,
"data": summary,
@@ -43,7 +46,7 @@ def ai_process(cfg: Config, acct_idx: int, mail: Email) -> Optional[dict]:
"account_email": mail.account_email,
"uid": mail.uid,
"folder": mail.folder,
"thinking_msg_id": thinking_msg_id,
"thinking_msg_id": msg_id,
}