feat: Add trigger count management to LoreParser and reset functionality in LorePlugin

Author: Raven95676

core/parser.py

@@ -41,6 +41,7 @@ class LoreParser:
         "_random_handler",
         "_logic_handler",
         "_save_handler",
+        "trigger_count",
     )
 
     def __init__(self, lorebook: dict[str, Any], scan_depth: int = 1):
@@ -120,6 +121,9 @@ def __init__(self, lorebook: dict[str, Any], scan_depth: int = 1):
         self._logic_handler = LogicHandler(self)
         self._save_handler = SaveHandler(self)
 
+        # 初始化触发器计数器
+        self.trigger_count: dict[str, int] = {}
+
     def __str__(self) -> str:
         """返回解析器的字符串表示"""
         return f"LoreParser(variables={self._vars},triggers={self._triggers},authors_notes={self._notes})"
@@ -370,6 +374,10 @@ def _process_trigger(
 
         return not trigger.block if can_trigger else True
 
+    def reset_trigger_count(self) -> None:
+        """重置所有触发器的计数器"""
+        self.trigger_count.clear()
+
     def process_chat(self) -> LoreResult:
         """处理聊天消息，应用所有适用的触发器和注释
 
@@ -381,7 +389,6 @@ def process_chat(self) -> LoreResult:
         """
         result = LoreResult()
         triged_lis: set[str] = set()
-        trigger_count: dict[str, int] = {}
         # 更新真实世界的空闲时间
         self._real_idle["before"] = self._real_idle["after"]
         self._real_idle["after"] = datetime.now()
@@ -390,7 +397,7 @@ def process_chat(self) -> LoreResult:
         for trigger in self._triggers:
             # 检查触发次数限制
             if trigger.max_trig != -1:
-                current_count = trigger_count.get(trigger.name, 0)
+                current_count = self.trigger_count.get(trigger.name, 0)
                 if current_count >= trigger.max_trig:
                     continue  # 跳过已达到最大触发次数的触发器
 
@@ -404,7 +411,9 @@ def process_chat(self) -> LoreResult:
             # 处理当前触发器
             if self._can_trigger(trigger, self.messages):
                 # 增加触发次数计数
-                trigger_count[trigger.name] = trigger_count.get(trigger.name, 0) + 1
+                self.trigger_count[trigger.name] = (
+                    self.trigger_count.get(trigger.name, 0) + 1
+                )
 
                 # 处理触发器, 如果返回 False，则停止处理下一个触发器
                 if not self._process_trigger(

main.py

@@ -15,7 +15,7 @@
 from .core.parser import LoreParser  # type: ignore
 
 
-@register("astrbot_plugin_lorebook_lite", "Raven95676", "lorebook插件", "0.1.7")
+@register("astrbot_plugin_lorebook_lite", "Raven95676", "lorebook插件", "0.1.8")
 class LorePlugin(Star):
     """Lorebook插件，用于根据预设规则处理聊天内容并修改LLM请求"""
 
@@ -227,6 +227,10 @@ async def on_llm_res(self, event: AstrMessageEvent, response: LLMResponse):
             msg_clean = " ".join(msg.split())
             self.lore_sessions[session_key].messages.append(msg_clean)
 
+        # 重置触发器计数器
+        if session_key in self.lore_sessions:
+            self.lore_sessions[session_key].reset_trigger_count()
+
         # 清除结果缓存并还原人格
         self._clear_session_results(session_key)
         await self._restore_persona(umo)

metadata.yaml

@@ -1,6 +1,6 @@
 name: astrbot-plugin-lorebook-lite
 desc: Astrbot 轻量世界书插件
 help: 无
-version: v0.1.7
+version: v0.1.8
 author: Raven95676
 repo: https://github.com/Raven95676/astrbot_plugin_lorebook_lite