s4u的一些修改,包括动作适配

This commit is contained in:
SengokuCola
2025-07-22 20:36:53 +08:00
parent e227a0158d
commit 75022b5d10
13 changed files with 148 additions and 53 deletions

View File

@@ -16,7 +16,7 @@ from src.chat.utils.prompt_builder import Prompt, global_prompt_manager
from src.plugin_system.core.component_registry import component_registry # 导入新插件系统
from src.plugin_system.base.base_command import BaseCommand
from src.mais4u.mais4u_chat.s4u_msg_processor import S4UMessageProcessor
from src.llm_models.utils_model import LLMRequest
# 定义日志配置
@@ -157,6 +157,40 @@ class ChatBot:
await self.s4u_message_processor.process_message(message)
return
async def hanle_notice_message(self, message: MessageRecv):
if message.message_info.message_id == "notice":
logger.info(f"收到notice消息暂时不支持处理")
return True
async def check_ban_content(self,message:MessageRecv):
content = message.processed_plain_text
model = LLMRequest(
model=global_config.model.utils_small,
temperature=0.3,
request_type="text_ban",
)
if content.startswith("[picid:"):
return False
prompt = f"""
你是一个直播内容审核者,请判断以下内容是否符合直播内容规范,如果符合,请输出"符合",如果不符合,请输出"不符合"以及理由
如果存在反动,敏感内容,请输出"不符合"
内容:{content}
现在请输出:
"""
response, (reasoning_content, model_name) = await model.generate_response_async(prompt=prompt)
logger.info(f"response: {response}")
logger.info(f"reasoning_content: {reasoning_content}")
if "不符合" in response:
return True
else:
return False
async def message_process(self, message_data: Dict[str, Any]) -> None:
@@ -174,6 +208,8 @@ class ChatBot:
- 性能计时
"""
try:
# 确保所有任务已启动
await self._ensure_started()
@@ -194,6 +230,11 @@ class ChatBot:
# print(message_data)
# logger.debug(str(message_data))
message = MessageRecv(message_data)
if await self.hanle_notice_message(message):
return
group_info = message.message_info.group_info
user_info = message.message_info.user_info
if message.message_info.additional_config:
@@ -214,6 +255,11 @@ class ChatBot:
# 处理消息内容,生成纯文本
await message.process()
if await self.check_ban_content(message):
logger.warning(f"检测到消息中含有违法,色情,暴力,反动,敏感内容,消息内容:{message.processed_plain_text},发送者:{message.message_info.user_info.user_nickname}")
return
# 过滤检查
if _check_ban_words(message.processed_plain_text, chat, user_info) or _check_ban_regex( # type: ignore