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

调整AI对话内容

Browse Source
This commit is contained in:
liuwei
2025-04-17 13:49:05 +08:00
parent 2a61b2d48f
commit 48c37767e7
1 changed files with 12 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
plugins/dify/main.py
View File

@@ -122,7 +122,6 @@ class DifyPlugin(MessagePluginInterface):
self.LOG.info(f"插件执行： {self.name}：{content}") self.LOG.info(f"插件执行： {self.name}：{content}")
sender = message.get("sender") sender = message.get("sender")
roomid = message.get("roomid", "") roomid = message.get("roomid", "")
wcf: Wcf = message.get("wcf")
gbm: GroupBotManager = message.get("gbm") gbm: GroupBotManager = message.get("gbm")
# 处理被@的消息 # 处理被@的消息
@@ -135,10 +134,10 @@ class DifyPlugin(MessagePluginInterface):
query = re.sub(r"@.*?[\u2005|\s]", "", content).strip() query = re.sub(r"@.*?[\u2005|\s]", "", content).strip()
if not query: if not query:
wcf.send_text("请在@我的同时提供问题内容", roomid, sender) self.message_util.send_text_msg("请在@我的同时提供问题内容", roomid, sender)
return True, "没有提供问题内容" return True, "没有提供问题内容"
self.message_util.send_text_msg("⏳AI 正在加油，请稍候… 😊", roomid, sender) # self.message_util.send_text_msg("⏳AI 正在加油，请稍候… 😊", roomid, sender)
try: try:
# 调用Dify API获取回复 # 调用Dify API获取回复
@@ -151,18 +150,18 @@ class DifyPlugin(MessagePluginInterface):
# 判断是否为本地文件路径 # 判断是否为本地文件路径
if os.path.isfile(response): if os.path.isfile(response):
# 如果是文件路径，使用发送文件方法 # 如果是文件路径，使用发送文件方法
wcf.send_file(response, roomid) self.message_util.send_file(response, roomid)
else: else:
# 如果是普通文本，使用发送文本方法 # 如果是普通文本，使用发送文本方法
wcf.send_text(response, roomid, sender) self.message_util.send_text_msg(response, roomid, sender)
return True, "发送成功" return True, "发送成功"
else: else:
wcf.send_text("❌未能获取到回复，请稍后再试", roomid, sender) self.message_util.send_text_msg("❌未能获取到回复，请稍后再试", roomid, sender)
return True, "未获取到回复" return True, "未获取到回复"
except Exception as e: except Exception as e:
self.LOG.error(f"处理Dify聊天请求出错: {e}") self.LOG.error(f"处理Dify聊天请求出错: {e}")
wcf.send_text(f"❌请求出错：{str(e)}", roomid, sender) self.message_util.send_text_msg(f"❌请求出错：{str(e)}", roomid, sender)
return True, f"处理出错: {e}" return True, f"处理出错: {e}"
# 原有的命令处理逻辑 # 原有的命令处理逻辑
@@ -171,7 +170,7 @@ class DifyPlugin(MessagePluginInterface):
# 检查命令格式 # 检查命令格式
if len(parts) < 2 or not parts[1].strip(): if len(parts) < 2 or not parts[1].strip():
wcf.send_text(f"{self.command_format}", self.message_util.send_text_msg(f"{self.command_format}",
(roomid if roomid else sender), sender) (roomid if roomid else sender), sender)
return True, "命令格式错误" return True, "命令格式错误"
@@ -179,8 +178,7 @@ class DifyPlugin(MessagePluginInterface):
if roomid and gbm.get_group_permission(roomid, Feature.AI_CAPABILITY) == PermissionStatus.DISABLED: if roomid and gbm.get_group_permission(roomid, Feature.AI_CAPABILITY) == PermissionStatus.DISABLED:
return False, "没有权限" return False, "没有权限"
self.message_util.send_text_msg("⏳AI 正在加油，请稍候… 😊", (roomid if roomid else sender), # self.message_util.send_text_msg("⏳AI 正在加油，请稍候… 😊", (roomid if roomid else sender),sender if roomid else "")
sender if roomid else "")
# 获取查询内容 # 获取查询内容
query = parts[1].strip() query = parts[1].strip()
@@ -200,18 +198,18 @@ class DifyPlugin(MessagePluginInterface):
# 判断是否为本地文件路径 # 判断是否为本地文件路径
if os.path.isfile(response): if os.path.isfile(response):
# 如果是文件路径，使用发送文件方法 # 如果是文件路径，使用发送文件方法
wcf.send_file(response, (roomid if roomid else sender)) self.message_util.send_file(response, (roomid if roomid else sender))
else: else:
# 如果是普通文本，使用发送文本方法 # 如果是普通文本，使用发送文本方法
wcf.send_text(response, (roomid if roomid else sender), sender if roomid else "") self.message_util.send_text_msg(response, (roomid if roomid else sender), sender if roomid else "")
return True, "发送成功" return True, "发送成功"
else: else:
wcf.send_text("❌未能获取到回复，请稍后再试", (roomid if roomid else sender), sender if roomid else "") self.message_util.send_text_msg("❌未能获取到回复，请稍后再试", (roomid if roomid else sender), sender if roomid else "")
return True, "未获取到回复" return True, "未获取到回复"
except Exception as e: except Exception as e:
self.LOG.error(f"处理Dify聊天请求出错: {e}") self.LOG.error(f"处理Dify聊天请求出错: {e}")
wcf.send_text(f"❌请求出错：{str(e)}", (roomid if roomid else sender), sender if roomid else "") self.message_util.send_text_msg(f"❌请求出错：{str(e)}", (roomid if roomid else sender), sender if roomid else "")
return True, f"处理出错: {e}" return True, f"处理出错: {e}"