@nocobase/plugin-ai 2.1.0-alpha.34 → 2.1.0-alpha.36

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.2.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.d.ts

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

package/dist/server/resource/aiConversations.js

@@ -43,6 +43,7 @@ var import_actions = __toESM(require("@nocobase/actions"));
 var import_database = require("@nocobase/database");
 var import_utils = require("../utils");
 var import_ai_employee = require("../ai-employees/ai-employee");
+var import_ai_chat_conversation = require("../manager/ai-chat-conversation");
 async function getAIEmployee(ctx, username) {
   var _a;
   const filter = {
@@ -67,12 +68,17 @@ function setupSSEHeaders(ctx) {
 function sendErrorResponse(ctx, errorMessage) {
   (0, import_utils.sendSSEError)(ctx, errorMessage);
 }
-async function parallelConversationsLimit(ctx, next) {
+async function loginInCheck(ctx, next) {
   var _a;
   const userId = (_a = ctx.auth) == null ? void 0 : _a.user.id;
   if (!userId) {
     return ctx.throw(403);
   }
+  await next();
+}
+const isReachParallelLimit = async (ctx) => {
+  var _a;
+  const userId = (_a = ctx.auth) == null ? void 0 : _a.user.id;
   const activeStreamCount = await ctx.db.getModel("aiConversations").count({
     where: {
       userId,
@@ -82,27 +88,32 @@ async function parallelConversationsLimit(ctx, next) {
       }
     }
   });
-  if (activeStreamCount > 2) {
-    sendErrorResponse(ctx, ctx.t("There are conversations in progress. Please try again later."));
+  return activeStreamCount > 2;
+};
+const saveUserMessages = async (ctx, sessionId, messages, messageId) => {
+  const userMessages = messages.filter((message) => message.role === "user");
+  if (!userMessages.length) {
     return;
   }
-  await next();
-}
+  const aiChatConversation = (0, import_ai_chat_conversation.createAIChatConversation)(ctx, sessionId);
+  await aiChatConversation.withTransaction(async (conversation) => {
+    if (messageId && await conversation.getMessage(messageId)) {
+      await conversation.removeMessages({ messageId });
+    }
+    await conversation.addMessages(userMessages);
+  });
+};
 var aiConversations_default = {
   name: "aiConversations",
   middlewares: [
     {
-      only: ["sendMessages", "resendMessages", "resumeStream"],
-      handler: parallelConversationsLimit
+      handler: loginInCheck
     }
   ],
   actions: {
     async list(ctx, next) {
       var _a;
       const userId = (_a = ctx.auth) == null ? void 0 : _a.user.id;
-      if (!userId) {
-        return ctx.throw(403);
-      }
       const filter = ctx.action.params.filter || {};
       ctx.action.mergeParams({
         filter: {
@@ -117,9 +128,6 @@ var aiConversations_default = {
     async unreadCount(ctx, next) {
       var _a;
       const userId = (_a = ctx.auth) == null ? void 0 : _a.user.id;
-      if (!userId) {
-        return ctx.throw(403);
-      }
       const count = await ctx.db.getModel("aiConversations").count({
         where: {
           userId,
@@ -136,9 +144,6 @@ var aiConversations_default = {
     async unreadCounts(ctx, next) {
       var _a;
       const userId = (_a = ctx.auth) == null ? void 0 : _a.user.id;
-      if (!userId) {
-        return ctx.throw(403);
-      }
       const [conversationUnreadCount, workflowTaskUnreadCount] = await Promise.all([
         ctx.db.getModel("aiConversations").count({
           where: {
@@ -165,10 +170,7 @@ var aiConversations_default = {
       var _a;
       const plugin = ctx.app.pm.get("ai");
       const userId = (_a = ctx.auth) == null ? void 0 : _a.user.id;
-      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 +182,8 @@ var aiConversations_default = {
           options: {
             systemMessage,
             skillSettings,
-            conversationSettings
+            conversationSettings,
+            modelSettings
           }
         });
       } catch (error) {
@@ -195,9 +198,6 @@ var aiConversations_default = {
       var _a;
       const plugin = ctx.app.pm.get("ai");
       const userId = (_a = ctx.auth) == null ? void 0 : _a.user.id;
-      if (!userId) {
-        return ctx.throw(403);
-      }
       const { filterByTk: sessionId } = ctx.action.params;
       const { title } = ctx.action.params.values || {};
       ctx.body = await plugin.aiConversationsManager.update({ userId, sessionId, title });
@@ -207,22 +207,19 @@ var aiConversations_default = {
       var _a;
       const plugin = ctx.app.pm.get("ai");
       const userId = (_a = ctx.auth) == null ? void 0 : _a.user.id;
-      if (!userId) {
-        return ctx.throw(403);
-      }
       const { filterByTk: sessionId } = ctx.action.params;
       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") {
@@ -235,9 +232,6 @@ var aiConversations_default = {
     async destroy(ctx, next) {
       var _a;
       const userId = (_a = ctx.auth) == null ? void 0 : _a.user.id;
-      if (!userId) {
-        return ctx.throw(403);
-      }
       ctx.action.mergeParams({
         filter: {
           userId
@@ -249,9 +243,6 @@ var aiConversations_default = {
       var _a, _b;
       const plugin = ctx.app.pm.get("ai");
       const userId = (_a = ctx.auth) == null ? void 0 : _a.user.id;
-      if (!userId) {
-        return ctx.throw(403);
-      }
       const { sessionId, cursor, updateRead: originalUpdateRead } = ctx.action.params || {};
       if (!sessionId) {
         ctx.throw(400);
@@ -278,9 +269,6 @@ var aiConversations_default = {
       var _a;
       const plugin = ctx.app.pm.get("ai");
       const userId = (_a = ctx.auth) == null ? void 0 : _a.user.id;
-      if (!userId) {
-        return ctx.throw(403);
-      }
       const {
         sessionId,
         messageId,
@@ -319,11 +307,8 @@ var aiConversations_default = {
     },
     async sendMessages(ctx, next) {
       var _a, _b, _c, _d;
-      const userId = (_a = ctx.auth) == null ? void 0 : _a.user.id;
-      if (!userId) {
-        return ctx.throw(403);
-      }
       const plugin = ctx.app.pm.get("ai");
+      const userId = (_a = ctx.auth) == null ? void 0 : _a.user.id;
       const {
         sessionId,
         aiEmployee: employeeName,
@@ -337,38 +322,27 @@ var aiConversations_default = {
       if (shouldStream) {
         setupSSEHeaders(ctx);
       }
-      if (!sessionId) {
-        if (shouldStream) {
-          sendErrorResponse(ctx, "sessionId is required");
-        } else {
-          ctx.status = 400;
-          ctx.body = { error: "sessionId is required" };
+      try {
+        if (!sessionId) {
+          throw new import_utils.ResourceActionError(400, ctx.t("sessionId is required"));
         }
-        return next();
-      }
-      const userMessage = messages.find((message) => message.role === "user");
-      if (!userMessage) {
-        if (shouldStream) {
-          sendErrorResponse(ctx, "user message is required");
-        } else {
-          ctx.status = 400;
-          ctx.body = { error: "user message is required" };
+        if (!Array.isArray(messages)) {
+          throw new import_utils.ResourceActionError(400, ctx.t("messages must be an array"));
+        }
+        const userMessage = messages.find((message) => message.role === "user");
+        if (!userMessage) {
+          throw new import_utils.ResourceActionError(400, ctx.t("user message is required"));
         }
-        return next();
-      }
-      try {
         const conversation = await plugin.aiConversationsManager.getConversation({
           sessionId,
           userId
         });
         if (!conversation) {
-          if (shouldStream) {
-            sendErrorResponse(ctx, "conversation not found");
-          } else {
-            ctx.status = 404;
-            ctx.body = { error: "conversation not found" };
-          }
-          return next();
+          throw new import_utils.ResourceActionError(400, ctx.t("conversation not found"));
+        }
+        const employee = await getAIEmployee(ctx, employeeName);
+        if (!employee) {
+          throw new import_utils.ResourceActionError(400, ctx.t("AI employee not found"));
         }
         if (!conversation.title) {
           const textUserMessage = messages.find(
@@ -383,17 +357,12 @@ var aiConversations_default = {
             await conversation.save();
           }
         }
-        const employee = await getAIEmployee(ctx, employeeName);
-        if (!employee) {
-          if (shouldStream) {
-            sendErrorResponse(ctx, "AI employee not found");
-          } else {
-            ctx.status = 404;
-            ctx.body = { error: "AI employee not found" };
-          }
-          return next();
+        if (await isReachParallelLimit(ctx)) {
+          await saveUserMessages(ctx, sessionId, messages, editingMessageId);
+          throw new import_utils.ResourceActionError(400, ctx.t("There are conversations in progress. Please try again later."));
         }
         const legacy = conversation.thread === 0;
+        const resolvedModel = await plugin.aiEmployeesManager.resolveModel(employee, model);
         const aiEmployee = new import_ai_employee.AIEmployee({
           ctx,
           employee,
@@ -402,7 +371,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) {
@@ -440,11 +409,19 @@ var aiConversations_default = {
         }
       } catch (err) {
         ctx.log.error(err);
+        let status = 500;
+        let message = ctx.t("Server unexpected error occur");
+        if (err instanceof import_utils.ResourceActionError) {
+          status = err.status;
+          message = err.message;
+        } else if (err instanceof Error) {
+          status = 500;
+          message = err.message;
+        }
         if (shouldStream) {
-          sendErrorResponse(ctx, err.message || "Tool call error");
+          sendErrorResponse(ctx, message);
         } else {
-          ctx.status = 500;
-          ctx.body = { error: err.message || "Tool call error" };
+          ctx.throw(status, message);
         }
       } finally {
         await next();
@@ -453,9 +430,6 @@ var aiConversations_default = {
     async abort(ctx, next) {
       var _a;
       const userId = (_a = ctx.auth) == null ? void 0 : _a.user.id;
-      if (!userId) {
-        return ctx.throw(403);
-      }
       const { sessionId } = ctx.action.params.values || {};
       const plugin = ctx.app.pm.get("ai");
       const conversation = await plugin.aiConversationsManager.getConversation({
@@ -472,30 +446,44 @@ var aiConversations_default = {
       var _a, _b, _c, _d, _e;
       const plugin = ctx.app.pm.get("ai");
       const userId = (_a = ctx.auth) == null ? void 0 : _a.user.id;
-      if (!userId) {
-        return ctx.throw(403);
-      }
+      const abortController = new AbortController();
+      const abortStream = () => abortController.abort();
+      const shouldStopStream = () => abortController.signal.aborted || ctx.res.destroyed || ctx.res.writableEnded;
       setupSSEHeaders(ctx);
       const sessionId = ((_b = ctx.action.params) == null ? void 0 : _b.sessionId) || ((_d = (_c = ctx.action.params) == null ? void 0 : _c.values) == null ? void 0 : _d.sessionId) || ((_e = ctx.action.params) == null ? void 0 : _e.filterByTk);
       if (!sessionId) {
         sendErrorResponse(ctx, "sessionId is required");
         return;
       }
+      ctx.req.once("aborted", abortStream);
+      ctx.res.once("close", abortStream);
       try {
         const conversation = await plugin.aiConversationsManager.getConversation({
           sessionId,
           userId
         });
+        if (shouldStopStream()) {
+          return;
+        }
         if (!conversation) {
           sendErrorResponse(ctx, "conversation not found");
           return;
         }
+        const reachLimit = await isReachParallelLimit(ctx);
+        if (shouldStopStream()) {
+          return;
+        }
         let hasChunks = false;
-        for await (const chunk of plugin.llmStreamCachedManager.getCached(sessionId).stream()) {
-          hasChunks = true;
-          ctx.res.write(chunk);
+        if (!reachLimit) {
+          for await (const chunk of plugin.llmStreamCachedManager.getCached(sessionId).stream({ signal: abortController.signal })) {
+            if (shouldStopStream()) {
+              break;
+            }
+            hasChunks = true;
+            ctx.res.write(chunk);
+          }
         }
-        if (!hasChunks) {
+        if (!hasChunks && !shouldStopStream()) {
           const currentConversation = await plugin.aiConversationsManager.getConversation({
             sessionId,
             userId
@@ -508,11 +496,16 @@ var aiConversations_default = {
           }
         }
       } catch (err) {
+        if (shouldStopStream()) {
+          return;
+        }
         ctx.log.error(err);
         sendErrorResponse(ctx, err.message || "Resume stream error");
         return;
       } finally {
-        if (!ctx.res.writableEnded) {
+        ctx.req.off("aborted", abortStream);
+        ctx.res.off("close", abortStream);
+        if (!shouldStopStream()) {
           ctx.res.end();
         }
         await next();
@@ -522,24 +515,26 @@ var aiConversations_default = {
       var _a, _b, _c, _d;
       const plugin = ctx.app.pm.get("ai");
       const userId = (_a = ctx.auth) == null ? void 0 : _a.user.id;
-      if (!userId) {
-        return ctx.throw(403);
-      }
-      setupSSEHeaders(ctx);
-      const { sessionId, webSearch, model } = ctx.action.params.values || {};
+      const { sessionId, webSearch, model, stream = true } = ctx.action.params.values || {};
       let { messageId } = ctx.action.params.values || {};
-      if (!sessionId) {
-        sendErrorResponse(ctx, "sessionId is required");
-        return next();
+      const shouldStream = stream !== false;
+      if (shouldStream) {
+        setupSSEHeaders(ctx);
       }
       try {
+        if (!sessionId) {
+          throw new import_utils.ResourceActionError(400, ctx.t("sessionId is required"));
+        }
         const conversation = await plugin.aiConversationsManager.getConversation({
           sessionId,
           userId
         });
         if (!conversation) {
-          sendErrorResponse(ctx, "conversation not found");
-          return next();
+          throw new import_utils.ResourceActionError(400, ctx.t("conversation not found"));
+        }
+        const employee = await getAIEmployee(ctx, conversation.aiEmployeeUsername);
+        if (!employee) {
+          throw new import_utils.ResourceActionError(400, ctx.t("AI employee not found"));
         }
         const resendMessages = [];
         if (messageId) {
@@ -549,8 +544,7 @@ var aiConversations_default = {
             }
           });
           if (!message) {
-            sendErrorResponse(ctx, "message not found");
-            return next();
+            throw new import_utils.ResourceActionError(400, ctx.t("message not found"));
           }
         } else {
           const message = await ctx.db.getRepository("aiConversations.messages", sessionId).findOne({
@@ -560,8 +554,7 @@ var aiConversations_default = {
             sort: ["-messageId"]
           });
           if (!message) {
-            sendErrorResponse(ctx, "No messages to resend");
-            return next();
+            throw new import_utils.ResourceActionError(400, ctx.t("message not found"));
           }
           messageId = message.messageId;
           if (["user", "tool"].includes(message.role)) {
@@ -575,11 +568,10 @@ var aiConversations_default = {
             });
           }
         }
-        const employee = await getAIEmployee(ctx, conversation.aiEmployeeUsername);
-        if (!employee) {
-          sendErrorResponse(ctx, "AI employee not found");
-          return next();
+        if (await isReachParallelLimit(ctx)) {
+          throw new import_utils.ResourceActionError(400, ctx.t("There are conversations in progress. Please try again later."));
         }
+        const resolvedModel = await plugin.aiEmployeesManager.resolveModel(employee, model);
         const aiEmployee = new import_ai_employee.AIEmployee({
           ctx,
           employee,
@@ -588,22 +580,40 @@ 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 });
+        if (shouldStream) {
+          await aiEmployee.stream({ messageId, userMessages: resendMessages.length ? resendMessages : void 0 });
+        } else {
+          ctx.body = await aiEmployee.invoke({
+            messageId,
+            userMessages: resendMessages.length ? resendMessages : void 0
+          });
+        }
       } catch (err) {
         ctx.log.error(err);
-        sendErrorResponse(ctx, err.message || "Chat error warning");
+        let status = 500;
+        let message = ctx.t("Server unexpected error occur");
+        if (err instanceof import_utils.ResourceActionError) {
+          status = err.status;
+          message = err.message;
+        } else if (err instanceof Error) {
+          status = 500;
+          message = err.message;
+        }
+        if (shouldStream) {
+          sendErrorResponse(ctx, message);
+        } else {
+          ctx.throw(status, message);
+        }
+      } finally {
+        await next();
       }
-      await next();
     },
     async updateUserDecision(ctx, next) {
       var _a;
       const plugin = ctx.app.pm.get("ai");
       const userId = (_a = ctx.auth) == null ? void 0 : _a.user.id;
-      if (!userId) {
-        return ctx.throw(403);
-      }
       const { sessionId, messageId, toolCallId, userDecision } = ctx.action.params.values || {};
       if (!sessionId) {
         ctx.throw(400);
@@ -683,9 +693,6 @@ var aiConversations_default = {
     async resumeToolCall(ctx, next) {
       var _a, _b, _c, _d;
       const userId = (_a = ctx.auth) == null ? void 0 : _a.user.id;
-      if (!userId) {
-        return ctx.throw(403);
-      }
       setupSSEHeaders(ctx);
       const plugin = ctx.app.pm.get("ai");
       const { sessionId, messageId, model, webSearch } = ctx.action.params.values || {};
@@ -728,6 +735,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 +744,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({