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:

9
plugins/xiuren_image/main.py
View File

@@ -58,7 +58,7 @@ class XiurenImagePlugin(MessagePluginInterface):
self._commands = self._config.get("XiurenImage", {}).get("command", ["图来", "秀人"])
self.command_format = self._config.get("XiurenImage", {}).get("command-format", "图来")
self.enable = self._config.get("XiurenImage", {}).get("enable", True)
self.image_folder =self._config.get("XiurenImage", {}).get("image_folder")
self.image_folder = self._config.get("XiurenImage", {}).get("image_folder")
# 检查图片文件夹是否存在
if not os.path.exists(self.image_folder):
self.LOG.warning(f"图片文件夹不存在: {self.image_folder}")
@@ -116,8 +116,11 @@ class XiurenImagePlugin(MessagePluginInterface):
# 发送图片
result = await bot.send_image_message((roomid if roomid else sender), Path(pic_path))
self.LOG.info(f"发送图片结果: {result}")
client_msg_id, create_time, new_msg_id = await bot.send_image_message((roomid if roomid else sender),
Path(pic_path))
revoke.add_message_to_revoke(roomid, client_msg_id, create_time, new_msg_id, 90)
self.LOG.info(
f"发送图片结果,client_msg_id= {client_msg_id},create_time={create_time},new_msg_id={new_msg_id}")
return True, "发送成功"
except Exception as e: