From 82bcf77b72a32190689de51792122c57729fed15 Mon Sep 17 00:00:00 2001 From: Bakadax Date: Wed, 26 Mar 2025 21:07:59 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A4=A7=E6=A6=82=E4=BF=AE=E5=A4=8D=E4=BA=86@/?= =?UTF-8?q?=E6=8F=90=E5=8F=8A=E7=9B=B8=E5=90=8C=E5=90=8D=E5=AD=97=E5=AF=BC?= =?UTF-8?q?=E8=87=B4=E5=85=B3=E9=94=AE=E8=AF=8D=E8=A7=A6=E5=8F=91=E7=9A=84?= =?UTF-8?q?bug?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/plugins/chat/utils.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/src/plugins/chat/utils.py b/src/plugins/chat/utils.py index fdc143aa..efe64f14 100644 --- a/src/plugins/chat/utils.py +++ b/src/plugins/chat/utils.py @@ -46,9 +46,13 @@ def is_mentioned_bot_in_message(message: MessageRecv) -> bool: """检查消息是否提到了机器人""" keywords = [global_config.BOT_NICKNAME] nicknames = global_config.BOT_ALIAS_NAMES + message_id = int(re.findall(r"\[CQ:reply,id=([0-9]*)\]", message.raw_message)[0]) + result = db.messages.find_one({"message_id": message_id}) message_content = re.sub(r'\[CQ:reply,[\s\S]*?\]','', message.raw_message) + message_content = re.sub(r'\[CQ:cq,[\s\S]*?\]','', message_content) + logger.exception(f"raw_message : {message.raw_message}") for keyword in keywords: - if f"[回复 {keyword} 的消息: " in message.processed_plain_text: + if (f"[回复 {keyword} 的消息: " in message.processed_plain_text) and result == None: return True if keyword in message_content: return True @@ -57,7 +61,6 @@ def is_mentioned_bot_in_message(message: MessageRecv) -> bool: return True return False - async def get_embedding(text, request_type="embedding"): """获取文本的embedding向量""" llm = LLM_request(model=global_config.embedding, request_type=request_type)