liuwei_x_x/abot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

加入自动撤回，防止骚扰

Browse Source
This commit is contained in:
liuwei
2025-05-08 09:24:20 +08:00
parent 0e856c9a0f
commit e18f391cab
2 changed files with 16 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
plugins/dify/main.py
View File

@@ -137,7 +137,7 @@ class DifyPlugin(MessagePluginInterface):
query = re.sub(r"@.*?[\u2005|\s]", "", content).strip()
if not query:
await bot.send_text_message(roomid, "请在@我的同时提供问题内容", sender)
await bot.send_at_message(roomid, "请在@我的同时提供问题内容", sender)
return False, "没有提供问题内容"
# self.message_util.send_text_msg("⏳AI 正在加油，请稍候… 😊", roomid, sender)
@@ -156,7 +156,7 @@ class DifyPlugin(MessagePluginInterface):
self.message_util.send_file(response, roomid)
else:
# 如果是普通文本，使用发送文本方法
await bot.send_text_message((roomid if roomid else sender), response, sender)
await bot.send_at_message((roomid if roomid else sender), response, sender)
return True, "发送成功"
else:
await bot.send_text_message((roomid if roomid else sender), "❌未能获取到回复，请稍后再试", sender)
@@ -204,15 +204,18 @@ class DifyPlugin(MessagePluginInterface):
# 判断是否为本地文件路径
if os.path.isfile(response):
# 如果是文件路径，使用发送文件方法
self.message_util.send_file(response, (roomid if roomid else sender))
await bot.send_image_message((roomid if roomid else sender), response)
else:
# 如果是普通文本，使用发送文本方法
await bot.send_text_message((roomid if roomid else sender), response,
sender if roomid else "")
await bot.send_at_message((roomid if roomid else sender), response,
sender if roomid else "")
return True, "发送成功"
else:
await bot.send_text_message((roomid if roomid else sender), "❌未能获取到回复，请稍后再试",
sender if roomid else "")
client_msg_id, create_time, new_msg_id = await bot.send_text_message((roomid if roomid else sender),
"❌未能获取到回复，请稍后再试",
sender if roomid else "")
revoke.add_message_to_revoke(roomid, client_msg_id, create_time, new_msg_id, 5)
return False, "未获取到回复"
except Exception as e: