@nocobase/plugin-ai 2.1.0-alpha.20 → 2.1.0-alpha.21

package/dist/server/ai-employees/ai-employee.js

@@ -45,8 +45,6 @@ var import_utils = require("../utils");
 var import_prompts = require("./prompts");
 var import_lodash = __toESM(require("lodash"));
 var import_ai_chat_conversation = require("../manager/ai-chat-conversation");
-var import_ai_feature_manager = require("../manager/ai-feature-manager");
-var import_prompts2 = require("@langchain/core/prompts");
 var import_middleware = require("./middleware");
 var import_checkpoints = require("./checkpoints");
 var import_langchain = require("langchain");
@@ -69,6 +67,8 @@ class AIEmployee {
   webSearch;
   model;
   legacy;
+  tools;
+  inWorkflow;
   constructor({
     ctx,
     employee,
@@ -78,7 +78,8 @@ class AIEmployee {
     webSearch,
     model,
     legacy,
-    from = "main-agent"
+    from = "main-agent",
+    tools = []
   }) {
     this.employee = employee;
     this.ctx = ctx;
@@ -91,6 +92,7 @@ class AIEmployee {
     this.model = model;
     this.legacy = legacy;
     this.from = from;
+    this.tools = tools;
     const builtInManager = this.plugin.builtInManager;
     builtInManager.setupBuiltInInfo(ctx, this.employee);
     this.webSearch = webSearch;
@@ -106,6 +108,14 @@ class AIEmployee {
     });
     return messages;
   }
+  async isInWorkflow() {
+    if (this.inWorkflow !== void 0) {
+      return this.inWorkflow;
+    }
+    const conversation = await this.aiConversationsRepo.findByTargetKey(this.sessionId);
+    this.inWorkflow = (conversation == null ? void 0 : conversation.category) === "task";
+    return this.inWorkflow;
+  }
   // === Chat flow ===
   buildState(messages) {
     return {
@@ -125,7 +135,7 @@ class AIEmployee {
         historyMessages: [],
         tools,
         resolvedTools,
-        middleware: this.getMiddleware({ tools, baseToolNames, model, providerName }),
+        middleware: await this.getMiddleware({ tools, baseToolNames, model, providerName }),
         config: void 0,
         state: void 0
       };
@@ -138,7 +148,7 @@ class AIEmployee {
       historyMessages,
       tools,
       resolvedTools,
-      middleware: this.getMiddleware({
+      middleware: await this.getMiddleware({
         tools,
         baseToolNames,
         model,
@@ -183,6 +193,12 @@ class AIEmployee {
     userMessages = [],
     userDecisions
   }) {
+    await this.aiConversationsRepo.update({
+      values: { llmActiveState: "streaming" },
+      filter: {
+        sessionId: this.sessionId
+      }
+    });
     try {
       const { providerName, model, provider, chatContext, config, state } = await this.buildChatContext({
         messageId,
@@ -209,6 +225,13 @@ class AIEmployee {
       this.ctx.log.error(err);
       this.sendErrorResponse(err.message || "Chat error warning");
       return false;
+    } finally {
+      await this.aiConversationsRepo.update({
+        values: { llmActiveState: "idle" },
+        filter: {
+          sessionId: this.sessionId
+        }
+      });
     }
   }
   async invoke({
@@ -218,29 +241,43 @@ class AIEmployee {
     writer,
     context
   }) {
+    var _a;
+    await this.aiConversationsRepo.update({
+      values: { llmActiveState: "invoking" },
+      filter: {
+        sessionId: this.sessionId
+      }
+    });
     try {
       const { provider, chatContext, config, state } = await this.buildChatContext({
         messageId,
         userMessages,
         userDecisions
       });
+      const { threadId } = await this.getCurrentThread();
       const invokeConfig = {
         context: { ctx: this.ctx, decisions: chatContext.decisions, ...context },
         recursionLimit: 100,
+        configurable: this.from === "main-agent" ? { thread_id: threadId } : void 0,
         writer,
         ...config
       };
-      if (this.from === "main-agent") {
-        const { threadId } = await this.getCurrentThread();
-        invokeConfig.configurable = { thread_id: threadId };
-      }
-      return await this.agentInvoke(provider, chatContext, invokeConfig, state);
+      const invokeResult = await this.agentInvoke(provider, chatContext, invokeConfig, state);
+      await this.handleInterruptedToolCalls((_a = invokeResult == null ? void 0 : invokeResult.__interrupt__) == null ? void 0 : _a[0], () => invokeResult == null ? void 0 : invokeResult.messageId);
+      return invokeResult;
     } catch (err) {
       if (err.name === "GraphInterrupt") {
         throw err;
       }
       this.ctx.log.error(err);
       throw err;
+    } finally {
+      await this.aiConversationsRepo.update({
+        values: { llmActiveState: "idle" },
+        filter: {
+          sessionId: this.sessionId
+        }
+      });
     }
   }
   // === Agent wiring & execution ===
@@ -322,7 +359,7 @@ class AIEmployee {
     }
   }
   async processChatStream(stream, options) {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p, _q, _r, _s, _t, _u, _v, _w, _x, _y;
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p, _q, _r, _s, _t, _u, _v, _w, _x, _y, _z;
     const aiMessageIdMap = /* @__PURE__ */ new Map();
     const { signal, providerName, model, provider, responseMetadata, allowEmpty = false } = options;
     let isReasoning = false;
@@ -384,35 +421,25 @@ class AIEmployee {
             }
           }
         } else if (mode === "updates") {
-          if ("__interrupt__" in chunks) {
-            const interruptId = chunks.__interrupt__[0].id;
-            const interruptActions = this.toInterruptActions(chunks.__interrupt__[0].value);
-            if (interruptActions.size) {
-              const toolsMap = await this.getToolsMap();
-              for (const interruptAction of interruptActions.values()) {
-                if (!interruptAction.currentConversation || !interruptAction.toolCall) {
-                  this.logger.warn("currentConversation or toolCall not exist in __interrupt__", interruptAction);
-                  continue;
-                }
-                const { sessionId, from, username } = interruptAction.currentConversation;
-                const { id: toolCallId, name: toolCallName } = interruptAction.toolCall;
-                const messageId = aiMessageIdMap.get(sessionId);
-                if (!messageId) {
-                  continue;
-                }
-                await this.updateToolCallInterrupted(sessionId, messageId, toolCallId, interruptId, interruptAction);
-                this.protocol.with(interruptAction.currentConversation).toolCallStatus({
+          const interrupt = (_b = chunks == null ? void 0 : chunks.__interrupt__) == null ? void 0 : _b[0];
+          if (interrupt) {
+            const toolsMap = await this.getToolsMap();
+            await this.handleInterruptedToolCalls(
+              interrupt,
+              (sessionId) => aiMessageIdMap.get(sessionId),
+              ({ messageId, interruptAction, toolCall, currentConversation }) => {
+                this.protocol.with(currentConversation).toolCallStatus({
                   toolCall: {
                     messageId,
-                    id: toolCallId,
-                    name: toolCallName,
-                    willInterrupt: this.shouldInterruptToolCall(toolsMap.get(toolCallName))
+                    id: toolCall.id,
+                    name: toolCall.name,
+                    willInterrupt: this.shouldInterruptToolCall(toolsMap.get(toolCall.name))
                   },
                   invokeStatus: "interrupted",
                   interruptAction
                 });
               }
-            }
+            );
           }
         } else if (mode === "custom") {
           const { currentConversation } = chunks;
@@ -448,31 +475,31 @@ class AIEmployee {
             this.protocol.with(currentConversation).toolCalls(chunks.body);
           } else if (chunks.action === "beforeToolCall") {
             const toolsMap = await this.getToolsMap();
-            const willInterrupt = this.shouldInterruptToolCall(toolsMap.get((_c = (_b = chunks.body) == null ? void 0 : _b.toolCall) == null ? void 0 : _c.name));
+            const willInterrupt = this.shouldInterruptToolCall(toolsMap.get((_d = (_c = chunks.body) == null ? void 0 : _c.toolCall) == null ? void 0 : _d.name));
             this.protocol.with(currentConversation).toolCallStatus({
               toolCall: {
-                messageId: (_e = (_d = chunks.body) == null ? void 0 : _d.toolCall) == null ? void 0 : _e.messageId,
-                id: (_g = (_f = chunks.body) == null ? void 0 : _f.toolCall) == null ? void 0 : _g.id,
-                name: (_i = (_h = chunks.body) == null ? void 0 : _h.toolCall) == null ? void 0 : _i.name,
+                messageId: (_f = (_e = chunks.body) == null ? void 0 : _e.toolCall) == null ? void 0 : _f.messageId,
+                id: (_h = (_g = chunks.body) == null ? void 0 : _g.toolCall) == null ? void 0 : _h.id,
+                name: (_j = (_i = chunks.body) == null ? void 0 : _i.toolCall) == null ? void 0 : _j.name,
                 willInterrupt
               },
               invokeStatus: "pending"
             });
           } else if (chunks.action === "afterToolCall") {
             const toolsMap = await this.getToolsMap();
-            const willInterrupt = this.shouldInterruptToolCall(toolsMap.get((_k = (_j = chunks.body) == null ? void 0 : _j.toolCall) == null ? void 0 : _k.name));
+            const willInterrupt = this.shouldInterruptToolCall(toolsMap.get((_l = (_k = chunks.body) == null ? void 0 : _k.toolCall) == null ? void 0 : _l.name));
             this.protocol.with(currentConversation).toolCallStatus({
               toolCall: {
-                messageId: (_m = (_l = chunks.body) == null ? void 0 : _l.toolCall) == null ? void 0 : _m.messageId,
-                id: (_o = (_n = chunks.body) == null ? void 0 : _n.toolCall) == null ? void 0 : _o.id,
-                name: (_q = (_p = chunks.body) == null ? void 0 : _p.toolCall) == null ? void 0 : _q.name,
+                messageId: (_n = (_m = chunks.body) == null ? void 0 : _m.toolCall) == null ? void 0 : _n.messageId,
+                id: (_p = (_o = chunks.body) == null ? void 0 : _o.toolCall) == null ? void 0 : _p.id,
+                name: (_r = (_q = chunks.body) == null ? void 0 : _q.toolCall) == null ? void 0 : _r.name,
                 willInterrupt
               },
               invokeStatus: "done",
-              status: (_s = (_r = chunks.body) == null ? void 0 : _r.toolCallResult) == null ? void 0 : _s.status,
-              invokeStartTime: (_u = (_t = chunks.body) == null ? void 0 : _t.toolCallResult) == null ? void 0 : _u.invokeStartTime,
-              invokeEndTime: (_w = (_v = chunks.body) == null ? void 0 : _v.toolCallResult) == null ? void 0 : _w.invokeEndTime,
-              content: (_y = (_x = chunks.body) == null ? void 0 : _x.toolCallResult) == null ? void 0 : _y.content
+              status: (_t = (_s = chunks.body) == null ? void 0 : _s.toolCallResult) == null ? void 0 : _t.status,
+              invokeStartTime: (_v = (_u = chunks.body) == null ? void 0 : _u.toolCallResult) == null ? void 0 : _v.invokeStartTime,
+              invokeEndTime: (_x = (_w = chunks.body) == null ? void 0 : _w.toolCallResult) == null ? void 0 : _x.invokeEndTime,
+              content: (_z = (_y = chunks.body) == null ? void 0 : _y.toolCallResult) == null ? void 0 : _z.content
             });
           } else if (chunks.action === "beforeSendToolMessage") {
             const { messageId, messages } = chunks.body ?? {};
@@ -524,6 +551,39 @@ class AIEmployee {
       }
     }
   }
+  async handleInterruptedToolCalls(interrupt, getMessageId, onInterrupted) {
+    const interruptId = interrupt == null ? void 0 : interrupt.id;
+    const interruptActions = this.toInterruptActions(interrupt == null ? void 0 : interrupt.value);
+    if (!interruptId || !interruptActions.size) {
+      return;
+    }
+    for (const interruptAction of interruptActions.values()) {
+      const currentConversation = interruptAction.currentConversation;
+      const toolCall = interruptAction.toolCall;
+      if (!currentConversation || !toolCall) {
+        this.logger.warn("currentConversation or toolCall not exist in __interrupt__", interruptAction);
+        continue;
+      }
+      const messageId = getMessageId(currentConversation.sessionId);
+      if (!messageId) {
+        continue;
+      }
+      await this.updateToolCallInterrupted(
+        currentConversation.sessionId,
+        messageId,
+        toolCall.id,
+        interruptId,
+        interruptAction
+      );
+      await (onInterrupted == null ? void 0 : onInterrupted({
+        messageId,
+        interruptId,
+        interruptAction,
+        toolCall,
+        currentConversation
+      }));
+    }
+  }
   // === Prompts & knowledge base ===
   // Notice: employee.dataSourceSettings is not used in the current version.
   getEmployeeDataSourceContext() {
@@ -567,10 +627,10 @@ Field: ${field.name}, Title: ${(_a = field.options.uiSchema) == null ? void 0 :
     return message;
   }
   async getSystemPrompt(userMessages) {
-    var _a;
+    var _a, _b, _c, _d, _e, _f, _g;
     const userConfig = await this.db.getRepository("usersAiEmployees").findOne({
       filter: {
-        userId: (_a = this.ctx.auth) == null ? void 0 : _a.user.id,
+        userId: ((_a = this.ctx.auth) == null ? void 0 : _a.user.id) ?? 0,
         aiEmployee: this.employee.username
       }
     });
@@ -596,14 +656,14 @@ ${workContextBackground.join("\n")}`;
 ${addSystemPrompt.map((it) => it.content).join("\n")}`;
     }
     let knowledgeBase;
-    if (this.isEnabledKnowledgeBase() && this.employee.knowledgeBasePrompt && (userMessages == null ? void 0 : userMessages.length)) {
+    const { knowledgeBaseManager } = this.plugin;
+    const employee = this.employee.toJSON();
+    if (await knowledgeBaseManager.isEnabledKnowledgeBase(employee) && employee.knowledgeBasePrompt && (userMessages == null ? void 0 : userMessages.length)) {
       const lastUserMessage = userMessages.filter((x) => x.role === "user").at(-1);
       if (lastUserMessage) {
-        const docs = await this.retrieveKnowledgeBase(lastUserMessage);
-        const knowledgeBaseData = docs.map((x) => x.content).join("\n");
-        const promptTemplate = import_prompts2.ChatPromptTemplate.fromTemplate(this.employee.knowledgeBasePrompt);
-        knowledgeBase = import_lodash.default.isEmpty(knowledgeBaseData) ? void 0 : await promptTemplate.format({
-          knowledgeBaseData
+        knowledgeBase = await knowledgeBaseManager.retrievePrompt({
+          employee,
+          query: lastUserMessage.content.content
         });
       }
     }
@@ -621,15 +681,15 @@ ${addSystemPrompt.map((it) => it.content).join("\n")}`;
       dataSources: dataSourceMessage,
       environment: {
         database: this.db.sequelize.getDialect(),
-        locale: this.ctx.getCurrentLocale() || "en-US",
-        currentDateTime: getCurrentDateTimeForPrompt(this.ctx.getCurrentLocale(), getCurrentTimezone(this.ctx)),
+        locale: ((_c = (_b = this.ctx).getCurrentLocale) == null ? void 0 : _c.call(_b)) || "en-US",
+        currentDateTime: getCurrentDateTimeForPrompt((_e = (_d = this.ctx).getCurrentLocale) == null ? void 0 : _e.call(_d), getCurrentTimezone(this.ctx)),
         timezone: getCurrentTimezone(this.ctx)
       },
       knowledgeBase,
       availableSkills,
       availableAIEmployees
     });
-    const { important } = this.ctx.action.params.values || {};
+    const { important } = ((_g = (_f = this.ctx.action) == null ? void 0 : _f.params) == null ? void 0 : _g.values) || {};
     if (important === "GraphRecursionError") {
       const importantPrompt = `<Important>You have already called tools multiple times and gathered sufficient information.
 First, provide a summary based on the existing information. Do not call additional tools.
@@ -639,92 +699,6 @@ If information is missing, clearly state it in the summary.</Important>`;
       return systemPrompt;
     }
   }
-  async retrieveKnowledgeBase(userMessage) {
-    const vectorStoreProvider = this.plugin.features.vectorStoreProvider;
-    let queryResult = [];
-    const queryString = userMessage.content.content;
-    if (!queryString || import_lodash.default.isEmpty(queryString)) {
-      return queryResult;
-    }
-    const { topK, score } = this.getAIEmployeeKnowledgeBaseConfig();
-    const knowledgeBaseGroup = await this.getKnowledgeBaseGroup();
-    for (const entry of knowledgeBaseGroup) {
-      const { vectorStoreConfig, knowledgeBaseType, knowledgeBaseList } = entry;
-      if (!knowledgeBaseList || import_lodash.default.isEmpty(knowledgeBaseList)) {
-        continue;
-      }
-      if (knowledgeBaseType === "LOCAL") {
-        const vectorStoreService = await vectorStoreProvider.createVectorStoreService(
-          vectorStoreConfig.vectorStoreProvider,
-          [
-            {
-              key: "vectorStoreConfigId",
-              value: vectorStoreConfig.vectorStoreConfigId
-            }
-          ]
-        );
-        const knowledgeBaseOuterIds = knowledgeBaseList.map((x) => x.knowledgeBaseOuterId);
-        const result = await vectorStoreService.search(queryString, {
-          topK,
-          score,
-          filter: {
-            knowledgeBaseOuterId: { in: knowledgeBaseOuterIds }
-          }
-        });
-        queryResult = [...queryResult, ...result];
-      } else if (knowledgeBaseType === "READONLY") {
-        for (const knowledgeBase of knowledgeBaseList) {
-          const vectorStoreService = await vectorStoreProvider.createVectorStoreService(
-            vectorStoreConfig.vectorStoreProvider,
-            [
-              ...knowledgeBase.vectorStoreProps,
-              {
-                key: "vectorStoreConfigId",
-                value: vectorStoreConfig.vectorStoreConfigId
-              }
-            ]
-          );
-          const result = await vectorStoreService.search(queryString, {
-            topK,
-            score
-          });
-          queryResult = [...queryResult, ...result];
-        }
-      } else if (knowledgeBaseType === "EXTERNAL") {
-        for (const knowledgeBase of knowledgeBaseList) {
-          const vectorStoreService = await vectorStoreProvider.createVectorStoreService(
-            vectorStoreConfig.vectorStoreProvider,
-            knowledgeBase.vectorStoreProps
-          );
-          const result = await vectorStoreService.search(queryString, {
-            topK,
-            score
-          });
-          queryResult = [...queryResult, ...result];
-        }
-      }
-    }
-    return queryResult;
-  }
-  isEnabledKnowledgeBase() {
-    const featureEnabled = this.plugin.features.isFeaturesEnabled(Object.values(import_ai_feature_manager.EEFeatures));
-    const knowledgeBaseEnabled = this.employee.enableKnowledgeBase;
-    return featureEnabled && knowledgeBaseEnabled;
-  }
-  getAIEmployeeKnowledgeBaseConfig() {
-    const { topK, score } = this.employee.knowledgeBase ?? {};
-    return {
-      topK,
-      score
-    };
-  }
-  async getKnowledgeBaseGroup() {
-    const { knowledgeBaseIds } = this.employee.knowledgeBase ?? {};
-    if (!knowledgeBaseIds || import_lodash.default.isEmpty(knowledgeBaseIds)) {
-      return [];
-    }
-    return await this.plugin.features.knowledgeBase.getKnowledgeBaseGroup(knowledgeBaseIds);
-  }
   // === Tool calls ===
   async initToolCall(transaction, messageId, toolCalls) {
     const nowTime = /* @__PURE__ */ new Date();
@@ -1123,7 +1097,7 @@ If information is missing, clearly state it in the summary.</Important>`;
   toInterruptActions(interrupt) {
     var _a;
     const result = /* @__PURE__ */ new Map();
-    const { actionRequests = [], reviewConfigs = [] } = interrupt;
+    const { actionRequests = [], reviewConfigs = [] } = interrupt ?? {};
     if (!actionRequests.length) {
       return result;
     }
@@ -1159,6 +1133,13 @@ If information is missing, clearly state it in the summary.</Important>`;
     const generalToolsNameSet = new Set(tools.map((x) => x.definition.name));
     const toolMap = await this.getToolsMap();
     const employeeTools = ((_a = this.employee.skillSettings) == null ? void 0 : _a.tools) ?? [];
+    employeeTools.push(...this.tools);
+    if (await this.plugin.knowledgeBaseManager.isEnabledKnowledgeBase(this.employee.toJSON())) {
+      const knowledgeBaseRetrieveTool = await this.toolsManager.getTools("knowledge-base-retrieve");
+      if (knowledgeBaseRetrieveTool) {
+        employeeTools.push({ name: "knowledge-base-retrieve" });
+      }
+    }
     for (const toolSetting of employeeTools) {
       if (generalToolsNameSet.has(toolSetting.name)) {
         continue;
@@ -1255,14 +1236,16 @@ If information is missing, clearly state it in the summary.</Important>`;
     );
     return availableAIEmployees;
   }
-  getMiddleware(options) {
+  async getMiddleware(options) {
     const { providerName, model, tools, baseToolNames, messageId, agentThread } = options;
+    const inWorkflow = await this.isInWorkflow();
     return [
       (0, import_middleware.skillToolBindingMiddleware)(this, {
         baseToolNames: Array.from(baseToolNames.values())
       }),
       (0, import_middleware.toolInteractionMiddleware)(this, tools),
       (0, import_middleware.toolCallStatusMiddleware)(this),
+      ...inWorkflow ? [(0, import_middleware.workflowHistoryMiddleware)(this, this.db)] : [],
       (0, import_middleware.conversationMiddleware)(this, { providerName, model, messageId, agentThread })
     ];
   }
@@ -1288,7 +1271,9 @@ If information is missing, clearly state it in the summary.</Important>`;
     throw new Error("Fail to create new agent thread");
   }
   async getToolsMap() {
-    const tools = await this.listTools();
+    const tools = await this.listTools({
+      sessionId: this.sessionId
+    });
     return new Map(tools.map((tool) => [tool.definition.name, tool]));
   }
   listTools(filter) {

package/dist/server/ai-employees/ai-knowledge-base.d.ts

@@ -0,0 +1,28 @@
+/**
+ * This file is part of the NocoBase (R) project.
+ * Copyright (c) 2020-2024 NocoBase Co., Ltd.
+ * Authors: NocoBase Team.
+ *
+ * This project is dual-licensed under AGPL-3.0 and NocoBase Commercial License.
+ * For more information, please refer to: https://www.nocobase.com/agreement.
+ */
+import { AIEmployee } from '../../collections/ai-employees';
+import { DocumentSegmentedWithScore } from '../features';
+import type PluginAIServer from '../plugin';
+export type KnowledgeBaseRetrieveOptions = {
+    username?: string;
+    employee?: AIEmployee;
+    query: string;
+};
+export declare class KnowledgeBaseManager {
+    private readonly plugin;
+    constructor(plugin: PluginAIServer);
+    retrievePrompt({ username, employee, query }: KnowledgeBaseRetrieveOptions): Promise<string>;
+    retrieve({ username, employee, query }: KnowledgeBaseRetrieveOptions): Promise<DocumentSegmentedWithScore[]>;
+    isEnabledKnowledgeBase(username: string): Promise<boolean>;
+    isEnabledKnowledgeBase(employee: AIEmployee): Promise<boolean>;
+    private getAIEmployeeKnowledgeBaseConfig;
+    private getKnowledgeBaseGroup;
+    private getEmployee;
+    private get aiEmployeeRepo();
+}

package/dist/server/ai-employees/ai-knowledge-base.js

@@ -0,0 +1,167 @@
+/**
+ * This file is part of the NocoBase (R) project.
+ * Copyright (c) 2020-2024 NocoBase Co., Ltd.
+ * Authors: NocoBase Team.
+ *
+ * This project is dual-licensed under AGPL-3.0 and NocoBase Commercial License.
+ * For more information, please refer to: https://www.nocobase.com/agreement.
+ */
+
+var __create = Object.create;
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var ai_knowledge_base_exports = {};
+__export(ai_knowledge_base_exports, {
+  KnowledgeBaseManager: () => KnowledgeBaseManager
+});
+module.exports = __toCommonJS(ai_knowledge_base_exports);
+var import_prompts = require("@langchain/core/prompts");
+var import_ai_feature_manager = require("../manager/ai-feature-manager");
+var import_lodash = __toESM(require("lodash"));
+class KnowledgeBaseManager {
+  constructor(plugin) {
+    this.plugin = plugin;
+  }
+  async retrievePrompt({ username, employee, query }) {
+    employee = employee ?? await this.getEmployee(username);
+    if (!employee) {
+      return "Specified knowledge base not existed";
+    }
+    const promptTemplate = import_prompts.ChatPromptTemplate.fromTemplate(employee.knowledgeBasePrompt);
+    const docs = await this.retrieve({
+      employee,
+      query
+    });
+    if (!(docs == null ? void 0 : docs.length)) {
+      return "No document match in knowledge base";
+    }
+    const knowledgeBaseData = docs.map((x) => x.content).join("\n");
+    return import_lodash.default.isEmpty(knowledgeBaseData) ? void 0 : await promptTemplate.format({
+      knowledgeBaseData
+    });
+  }
+  async retrieve({ username, employee, query }) {
+    const vectorStoreProvider = this.plugin.features.vectorStoreProvider;
+    let queryResult = [];
+    if (!query || import_lodash.default.isEmpty(query)) {
+      return queryResult;
+    }
+    employee = employee ?? await this.getEmployee(username);
+    if (!employee) {
+      return queryResult;
+    }
+    const { topK, score } = this.getAIEmployeeKnowledgeBaseConfig(employee);
+    const knowledgeBaseGroup = await this.getKnowledgeBaseGroup(employee);
+    for (const entry of knowledgeBaseGroup) {
+      const { vectorStoreConfig, knowledgeBaseType, knowledgeBaseList } = entry;
+      if (!knowledgeBaseList || import_lodash.default.isEmpty(knowledgeBaseList)) {
+        continue;
+      }
+      if (knowledgeBaseType === "LOCAL") {
+        const vectorStoreService = await vectorStoreProvider.createVectorStoreService(
+          vectorStoreConfig.vectorStoreProvider,
+          [
+            {
+              key: "vectorStoreConfigId",
+              value: vectorStoreConfig.vectorStoreConfigId
+            }
+          ]
+        );
+        const knowledgeBaseOuterIds = knowledgeBaseList.map((x) => x.knowledgeBaseOuterId);
+        const result = await vectorStoreService.search(query, {
+          topK,
+          score,
+          filter: {
+            knowledgeBaseOuterId: { in: knowledgeBaseOuterIds }
+          }
+        });
+        queryResult = [...queryResult, ...result];
+      } else if (knowledgeBaseType === "READONLY") {
+        for (const knowledgeBase of knowledgeBaseList) {
+          const vectorStoreService = await vectorStoreProvider.createVectorStoreService(
+            vectorStoreConfig.vectorStoreProvider,
+            [
+              ...knowledgeBase.vectorStoreProps,
+              {
+                key: "vectorStoreConfigId",
+                value: vectorStoreConfig.vectorStoreConfigId
+              }
+            ]
+          );
+          const result = await vectorStoreService.search(query, {
+            topK,
+            score
+          });
+          queryResult = [...queryResult, ...result];
+        }
+      } else if (knowledgeBaseType === "EXTERNAL") {
+        for (const knowledgeBase of knowledgeBaseList) {
+          const vectorStoreService = await vectorStoreProvider.createVectorStoreService(
+            vectorStoreConfig.vectorStoreProvider,
+            knowledgeBase.vectorStoreProps
+          );
+          const result = await vectorStoreService.search(query, {
+            topK,
+            score
+          });
+          queryResult = [...queryResult, ...result];
+        }
+      }
+    }
+    return queryResult;
+  }
+  async isEnabledKnowledgeBase(usernameOrEmployee) {
+    const featureEnabled = this.plugin.features.isFeaturesEnabled(Object.values(import_ai_feature_manager.EEFeatures));
+    const employee = typeof usernameOrEmployee === "string" ? await this.getEmployee(usernameOrEmployee) : usernameOrEmployee;
+    const knowledgeBaseEnabled = employee == null ? void 0 : employee.enableKnowledgeBase;
+    return featureEnabled && knowledgeBaseEnabled;
+  }
+  getAIEmployeeKnowledgeBaseConfig(employee) {
+    const { topK, score } = (employee == null ? void 0 : employee.knowledgeBase) ?? {};
+    return {
+      topK,
+      score
+    };
+  }
+  async getKnowledgeBaseGroup(employee) {
+    const { knowledgeBaseIds } = (employee == null ? void 0 : employee.knowledgeBase) ?? {};
+    if (!knowledgeBaseIds || import_lodash.default.isEmpty(knowledgeBaseIds)) {
+      return [];
+    }
+    return await this.plugin.features.knowledgeBase.getKnowledgeBaseGroup(knowledgeBaseIds);
+  }
+  async getEmployee(username) {
+    return await this.aiEmployeeRepo.findOne({ filter: { username } });
+  }
+  get aiEmployeeRepo() {
+    return this.plugin.db.getRepository("aiEmployees");
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  KnowledgeBaseManager
+});

package/dist/server/ai-employees/middleware/index.d.ts

@@ -9,3 +9,4 @@
 export * from './conversation';
 export * from './skill-tools';
 export * from './tools';
+export * from './workflow-history';