@nocobase/plugin-ai 2.2.18 → 2.2.20

package/dist/ai/tools/knowledge-base-retrieve.js

@@ -61,6 +61,7 @@ var knowledge_base_retrieve_default = (0, import_ai.defineTools)({
     })
   },
   async invoke(ctx, { query }, runtime) {
+    var _a;
     const toolCallId = runtime == null ? void 0 : runtime.toolCallId;
     if (!toolCallId) {
       throw new Error("Missing tool call context");
@@ -83,14 +84,24 @@ var knowledge_base_retrieve_default = (0, import_ai.defineTools)({
       throw new Error(`AI conversation not found for session "${aiToolMessage.sessionId}"`);
     }
     const plugin = (0, import_shared.getAIPlugin)(ctx);
-    const content = await plugin.knowledgeBaseManager.retrievePrompt({
+    const { prompt, documents } = await plugin.knowledgeBaseManager.retrievePrompt({
       username,
       query,
       roleNames: (0, import_ai_knowledge_base.getCurrentRoleNames)(ctx.state)
     });
+    if (documents.length) {
+      (_a = runtime.writer) == null ? void 0 : _a.call(runtime, {
+        action: "knowledgeBaseRetrieved",
+        body: { toolCallId, documents },
+        currentConversation: {
+          sessionId: aiToolMessage.sessionId,
+          username
+        }
+      });
+    }
     return {
       status: "success",
-      content
+      content: prompt
     };
   }
 });

package/dist/ai/tools/subAgentWebSearch.js

@@ -31,6 +31,7 @@ __export(subAgentWebSearch_exports, {
 module.exports = __toCommonJS(subAgentWebSearch_exports);
 var import_ai = require("@nocobase/ai");
 var import_zod = require("zod");
+var import_node_crypto = require("node:crypto");
 var subAgentWebSearch_default = (0, import_ai.defineTools)({
   scope: "SPECIFIED",
   defaultPermission: "ALLOW",
@@ -51,12 +52,15 @@ var subAgentWebSearch_default = (0, import_ai.defineTools)({
   invoke: async (ctx, args, id) => {
     var _a, _b, _c;
     const pluginAI = ctx.app.pm.get("ai");
-    const { model } = ((_b = (_a = ctx.action) == null ? void 0 : _a.params) == null ? void 0 : _b.values) ?? {};
-    const { provider } = await pluginAI.aiManager.getLLMService({
-      ...model,
-      webSearch: true,
-      reasoning: { mode: "off" }
-    });
+    const { model, sessionId } = ((_b = (_a = ctx.action) == null ? void 0 : _a.params) == null ? void 0 : _b.values) ?? {};
+    const { provider } = await pluginAI.aiManager.getLLMService(
+      {
+        ...model,
+        webSearch: true,
+        reasoning: { mode: "off" }
+      },
+      { sessionId: typeof sessionId === "string" && sessionId ? sessionId : `web-search-${(0, import_node_crypto.randomUUID)()}` }
+    );
     if (!((_c = args.query) == null ? void 0 : _c.length)) {
       return {
         status: "success",