feat: 长消息处理策略可扩展性

This commit is contained in:
RockChinQ
2024-03-08 20:31:22 +08:00
parent 22cb8a6a06
commit 4347ddd42a
8 changed files with 39 additions and 11 deletions
+3 -3
View File
@@ -25,14 +25,14 @@ class ContentFilterStage(stage.PipelineStage):
async def initialize(self):
filters_required = [
"ContentIgnore"
"content-filter"
]
if self.ap.pipeline_cfg.data['check-sensitive-words']:
filters_required.append("BanWordFilter")
filters_required.append("ban-word-filter")
if self.ap.pipeline_cfg.data['baidu-cloud-examine']['enable']:
filters_required.append("BaiduCloudExamine")
filters_required.append("baidu-cloud-examine")
for filter in filter_model.preregistered_filters:
if filter.name in filters_required:
@@ -10,7 +10,7 @@ BAIDU_EXAMINE_URL = "https://aip.baidubce.com/rest/2.0/solution/v1/text_censor/v
BAIDU_EXAMINE_TOKEN_URL = "https://aip.baidubce.com/oauth/2.0/token"
@filter_model.filter_class("BaiduCloudExamine")
@filter_model.filter_class("baidu-cloud-examine")
class BaiduCloudExamine(filter_model.ContentFilter):
"""百度云内容审核"""
+1 -1
View File
@@ -6,7 +6,7 @@ from .. import entities
from ....config import manager as cfg_mgr
@filter_model.filter_class("BanWordFilter")
@filter_model.filter_class("ban-word-filter")
class BanWordFilter(filter_model.ContentFilter):
"""根据内容禁言"""
+1 -1
View File
@@ -5,7 +5,7 @@ from .. import entities
from .. import filter as filter_model
@filter_model.filter_class("ContentIgnore")
@filter_model.filter_class("content-ignore")
class ContentIgnore(filter_model.ContentFilter):
"""根据内容忽略消息"""