From f1ebe1b526707f28933c0a56496b8a5419883cc0 Mon Sep 17 00:00:00 2001
From: 18586361686 <1207669650@qq.com>
Date: 星期三, 14 五月 2025 12:41:50 +0800
Subject: [PATCH] chore: 优化 bot 调用插件方法
---
aiflowy-modules/aiflowy-module-ai/src/main/java/tech/aiflowy/ai/controller/AiBotController.java | 11 ++++-------
1 files changed, 4 insertions(+), 7 deletions(-)
diff --git a/aiflowy-modules/aiflowy-module-ai/src/main/java/tech/aiflowy/ai/controller/AiBotController.java b/aiflowy-modules/aiflowy-module-ai/src/main/java/tech/aiflowy/ai/controller/AiBotController.java
index d26169f..1d9c920 100644
--- a/aiflowy-modules/aiflowy-module-ai/src/main/java/tech/aiflowy/ai/controller/AiBotController.java
+++ b/aiflowy-modules/aiflowy-module-ai/src/main/java/tech/aiflowy/ai/controller/AiBotController.java
@@ -522,21 +522,18 @@
// }
// }
private void appendPluginToolFunction(BigInteger botId, HumanMessage humanMessage) {
- QueryWrapper queryWrapper = QueryWrapper.create().eq(AiBotPlugins::getBotId, botId);
- List<AiBotPlugins> aiBotPlugins = aiBotPluginsService.getMapper().selectListWithRelationsByQuery(queryWrapper);
- // 鏍规嵁鎻掍欢iD鏌ヨ璇ユ彃浠朵笅闈㈡湁鍝簺鎻掍欢宸ュ叿锛岃浆鎹㈡垚Function
- for (AiBotPlugins aiBotPlugin: aiBotPlugins){
- BigInteger pluginId = aiBotPlugin.getPluginToolId();
+ QueryWrapper queryWrapper = QueryWrapper.create().select("plugin_tool_id").eq(AiBotPlugins::getBotId, botId);
+ List<BigInteger> pluginToolIds = aiBotPluginsService.getMapper().selectListWithRelationsByQueryAs(queryWrapper, BigInteger.class);
+
QueryWrapper queryTool = QueryWrapper.create()
.select("*")
.from("tb_ai_plugin_tool")
- .where("plugin_id = ?", pluginId);
+ .in("id", pluginToolIds);
List<AiPluginTool> aiPluginTools = aiPluginToolService.getMapper().selectListWithRelationsByQuery(queryTool);
for (AiPluginTool item: aiPluginTools){
humanMessage.addFunction(item.toFunction());
}
- }
}
}
--
Gitblit v1.8.0