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

package/dist/node_modules/openai/package.json

@@ -1 +1 @@
-{"name":"openai","version":"6.22.0","description":"The official TypeScript library for the OpenAI API","author":"OpenAI <support@openai.com>","types":"./index.d.ts","main":"./index.js","type":"commonjs","repository":"github:openai/openai-node","license":"Apache-2.0","packageManager":"yarn@1.22.22","files":["**/*"],"private":false,"publishConfig":{"access":"public"},"scripts":{"test":"./scripts/test","build":"./scripts/build","format":"./scripts/format","tsn":"ts-node -r tsconfig-paths/register","lint":"./scripts/lint","fix":"./scripts/format"},"dependencies":{},"bin":{"openai":"bin/cli"},"exports":{".":{"require":{"types":"./index.d.ts","default":"./index.js"},"types":"./index.d.mts","default":"./index.mjs"},"./_vendor/*.mjs":{"default":"./_vendor/*.mjs"},"./_vendor/*.js":{"default":"./_vendor/*.js"},"./_vendor/*":{"import":"./_vendor/*.mjs","require":"./_vendor/*.js"},"./api-promise":{"import":"./api-promise.mjs","require":"./api-promise.js"},"./api-promise.js":{"default":"./api-promise.js"},"./api-promise.mjs":{"default":"./api-promise.mjs"},"./azure":{"import":"./azure.mjs","require":"./azure.js"},"./azure.js":{"default":"./azure.js"},"./azure.mjs":{"default":"./azure.mjs"},"./beta/*.mjs":{"default":"./beta/*.mjs"},"./beta/*.js":{"default":"./beta/*.js"},"./beta/*":{"import":"./beta/*.mjs","require":"./beta/*.js"},"./client":{"import":"./client.mjs","require":"./client.js"},"./client.js":{"default":"./client.js"},"./client.mjs":{"default":"./client.mjs"},"./core/*.mjs":{"default":"./core/*.mjs"},"./core/*.js":{"default":"./core/*.js"},"./core/*":{"import":"./core/*.mjs","require":"./core/*.js"},"./error":{"import":"./error.mjs","require":"./error.js"},"./error.js":{"default":"./error.js"},"./error.mjs":{"default":"./error.mjs"},"./helpers/*.mjs":{"default":"./helpers/*.mjs"},"./helpers/*.js":{"default":"./helpers/*.js"},"./helpers/*":{"import":"./helpers/*.mjs","require":"./helpers/*.js"},"./index":{"import":"./index.mjs","require":"./index.js"},"./index.js":{"default":"./index.js"},"./index.mjs":{"default":"./index.mjs"},"./lib/*.mjs":{"default":"./lib/*.mjs"},"./lib/*.js":{"default":"./lib/*.js"},"./lib/*":{"import":"./lib/*.mjs","require":"./lib/*.js"},"./pagination":{"import":"./pagination.mjs","require":"./pagination.js"},"./pagination.js":{"default":"./pagination.js"},"./pagination.mjs":{"default":"./pagination.mjs"},"./realtime/*.mjs":{"default":"./realtime/*.mjs"},"./realtime/*.js":{"default":"./realtime/*.js"},"./realtime/*":{"import":"./realtime/*.mjs","require":"./realtime/*.js"},"./resource":{"import":"./resource.mjs","require":"./resource.js"},"./resource.js":{"default":"./resource.js"},"./resource.mjs":{"default":"./resource.mjs"},"./resources/*.mjs":{"default":"./resources/*.mjs"},"./resources/*.js":{"default":"./resources/*.js"},"./resources/*":{"import":"./resources/*.mjs","require":"./resources/*.js"},"./resources":{"import":"./resources.mjs","require":"./resources.js"},"./resources.js":{"default":"./resources.js"},"./resources.mjs":{"default":"./resources.mjs"},"./streaming":{"import":"./streaming.mjs","require":"./streaming.js"},"./streaming.js":{"default":"./streaming.js"},"./streaming.mjs":{"default":"./streaming.mjs"},"./uploads":{"import":"./uploads.mjs","require":"./uploads.js"},"./uploads.js":{"default":"./uploads.js"},"./uploads.mjs":{"default":"./uploads.mjs"},"./version":{"import":"./version.mjs","require":"./version.js"},"./version.js":{"default":"./version.js"},"./version.mjs":{"default":"./version.mjs"}},"peerDependencies":{"ws":"^8.18.0","zod":"^3.25 || ^4.0"},"peerDependenciesMeta":{"ws":{"optional":true},"zod":{"optional":true}},"_lastModified":"2026-05-11T23:08:12.732Z"}
+{"name":"openai","version":"6.22.0","description":"The official TypeScript library for the OpenAI API","author":"OpenAI <support@openai.com>","types":"./index.d.ts","main":"./index.js","type":"commonjs","repository":"github:openai/openai-node","license":"Apache-2.0","packageManager":"yarn@1.22.22","files":["**/*"],"private":false,"publishConfig":{"access":"public"},"scripts":{"test":"./scripts/test","build":"./scripts/build","format":"./scripts/format","tsn":"ts-node -r tsconfig-paths/register","lint":"./scripts/lint","fix":"./scripts/format"},"dependencies":{},"bin":{"openai":"bin/cli"},"exports":{".":{"require":{"types":"./index.d.ts","default":"./index.js"},"types":"./index.d.mts","default":"./index.mjs"},"./_vendor/*.mjs":{"default":"./_vendor/*.mjs"},"./_vendor/*.js":{"default":"./_vendor/*.js"},"./_vendor/*":{"import":"./_vendor/*.mjs","require":"./_vendor/*.js"},"./api-promise":{"import":"./api-promise.mjs","require":"./api-promise.js"},"./api-promise.js":{"default":"./api-promise.js"},"./api-promise.mjs":{"default":"./api-promise.mjs"},"./azure":{"import":"./azure.mjs","require":"./azure.js"},"./azure.js":{"default":"./azure.js"},"./azure.mjs":{"default":"./azure.mjs"},"./beta/*.mjs":{"default":"./beta/*.mjs"},"./beta/*.js":{"default":"./beta/*.js"},"./beta/*":{"import":"./beta/*.mjs","require":"./beta/*.js"},"./client":{"import":"./client.mjs","require":"./client.js"},"./client.js":{"default":"./client.js"},"./client.mjs":{"default":"./client.mjs"},"./core/*.mjs":{"default":"./core/*.mjs"},"./core/*.js":{"default":"./core/*.js"},"./core/*":{"import":"./core/*.mjs","require":"./core/*.js"},"./error":{"import":"./error.mjs","require":"./error.js"},"./error.js":{"default":"./error.js"},"./error.mjs":{"default":"./error.mjs"},"./helpers/*.mjs":{"default":"./helpers/*.mjs"},"./helpers/*.js":{"default":"./helpers/*.js"},"./helpers/*":{"import":"./helpers/*.mjs","require":"./helpers/*.js"},"./index":{"import":"./index.mjs","require":"./index.js"},"./index.js":{"default":"./index.js"},"./index.mjs":{"default":"./index.mjs"},"./lib/*.mjs":{"default":"./lib/*.mjs"},"./lib/*.js":{"default":"./lib/*.js"},"./lib/*":{"import":"./lib/*.mjs","require":"./lib/*.js"},"./pagination":{"import":"./pagination.mjs","require":"./pagination.js"},"./pagination.js":{"default":"./pagination.js"},"./pagination.mjs":{"default":"./pagination.mjs"},"./realtime/*.mjs":{"default":"./realtime/*.mjs"},"./realtime/*.js":{"default":"./realtime/*.js"},"./realtime/*":{"import":"./realtime/*.mjs","require":"./realtime/*.js"},"./resource":{"import":"./resource.mjs","require":"./resource.js"},"./resource.js":{"default":"./resource.js"},"./resource.mjs":{"default":"./resource.mjs"},"./resources/*.mjs":{"default":"./resources/*.mjs"},"./resources/*.js":{"default":"./resources/*.js"},"./resources/*":{"import":"./resources/*.mjs","require":"./resources/*.js"},"./resources":{"import":"./resources.mjs","require":"./resources.js"},"./resources.js":{"default":"./resources.js"},"./resources.mjs":{"default":"./resources.mjs"},"./streaming":{"import":"./streaming.mjs","require":"./streaming.js"},"./streaming.js":{"default":"./streaming.js"},"./streaming.mjs":{"default":"./streaming.mjs"},"./uploads":{"import":"./uploads.mjs","require":"./uploads.js"},"./uploads.js":{"default":"./uploads.js"},"./uploads.mjs":{"default":"./uploads.mjs"},"./version":{"import":"./version.mjs","require":"./version.js"},"./version.js":{"default":"./version.js"},"./version.mjs":{"default":"./version.mjs"}},"peerDependencies":{"ws":"^8.18.0","zod":"^3.25 || ^4.0"},"peerDependenciesMeta":{"ws":{"optional":true},"zod":{"optional":true}},"_lastModified":"2026-05-13T05:25:52.727Z"}

package/dist/node_modules/zod/package.json

@@ -1 +1 @@
-{"name":"zod","version":"4.3.5","type":"module","license":"MIT","author":"Colin McDonnell <zod@colinhacks.com>","description":"TypeScript-first schema declaration and validation library with static type inference","homepage":"https://zod.dev","llms":"https://zod.dev/llms.txt","llmsFull":"https://zod.dev/llms-full.txt","mcpServer":"https://mcp.inkeep.com/zod/mcp","funding":"https://github.com/sponsors/colinhacks","sideEffects":false,"files":["src","**/*.js","**/*.mjs","**/*.cjs","**/*.d.ts","**/*.d.mts","**/*.d.cts","**/package.json"],"keywords":["typescript","schema","validation","type","inference"],"main":"./index.cjs","types":"./index.d.cts","module":"./index.js","zshy":{"exports":{"./package.json":"./package.json",".":"./src/index.ts","./mini":"./src/mini/index.ts","./locales":"./src/locales/index.ts","./v3":"./src/v3/index.ts","./v4":"./src/v4/index.ts","./v4-mini":"./src/v4-mini/index.ts","./v4/mini":"./src/v4/mini/index.ts","./v4/core":"./src/v4/core/index.ts","./v4/locales":"./src/v4/locales/index.ts","./v4/locales/*":"./src/v4/locales/*"},"conditions":{"@zod/source":"src"}},"exports":{"./package.json":"./package.json",".":{"@zod/source":"./src/index.ts","types":"./index.d.cts","import":"./index.js","require":"./index.cjs"},"./mini":{"@zod/source":"./src/mini/index.ts","types":"./mini/index.d.cts","import":"./mini/index.js","require":"./mini/index.cjs"},"./locales":{"@zod/source":"./src/locales/index.ts","types":"./locales/index.d.cts","import":"./locales/index.js","require":"./locales/index.cjs"},"./v3":{"@zod/source":"./src/v3/index.ts","types":"./v3/index.d.cts","import":"./v3/index.js","require":"./v3/index.cjs"},"./v4":{"@zod/source":"./src/v4/index.ts","types":"./v4/index.d.cts","import":"./v4/index.js","require":"./v4/index.cjs"},"./v4-mini":{"@zod/source":"./src/v4-mini/index.ts","types":"./v4-mini/index.d.cts","import":"./v4-mini/index.js","require":"./v4-mini/index.cjs"},"./v4/mini":{"@zod/source":"./src/v4/mini/index.ts","types":"./v4/mini/index.d.cts","import":"./v4/mini/index.js","require":"./v4/mini/index.cjs"},"./v4/core":{"@zod/source":"./src/v4/core/index.ts","types":"./v4/core/index.d.cts","import":"./v4/core/index.js","require":"./v4/core/index.cjs"},"./v4/locales":{"@zod/source":"./src/v4/locales/index.ts","types":"./v4/locales/index.d.cts","import":"./v4/locales/index.js","require":"./v4/locales/index.cjs"},"./v4/locales/*":{"@zod/source":"./src/v4/locales/*","types":"./v4/locales/*","import":"./v4/locales/*","require":"./v4/locales/*"}},"repository":{"type":"git","url":"git+https://github.com/colinhacks/zod.git"},"bugs":{"url":"https://github.com/colinhacks/zod/issues"},"support":{"backing":{"npm-funding":true}},"scripts":{"clean":"git clean -xdf . -e node_modules","build":"zshy --project tsconfig.build.json","postbuild":"tsx ../../scripts/write-stub-package-jsons.ts && pnpm biome check --write .","test:watch":"pnpm vitest","test":"pnpm vitest run","prepublishOnly":"tsx ../../scripts/check-versions.ts"},"_lastModified":"2026-05-11T23:08:10.845Z"}
+{"name":"zod","version":"4.3.5","type":"module","license":"MIT","author":"Colin McDonnell <zod@colinhacks.com>","description":"TypeScript-first schema declaration and validation library with static type inference","homepage":"https://zod.dev","llms":"https://zod.dev/llms.txt","llmsFull":"https://zod.dev/llms-full.txt","mcpServer":"https://mcp.inkeep.com/zod/mcp","funding":"https://github.com/sponsors/colinhacks","sideEffects":false,"files":["src","**/*.js","**/*.mjs","**/*.cjs","**/*.d.ts","**/*.d.mts","**/*.d.cts","**/package.json"],"keywords":["typescript","schema","validation","type","inference"],"main":"./index.cjs","types":"./index.d.cts","module":"./index.js","zshy":{"exports":{"./package.json":"./package.json",".":"./src/index.ts","./mini":"./src/mini/index.ts","./locales":"./src/locales/index.ts","./v3":"./src/v3/index.ts","./v4":"./src/v4/index.ts","./v4-mini":"./src/v4-mini/index.ts","./v4/mini":"./src/v4/mini/index.ts","./v4/core":"./src/v4/core/index.ts","./v4/locales":"./src/v4/locales/index.ts","./v4/locales/*":"./src/v4/locales/*"},"conditions":{"@zod/source":"src"}},"exports":{"./package.json":"./package.json",".":{"@zod/source":"./src/index.ts","types":"./index.d.cts","import":"./index.js","require":"./index.cjs"},"./mini":{"@zod/source":"./src/mini/index.ts","types":"./mini/index.d.cts","import":"./mini/index.js","require":"./mini/index.cjs"},"./locales":{"@zod/source":"./src/locales/index.ts","types":"./locales/index.d.cts","import":"./locales/index.js","require":"./locales/index.cjs"},"./v3":{"@zod/source":"./src/v3/index.ts","types":"./v3/index.d.cts","import":"./v3/index.js","require":"./v3/index.cjs"},"./v4":{"@zod/source":"./src/v4/index.ts","types":"./v4/index.d.cts","import":"./v4/index.js","require":"./v4/index.cjs"},"./v4-mini":{"@zod/source":"./src/v4-mini/index.ts","types":"./v4-mini/index.d.cts","import":"./v4-mini/index.js","require":"./v4-mini/index.cjs"},"./v4/mini":{"@zod/source":"./src/v4/mini/index.ts","types":"./v4/mini/index.d.cts","import":"./v4/mini/index.js","require":"./v4/mini/index.cjs"},"./v4/core":{"@zod/source":"./src/v4/core/index.ts","types":"./v4/core/index.d.cts","import":"./v4/core/index.js","require":"./v4/core/index.cjs"},"./v4/locales":{"@zod/source":"./src/v4/locales/index.ts","types":"./v4/locales/index.d.cts","import":"./v4/locales/index.js","require":"./v4/locales/index.cjs"},"./v4/locales/*":{"@zod/source":"./src/v4/locales/*","types":"./v4/locales/*","import":"./v4/locales/*","require":"./v4/locales/*"}},"repository":{"type":"git","url":"git+https://github.com/colinhacks/zod.git"},"bugs":{"url":"https://github.com/colinhacks/zod/issues"},"support":{"backing":{"npm-funding":true}},"scripts":{"clean":"git clean -xdf . -e node_modules","build":"zshy --project tsconfig.build.json","postbuild":"tsx ../../scripts/write-stub-package-jsons.ts && pnpm biome check --write .","test:watch":"pnpm vitest","test":"pnpm vitest run","prepublishOnly":"tsx ../../scripts/check-versions.ts"},"_lastModified":"2026-05-13T05:25:51.099Z"}

package/dist/server/ai-employees/ai-conversations.d.ts

@@ -13,6 +13,7 @@ export type AIConversationsOptions = {
     systemMessage?: unknown;
     skillSettings?: unknown;
     conversationSettings?: unknown;
+    modelSettings?: unknown;
     [key: string]: unknown;
 };
 export type AIConversationFilterParams = {

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

@@ -90,7 +90,7 @@ class AIConversationsManager {
     if (!conversation) {
       throw new Error("invalid sessionId");
     }
-    const { systemMessage, skillSettings, conversationSettings } = inputOptions ?? {};
+    const { systemMessage, skillSettings, conversationSettings, modelSettings } = inputOptions ?? {};
     const options = conversation.options ?? {};
     if (systemMessage) {
       options["systemMessage"] = systemMessage;
@@ -101,6 +101,9 @@ class AIConversationsManager {
     if (conversationSettings) {
       options["conversationSettings"] = conversationSettings;
     }
+    if (modelSettings) {
+      options["modelSettings"] = modelSettings;
+    }
     const values = { options };
     if (title) {
       values.title = title;

package/dist/server/ai-employees/ai-employee.d.ts

@@ -9,6 +9,8 @@
 import { Model, Transaction } from '@nocobase/database';
 import { LLMProvider } from '../llm-providers/provider';
 import { AIChatContext, AIChatConversation, AIMessageInput, AIToolCall, UserDecision } from '../types';
+import { DocumentSegmentedWithScore } from '../features';
+import { KnowledgeBaseGroup } from '../types';
 import { ToolsEntry } from '@nocobase/ai';
 import { AIToolMessage } from '../types/ai-message.type';
 import { Context } from '@nocobase/actions';
@@ -48,6 +50,10 @@ export declare class AIEmployee {
     private inWorkflow?;
     private streamCached;
     constructor({ ctx, employee, sessionId, systemMessage, skillSettings, webSearch, model, legacy, from, tools, }: AIEmployeeOptions);
+    private get chatSettings();
+    private get systemPromptMode();
+    private areSkillsEnabled;
+    private areToolsEnabled;
     getFormatMessages(userMessages: AIMessageInput[]): Promise<{
         role: "user" | "assistant" | "tool" | "system";
         content: any;
@@ -100,6 +106,7 @@ export declare class AIEmployee {
     processChatStream(stream: any, options: {
         signal: AbortSignal;
         providerName: string;
+        llmService?: string;
         model: string;
         provider: LLMProvider;
         allowEmpty?: boolean;
@@ -107,7 +114,14 @@ export declare class AIEmployee {
     }): Promise<void>;
     private handleInterruptedToolCalls;
     getEmployeeDataSourceContext(): string;
-    getSystemPrompt(userMessages: AIMessageInput[]): Promise<string>;
+    getSystemPrompt(userMessages: AIMessageInput[]): Promise<any>;
+    retrieveKnowledgeBase(userMessage: AIMessageInput): Promise<DocumentSegmentedWithScore[]>;
+    isEnabledKnowledgeBase(): boolean;
+    getAIEmployeeKnowledgeBaseConfig(): {
+        topK: number;
+        score: string;
+    };
+    getKnowledgeBaseGroup(): Promise<KnowledgeBaseGroup[]>;
     initToolCall(transaction: Transaction, messageId: string, toolCalls: {
         id: string;
         name: string;

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

@@ -45,6 +45,7 @@ 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_middleware = require("./middleware");
 var import_ai = require("@nocobase/ai");
 var import_checkpoints = require("./checkpoints");
@@ -107,6 +108,19 @@ class AIEmployee {
       }
     });
   }
+  get chatSettings() {
+    var _a, _b, _c;
+    return ((_b = (_a = this.employee) == null ? void 0 : _a.get) == null ? void 0 : _b.call(_a, "chatSettings")) ?? ((_c = this.employee) == null ? void 0 : _c.chatSettings) ?? {};
+  }
+  get systemPromptMode() {
+    return this.chatSettings.systemPromptMode ?? "default";
+  }
+  areSkillsEnabled() {
+    return this.chatSettings.enableSkills !== false;
+  }
+  areToolsEnabled() {
+    return this.chatSettings.enableTools !== false;
+  }
   async getFormatMessages(userMessages) {
     const { provider } = await this.plugin.aiManager.getLLMService({
       ...this.model
@@ -136,7 +150,7 @@ class AIEmployee {
       }
     };
   }
-  async initSession({ messageId, provider, model, providerName }) {
+  async initSession({ messageId, provider, model, providerName, llmService }) {
     const { tools, baseToolNames } = await this.getAgentTools();
     const resolvedTools = provider.resolveTools(tools.map(import_utils.buildTool));
     if (!messageId && this.legacy !== true) {
@@ -144,7 +158,7 @@ class AIEmployee {
         historyMessages: [],
         tools,
         resolvedTools,
-        middleware: await this.getMiddleware({ tools, baseToolNames, model, providerName }),
+        middleware: await this.getMiddleware({ tools, baseToolNames, model, providerName, llmService }),
         config: void 0,
         state: void 0
       };
@@ -162,6 +176,7 @@ class AIEmployee {
         baseToolNames,
         model,
         providerName,
+        llmService,
         messageId,
         agentThread
       }),
@@ -178,6 +193,7 @@ class AIEmployee {
     userMessages,
     userDecisions
   }) {
+    var _a, _b;
     const { provider, model, service } = await this.plugin.aiManager.getLLMService({
       ...this.model
     });
@@ -185,7 +201,8 @@ class AIEmployee {
       messageId,
       provider,
       model,
-      providerName: service.provider
+      providerName: service.provider,
+      llmService: ((_a = service.get) == null ? void 0 : _a.call(service, "name")) || service.name
     });
     const chatContext = await this.aiChatConversation.getChatContext({
       userMessages: [...historyMessages, ...userMessages ?? []],
@@ -195,7 +212,15 @@ class AIEmployee {
       getSystemPrompt: (userMessages2) => this.getSystemPrompt(userMessages2),
       formatMessages: (messages) => this.formatMessages({ messages, provider })
     });
-    return { providerName: service.provider, model, provider, chatContext, config, state };
+    return {
+      providerName: service.provider,
+      llmService: ((_b = service.get) == null ? void 0 : _b.call(service, "name")) || service.name,
+      model,
+      provider,
+      chatContext,
+      config,
+      state
+    };
   }
   async stream({
     messageId,
@@ -210,7 +235,7 @@ class AIEmployee {
       }
     });
     try {
-      const { providerName, model, provider, chatContext, config, state } = await this.buildChatContext({
+      const { providerName, llmService, model, provider, chatContext, config, state } = await this.buildChatContext({
         messageId,
         userMessages,
         userDecisions
@@ -226,6 +251,7 @@ class AIEmployee {
       await this.processChatStream(stream, {
         signal,
         providerName,
+        llmService,
         model,
         provider,
         responseMetadata
@@ -375,7 +401,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, _z;
     const aiMessageIdMap = /* @__PURE__ */ new Map();
-    const { signal, providerName, model, provider, responseMetadata, allowEmpty = false } = options;
+    const { signal, providerName, llmService, model, provider, responseMetadata, allowEmpty = false } = options;
     let isReasoning = false;
     let gathered;
     signal.addEventListener("abort", async () => {
@@ -384,6 +410,7 @@ class AIEmployee {
           const values = (0, import_utils2.convertAIMessage)({
             aiEmployee: this,
             providerName,
+            llmService,
             model,
             aiMessage: gathered
           });
@@ -651,13 +678,19 @@ Field: ${field.name}, Title: ${(_a = field.options.uiSchema) == null ? void 0 :
   }
   async getSystemPrompt(userMessages) {
     var _a, _b, _c, _d, _e, _f, _g;
+    if (this.systemPromptMode === "none") {
+      return "";
+    }
+    const about = await (0, import_utils.parseVariables)(this.ctx, this.employee.about ?? this.employee.defaultPrompt ?? "");
+    if (this.systemPromptMode === "raw") {
+      return about;
+    }
     const userConfig = await this.db.getRepository("usersAiEmployees").findOne({
       filter: {
         userId: ((_a = this.ctx.auth) == null ? void 0 : _a.user.id) ?? 0,
         aiEmployee: this.employee.username
       }
     });
-    const about = await (0, import_utils.parseVariables)(this.ctx, this.employee.about ?? this.employee.defaultPrompt ?? "");
     let background = "";
     if (this.systemMessage) {
       background = await (0, import_utils.parseVariables)(this.ctx, this.systemMessage);
@@ -721,6 +754,92 @@ 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: "vectorStoreConfigKey",
+              value: vectorStoreConfig.vectorStoreConfigKey
+            }
+          ]
+        );
+        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: "vectorStoreConfigKey",
+                value: vectorStoreConfig.vectorStoreConfigKey
+              }
+            ]
+          );
+          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 { knowledgeBaseKeys } = this.employee.knowledgeBase ?? {};
+    if (!knowledgeBaseKeys || import_lodash.default.isEmpty(knowledgeBaseKeys)) {
+      return [];
+    }
+    return await this.plugin.features.knowledgeBase.getKnowledgeBaseGroup(knowledgeBaseKeys);
+  }
   // === Tool calls ===
   async initToolCall(transaction, messageId, toolCalls) {
     const nowTime = /* @__PURE__ */ new Date();
@@ -1147,6 +1266,9 @@ If information is missing, clearly state it in the summary.</Important>`;
   }
   async getAIEmployeeTools() {
     var _a;
+    if (!this.areToolsEnabled()) {
+      return [];
+    }
     const tools = await this.listTools({ scope: "GENERAL" });
     if (this.webSearch === true) {
       const subAgentWebSearch = await this.toolsManager.getTools(import_ai.SYSTEM_TOOLS.WEB_SEARCH);
@@ -1191,6 +1313,9 @@ If information is missing, clearly state it in the summary.</Important>`;
   }
   async getAvailableSkills() {
     var _a;
+    if (!this.areSkillsEnabled()) {
+      return [];
+    }
     const { skillsManager } = this.plugin.ai;
     const aIEmployeeTools = await this.getAIEmployeeTools();
     const getSkill = aIEmployeeTools.find((it) => it.definition.name === "getSkill");
@@ -1216,6 +1341,12 @@ If information is missing, clearly state it in the summary.</Important>`;
     }
   }
   async getAgentTools() {
+    if (!this.areToolsEnabled()) {
+      return {
+        tools: [],
+        baseToolNames: /* @__PURE__ */ new Set()
+      };
+    }
     const baseTools = await this.getAIEmployeeTools();
     const toolMap = await this.getToolsMap();
     const availableSkills = await this.getAvailableSkills();
@@ -1286,7 +1417,7 @@ If information is missing, clearly state it in the summary.</Important>`;
     return availableAIEmployees;
   }
   async getMiddleware(options) {
-    const { providerName, model, tools, baseToolNames, messageId, agentThread } = options;
+    const { providerName, llmService, model, tools, baseToolNames, messageId, agentThread } = options;
     const inWorkflow = await this.isInWorkflow();
     return [
       (0, import_middleware.skillToolBindingMiddleware)(this, {
@@ -1295,7 +1426,7 @@ If information is missing, clearly state it in the summary.</Important>`;
       (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 })
+      (0, import_middleware.conversationMiddleware)(this, { providerName, llmService, model, messageId, agentThread })
     ];
   }
   async getCurrentThread() {

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

@@ -7,10 +7,14 @@
  * For more information, please refer to: https://www.nocobase.com/agreement.
  */
 import PluginAIServer from '../plugin';
+import type { Model } from '@nocobase/database';
+import type { ModelRef } from './ai-employee';
 export declare class AIEmployeesManager {
     protected plugin: PluginAIServer;
     conversationController: Map<string, AbortController>;
     constructor(plugin: PluginAIServer);
+    getEmployee(username: string): Promise<Model | null>;
+    resolveModel(employee: Model, model?: ModelRef | null): Promise<ModelRef>;
     onAbortConversation(sessionId: string): boolean;
     abortConversation(sessionId: string): void;
 }

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

@@ -34,6 +34,47 @@ class AIEmployeesManager {
     this.plugin = plugin;
   }
   conversationController = /* @__PURE__ */ new Map();
+  async getEmployee(username) {
+    return await this.plugin.db.getRepository("aiEmployees").findOne({
+      filter: {
+        username
+      }
+    });
+  }
+  async resolveModel(employee, model) {
+    var _a;
+    const modelSettings = ((_a = employee.get) == null ? void 0 : _a.call(employee, "modelSettings")) || employee.modelSettings;
+    if (modelSettings == null ? void 0 : modelSettings.enabled) {
+      const models = Array.isArray(modelSettings.models) ? modelSettings.models : [];
+      const configuredModels = models.filter((item) => (item == null ? void 0 : item.llmService) && (item == null ? void 0 : item.model)).map((item) => ({
+        llmService: item.llmService,
+        model: item.model
+      }));
+      if (!configuredModels.length && modelSettings.llmService && modelSettings.model) {
+        configuredModels.push({
+          llmService: modelSettings.llmService,
+          model: modelSettings.model
+        });
+      }
+      if (!configuredModels.length) {
+        throw new Error("AI employee model not configured");
+      }
+      if ((model == null ? void 0 : model.llmService) && (model == null ? void 0 : model.model) && configuredModels.some((item) => item.llmService === model.llmService && item.model === model.model)) {
+        return model;
+      }
+      const firstModel = configuredModels[0];
+      if ((firstModel == null ? void 0 : firstModel.llmService) && (firstModel == null ? void 0 : firstModel.model)) {
+        return {
+          llmService: firstModel.llmService,
+          model: firstModel.model
+        };
+      }
+    }
+    if ((model == null ? void 0 : model.llmService) && (model == null ? void 0 : model.model)) {
+      return model;
+    }
+    return await this.plugin.aiManager.resolveModel();
+  }
   onAbortConversation(sessionId) {
     const controller = this.conversationController.get(sessionId);
     if (controller) {

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

@@ -86,8 +86,8 @@ class KnowledgeBaseManager {
           vectorStoreConfig.vectorStoreProvider,
           [
             {
-              key: "vectorStoreConfigId",
-              value: vectorStoreConfig.vectorStoreConfigId
+              key: "vectorStoreConfigKey",
+              value: vectorStoreConfig.vectorStoreConfigKey
             }
           ]
         );
@@ -107,8 +107,8 @@ class KnowledgeBaseManager {
             [
               ...knowledgeBase.vectorStoreProps,
               {
-                key: "vectorStoreConfigId",
-                value: vectorStoreConfig.vectorStoreConfigId
+                key: "vectorStoreConfigKey",
+                value: vectorStoreConfig.vectorStoreConfigKey
               }
             ]
           );
@@ -148,11 +148,11 @@ class KnowledgeBaseManager {
     };
   }
   async getKnowledgeBaseGroup(employee) {
-    const { knowledgeBaseIds } = (employee == null ? void 0 : employee.knowledgeBase) ?? {};
-    if (!knowledgeBaseIds || import_lodash.default.isEmpty(knowledgeBaseIds)) {
+    const { knowledgeBaseKeys } = (employee == null ? void 0 : employee.knowledgeBase) ?? {};
+    if (!knowledgeBaseKeys || import_lodash.default.isEmpty(knowledgeBaseKeys)) {
       return [];
     }
-    return await this.plugin.features.knowledgeBase.getKnowledgeBaseGroup(knowledgeBaseIds);
+    return await this.plugin.features.knowledgeBase.getKnowledgeBaseGroup(knowledgeBaseKeys);
   }
   async getEmployee(username) {
     return await this.aiEmployeeRepo.findOne({ filter: { username } });

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

@@ -10,6 +10,7 @@ import { AIEmployee } from '../ai-employee';
 import z from 'zod';
 export declare const conversationMiddleware: (aiEmployee: AIEmployee, options: {
     providerName: string;
+    llmService?: string;
     model: string;
     messageId?: string;
     agentThread?: {

package/dist/server/ai-employees/middleware/conversation.js

@@ -43,16 +43,18 @@ var import_langchain = require("langchain");
 var import_zod = __toESM(require("zod"));
 var import_utils = require("../utils");
 const conversationMiddleware = (aiEmployee, options) => {
-  const { providerName, model, messageId, agentThread } = options;
+  const { providerName, llmService, model, messageId, agentThread } = options;
   const convertAIMessage = (aiMessage) => (0, import_utils.convertAIMessage)({
     aiEmployee,
     providerName,
+    llmService,
     model,
     aiMessage
   });
-  const convertHumanMessage = (humanMessage) => (0, import_utils.convertHumanMessage)({ providerName, model, humanMessage });
+  const convertHumanMessage = (humanMessage) => (0, import_utils.convertHumanMessage)({ providerName, llmService, model, humanMessage });
   const convertToolMessage = (toolMessage) => (0, import_utils.convertToolMessage)({
     providerName,
+    llmService,
     model,
     toolMessage
   });

package/dist/server/ai-employees/sub-agents/dispatcher.js

@@ -114,16 +114,14 @@ class SubAgentsDispatcher {
     if (!userId) {
       throw new Error("User not authenticated");
     }
-    if (!(model == null ? void 0 : model.llmService) || !(model == null ? void 0 : model.model)) {
-      throw new Error("LLM service not configured");
-    }
+    const resolvedModel = await plugin.aiEmployeesManager.resolveModel(employee, model);
     const aiEmployee = new import_ai_employee.AIEmployee({
       ctx,
       employee,
       sessionId,
       skillSettings,
       webSearch,
-      model,
+      model: resolvedModel,
       from: "sub-agent"
     });
     const lastMessage = await ctx.db.getRepository("aiMessages").findOne({

package/dist/server/ai-employees/utils.d.ts

@@ -9,19 +9,22 @@
 import { AIMessage, HumanMessage, ToolMessage } from 'langchain';
 import { AIMessageInput } from '../types';
 import { AIEmployee } from './ai-employee';
-export declare const convertAIMessage: ({ aiEmployee, providerName: provider, model, aiMessage, }: {
+export declare const convertAIMessage: ({ aiEmployee, providerName: provider, llmService, model, aiMessage, }: {
     aiEmployee: AIEmployee;
     providerName: string;
+    llmService?: string;
     model: string;
     aiMessage: AIMessage;
 }) => AIMessageInput;
-export declare const convertHumanMessage: ({ providerName: provider, model, humanMessage, }: {
+export declare const convertHumanMessage: ({ providerName: provider, llmService, model, humanMessage, }: {
     providerName: string;
+    llmService?: string;
     model: string;
     humanMessage: HumanMessage;
 }) => AIMessageInput;
-export declare const convertToolMessage: ({ providerName: provider, model, toolMessage, }: {
+export declare const convertToolMessage: ({ providerName: provider, llmService, model, toolMessage, }: {
     providerName: string;
+    llmService?: string;
     model: string;
     toolMessage: ToolMessage;
 }) => AIMessageInput;

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

@@ -34,6 +34,7 @@ module.exports = __toCommonJS(utils_exports);
 const convertAIMessage = ({
   aiEmployee,
   providerName: provider,
+  llmService,
   model,
   aiMessage
 }) => {
@@ -71,6 +72,7 @@ const convertAIMessage = ({
       id: aiMessage.id,
       model,
       provider,
+      llmService,
       usage_metadata: {}
     },
     toolCalls: null
@@ -94,6 +96,7 @@ const convertAIMessage = ({
 };
 const convertHumanMessage = ({
   providerName: provider,
+  llmService,
   model,
   humanMessage
 }) => {
@@ -107,7 +110,8 @@ const convertHumanMessage = ({
     metadata: {
       id: humanMessage.id,
       model,
-      provider
+      provider,
+      llmService
     }
   };
   values.attachments = humanMessage.additional_kwargs.attachments;
@@ -116,6 +120,7 @@ const convertHumanMessage = ({
 };
 const convertToolMessage = ({
   providerName: provider,
+  llmService,
   model,
   toolMessage
 }) => {
@@ -129,6 +134,7 @@ const convertToolMessage = ({
       id: toolMessage.id,
       model,
       provider,
+      llmService,
       toolCallId: toolMessage.tool_call_id,
       toolName: toolMessage.name
     }

package/dist/server/features/knowledge-base.d.ts

@@ -6,7 +6,8 @@
  * 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 { KnowledgeBaseGroup } from '../types';
+import { KnowledgeBase, KnowledgeBaseGroup } from '../types';
 export interface KnowledgeBaseFeature {
-    getKnowledgeBaseGroup(knowledgeBaseIds: string[]): Promise<KnowledgeBaseGroup[]>;
+    getKnowledgeBase(knowledgeBaseKeys: string[]): Promise<KnowledgeBase[]>;
+    getKnowledgeBaseGroup(knowledgeBaseKeys: string[]): Promise<KnowledgeBaseGroup[]>;
 }