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

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");
@@ -70,6 +71,7 @@ class AIEmployee {
   legacy;
   tools;
   inWorkflow;
+  streamCached;
   constructor({
     ctx,
     employee,
@@ -94,10 +96,30 @@ class AIEmployee {
     this.legacy = legacy;
     this.from = from;
     this.tools = tools;
+    this.streamCached = this.plugin.llmStreamCachedManager.getCached(sessionId);
     const builtInManager = this.plugin.builtInManager;
     builtInManager.setupBuiltInInfo(ctx, this.employee);
     this.webSearch = webSearch;
-    this.protocol = ChatStreamProtocol.fromContext(ctx);
+    this.protocol = ChatStreamProtocol.fromContext(ctx, async (chunk) => {
+      try {
+        await this.streamCached.append(chunk);
+      } catch (error) {
+        this.logger.warn("Failed to append LLM stream cache", { sessionId: this.sessionId, error });
+      }
+    });
+  }
+  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({
@@ -128,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) {
@@ -136,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
       };
@@ -154,6 +176,7 @@ class AIEmployee {
         baseToolNames,
         model,
         providerName,
+        llmService,
         messageId,
         agentThread
       }),
@@ -170,6 +193,7 @@ class AIEmployee {
     userMessages,
     userDecisions
   }) {
+    var _a, _b;
     const { provider, model, service } = await this.plugin.aiManager.getLLMService({
       ...this.model
     });
@@ -177,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 ?? []],
@@ -187,13 +212,22 @@ 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,
     userMessages = [],
     userDecisions
   }) {
+    await this.streamCached.clear();
     await this.aiConversationsRepo.update({
       values: { llmActiveState: "streaming" },
       filter: {
@@ -201,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
@@ -217,6 +251,7 @@ class AIEmployee {
       await this.processChatStream(stream, {
         signal,
         providerName,
+        llmService,
         model,
         provider,
         responseMetadata
@@ -228,11 +263,12 @@ class AIEmployee {
       return false;
     } finally {
       await this.aiConversationsRepo.update({
-        values: { llmActiveState: "idle" },
+        values: { llmActiveState: "idle", read: false },
         filter: {
           sessionId: this.sessionId
         }
       });
+      await this.streamCached.clear();
     }
   }
   async invoke({
@@ -365,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 () => {
@@ -374,6 +410,7 @@ class AIEmployee {
           const values = (0, import_utils2.convertAIMessage)({
             aiEmployee: this,
             providerName,
+            llmService,
             model,
             aiMessage: gathered
           });
@@ -386,6 +423,14 @@ class AIEmployee {
         }
       } catch (e) {
         this.logger.error("Fail to save message after conversation abort", gathered);
+      } finally {
+        await this.aiConversationsRepo.update({
+          values: { llmActiveState: "idle", read: true },
+          filter: {
+            sessionId: this.sessionId
+          }
+        });
+        await this.streamCached.clear();
       }
     });
     try {
@@ -394,7 +439,7 @@ class AIEmployee {
         from: this.from,
         username: this.employee.username
       };
-      this.protocol.with(aiEmployeeConversation).startStream();
+      await this.protocol.with(aiEmployeeConversation).startStream();
       for await (const [mode, chunks] of stream) {
         if (mode === "messages") {
           const [chunk, metadata] = chunks;
@@ -404,24 +449,24 @@ class AIEmployee {
             if (chunk.content) {
               if (isReasoning) {
                 isReasoning = false;
-                this.protocol.with(currentConversation).stopReasoning();
+                await this.protocol.with(currentConversation).stopReasoning();
               }
               const parsedContent = provider.parseResponseChunk(chunk.content);
               if (parsedContent) {
-                this.protocol.with(currentConversation).content(parsedContent);
+                await this.protocol.with(currentConversation).content(parsedContent);
               }
             }
             if ((_a = chunk.tool_call_chunks) == null ? void 0 : _a.length) {
-              this.protocol.with(currentConversation).toolCallChunks(chunk.tool_call_chunks);
+              await this.protocol.with(currentConversation).toolCallChunks(chunk.tool_call_chunks);
             }
             const webSearch = provider.parseWebSearchAction(chunk);
             if (webSearch == null ? void 0 : webSearch.length) {
-              this.protocol.with(currentConversation).webSearch(webSearch);
+              await this.protocol.with(currentConversation).webSearch(webSearch);
             }
             const reasoningContent = provider.parseReasoningContent(chunk);
             if (reasoningContent) {
               isReasoning = true;
-              this.protocol.with(currentConversation).reasoning(reasoningContent);
+              await this.protocol.with(currentConversation).reasoning(reasoningContent);
             }
           }
         } else if (mode === "updates") {
@@ -431,8 +476,8 @@ class AIEmployee {
             await this.handleInterruptedToolCalls(
               interrupt,
               (sessionId) => aiMessageIdMap.get(sessionId),
-              ({ messageId, interruptAction, toolCall, currentConversation }) => {
-                this.protocol.with(currentConversation).toolCallStatus({
+              async ({ messageId, interruptAction, toolCall, currentConversation }) => {
+                await this.protocol.with(currentConversation).toolCallStatus({
                   toolCall: {
                     messageId,
                     id: toolCall.id,
@@ -448,6 +493,7 @@ class AIEmployee {
         } else if (mode === "custom") {
           const { currentConversation } = chunks;
           if (chunks.action === "AfterAIMessageSaved") {
+            await this.streamCached.skipped();
             aiMessageIdMap.set(currentConversation.sessionId, chunks.body.messageId);
             const data = responseMetadata.get(chunks.body.id);
             if (data) {
@@ -476,11 +522,11 @@ class AIEmployee {
               }
             }
           } else if (chunks.action === "initToolCalls") {
-            this.protocol.with(currentConversation).toolCalls(chunks.body);
+            await this.protocol.with(currentConversation).toolCalls(chunks.body);
           } else if (chunks.action === "beforeToolCall") {
             const toolsMap = await this.getToolsMap();
             const willInterrupt = this.shouldInterruptToolCall(toolsMap.get((_d = (_c = chunks.body) == null ? void 0 : _c.toolCall) == null ? void 0 : _d.name));
-            this.protocol.with(currentConversation).toolCallStatus({
+            await this.protocol.with(currentConversation).toolCallStatus({
               toolCall: {
                 messageId: (_f = (_e = chunks.body) == null ? void 0 : _e.toolCall) == null ? void 0 : _f.messageId,
                 id: (_h = (_g = chunks.body) == null ? void 0 : _g.toolCall) == null ? void 0 : _h.id,
@@ -492,7 +538,7 @@ class AIEmployee {
           } else if (chunks.action === "afterToolCall") {
             const toolsMap = await this.getToolsMap();
             const willInterrupt = this.shouldInterruptToolCall(toolsMap.get((_l = (_k = chunks.body) == null ? void 0 : _k.toolCall) == null ? void 0 : _l.name));
-            this.protocol.with(currentConversation).toolCallStatus({
+            await this.protocol.with(currentConversation).toolCallStatus({
               toolCall: {
                 messageId: (_n = (_m = chunks.body) == null ? void 0 : _m.toolCall) == null ? void 0 : _n.messageId,
                 id: (_p = (_o = chunks.body) == null ? void 0 : _o.toolCall) == null ? void 0 : _p.id,
@@ -516,7 +562,7 @@ class AIEmployee {
               for (const { metadata } of messages) {
                 const tools = toolsMap.get(metadata.toolName);
                 const toolCallResult = toolCallResultMap.get(metadata.toolCallId);
-                this.protocol.with(currentConversation).toolCallStatus({
+                await this.protocol.with(currentConversation).toolCallStatus({
                   toolCall: {
                     messageId,
                     id: metadata.toolCallId,
@@ -531,9 +577,9 @@ class AIEmployee {
                 });
               }
             }
-            this.protocol.with(currentConversation).newMessage();
+            await this.protocol.with(currentConversation).newMessage();
           } else if (chunks.action === "afterSubAgentInvoke") {
-            this.protocol.with(currentConversation).subAgentCompleted();
+            await this.protocol.with(currentConversation).subAgentCompleted();
           }
         }
       }
@@ -541,7 +587,7 @@ class AIEmployee {
         this.sendErrorResponse("Empty message");
         return;
       }
-      this.protocol.with(aiEmployeeConversation).endStream();
+      await this.protocol.with(aiEmployeeConversation).endStream();
     } catch (err) {
       this.ctx.log.error(err);
       if (err.name === "GraphRecursionError") {
@@ -632,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);
@@ -702,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();
@@ -1128,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);
@@ -1172,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");
@@ -1197,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();
@@ -1267,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, {
@@ -1276,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() {
@@ -1393,8 +1543,9 @@ class AgentThread {
   }
 }
 class ChatStreamProtocol {
-  constructor(streamConsumer) {
+  constructor(streamConsumer, onWrite) {
     this.streamConsumer = streamConsumer;
+    this.onWrite = onWrite;
   }
   _statistics = {
     sent: 0,
@@ -1405,43 +1556,45 @@ class ChatStreamProtocol {
       this._statistics.sent = 0;
     }
   };
-  static fromContext(ctx) {
-    return new ChatStreamProtocol(ctx.res);
+  static fromContext(ctx, onWrite) {
+    return new ChatStreamProtocol(ctx.res, onWrite);
   }
   with(conversation) {
-    const write = ({ type, body }) => {
+    const write = async ({ type, body }) => {
+      var _a;
       const { sessionId, from, username } = conversation;
       const data = `data: ${JSON.stringify({ sessionId, from, username, type, body })}
 
 `;
+      await ((_a = this.onWrite) == null ? void 0 : _a.call(this, data));
       this.streamConsumer.write(data);
       this._statistics.addSent(data.length);
     };
     return {
-      startStream: () => {
+      startStream: async () => {
         this._statistics.reset();
-        write({ type: "stream_start" });
+        await write({ type: "stream_start" });
       },
-      endStream: () => {
-        write({ type: "stream_end" });
+      endStream: async () => {
+        await write({ type: "stream_end" });
       },
-      subAgentCompleted: () => {
-        write({ type: "sub_agent_completed" });
+      subAgentCompleted: async () => {
+        await write({ type: "sub_agent_completed" });
       },
-      newMessage: (content) => {
-        write({ type: "new_message", body: content });
+      newMessage: async (content) => {
+        await write({ type: "new_message", body: content });
       },
-      content: (content) => {
-        write({ type: "content", body: content });
+      content: async (content) => {
+        await write({ type: "content", body: content });
       },
-      webSearch: (content) => {
-        write({ type: "web_search", body: content });
+      webSearch: async (content) => {
+        await write({ type: "web_search", body: content });
       },
-      reasoning: (content) => {
-        write({ type: "reasoning", body: content });
+      reasoning: async (content) => {
+        await write({ type: "reasoning", body: content });
       },
-      stopReasoning: () => {
-        write({
+      stopReasoning: async () => {
+        await write({
           type: "reasoning",
           body: {
             status: "stop",
@@ -1449,13 +1602,13 @@ class ChatStreamProtocol {
           }
         });
       },
-      toolCallChunks: (content) => {
-        write({ type: "tool_call_chunks", body: content });
+      toolCallChunks: async (content) => {
+        await write({ type: "tool_call_chunks", body: content });
       },
-      toolCalls: (content) => {
-        write({ type: "tool_calls", body: content });
+      toolCalls: async (content) => {
+        await write({ type: "tool_calls", body: content });
       },
-      toolCallStatus: ({
+      toolCallStatus: async ({
         toolCall,
         invokeStatus,
         status,
@@ -1464,7 +1617,7 @@ class ChatStreamProtocol {
         content,
         interruptAction
       }) => {
-        write({
+        await write({
           type: "tool_call_status",
           body: {
             toolCall,

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?: {