Update main.py

Author: lingyun14beta

astrbot/builtin_stars/astrbot/main.py

@@ -6,7 +6,7 @@
 from astrbot.api.provider import LLMResponse, ProviderRequest
 from astrbot.core import logger
 
-from .constants import LTM_ACTIVE_REPLY_KEY
+from .constants import LTM_ACTIVE_REPLY_IN_PROGRESS_KEY, LTM_ACTIVE_REPLY_KEY
 from .long_term_memory import LongTermMemory
 
 
@@ -99,7 +99,7 @@ async def decorate_llm_req(
             # so on_llm_response can correctly identify the active reply response
             active_reply_req_id = event.get_extra(LTM_ACTIVE_REPLY_KEY, None)
             if active_reply_req_id is not None and id(req) == active_reply_req_id:
-                event.set_extra("_ltm_active_reply_in_progress", True)
+                event.set_extra(LTM_ACTIVE_REPLY_IN_PROGRESS_KEY, True)
             try:
                 await self.ltm.on_req_llm(event, req)
             except BaseException as e:
@@ -112,7 +112,7 @@ async def record_llm_resp_to_ltm(
         """在 LLM 响应后记录对话"""
         if self.ltm and self.ltm_enabled(event):
             # Skip recording if this response is from an active reply request
-            if event.get_extra("_ltm_active_reply_in_progress", False):
+            if event.get_extra(LTM_ACTIVE_REPLY_IN_PROGRESS_KEY, False):
                 return
             # Only record if group_icl_enable is on, to keep session_chats consistent
             # (handle_message is also guarded by group_icl_enable)
@@ -136,4 +136,4 @@ async def after_message_sent(self, event: AstrMessageEvent) -> None:
                 logger.error(f"ltm: {e}")
         # 清除主动回复标记，避免 event 被复用时意外影响后续流程
         event.set_extra(LTM_ACTIVE_REPLY_KEY, None)
-        event.set_extra("_ltm_active_reply_in_progress", False)
+        event.set_extra(LTM_ACTIVE_REPLY_IN_PROGRESS_KEY, False)