mirror of https://github.com/Mai-with-u/MaiBot.git
大概修复了@/提及相同名字导致关键词触发的bug
parent
63a2f1f7ee
commit
82bcf77b72
|
|
@ -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)
|
||||
|
|
|
|||
Loading…
Reference in New Issue