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

加入dify 自动撤回

Browse Source
This commit is contained in:
liuwei
2025-05-06 09:55:33 +08:00
parent 64c2314adb
commit 76764bb30a
1 changed files with 13 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
plugins/dify/main.py
View File

@@ -12,6 +12,7 @@ from message_util import MessageUtil
from plugin_common.message_plugin_interface import MessagePluginInterface from plugin_common.message_plugin_interface import MessagePluginInterface
from plugin_common.plugin_interface import PluginStatus from plugin_common.plugin_interface import PluginStatus
from utils.decorator.plugin_decorators import plugin_stats_decorator from utils.decorator.plugin_decorators import plugin_stats_decorator
from utils.revoke.message_auto_revoke import MessageAutoRevoke
from utils.robot_cmd.robot_command import Feature, PermissionStatus, GroupBotManager from utils.robot_cmd.robot_command import Feature, PermissionStatus, GroupBotManager
from utils.decorator.points_decorator import plugin_points_cost from utils.decorator.points_decorator import plugin_points_cost
from utils.media_downloader import MediaDownloader from utils.media_downloader import MediaDownloader
@@ -124,6 +125,8 @@ class DifyPlugin(MessagePluginInterface):
gbm: GroupBotManager = message.get("gbm") gbm: GroupBotManager = message.get("gbm")
bot: WechatAPIClient = message.get("bot") bot: WechatAPIClient = message.get("bot")
revoke: MessageAutoRevoke = message.get("revoke")
# 处理被@的消息 # 处理被@的消息
if message.get("is_at", False) and roomid: if message.get("is_at", False) and roomid:
# 检查权限 # 检查权限
@@ -161,7 +164,11 @@ class DifyPlugin(MessagePluginInterface):
except Exception as e: except Exception as e:
self.LOG.error(f"处理Dify聊天请求出错: {e}") self.LOG.error(f"处理Dify聊天请求出错: {e}")
await bot.send_text_message((roomid if roomid else sender), "❌未能获取到回复，请稍后再试", sender) client_msg_id, create_time, new_msg_id = await bot.send_text_message((roomid if roomid else sender),
"❌未能获取到回复，请稍后再试",
sender)
revoke.add_message_to_revoke(roomid, client_msg_id, create_time, new_msg_id, 5)
return False, f"处理出错: {e}" return False, f"处理出错: {e}"
# 原有的命令处理逻辑 # 原有的命令处理逻辑
@@ -210,8 +217,11 @@ class DifyPlugin(MessagePluginInterface):
except Exception as e: except Exception as e:
self.LOG.error(f"处理Dify聊天请求出错: {e}") self.LOG.error(f"处理Dify聊天请求出错: {e}")
await bot.send_text_message((roomid if roomid else sender), f"❌请求出错：{str(e)}", client_msg_id, create_time, new_msg_id = await bot.send_text_message((roomid if roomid else sender),
f"❌请求出错：{str(e)}",
sender if roomid else "") sender if roomid else "")
revoke.add_message_to_revoke(roomid, client_msg_id, create_time, new_msg_id, 5)
return False, f"处理出错: {e}" return False, f"处理出错: {e}"
def _chat_with_dify(self, session_id: str, user_id: str, query: str) -> Optional[str]: def _chat_with_dify(self, session_id: str, user_id: str, query: str) -> Optional[str]: