Austin/QChatGPT
1
0
Fork 0
mirror of https://github.com/RockChinQ/QChatGPT.git synced 2024-11-15 19:22:24 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: 消息忽略规则失效 (#854)

Browse Source
This commit is contained in:
RockChinQ 2024-08-01 23:01:28 +08:00
parent 47e281fb61
commit c9c8603ccc
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
pkg/pipeline/cntfilter/cntfilter.py
View File

@ -1,6 +1,8 @@
from __future__ import annotations from __future__ import annotations
import mirai import mirai
import mirai.models
import mirai.models.message
from ...core import app from ...core import app
@ -63,6 +65,7 @@ class ContentFilterStage(stage.PipelineStage):
"""请求llm前处理消息 """请求llm前处理消息
只要有一个不通过就不放行，只放行 PASS 的消息 只要有一个不通过就不放行，只放行 PASS 的消息
""" """
if not self.ap.pipeline_cfg.data['income-msg-check']: if not self.ap.pipeline_cfg.data['income-msg-check']:
return entities.StageProcessResult( return entities.StageProcessResult(
result_type=entities.ResultType.CONTINUE, result_type=entities.ResultType.CONTINUE,
@ -145,11 +148,13 @@ class ContentFilterStage(stage.PipelineStage):
contain_non_text = False contain_non_text = False
text_components = [mirai.Plain, mirai.models.message.Source]
for me in query.message_chain: for me in query.message_chain:
if not isinstance(me, mirai.Plain): if type(me) not in text_components:
contain_non_text = True contain_non_text = True
break break
if contain_non_text: if contain_non_text:
self.ap.logger.debug(f"消息中包含非文本消息，跳过内容过滤器检查。") self.ap.logger.debug(f"消息中包含非文本消息，跳过内容过滤器检查。")
return entities.StageProcessResult( return entities.StageProcessResult(