@nocobase/plugin-ai 2.1.0-beta.29 → 2.1.0-beta.32

package/dist/server/manager/llm-stream-manager.js

@@ -0,0 +1,142 @@
+/**
+ * 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 __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+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 __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var llm_stream_manager_exports = {};
+__export(llm_stream_manager_exports, {
+  LLMStreamCached: () => LLMStreamCached,
+  LLMStreamCachedManager: () => LLMStreamCachedManager
+});
+module.exports = __toCommonJS(llm_stream_manager_exports);
+var import_utils = require("@nocobase/utils");
+const CACHE_NAME = "ai-llm-stream-cache";
+const LOCK_KEY_PREFIX = "ai-llm-stream-lock";
+const WRITE_LOCK_TTL = 3e3;
+const CACHE_TTL = 10 * 60 * 1e3;
+const STREAM_END_MARK = '"type":"stream_end"';
+const SKIPPED_MARK = "__skipped__";
+const DEFAULT_POLL_INTERVAL = 50;
+const DEFAULT_INITIAL_WAIT_TIMEOUT = 1e3;
+class LLMStreamCachedManager {
+  constructor(plugin) {
+    this.plugin = plugin;
+  }
+  cachePromise;
+  getCached(sessionId) {
+    return new LLMStreamCached(sessionId, this);
+  }
+  async clear(sessionId) {
+    await this.withLock(sessionId, async () => {
+      const cache = await this.getCache();
+      await cache.del(sessionId);
+    });
+  }
+  async append(sessionId, chunk) {
+    await this.withLock(sessionId, async () => {
+      const cache = await this.getCache();
+      const chunks = await cache.get(sessionId) ?? [];
+      chunks.push(chunk);
+      await cache.set(sessionId, chunks, CACHE_TTL);
+    });
+  }
+  async *stream(sessionId, options) {
+    const pollInterval = (options == null ? void 0 : options.pollInterval) ?? DEFAULT_POLL_INTERVAL;
+    const initialWaitTimeout = (options == null ? void 0 : options.initialWaitTimeout) ?? DEFAULT_INITIAL_WAIT_TIMEOUT;
+    let offset = 0;
+    let waited = 0;
+    let completed = false;
+    while (!completed) {
+      const chunks = await this.getChunks(sessionId);
+      const lastSkippedIndex = chunks.lastIndexOf(SKIPPED_MARK);
+      if (lastSkippedIndex >= offset) {
+        offset = lastSkippedIndex + 1;
+      }
+      while (offset < chunks.length) {
+        const chunk = chunks[offset++];
+        yield chunk;
+        waited = 0;
+        if (chunk.includes(STREAM_END_MARK)) {
+          completed = true;
+          break;
+        }
+      }
+      if (completed) {
+        return;
+      }
+      if (!chunks.length) {
+        if (offset > 0) {
+          return;
+        }
+        if (waited >= initialWaitTimeout) {
+          return;
+        }
+        waited += pollInterval;
+      }
+      await (0, import_utils.sleep)(pollInterval);
+    }
+  }
+  async getChunks(sessionId) {
+    const cache = await this.getCache();
+    return await cache.get(sessionId) ?? [];
+  }
+  async getCache() {
+    this.cachePromise ??= this.plugin.app.cacheManager.createCache({
+      name: CACHE_NAME,
+      store: this.plugin.app.cacheManager.defaultStore
+    });
+    return this.cachePromise;
+  }
+  async withLock(sessionId, fn) {
+    return await this.plugin.app.lockManager.runExclusive(this.getLockKey(sessionId), fn, WRITE_LOCK_TTL);
+  }
+  getLockKey(sessionId) {
+    return `${LOCK_KEY_PREFIX}:${sessionId}`;
+  }
+}
+class LLMStreamCached {
+  constructor(sessionId, manager) {
+    this.sessionId = sessionId;
+    this.manager = manager;
+  }
+  async clear() {
+    await this.manager.clear(this.sessionId);
+  }
+  async append(chunk) {
+    await this.manager.append(this.sessionId, chunk);
+  }
+  async skipped() {
+    await this.append(SKIPPED_MARK);
+  }
+  async *stream(options) {
+    for await (const chunk of this.manager.stream(this.sessionId, options)) {
+      yield chunk;
+    }
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  LLMStreamCached,
+  LLMStreamCachedManager
+});

package/dist/server/migrations/20260407170416-ai-employee-knowledge-base-add-key.d.ts

@@ -0,0 +1,14 @@
+/**
+ * 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 { Migration } from '@nocobase/server';
+export default class extends Migration {
+    on: string;
+    appVersion: string;
+    up(): Promise<void>;
+}

package/dist/server/migrations/20260407170416-ai-employee-knowledge-base-add-key.js

@@ -0,0 +1,61 @@
+/**
+ * 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 __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+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 __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var ai_employee_knowledge_base_add_key_exports = {};
+__export(ai_employee_knowledge_base_add_key_exports, {
+  default: () => ai_employee_knowledge_base_add_key_default
+});
+module.exports = __toCommonJS(ai_employee_knowledge_base_add_key_exports);
+var import_server = require("@nocobase/server");
+class ai_employee_knowledge_base_add_key_default extends import_server.Migration {
+  on = "afterSync";
+  // 'beforeLoad' or 'afterLoad'
+  appVersion = "<2.1.0";
+  async up() {
+    var _a, _b, _c, _d;
+    const aiEmployeesRepo = this.app.db.getRepository("aiEmployees");
+    const aiEmployeeList = await aiEmployeesRepo.find();
+    for (const item of aiEmployeeList) {
+      if (!((_b = (_a = item.knowledgeBase) == null ? void 0 : _a.knowledgeBaseIds) == null ? void 0 : _b.length)) {
+        continue;
+      }
+      if ((_d = (_c = item.knowledgeBase) == null ? void 0 : _c.knowledgeBaseKeys) == null ? void 0 : _d.length) {
+        continue;
+      }
+      await aiEmployeesRepo.update({
+        values: {
+          knowledgeBase: {
+            ...item.knowledgeBase,
+            knowledgeBaseKeys: item.knowledgeBase.knowledgeBaseIds.map((it) => String(it))
+          }
+        },
+        filter: {
+          username: item.username
+        }
+      });
+    }
+  }
+}

package/dist/server/plugin.d.ts

@@ -20,11 +20,13 @@ import { DocumentLoaders } from './document-loader';
 import type PluginFileManagerServer from '@nocobase/plugin-file-manager';
 import { SubAgentsDispatcher } from './ai-employees/sub-agents';
 import { KnowledgeBaseManager } from './ai-employees/ai-knowledge-base';
+import { LLMStreamCachedManager } from './manager/llm-stream-manager';
 export declare class PluginAIServer extends Plugin {
     features: AIPluginFeatureManagerImpl;
     aiManager: AIManager;
     aiEmployeesManager: AIEmployeesManager;
     aiConversationsManager: AIConversationsManager;
+    llmStreamCachedManager: LLMStreamCachedManager;
     builtInManager: BuiltInManager;
     aiContextDatasourceManager: AIContextDatasourceManager;
     aiCodingManager: AICodingManager;

package/dist/server/plugin.js

@@ -77,11 +77,13 @@ var import_mimo = require("./llm-providers/mimo");
 var import_sub_agents = require("./ai-employees/sub-agents");
 var import_employee = require("./workflow/nodes/employee");
 var import_ai_knowledge_base = require("./ai-employees/ai-knowledge-base");
+var import_llm_stream_manager = require("./manager/llm-stream-manager");
 class PluginAIServer extends import_server.Plugin {
   features = new import_ai_feature_manager.AIPluginFeatureManagerImpl();
   aiManager = new import_ai_manager.AIManager(this);
   aiEmployeesManager = new import_ai_employees_manager.AIEmployeesManager(this);
   aiConversationsManager = new import_ai_conversations.AIConversationsManager(this);
+  llmStreamCachedManager = new import_llm_stream_manager.LLMStreamCachedManager(this);
   builtInManager = new import_built_in_manager.BuiltInManager(this);
   aiContextDatasourceManager = new import_ai_context_datasource_manager.AIContextDatasourceManager(this);
   aiCodingManager = new import_ai_coding_manager.AICodingManager(this);
@@ -135,6 +137,7 @@ class PluginAIServer extends import_server.Plugin {
     this.registerWorkflow();
     this.registerWorkContextResolveStrategy();
     (0, import_employee.registerAIEmployeeTaskNotification)(this);
+    (0, import_ai_conversations.registerAIConversationReadNotification)(this);
     (0, import_employee.registerOnJobAbortedHandler)(this);
   }
   registerLLMProviders() {

package/dist/server/resource/ai.js

@@ -40,7 +40,6 @@ __export(ai_exports, {
 });
 module.exports = __toCommonJS(ai_exports);
 var import_lodash = __toESM(require("lodash"));
-var import_recommended_models = require("../../common/recommended-models");
 const aiResource = {
   name: "ai",
   actions: {
@@ -157,46 +156,7 @@ const aiResource = {
     },
     listAllEnabledModels: async (ctx, next) => {
       const plugin = ctx.app.pm.get("ai");
-      const services = await ctx.db.getRepository("llmServices").find({ sort: "sort" });
-      const llmServices = services.filter((service) => service.enabled !== false).map((service) => {
-        const raw = service.enabledModels;
-        let enabledModels;
-        if (raw && typeof raw === "object" && !Array.isArray(raw) && raw.mode) {
-          if (raw.mode === "recommended") {
-            enabledModels = (0, import_recommended_models.getRecommendedModels)(service.provider);
-          } else {
-            enabledModels = (raw.models || []).filter((m) => m.value).map((m) => ({
-              label: m.label || m.value,
-              value: m.value
-            }));
-          }
-        } else if (Array.isArray(raw)) {
-          if (raw.length === 0) {
-            enabledModels = (0, import_recommended_models.getRecommendedModels)(service.provider);
-          } else {
-            enabledModels = raw.map((id) => ({ label: id, value: id }));
-          }
-        } else {
-          enabledModels = (0, import_recommended_models.getRecommendedModels)(service.provider);
-        }
-        if (enabledModels.length === 0) {
-          return null;
-        }
-        const providerMeta = plugin.aiManager.llmProviders.get(service.provider);
-        const P = providerMeta.provider;
-        const p = new P({ app: ctx.app });
-        const isToolConflict = p.isToolConflict();
-        return {
-          llmService: service.name,
-          llmServiceTitle: service.title,
-          provider: service.provider,
-          providerTitle: providerMeta == null ? void 0 : providerMeta.title,
-          enabledModels,
-          supportWebSearch: (providerMeta == null ? void 0 : providerMeta.supportWebSearch) ?? false,
-          isToolConflict
-        };
-      }).filter(Boolean);
-      ctx.body = llmServices;
+      ctx.body = await plugin.aiManager.listAllEnabledModels();
       await next();
     }
   }

package/dist/server/resource/aiConversations.d.ts

@@ -7,10 +7,17 @@
  * For more information, please refer to: https://www.nocobase.com/agreement.
  */
 import { Context, Next } from '@nocobase/actions';
+declare function parallelConversationsLimit(ctx: Context, next: Next): Promise<never>;
 declare const _default: {
     name: string;
+    middlewares: {
+        only: string[];
+        handler: typeof parallelConversationsLimit;
+    }[];
     actions: {
         list(ctx: Context, next: Next): Promise<void>;
+        unreadCount(ctx: Context, next: Next): Promise<never>;
+        unreadCounts(ctx: Context, next: Next): Promise<never>;
         create(ctx: Context, next: Next): Promise<never>;
         update(ctx: Context, next: Next): Promise<never>;
         updateOptions(ctx: Context, next: Next): Promise<never>;
@@ -19,6 +26,7 @@ declare const _default: {
         updateToolArgs(ctx: Context, next: Next): Promise<any>;
         sendMessages(ctx: Context, next: Next): Promise<any>;
         abort(ctx: Context, next: Next): Promise<never>;
+        resumeStream(ctx: Context, next: Next): Promise<never>;
         resendMessages(ctx: Context, next: Next): Promise<any>;
         updateUserDecision(ctx: Context, next: Next): Promise<never>;
         resumeToolCall(ctx: Context, next: Next): Promise<any>;

package/dist/server/resource/aiConversations.js

@@ -40,6 +40,7 @@ __export(aiConversations_exports, {
 });
 module.exports = __toCommonJS(aiConversations_exports);
 var import_actions = __toESM(require("@nocobase/actions"));
+var import_database = require("@nocobase/database");
 var import_utils = require("../utils");
 var import_ai_employee = require("../ai-employees/ai-employee");
 async function getAIEmployee(ctx, username) {
@@ -66,8 +67,35 @@ function setupSSEHeaders(ctx) {
 function sendErrorResponse(ctx, errorMessage) {
   (0, import_utils.sendSSEError)(ctx, errorMessage);
 }
+async function parallelConversationsLimit(ctx, next) {
+  var _a;
+  const userId = (_a = ctx.auth) == null ? void 0 : _a.user.id;
+  if (!userId) {
+    return ctx.throw(403);
+  }
+  const activeStreamCount = await ctx.db.getModel("aiConversations").count({
+    where: {
+      userId,
+      llmActiveState: "streaming",
+      updatedAt: {
+        [import_database.Op.gte]: new Date(Date.now() - 10 * 60 * 1e3)
+      }
+    }
+  });
+  if (activeStreamCount > 2) {
+    sendErrorResponse(ctx, ctx.t("There are conversations in progress. Please try again later."));
+    return;
+  }
+  await next();
+}
 var aiConversations_default = {
   name: "aiConversations",
+  middlewares: [
+    {
+      only: ["sendMessages", "resendMessages", "resumeStream"],
+      handler: parallelConversationsLimit
+    }
+  ],
   actions: {
     async list(ctx, next) {
       var _a;
@@ -86,6 +114,53 @@ var aiConversations_default = {
       });
       return import_actions.default.list(ctx, next);
     },
+    async unreadCount(ctx, next) {
+      var _a;
+      const userId = (_a = ctx.auth) == null ? void 0 : _a.user.id;
+      if (!userId) {
+        return ctx.throw(403);
+      }
+      const count = await ctx.db.getModel("aiConversations").count({
+        where: {
+          userId,
+          read: false,
+          from: "main-agent",
+          category: "chat"
+        }
+      });
+      ctx.body = {
+        count
+      };
+      await next();
+    },
+    async unreadCounts(ctx, next) {
+      var _a;
+      const userId = (_a = ctx.auth) == null ? void 0 : _a.user.id;
+      if (!userId) {
+        return ctx.throw(403);
+      }
+      const [conversationUnreadCount, workflowTaskUnreadCount] = await Promise.all([
+        ctx.db.getModel("aiConversations").count({
+          where: {
+            userId,
+            read: false,
+            from: "main-agent",
+            category: "chat"
+          }
+        }),
+        ctx.db.getModel("usersAiWorkflowTasks").count({
+          where: {
+            userId,
+            read: false
+          }
+        })
+      ]);
+      ctx.body = {
+        conversationUnreadCount,
+        workflowTaskUnreadCount
+      };
+      await next();
+    },
     async create(ctx, next) {
       var _a;
       const plugin = ctx.app.pm.get("ai");
@@ -93,7 +168,7 @@ var aiConversations_default = {
       if (!userId) {
         return ctx.throw(403);
       }
-      const { aiEmployee, systemMessage, skillSettings, conversationSettings } = ctx.action.params.values || {};
+      const { aiEmployee, systemMessage, skillSettings, conversationSettings, modelSettings } = ctx.action.params.values || {};
       const employee = await getAIEmployee(ctx, aiEmployee.username);
       if (!employee) {
         ctx.throw(400, "AI employee not found");
@@ -105,7 +180,8 @@ var aiConversations_default = {
           options: {
             systemMessage,
             skillSettings,
-            conversationSettings
+            conversationSettings,
+            modelSettings
           }
         });
       } catch (error) {
@@ -139,15 +215,15 @@ var aiConversations_default = {
       if (!sessionId) {
         return ctx.throw(400, "invalid sessionId");
       }
-      const { systemMessage, skillSettings, conversationSettings } = ctx.action.params.values || {};
-      if (!systemMessage && !skillSettings && !conversationSettings) {
+      const { systemMessage, skillSettings, conversationSettings, modelSettings } = ctx.action.params.values || {};
+      if (!systemMessage && !skillSettings && !conversationSettings && !modelSettings) {
         return ctx.throw(400, "invalid options");
       }
       try {
         ctx.body = await plugin.aiConversationsManager.update({
           userId,
           sessionId,
-          options: { systemMessage, skillSettings, conversationSettings }
+          options: { systemMessage, skillSettings, conversationSettings, modelSettings }
         });
       } catch (error) {
         if (error.message === "invalid sessionId") {
@@ -177,17 +253,19 @@ var aiConversations_default = {
       if (!userId) {
         return ctx.throw(403);
       }
-      const { sessionId, cursor } = ctx.action.params || {};
+      const { sessionId, cursor, updateRead: originalUpdateRead } = ctx.action.params || {};
       if (!sessionId) {
         ctx.throw(400);
       }
       const paginate = ((_b = ctx.action.params) == null ? void 0 : _b.paginate) === "false" ? false : true;
+      const updateRead = originalUpdateRead === "true" || originalUpdateRead === true;
       try {
         ctx.body = await plugin.aiConversationsManager.getMessages({
           userId,
           sessionId,
           cursor,
-          paginate
+          paginate,
+          updateRead
         });
       } catch (error) {
         if (error.message === "invalid sessionId") {
@@ -317,6 +395,7 @@ var aiConversations_default = {
           return next();
         }
         const legacy = conversation.thread === 0;
+        const resolvedModel = await plugin.aiEmployeesManager.resolveModel(employee, model);
         const aiEmployee = new import_ai_employee.AIEmployee({
           ctx,
           employee,
@@ -325,7 +404,7 @@ var aiConversations_default = {
           skillSettings: (_c = conversation.options) == null ? void 0 : _c.skillSettings,
           tools: (_d = conversation.options) == null ? void 0 : _d.tools,
           webSearch,
-          model,
+          model: resolvedModel,
           legacy
         });
         if (!editingMessageId) {
@@ -391,6 +470,56 @@ var aiConversations_default = {
       plugin.aiEmployeesManager.abortConversation(sessionId);
       await next();
     },
+    async resumeStream(ctx, next) {
+      var _a, _b, _c, _d, _e;
+      const plugin = ctx.app.pm.get("ai");
+      const userId = (_a = ctx.auth) == null ? void 0 : _a.user.id;
+      if (!userId) {
+        return ctx.throw(403);
+      }
+      setupSSEHeaders(ctx);
+      const sessionId = ((_b = ctx.action.params) == null ? void 0 : _b.sessionId) || ((_d = (_c = ctx.action.params) == null ? void 0 : _c.values) == null ? void 0 : _d.sessionId) || ((_e = ctx.action.params) == null ? void 0 : _e.filterByTk);
+      if (!sessionId) {
+        sendErrorResponse(ctx, "sessionId is required");
+        return;
+      }
+      try {
+        const conversation = await plugin.aiConversationsManager.getConversation({
+          sessionId,
+          userId
+        });
+        if (!conversation) {
+          sendErrorResponse(ctx, "conversation not found");
+          return;
+        }
+        let hasChunks = false;
+        for await (const chunk of plugin.llmStreamCachedManager.getCached(sessionId).stream()) {
+          hasChunks = true;
+          ctx.res.write(chunk);
+        }
+        if (!hasChunks) {
+          const currentConversation = await plugin.aiConversationsManager.getConversation({
+            sessionId,
+            userId
+          });
+          const llmActiveState = currentConversation == null ? void 0 : currentConversation.llmActiveState;
+          if (llmActiveState && llmActiveState !== "idle") {
+            ctx.res.write(`data: ${JSON.stringify({ type: "chunks_cache_missing", body: { llmActiveState } })}
+
+`);
+          }
+        }
+      } catch (err) {
+        ctx.log.error(err);
+        sendErrorResponse(ctx, err.message || "Resume stream error");
+        return;
+      } finally {
+        if (!ctx.res.writableEnded) {
+          ctx.res.end();
+        }
+        await next();
+      }
+    },
     async resendMessages(ctx, next) {
       var _a, _b, _c, _d;
       const plugin = ctx.app.pm.get("ai");
@@ -453,6 +582,7 @@ var aiConversations_default = {
           sendErrorResponse(ctx, "AI employee not found");
           return next();
         }
+        const resolvedModel = await plugin.aiEmployeesManager.resolveModel(employee, model);
         const aiEmployee = new import_ai_employee.AIEmployee({
           ctx,
           employee,
@@ -461,7 +591,7 @@ var aiConversations_default = {
           skillSettings: (_c = conversation.options) == null ? void 0 : _c.skillSettings,
           tools: (_d = conversation.options) == null ? void 0 : _d.tools,
           webSearch,
-          model
+          model: resolvedModel
         });
         await aiEmployee.stream({ messageId, userMessages: resendMessages.length ? resendMessages : void 0 });
       } catch (err) {
@@ -601,6 +731,7 @@ var aiConversations_default = {
           sendErrorResponse(ctx, "No tool calls found");
           return next();
         }
+        const resolvedModel = await plugin.aiEmployeesManager.resolveModel(employee, model);
         const aiEmployee = new import_ai_employee.AIEmployee({
           ctx,
           employee,
@@ -609,7 +740,7 @@ var aiConversations_default = {
           skillSettings: (_c = conversation.options) == null ? void 0 : _c.skillSettings,
           tools: (_d = conversation.options) == null ? void 0 : _d.tools,
           webSearch,
-          model
+          model: resolvedModel
         });
         const userDecisions = await plugin.aiConversationsManager.getUserDecisions(messageId);
         await aiEmployee.stream({

package/dist/server/resource/aiEmployees.js

@@ -45,7 +45,9 @@ module.exports = __toCommonJS(aiEmployees_exports);
 var import_actions = __toESM(require("@nocobase/actions"));
 var templates = __toESM(require("../ai-employees/templates"));
 var import_lodash = __toESM(require("lodash"));
+var import_ai_feature_manager = require("../manager/ai-feature-manager");
 const list = async (ctx, next) => {
+  var _a, _b;
   const { paginate } = ctx.action.params || {};
   const plugin = ctx.app.pm.get("ai");
   const builtInManager = plugin.builtInManager;
@@ -58,9 +60,36 @@ const list = async (ctx, next) => {
   });
   await import_actions.default.list(ctx, () => {
   });
-  let data = ctx.body.rows;
+  let data;
   if (paginate === "false" || paginate === false) {
+    ctx.body = ctx.body.map((it) => it.toJSON());
     data = ctx.body;
+  } else {
+    ctx.body.rows = ctx.body.rows.map((it) => it.toJSON());
+    data = ctx.body.rows;
+  }
+  const featureEnabled = plugin.features.isFeaturesEnabled(Object.values(import_ai_feature_manager.EEFeatures));
+  if (featureEnabled) {
+    const knowledgeBaseKeys = import_lodash.default.uniq(
+      data.map((it) => {
+        var _a2;
+        return ((_a2 = it.knowledgeBase) == null ? void 0 : _a2.knowledgeBaseKeys) ?? [];
+      }).flatMap((it) => it)
+    );
+    const knowledgeBaseList = await plugin.features.knowledgeBase.getKnowledgeBase(knowledgeBaseKeys);
+    const existedKnowledgeBaseKeys = (knowledgeBaseList == null ? void 0 : knowledgeBaseList.map((it) => it.key)) ?? [];
+    for (const row of data) {
+      row.missingKnowledgeBaseKeys = [];
+      if (!((_b = (_a = row.knowledgeBase) == null ? void 0 : _a.knowledgeBaseKeys) == null ? void 0 : _b.length)) {
+        continue;
+      }
+      for (const k of row.knowledgeBase.knowledgeBaseKeys) {
+        if (existedKnowledgeBaseKeys.includes(k)) {
+          continue;
+        }
+        row.missingKnowledgeBaseKeys.push(k);
+      }
+    }
   }
   data.forEach((row) => {
     if (row.builtIn) {
@@ -153,6 +182,8 @@ const listByUser = async (ctx, next) => {
         prompt: (_b = (_a = row.userConfigs) == null ? void 0 : _a[0]) == null ? void 0 : _b.prompt
       },
       skillSettings,
+      chatSettings: row.chatSettings,
+      modelSettings: row.modelSettings,
       builtIn: row.builtIn,
       category: row.category,
       deprecated: row.deprecated

package/dist/server/types/knowledge-base.type.d.ts

@@ -15,16 +15,17 @@ export type VectorStoreProp = {
 export type KnowledgeBase = {
     knowledgeBaseType: KnowledgeBaseType;
     knowledgeBaseOuterId: string;
+    key: string;
     name: string;
     description: string;
     vectorStoreProvider: string;
-    vectorStoreConfigId?: string;
+    vectorStoreConfigKey?: string;
     vectorStoreProps?: VectorStoreProp[];
     enabled: boolean;
 };
 export type VectorStoreConfig = {
     vectorStoreProvider: string;
-    vectorStoreConfigId?: string;
+    vectorStoreConfigKey?: string;
 };
 export type KnowledgeBaseGroup = {
     vectorStoreConfig: VectorStoreConfig;