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

package/dist/server/index.d.ts

@@ -11,6 +11,8 @@ export { LLMProvider, LLMProviderOptions } from './llm-providers/provider';
 export { LLMProviderMeta } from './manager/ai-manager';
 export { ToolOptions } from './manager/tool-manager';
 export { DocumentLoader } from './document-loader';
+export { AIEmployee } from './ai-employees/ai-employee';
+export type { AIEmployeeOptions, ModelRef } from './ai-employees/ai-employee';
 export type { ParsedDocumentResult, ParseableFile, DocumentParseMeta } from './document-loader';
 export type * from './features';
 export type * from './types';

package/dist/server/index.js

@@ -36,6 +36,7 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 var server_exports = {};
 __export(server_exports, {
+  AIEmployee: () => import_ai_employee.AIEmployee,
   DocumentLoader: () => import_document_loader.DocumentLoader,
   LLMProvider: () => import_provider.LLMProvider,
   LLMProviderMeta: () => import_ai_manager.LLMProviderMeta,
@@ -49,8 +50,10 @@ var import_provider = require("./llm-providers/provider");
 var import_ai_manager = require("./manager/ai-manager");
 var import_tool_manager = require("./manager/tool-manager");
 var import_document_loader = require("./document-loader");
+var import_ai_employee = require("./ai-employees/ai-employee");
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  AIEmployee,
   DocumentLoader,
   LLMProvider,
   LLMProviderMeta,

package/dist/server/llm-providers/common/reasoning.d.ts

@@ -10,8 +10,10 @@ import { BaseMessage } from '@langchain/core/messages';
 import { ChatOpenAICompletions } from '@langchain/openai';
 import type OpenAI from 'openai';
 export declare const REASONING_MAP_KEY = "__nb_reasoning_map";
+export declare const MODEL_KWARGS_KEY = "__nb_model_kwargs";
 export declare const collectReasoningMap: (messages: BaseMessage[]) => Map<string, string>;
 export declare const patchRequestMessagesReasoning: (request: any, reasoningMap?: Map<string, string>) => void;
+export declare const patchRequestModelKwargs: (request: any, modelKwargs?: Record<string, any>) => void;
 export declare class ReasoningChatOpenAI extends ChatOpenAICompletions {
     _generate(messages: BaseMessage[], options: any, runManager?: any): Promise<import("@langchain/core/dist/outputs").ChatResult>;
     _streamResponseChunks(messages: BaseMessage[], options: any, runManager?: any): AsyncGenerator<import("@langchain/core/dist/outputs").ChatGenerationChunk, void, unknown>;

package/dist/server/llm-providers/common/reasoning.js

@@ -26,15 +26,18 @@ var __copyProps = (to, from, except, desc) => {
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 var reasoning_exports = {};
 __export(reasoning_exports, {
+  MODEL_KWARGS_KEY: () => MODEL_KWARGS_KEY,
   REASONING_MAP_KEY: () => REASONING_MAP_KEY,
   ReasoningChatOpenAI: () => ReasoningChatOpenAI,
   collectReasoningMap: () => collectReasoningMap,
-  patchRequestMessagesReasoning: () => patchRequestMessagesReasoning
+  patchRequestMessagesReasoning: () => patchRequestMessagesReasoning,
+  patchRequestModelKwargs: () => patchRequestModelKwargs
 });
 module.exports = __toCommonJS(reasoning_exports);
 var import_messages = require("@langchain/core/messages");
 var import_openai = require("@langchain/openai");
 const REASONING_MAP_KEY = "__nb_reasoning_map";
+const MODEL_KWARGS_KEY = "__nb_model_kwargs";
 const collectReasoningMap = (messages) => {
   var _a;
   const reasoningMap = /* @__PURE__ */ new Map();
@@ -69,6 +72,12 @@ const patchRequestMessagesReasoning = (request, reasoningMap) => {
     }
   }
 };
+const patchRequestModelKwargs = (request, modelKwargs) => {
+  if (!modelKwargs || typeof modelKwargs !== "object") {
+    return;
+  }
+  Object.assign(request, modelKwargs);
+};
 class ReasoningChatOpenAI extends import_openai.ChatOpenAICompletions {
   async _generate(messages, options, runManager) {
     const reasoningMap = collectReasoningMap(messages);
@@ -108,7 +117,9 @@ class ReasoningChatOpenAI extends import_openai.ChatOpenAICompletions {
   }
   async completionWithRetry(request, requestOptions) {
     const reasoningMap = requestOptions == null ? void 0 : requestOptions[REASONING_MAP_KEY];
+    const modelKwargs = requestOptions == null ? void 0 : requestOptions[MODEL_KWARGS_KEY];
     patchRequestMessagesReasoning(request, reasoningMap);
+    patchRequestModelKwargs(request, modelKwargs);
     if (request.stream) {
       return super.completionWithRetry(request, requestOptions);
     }
@@ -117,8 +128,10 @@ class ReasoningChatOpenAI extends import_openai.ChatOpenAICompletions {
 }
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  MODEL_KWARGS_KEY,
   REASONING_MAP_KEY,
   ReasoningChatOpenAI,
   collectReasoningMap,
-  patchRequestMessagesReasoning
+  patchRequestMessagesReasoning,
+  patchRequestModelKwargs
 });

package/dist/server/llm-providers/dashscope.d.ts

@@ -7,7 +7,7 @@
  * For more information, please refer to: https://www.nocobase.com/agreement.
  */
 import { AIMessageChunk } from '@langchain/core/messages';
-import { EmbeddingProvider, LLMProvider } from './provider';
+import { EmbeddingProvider, LLMModelRequestBuilder, LLMProvider } from './provider';
 import { EmbeddingsInterface } from '@langchain/core/embeddings';
 import { SupportedModel } from '../manager/ai-manager';
 import { Model } from '@nocobase/database';
@@ -19,6 +19,7 @@ export declare class DashscopeProvider extends LLMProvider {
     createModel(): ReasoningChatOpenAI;
     isToolConflict(): boolean;
     resolveTools(toolDefinitions: any[]): any[];
+    protected getModelRequestBuilder(model?: string): LLMModelRequestBuilder | null;
     parseResponseMessage(message: Model): {
         key: any;
         createdAt: any;

package/dist/server/llm-providers/dashscope.js

@@ -41,6 +41,7 @@ __export(dashscope_exports, {
   dashscopeProviderOptions: () => dashscopeProviderOptions
 });
 module.exports = __toCommonJS(dashscope_exports);
+var import_messages = require("@langchain/core/messages");
 var import_openai = require("@langchain/openai");
 var import_provider = require("./provider");
 var import_ai_manager = require("../manager/ai-manager");
@@ -94,6 +95,44 @@ class DashscopeProvider extends import_provider.LLMProvider {
       return toolDefinitions;
     }
   }
+  getModelRequestBuilder(model) {
+    if (!/^qwen-mt-/i.test(model || "")) {
+      return null;
+    }
+    return ({ context, options }) => {
+      var _a, _b;
+      const { sourceText, sourceLang, targetLang, terms } = (options == null ? void 0 : options.modelRequestParams) || {};
+      if (!sourceText || !sourceLang || !targetLang) {
+        return {
+          context,
+          options
+        };
+      }
+      (_b = this.app.logger) == null ? void 0 : _b.debug("Dashscope Qwen-MT model request builder matched", {
+        model: (_a = this.modelOptions) == null ? void 0 : _a.model,
+        sourceTextLength: (sourceText == null ? void 0 : sourceText.length) ?? 0,
+        sourceLang,
+        targetLang,
+        terms: Array.isArray(terms) ? terms.length : 0
+      });
+      return {
+        context: {
+          messages: [new import_messages.HumanMessage(sourceText)]
+        },
+        options: {
+          ...options,
+          modelKwargs: {
+            ...(options == null ? void 0 : options.modelKwargs) || {},
+            translation_options: {
+              source_lang: sourceLang,
+              target_lang: targetLang,
+              ...Array.isArray(terms) && terms.length ? { terms } : {}
+            }
+          }
+        }
+      };
+    };
+  }
   parseResponseMessage(message) {
     var _a;
     const result = super.parseResponseMessage(message);

package/dist/server/llm-providers/deepseek.js

@@ -64,7 +64,9 @@ class ReasoningDeepSeek extends import_deepseek.ChatDeepSeek {
   }
   async completionWithRetry(request, requestOptions) {
     const reasoningMap = requestOptions == null ? void 0 : requestOptions[import_reasoning.REASONING_MAP_KEY];
+    const modelKwargs = requestOptions == null ? void 0 : requestOptions[import_reasoning.MODEL_KWARGS_KEY];
     (0, import_reasoning.patchRequestMessagesReasoning)(request, reasoningMap);
+    (0, import_reasoning.patchRequestModelKwargs)(request, modelKwargs);
     if (request.stream) {
       return super.completionWithRetry(request, requestOptions);
     }

package/dist/server/llm-providers/provider.d.ts

@@ -22,6 +22,19 @@ export type ParsedAttachmentResult = {
     placement: string;
     content: any;
 };
+export type LLMProviderInvokeOptions = {
+    modelKwargs?: Record<string, any>;
+    modelRequestParams?: Record<string, any>;
+    [key: string]: any;
+};
+export type LLMModelRequestBuilderResult = {
+    context: AIChatContext;
+    options?: LLMProviderInvokeOptions;
+};
+export type LLMModelRequestBuilder = (input: {
+    context: AIChatContext;
+    options?: LLMProviderInvokeOptions;
+}) => LLMModelRequestBuilderResult;
 export interface LLMProviderOptions {
     app: Application;
     serviceOptions?: Record<string, any>;
@@ -35,8 +48,9 @@ export declare abstract class LLMProvider {
     abstract createModel(): BaseChatModel | any;
     get baseURL(): string | null;
     constructor(opts: LLMProviderOptions);
+    protected getModelRequestBuilder(_model?: string): LLMModelRequestBuilder | null;
     prepareChain(context: AIChatContext): any;
-    invoke(context: AIChatContext, options?: any): Promise<any>;
+    invoke(context: AIChatContext, options?: LLMProviderInvokeOptions): Promise<any>;
     stream(context: AIChatContext, options?: any): Promise<any>;
     listModels(): Promise<{
         models?: {

package/dist/server/llm-providers/provider.js

@@ -45,6 +45,7 @@ var import_utils = require("../utils");
 var import_stream = require("@langchain/core/utils/stream");
 var import_document_loader = require("../document-loader");
 var import_node_path = __toESM(require("node:path"));
+var import_reasoning = require("./common/reasoning");
 class LLMProvider {
   app;
   serviceOptions;
@@ -62,6 +63,9 @@ class LLMProvider {
       this.chatModel = this.createModel();
     }
   }
+  getModelRequestBuilder(_model) {
+    return null;
+  }
   prepareChain(context) {
     var _a, _b, _c, _d;
     let chain = this.chatModel;
@@ -84,8 +88,23 @@ class LLMProvider {
     return chain;
   }
   async invoke(context, options) {
-    const chain = this.prepareChain(context);
-    return chain.invoke(context.messages, options);
+    var _a;
+    const builder = this.getModelRequestBuilder((_a = this.modelOptions) == null ? void 0 : _a.model);
+    const request = (builder == null ? void 0 : builder({ context, options })) || { context, options };
+    const chain = this.prepareChain(request.context);
+    const { modelKwargs, modelRequestParams, options: requestOptions, ...restOptions } = request.options || {};
+    const invokeOptions = modelKwargs ? {
+      ...restOptions,
+      [import_reasoning.MODEL_KWARGS_KEY]: modelKwargs,
+      options: {
+        ...requestOptions || {},
+        [import_reasoning.MODEL_KWARGS_KEY]: modelKwargs
+      }
+    } : {
+      ...restOptions,
+      ...requestOptions ? { options: requestOptions } : {}
+    };
+    return chain.invoke(request.context.messages, invokeOptions);
   }
   async stream(context, options) {
     const chain = this.prepareChain(context);

package/dist/server/manager/ai-chat-conversation.js

@@ -138,11 +138,10 @@ class AIChatConversationImpl {
       formatMessages
     } = options ?? {};
     let messages = userMessages ? await (formatMessages == null ? void 0 : formatMessages(userMessages)) ?? [] : void 0;
-    const additionSystemPrompt = messages == null ? void 0 : messages.filter((it) => it.role === "system").map((it) => it.content).join("\n");
+    const additionSystemPrompt = messages == null ? void 0 : messages.filter((it) => it.role === "system").map((it) => it.content).filter(Boolean).join("\n");
     messages = messages == null ? void 0 : messages.filter((it) => it.role !== "system");
-    const systemPrompt = `${await (getSystemPrompt == null ? void 0 : getSystemPrompt(userMessages ?? [])) ?? ""}
-
-${additionSystemPrompt}`;
+    const baseSystemPrompt = await (getSystemPrompt == null ? void 0 : getSystemPrompt(userMessages ?? []));
+    const systemPrompt = [baseSystemPrompt, additionSystemPrompt].filter(Boolean).join("\n\n") || void 0;
     const chatContext = {
       systemPrompt,
       messages,

package/dist/server/manager/ai-manager.d.ts

@@ -26,6 +26,19 @@ export type LLMModelOptions = {
     model: string;
     webSearch?: boolean;
 };
+export type EnabledLLMModel = {
+    label: string;
+    value: string;
+};
+export type EnabledLLMService = {
+    llmService: string;
+    llmServiceTitle: string;
+    provider: string;
+    providerTitle?: string;
+    enabledModels: EnabledLLMModel[];
+    supportWebSearch: boolean;
+    isToolConflict: boolean;
+};
 export declare class AIManager {
     protected plugin: PluginAIServer;
     llmProviders: Map<string, LLMProviderMeta>;
@@ -39,6 +52,10 @@ export declare class AIManager {
         supportWebSearch: boolean;
     }[];
     getSupportedProvider(model: SupportedModel): string[];
+    listAllEnabledModels(): Promise<EnabledLLMService[]>;
+    resolveModel(model?: LLMModelOptions | null): Promise<LLMModelOptions>;
+    private toEnabledLLMService;
+    private getEnabledModels;
     getLLMService(options: LLMModelOptions): Promise<{
         provider: LLMProvider;
         model: string;

package/dist/server/manager/ai-manager.js

@@ -31,6 +31,7 @@ __export(ai_manager_exports, {
 });
 module.exports = __toCommonJS(ai_manager_exports);
 var import_tool_manager = require("./tool-manager");
+var import_recommended_models = require("../../common/recommended-models");
 var SupportedModel = /* @__PURE__ */ ((SupportedModel2) => {
   SupportedModel2["LLM"] = "LLM";
   SupportedModel2["EMBEDDING"] = "EMBEDDING";
@@ -57,6 +58,70 @@ class AIManager {
   getSupportedProvider(model) {
     return Array.from(this.llmProviders.entries()).filter(([_2, { supportedModel }]) => supportedModel && supportedModel.includes(model)).map(([name]) => name);
   }
+  async listAllEnabledModels() {
+    const services = await this.plugin.db.getRepository("llmServices").find({ sort: "sort" });
+    return services.filter((service) => service.enabled !== false).map((service) => this.toEnabledLLMService(service)).filter(Boolean);
+  }
+  async resolveModel(model) {
+    var _a;
+    if ((model == null ? void 0 : model.llmService) && (model == null ? void 0 : model.model)) {
+      return model;
+    }
+    const services = await this.listAllEnabledModels();
+    const service = services.find((service2) => service2.enabledModels.length);
+    const firstModel = (_a = service == null ? void 0 : service.enabledModels[0]) == null ? void 0 : _a.value;
+    if ((service == null ? void 0 : service.llmService) && firstModel) {
+      return {
+        llmService: service.llmService,
+        model: firstModel
+      };
+    }
+    throw new Error("LLM service not configured");
+  }
+  toEnabledLLMService(service) {
+    var _a, _b, _c;
+    const provider = ((_a = service.get) == null ? void 0 : _a.call(service, "provider")) || service.provider;
+    const providerMeta = this.llmProviders.get(provider);
+    if (!providerMeta) {
+      return null;
+    }
+    const enabledModels = this.getEnabledModels(service);
+    if (!enabledModels.length) {
+      return null;
+    }
+    const Provider = providerMeta.provider;
+    const providerClient = new Provider({ app: this.plugin.app });
+    return {
+      llmService: ((_b = service.get) == null ? void 0 : _b.call(service, "name")) || service.name,
+      llmServiceTitle: ((_c = service.get) == null ? void 0 : _c.call(service, "title")) || service.title,
+      provider,
+      providerTitle: providerMeta.title,
+      enabledModels,
+      supportWebSearch: providerMeta.supportWebSearch ?? false,
+      isToolConflict: providerClient.isToolConflict()
+    };
+  }
+  getEnabledModels(service) {
+    var _a, _b;
+    const provider = ((_a = service.get) == null ? void 0 : _a.call(service, "provider")) || service.provider;
+    const raw = ((_b = service.get) == null ? void 0 : _b.call(service, "enabledModels")) || service.enabledModels;
+    if (raw && typeof raw === "object" && !Array.isArray(raw) && raw.mode) {
+      if (raw.mode === "recommended") {
+        return (0, import_recommended_models.getRecommendedModels)(provider);
+      }
+      return (raw.models || []).filter((model) => model.value).map((model) => ({
+        label: model.label || model.value,
+        value: model.value
+      }));
+    }
+    if (Array.isArray(raw)) {
+      if (!raw.length) {
+        return (0, import_recommended_models.getRecommendedModels)(provider);
+      }
+      return raw.map((id) => ({ label: id, value: id }));
+    }
+    return (0, import_recommended_models.getRecommendedModels)(provider);
+  }
   async getLLMService(options) {
     const { llmService, model, webSearch } = options ?? {};
     if (!llmService || !model) {

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/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.js

@@ -168,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");
@@ -180,7 +180,8 @@ var aiConversations_default = {
           options: {
             systemMessage,
             skillSettings,
-            conversationSettings
+            conversationSettings,
+            modelSettings
           }
         });
       } catch (error) {
@@ -214,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") {
@@ -394,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,
@@ -402,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) {
@@ -580,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,
@@ -588,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) {
@@ -728,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,
@@ -736,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