fix: improve content safety check handling for at or wake commands

2025-02-12 23:42:32 +08:00
parent 9cd243da47
commit 7fe3b97d00
1 changed files with 3 additions and 2 deletions
@@ -22,8 +22,9 @@ class ContentSafetyCheckStage(Stage):
        text = check_text if check_text else event.get_message_str()
        ok, info = self.strategy_selector.check(text)
        if not ok:
-            event.set_result(MessageEventResult().message("你的消息或者大模型的响应中包含不适当的内容，已被屏蔽。"))
-            yield 
+            if event.is_at_or_wake_command:
+                event.set_result(MessageEventResult().message("你的消息或者大模型的响应中包含不适当的内容，已被屏蔽。"))
+                yield 
            event.stop_event()
            logger.info(f"内容安全检查不通过，原因：{info}")
            return