@nocobase/plugin-ai 2.2.0-beta.8 → 2.2.0

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

@@ -0,0 +1,195 @@
+/**
+ * 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_usage_events_exports = {};
+__export(ai_usage_events_exports, {
+  buildAIUsageEventValues: () => buildAIUsageEventValues,
+  normalizeUsageMetadata: () => normalizeUsageMetadata,
+  recordAIUsageEventsForMessages: () => recordAIUsageEventsForMessages
+});
+module.exports = __toCommonJS(ai_usage_events_exports);
+const SKIPPED_MESSAGE_ROLES = /* @__PURE__ */ new Set(["user", "tool", "system"]);
+function normalizeUsageMetadata(usageMetadata) {
+  const inputTokens = getNumericValue(usageMetadata, ["input_tokens", "prompt_tokens", "inputTokens", "promptTokens"]) ?? 0;
+  const outputTokens = getNumericValue(usageMetadata, ["output_tokens", "completion_tokens", "outputTokens", "completionTokens"]) ?? 0;
+  const totalTokens = getNumericValue(usageMetadata, ["total_tokens", "totalTokens"]) ?? inputTokens + outputTokens;
+  const cachedTokens = getNumericValue(usageMetadata, ["cached_tokens", "cachedTokens"]) ?? getNestedNumericValue(usageMetadata, ["input_token_details", "cache_read"]) ?? getNestedNumericValue(usageMetadata, ["input_token_details", "cached_tokens"]) ?? getNestedNumericValue(usageMetadata, ["prompt_tokens_details", "cached_tokens"]) ?? 0;
+  const reasoningTokens = getNumericValue(usageMetadata, ["reasoning_tokens", "reasoningTokens"]) ?? getNestedNumericValue(usageMetadata, ["output_token_details", "reasoning"]) ?? getNestedNumericValue(usageMetadata, ["completion_tokens_details", "reasoning_tokens"]) ?? 0;
+  return {
+    inputTokens,
+    outputTokens,
+    totalTokens,
+    cachedTokens,
+    reasoningTokens
+  };
+}
+function buildAIUsageEventValues(sessionId, message, conversation) {
+  const role = readString(message, "role");
+  if (!role || SKIPPED_MESSAGE_ROLES.has(role)) {
+    return null;
+  }
+  const metadata = readRecord(message, "metadata");
+  const provider = readString(metadata, "provider");
+  const model = readString(metadata, "model");
+  if (!provider || !model) {
+    return null;
+  }
+  const usageMetadata = readRecord(metadata, "usage_metadata") ?? {};
+  const responseMetadata = readRecord(metadata, "response_metadata");
+  const normalizedUsage = normalizeUsageMetadata(usageMetadata);
+  const toolCalls = readArray(message, "toolCalls");
+  const autoCallTools = readArray(metadata, "autoCallTools");
+  const llmService = readString(metadata, "llmService");
+  return {
+    occurredAt: normalizeDate(readValue(message, "createdAt")),
+    sessionId,
+    messageId: String(readValue(message, "messageId") ?? ""),
+    userId: readValue(conversation, "userId"),
+    aiEmployeeUsername: readValue(conversation, "aiEmployeeUsername"),
+    from: readString(conversation, "from") ?? "main-agent",
+    category: readString(conversation, "category") ?? "chat",
+    eventType: "llm_message",
+    role,
+    provider,
+    ...llmService ? { llmService } : {},
+    model,
+    ...normalizedUsage,
+    toolCallCount: (toolCalls == null ? void 0 : toolCalls.length) ?? 0,
+    autoToolCallCount: (autoCallTools == null ? void 0 : autoCallTools.length) ?? 0,
+    status: "success",
+    rawUsageMetadata: usageMetadata,
+    ...responseMetadata ? { rawResponseMetadata: responseMetadata } : {}
+  };
+}
+async function recordAIUsageEventsForMessages(ctx, sessionId, messages, transaction) {
+  const recordableMessages = messages.filter(isRecordableLLMMessage);
+  if (recordableMessages.length === 0) {
+    return;
+  }
+  const conversation = await ctx.db.getRepository("aiConversations").findOne({
+    filter: {
+      sessionId
+    },
+    transaction
+  });
+  if (!conversation) {
+    return;
+  }
+  const usageEvents = recordableMessages.map((message) => buildAIUsageEventValues(sessionId, message, conversation)).filter(isUsageEventValues);
+  if (usageEvents.length === 0) {
+    return;
+  }
+  const usageEventsRepo = ctx.db.getRepository("aiUsageEvents");
+  for (const values of usageEvents) {
+    await usageEventsRepo.updateOrCreate({
+      filterKeys: ["messageId", "eventType"],
+      values,
+      transaction
+    });
+  }
+}
+function isUsageEventValues(value) {
+  return value !== null && value.messageId !== "";
+}
+function isRecordableLLMMessage(message) {
+  const role = readString(message, "role");
+  if (!role || SKIPPED_MESSAGE_ROLES.has(role)) {
+    return false;
+  }
+  const metadata = readRecord(message, "metadata");
+  return Boolean(readString(metadata, "provider") && readString(metadata, "model"));
+}
+function isRecord(value) {
+  return Boolean(value) && typeof value === "object" && !Array.isArray(value);
+}
+function readValue(source, key) {
+  if (!isRecord(source)) {
+    return void 0;
+  }
+  const value = source[key];
+  if (value !== void 0) {
+    return value;
+  }
+  if (typeof source.get === "function") {
+    return source.get(key);
+  }
+  return void 0;
+}
+function readString(source, key) {
+  const value = readValue(source, key);
+  return typeof value === "string" && value !== "" ? value : void 0;
+}
+function readRecord(source, key) {
+  const value = readValue(source, key);
+  return isRecord(value) ? value : void 0;
+}
+function readArray(source, key) {
+  const value = readValue(source, key);
+  return Array.isArray(value) ? value : void 0;
+}
+function getNumericValue(source, keys) {
+  for (const key of keys) {
+    const normalized = normalizeNumber(source[key]);
+    if (normalized !== void 0) {
+      return normalized;
+    }
+  }
+}
+function getNestedNumericValue(source, path) {
+  let current = source;
+  for (const key of path) {
+    if (!isRecord(current)) {
+      return void 0;
+    }
+    current = current[key];
+  }
+  return normalizeNumber(current);
+}
+function normalizeNumber(value) {
+  const numeric = typeof value === "number" ? value : typeof value === "string" ? Number(value) : NaN;
+  if (!Number.isFinite(numeric)) {
+    return void 0;
+  }
+  return Math.max(0, Math.trunc(numeric));
+}
+function normalizeDate(value) {
+  if (value instanceof Date && Number.isFinite(value.getTime())) {
+    return value;
+  }
+  if (typeof value === "string" || typeof value === "number") {
+    const date = new Date(value);
+    if (Number.isFinite(date.getTime())) {
+      return date;
+    }
+  }
+  return /* @__PURE__ */ new Date();
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  buildAIUsageEventValues,
+  normalizeUsageMetadata,
+  recordAIUsageEventsForMessages
+});

package/dist/server/manager/work-context-handler.js

@@ -41,6 +41,7 @@ __export(work_context_handler_exports, {
 module.exports = __toCommonJS(work_context_handler_exports);
 var import_utils = require("@nocobase/utils");
 var import_lodash = __toESM(require("lodash"));
+const MODEL_EXCLUDED_FIELDS = /* @__PURE__ */ new Set(["frontendTools"]);
 const createWorkContextHandler = (plugin) => new WorkContextHandlerImpl(plugin);
 class WorkContextHandlerImpl {
   constructor(plugin) {
@@ -82,11 +83,17 @@ class WorkContextHandlerImpl {
     if (!contextItem) {
       return "";
     }
-    const { resolve } = this.strategies.get(contextItem.type) ?? {};
+    const modelContext = this.excludeInternalFields(contextItem);
+    const { resolve } = this.strategies.get(modelContext.type) ?? {};
     if (resolve) {
-      return await resolve(ctx, contextItem);
+      return await resolve(ctx, modelContext);
     }
-    return await ((_b = (_a = this.defaultStrategy).resolve) == null ? void 0 : _b.call(_a, ctx, contextItem));
+    return await ((_b = (_a = this.defaultStrategy).resolve) == null ? void 0 : _b.call(_a, ctx, modelContext));
+  }
+  excludeInternalFields(contextItem) {
+    return Object.fromEntries(
+      Object.entries(contextItem).filter(([field]) => !MODEL_EXCLUDED_FIELDS.has(field))
+    );
   }
 }
 const Stringify = (_plugin) => async (_ctx, contextItem) => {

package/dist/server/migrations/20260501000000-ai-employee-knowledge-base-retrieval-strategy.d.ts

@@ -0,0 +1,14 @@
+/**
+ * This file is part of the NocoBase (R) project.
+ * Copyright (c) 2020-2024 NocoBase Co., Ltd.
+ * Authors: NocoBase Team.
+ *
+ * This project is dual-licensed under AGPL-3.0 and NocoBase Commercial License.
+ * For more information, please refer to: https://www.nocobase.com/agreement.
+ */
+import { Migration } from '@nocobase/server';
+export default class extends Migration {
+    on: string;
+    appVersion: string;
+    up(): Promise<void>;
+}

package/dist/server/migrations/20260501000000-ai-employee-knowledge-base-retrieval-strategy.js

@@ -0,0 +1,49 @@
+/**
+ * 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_retrieval_strategy_exports = {};
+__export(ai_employee_knowledge_base_retrieval_strategy_exports, {
+  default: () => ai_employee_knowledge_base_retrieval_strategy_default
+});
+module.exports = __toCommonJS(ai_employee_knowledge_base_retrieval_strategy_exports);
+var import_server = require("@nocobase/server");
+var import_ai_knowledge_base = require("../ai-employees/ai-knowledge-base");
+const getKnowledgeBaseSettings = (value) => value !== null && typeof value === "object" && !Array.isArray(value) ? { ...value } : {};
+class ai_employee_knowledge_base_retrieval_strategy_default extends import_server.Migration {
+  on = "afterSync";
+  appVersion = "<3.0.0";
+  async up() {
+    const repository = this.db.getRepository("aiEmployees");
+    const employees = await repository.find();
+    for (const employee of employees) {
+      const knowledgeBase = getKnowledgeBaseSettings(employee.get("knowledgeBase"));
+      if ((0, import_ai_knowledge_base.isKnowledgeBaseRetrievalStrategy)(knowledgeBase.retrievalStrategy)) {
+        continue;
+      }
+      await employee.update({ knowledgeBase: { ...knowledgeBase, retrievalStrategy: "always" } });
+    }
+  }
+}

package/dist/server/plugin.d.ts

@@ -62,6 +62,7 @@ export declare class PluginAIServer extends Plugin {
     get repositories(): {
         aiContextDatasources: import("@nocobase/database").Repository<any, any>;
     };
+    private registerAIFileAccessAuthorizer;
     get fileManager(): PluginFileManagerServer;
     private repository;
 }

package/dist/server/plugin.js

@@ -56,6 +56,7 @@ var import_ai_employees_manager = require("./ai-employees/ai-employees-manager")
 var import_ai_conversations = require("./ai-employees/ai-conversations");
 var import_snowflake = __toESM(require("./snowflake"));
 var aiEmployeeActions = __toESM(require("./resource/aiEmployees"));
+var llmServiceActions = __toESM(require("./resource/llmServices"));
 var import_google_genai = require("./llm-providers/google-genai");
 var import_ai_employee = require("./workflow/triggers/ai-employee");
 var import_tools = require("./tools");
@@ -74,10 +75,15 @@ var import_xai = require("./llm-providers/xai");
 var import_document_loader = require("./document-loader");
 var import_checkpoints = require("./ai-employees/checkpoints");
 var import_mimo = require("./llm-providers/mimo");
+var import_mistral = require("./llm-providers/mistral");
+var import_orcarouter = require("./llm-providers/orcarouter");
+var import_shengsuanyun = require("./llm-providers/shengsuanyun");
 var import_sub_agents = require("./ai-employees/sub-agents");
 var import_employee = require("./workflow/nodes/employee");
 var import_ai_knowledge_base = require("./ai-employees/ai-knowledge-base");
 var import_llm_stream_manager = require("./manager/llm-stream-manager");
+var import_attachments = require("./attachments");
+var import_ai_knowledge_base2 = require("./ai-employees/ai-knowledge-base");
 class PluginAIServer extends import_server.Plugin {
   features = new import_ai_feature_manager.AIPluginFeatureManagerImpl();
   aiManager = new import_ai_manager.AIManager(this);
@@ -131,6 +137,9 @@ class PluginAIServer extends import_server.Plugin {
     this.app.on("afterUpgrade", async () => {
       await this.resetConversationThreadsWhenCheckpointsEmpty();
     });
+    this.db.on("aiEmployees.beforeCreate", (employee) => {
+      employee.set("knowledgeBase", (0, import_ai_knowledge_base2.withDefaultKnowledgeBaseRetrievalStrategy)(employee.get("knowledgeBase")));
+    });
   }
   async resetConversationThreadsWhenCheckpointsEmpty() {
     const [checkpointCount, checkpointWriteCount, checkpointBlobCount, conversationsWithThreadCount] = await Promise.all([
@@ -166,6 +175,7 @@ class PluginAIServer extends import_server.Plugin {
     this.defineResources();
     this.registerMcpClientEvents();
     this.setPermissions();
+    this.registerAIFileAccessAuthorizer();
     this.registerWorkflow();
     this.registerWorkContextResolveStrategy();
     (0, import_employee.registerAIEmployeeTaskNotification)(this);
@@ -180,10 +190,13 @@ class PluginAIServer extends import_server.Plugin {
     this.aiManager.registerLLMProvider("dashscope", import_dashscope.dashscopeProviderOptions);
     this.aiManager.registerLLMProvider("kimi", import_kimi.kimiProviderOptions);
     this.aiManager.registerLLMProvider("mimo", import_mimo.mimoProviderOptions);
+    this.aiManager.registerLLMProvider("mistral", import_mistral.mistralProviderOptions);
     this.aiManager.registerLLMProvider("ollama", import_ollama.ollamaProviderOptions);
     this.aiManager.registerLLMProvider("openai-completions", import_openai2.openaiCompletionsProviderOptions);
     this.aiManager.registerLLMProvider("kimi", import_kimi.kimiProviderOptions);
     this.aiManager.registerLLMProvider("xai", import_xai.xaiProviderOptions);
+    this.aiManager.registerLLMProvider("orcarouter", import_orcarouter.orcarouterProviderOptions);
+    this.aiManager.registerLLMProvider("shengsuanyun", import_shengsuanyun.shengsuanyunProviderOptions);
   }
   registerTools() {
     const toolsManager = this.ai.toolsManager;
@@ -214,9 +227,25 @@ class PluginAIServer extends import_server.Plugin {
       },
       { before: "createMiddleware" }
     );
+    this.app.resourceManager.use(
+      async (ctx, next) => {
+        const { resourceName, actionName } = ctx.action;
+        if (resourceName === "aiFiles" && actionName === "create") {
+          (0, import_attachments.appendAIFileAttachmentSource)(ctx.action.params.values);
+        }
+        await next();
+        if (resourceName === "aiFiles" && actionName === "create") {
+          (0, import_attachments.appendAIFileAttachmentSource)(ctx.body);
+        }
+      },
+      { after: "createMiddleware" }
+    );
     Object.entries(aiEmployeeActions).forEach(([name, action]) => {
       this.app.resourceManager.registerActionHandler(`aiEmployees:${name}`, action);
     });
+    Object.entries(llmServiceActions).forEach(([name, action]) => {
+      this.app.resourceManager.registerActionHandler(`llmServices:${name}`, action);
+    });
   }
   registerMcpClientEvents() {
     this.db.on("aiMcpClients.afterCreate", (model, { transaction } = {}) => {
@@ -352,6 +381,26 @@ class PluginAIServer extends import_server.Plugin {
       aiContextDatasources: this.repository("aiContextDatasources")
     };
   }
+  registerAIFileAccessAuthorizer() {
+    this.fileManager.registerFileAccessAuthorizer({
+      name: "ai-files",
+      authorize: async (ctx, params) => {
+        var _a;
+        const currentUserId = (_a = ctx.state.currentUser) == null ? void 0 : _a.id;
+        if (params.dataSourceKey !== "main" || params.collectionName !== "aiFiles" || !currentUserId) {
+          return false;
+        }
+        const file = await ctx.db.getRepository("aiFiles").findOne({
+          filter: {
+            id: params.id,
+            createdById: currentUserId
+          },
+          fields: ["id"]
+        });
+        return Boolean(file);
+      }
+    });
+  }
   get fileManager() {
     return this.app.pm.get("file-manager");
   }

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

@@ -7,6 +7,8 @@
  * For more information, please refer to: https://www.nocobase.com/agreement.
  */
 import { Context, Next } from '@nocobase/actions';
+import { Model } from '@nocobase/database';
+export declare function isAIEmployeeEnabled(employee: Model | null | undefined): boolean;
 declare function loginInCheck(ctx: Context, next: Next): Promise<never>;
 declare const _default: {
     name: string;

package/dist/server/resource/aiConversations.js

@@ -36,7 +36,8 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 var aiConversations_exports = {};
 __export(aiConversations_exports, {
-  default: () => aiConversations_default
+  default: () => aiConversations_default,
+  isAIEmployeeEnabled: () => isAIEmployeeEnabled
 });
 module.exports = __toCommonJS(aiConversations_exports);
 var import_actions = __toESM(require("@nocobase/actions"));
@@ -44,6 +45,8 @@ 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");
+var import_frontend_tools = require("../../common/frontend-tools");
+var import_frontend_tools2 = require("../frontend-tools");
 async function getAIEmployee(ctx, username) {
   var _a;
   const filter = {
@@ -57,6 +60,10 @@ async function getAIEmployee(ctx, username) {
   });
   return employee;
 }
+function isAIEmployeeEnabled(employee) {
+  var _a;
+  return ((_a = employee == null ? void 0 : employee.get) == null ? void 0 : _a.call(employee, "enabled")) !== false;
+}
 function setupSSEHeaders(ctx) {
   ctx.set({
     "Content-Type": "text/event-stream",
@@ -103,6 +110,33 @@ const saveUserMessages = async (ctx, sessionId, messages, messageId) => {
     await conversation.addMessages(userMessages);
   });
 };
+function isRecord(value) {
+  return Boolean(value && typeof value === "object" && !Array.isArray(value));
+}
+function normalizeIncomingMessageAttachments(ctx, messages) {
+  for (const message of messages) {
+    if (message.attachments == null) {
+      continue;
+    }
+    if (!Array.isArray(message.attachments)) {
+      throw new import_utils.ResourceActionError(400, ctx.t("Invalid attachment"));
+    }
+    message.attachments = message.attachments.map((attachment) => {
+      if (!isRecord(attachment) || !isRecord(attachment.source)) {
+        throw new import_utils.ResourceActionError(400, ctx.t("Invalid attachment"));
+      }
+      const source = { ...attachment.source };
+      delete source.trustworthy;
+      if (typeof source.collectionName !== "string" || !source.collectionName) {
+        throw new import_utils.ResourceActionError(400, ctx.t("Invalid attachment"));
+      }
+      return {
+        ...attachment,
+        source
+      };
+    });
+  }
+}
 var aiConversations_default = {
   name: "aiConversations",
   middlewares: [
@@ -115,12 +149,14 @@ var aiConversations_default = {
       var _a;
       const userId = (_a = ctx.auth) == null ? void 0 : _a.user.id;
       const filter = ctx.action.params.filter || {};
+      const scope = ctx.action.params.scope;
       ctx.action.mergeParams({
         filter: {
           ...filter,
           userId,
           from: filter.from ?? "main-agent",
-          category: "chat"
+          category: "chat",
+          ...typeof scope === "string" && scope ? { scope } : {}
         }
       });
       return import_actions.default.list(ctx, next);
@@ -170,15 +206,20 @@ var aiConversations_default = {
       var _a;
       const plugin = ctx.app.pm.get("ai");
       const userId = (_a = ctx.auth) == null ? void 0 : _a.user.id;
-      const { aiEmployee, systemMessage, skillSettings, conversationSettings, modelSettings } = ctx.action.params.values || {};
+      const { aiEmployee, systemMessage, skillSettings, conversationSettings, modelSettings, scope } = ctx.action.params.values || {};
+      const normalizedScope = typeof scope === "string" ? scope : void 0;
       const employee = await getAIEmployee(ctx, aiEmployee.username);
       if (!employee) {
         ctx.throw(400, "AI employee not found");
       }
+      if (!isAIEmployeeEnabled(employee)) {
+        ctx.throw(400, "AI employee is disabled");
+      }
       try {
         ctx.body = await plugin.aiConversationsManager.create({
           userId,
           aiEmployee,
+          scope: normalizedScope,
           options: {
             systemMessage,
             skillSettings,
@@ -329,6 +370,7 @@ var aiConversations_default = {
         if (!Array.isArray(messages)) {
           throw new import_utils.ResourceActionError(400, ctx.t("messages must be an array"));
         }
+        normalizeIncomingMessageAttachments(ctx, messages);
         const userMessage = messages.find((message) => message.role === "user");
         if (!userMessage) {
           throw new import_utils.ResourceActionError(400, ctx.t("user message is required"));
@@ -633,10 +675,28 @@ var aiConversations_default = {
       if (!message) {
         ctx.throw(400);
       }
+      const messageConversation = await plugin.aiConversationsManager.getConversation({
+        sessionId: message.sessionId,
+        userId
+      });
+      if (!messageConversation) {
+        ctx.throw(400);
+      }
       const toolCalls = message.toolCalls;
       if (!(toolCalls == null ? void 0 : toolCalls.length)) {
         ctx.throw(400);
       }
+      const selectedToolCall = toolCalls.find((toolCall2) => toolCall2.id === toolCallId);
+      if (!selectedToolCall) {
+        ctx.throw(400);
+      }
+      if (selectedToolCall.name === import_frontend_tools.EXECUTE_FRONTEND_TOOL_NAME) {
+        const toolId = isRecord(selectedToolCall.args) ? selectedToolCall.args.toolId : void 0;
+        const frontendTool = typeof toolId === "string" ? await (0, import_frontend_tools2.findCurrentFrontendTool)(ctx, toolId, message.sessionId) : void 0;
+        if (!frontendTool) {
+          ctx.throw(400, ctx.t("Frontend tool is unavailable"));
+        }
+      }
       const aiToolMessagesModel = ctx.db.getModel("aiToolMessages");
       const toolCall = await aiToolMessagesModel.findOne({
         where: { sessionId: message.sessionId, messageId: message.messageId, toolCallId }
@@ -680,6 +740,7 @@ var aiConversations_default = {
         toolCall2.auto = toolMessage == null ? void 0 : toolMessage.auto;
         toolCall2.status = toolMessage == null ? void 0 : toolMessage.status;
         toolCall2.content = toolMessage == null ? void 0 : toolMessage.content;
+        toolCall2.userDecision = toolMessage == null ? void 0 : toolMessage.userDecision;
         toolCall2.execution = tools == null ? void 0 : tools.execution;
         toolCall2.willInterrupt = (tools == null ? void 0 : tools.execution) === "frontend" || (toolMessage == null ? void 0 : toolMessage.auto) === false;
         toolCall2.defaultPermission = tools == null ? void 0 : tools.defaultPermission;
@@ -730,6 +791,14 @@ var aiConversations_default = {
           sendErrorResponse(ctx, "message not found");
           return next();
         }
+        const messageConversation = await plugin.aiConversationsManager.getConversation({
+          sessionId: message.sessionId,
+          userId
+        });
+        if (!messageConversation) {
+          sendErrorResponse(ctx, "conversation not found");
+          return next();
+        }
         const tools = message.toolCalls;
         if (!(tools == null ? void 0 : tools.length)) {
           sendErrorResponse(ctx, "No tool calls found");
@@ -746,7 +815,7 @@ var aiConversations_default = {
           webSearch,
           model: resolvedModel
         });
-        const userDecisions = await plugin.aiConversationsManager.getUserDecisions(messageId);
+        const userDecisions = await plugin.aiConversationsManager.getUserDecisions(message.messageId);
         await aiEmployee.stream({
           userDecisions
         });
@@ -758,3 +827,7 @@ var aiConversations_default = {
     }
   }
 };
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  isAIEmployeeEnabled
+});

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

@@ -7,6 +7,8 @@
  * For more information, please refer to: https://www.nocobase.com/agreement.
  */
 import { Context, Next } from '@nocobase/actions';
+export declare const create: (ctx: Context, next: Next) => Promise<void>;
+export declare const update: (ctx: Context, next: Next) => Promise<void>;
 export declare const list: (ctx: Context, next: Next) => Promise<void>;
 export declare const listByUser: (ctx: Context, next: Next) => Promise<any>;
 export declare const updateUserPrompt: (ctx: Context, next: Next) => Promise<any>;