From 7fe3b97d00583b5ceca2ed693ba8d677f59daf5f Mon Sep 17 00:00:00 2001
From: Soulter <905617992@qq.com>
Date: Wed, 12 Feb 2025 23:42:32 +0800
Subject: [PATCH] fix: improve content safety check handling for at or wake
 commands

---
 astrbot/core/pipeline/content_safety_check/stage.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/astrbot/core/pipeline/content_safety_check/stage.py b/astrbot/core/pipeline/content_safety_check/stage.py
index 1cdd5a3ed..ffd7689a7 100644
--- a/astrbot/core/pipeline/content_safety_check/stage.py
+++ b/astrbot/core/pipeline/content_safety_check/stage.py
@@ -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