feat: 支持拒绝回复包含敏感词的提问 (#210)

pkg/qqbot/filter.py

@@ -23,6 +23,12 @@ class ReplyFilter:
             self.baidu_secret_key = config.baidu_secret_key
             self.inappropriate_message_tips = config.inappropriate_message_tips
 
+    def is_illegal(self, message: str) -> bool:
+        processed = self.process(message)
+        if processed != message:
+            return True
+        return False
+
     def process(self, message: str) -> str:
 
         # 本地关键词屏蔽

pkg/qqbot/process.py

@@ -66,6 +66,11 @@ def process_message(launcher_type: str, launcher_id: int, text_message: str, mes
                                                                                 result.mute_time_remaining))
             return reply
 
+    import config
+    if hasattr(config, 'income_msg_check') and config.income_msg_check:
+        if mgr.reply_filter.is_illegal(text_message):
+            return MessageChain(Plain("[bot] 你的提问中有不合适的内容, 请更换措辞~"))
+
     pkg.openai.session.get_session(session_name).acquire_response_lock()
 
     text_message = text_message.strip()