@nocobase/plugin-ai 2.3.0-alpha.1 → 2.3.0-beta.10

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");
@@ -76,11 +77,13 @@ 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);
@@ -134,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([
@@ -169,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);
@@ -189,6 +196,7 @@ class PluginAIServer extends import_server.Plugin {
     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;
@@ -235,6 +243,9 @@ class PluginAIServer extends import_server.Plugin {
     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 } = {}) => {
@@ -370,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",
@@ -142,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);
@@ -197,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,
@@ -672,6 +686,17 @@ var aiConversations_default = {
       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 }
@@ -715,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;
@@ -801,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>;

package/dist/server/resource/aiEmployees.js

@@ -36,16 +36,110 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 var aiEmployees_exports = {};
 __export(aiEmployees_exports, {
+  create: () => create,
   getTemplates: () => getTemplates,
   list: () => list,
   listByUser: () => listByUser,
+  update: () => update,
   updateUserPrompt: () => updateUserPrompt
 });
 module.exports = __toCommonJS(aiEmployees_exports);
 var import_actions = __toESM(require("@nocobase/actions"));
+var import_database = require("@nocobase/database");
 var templates = __toESM(require("../ai-employees/templates"));
 var import_lodash = __toESM(require("lodash"));
 var import_ai_feature_manager = require("../manager/ai-feature-manager");
+var import_error_codes = require("../../common/error-codes");
+var import_ai_employee_validation = require("../../common/ai-employee-validation");
+var import_ai_knowledge_base = require("../ai-employees/ai-knowledge-base");
+const isUniqueConstraintError = (error) => error instanceof import_database.UniqueConstraintError || typeof error === "object" && error !== null && "name" in error && error.name === "SequelizeUniqueConstraintError";
+const throwUsernameConflict = (ctx) => ctx.throw(409, {
+  code: import_error_codes.AI_EMPLOYEE_USERNAME_CONFLICT,
+  message: ctx.t("Username already exists")
+});
+const isRecord = (value) => value !== null && typeof value === "object" && !Array.isArray(value);
+const validateAndNormalizeProfileValues = (ctx) => {
+  const values = ctx.action.params.values;
+  if (!isRecord(values)) {
+    return;
+  }
+  if ("username" in values) {
+    if (typeof values.username !== "string" || !(0, import_ai_employee_validation.isValidAIEmployeeUsername)(values.username)) {
+      ctx.throw(400, {
+        code: import_error_codes.AI_EMPLOYEE_USERNAME_INVALID,
+        message: ctx.t("Use 1-64 letters, numbers, underscores, or hyphens.")
+      });
+    }
+    values.username = (0, import_ai_employee_validation.normalizeAIEmployeeName)(values.username);
+  }
+  if ("nickname" in values) {
+    if (typeof values.nickname !== "string" || !(0, import_ai_employee_validation.isValidAIEmployeeNickname)(values.nickname)) {
+      ctx.throw(400, {
+        code: import_error_codes.AI_EMPLOYEE_NICKNAME_INVALID,
+        message: ctx.t("Use 1-64 letters, numbers, spaces, or . _ - ' ( ) & \xB7.")
+      });
+    }
+    values.nickname = (0, import_ai_employee_validation.normalizeAIEmployeeName)(values.nickname);
+  }
+};
+const setDefaultKnowledgeBaseRetrievalStrategy = (ctx) => {
+  const values = ctx.action.params.values;
+  if (isRecord(values)) {
+    values.knowledgeBase = (0, import_ai_knowledge_base.withDefaultKnowledgeBaseRetrievalStrategy)(values.knowledgeBase);
+  }
+};
+const readEmployeeValue = (employee, key) => typeof (employee == null ? void 0 : employee.get) === "function" ? employee.get(key) : employee == null ? void 0 : employee[key];
+const validateKnowledgeBasePrompt = (ctx, employee) => {
+  const values = ctx.action.params.values;
+  if (!isRecord(values)) {
+    return;
+  }
+  const enableKnowledgeBase = typeof values.enableKnowledgeBase === "boolean" ? values.enableKnowledgeBase : readEmployeeValue(employee, "enableKnowledgeBase") === true;
+  const knowledgeBasePrompt = "knowledgeBasePrompt" in values ? values.knowledgeBasePrompt : readEmployeeValue(employee, "knowledgeBasePrompt");
+  if (enableKnowledgeBase && !(0, import_ai_employee_validation.hasKnowledgeBaseDataPlaceholder)(knowledgeBasePrompt)) {
+    ctx.throw(400, {
+      code: import_error_codes.AI_EMPLOYEE_KNOWLEDGE_BASE_PROMPT_INVALID,
+      message: ctx.t("The Knowledge Base Prompt must include {knowledgeBaseData} before you can save it."),
+      data: { field: "knowledgeBasePrompt" }
+    });
+  }
+};
+const hasKnowledgeBaseConfigurationChanges = (values) => isRecord(values) && ["enableKnowledgeBase", "knowledgeBase", "knowledgeBasePrompt"].some(
+  (key) => Object.prototype.hasOwnProperty.call(values, key)
+);
+const create = async (ctx, next) => {
+  var _a;
+  validateAndNormalizeProfileValues(ctx);
+  setDefaultKnowledgeBaseRetrievalStrategy(ctx);
+  validateKnowledgeBasePrompt(ctx);
+  const username = (_a = ctx.action.params.values) == null ? void 0 : _a.username;
+  if (typeof username === "string") {
+    const existingEmployee = await ctx.db.getRepository("aiEmployees").findOne({
+      filter: { username }
+    });
+    if (existingEmployee) {
+      throwUsernameConflict(ctx);
+    }
+  }
+  try {
+    await import_actions.default.create(ctx, next);
+  } catch (error) {
+    if (isUniqueConstraintError(error)) {
+      throwUsernameConflict(ctx);
+    }
+    throw error;
+  }
+};
+const update = async (ctx, next) => {
+  validateAndNormalizeProfileValues(ctx);
+  if (hasKnowledgeBaseConfigurationChanges(ctx.action.params.values)) {
+    const employee = await ctx.db.getRepository("aiEmployees").findOne({
+      filterByTk: ctx.action.params.filterByTk
+    });
+    validateKnowledgeBasePrompt(ctx, employee);
+  }
+  await import_actions.default.update(ctx, next);
+};
 const list = async (ctx, next) => {
   var _a, _b;
   const { paginate } = ctx.action.params || {};
@@ -227,8 +321,10 @@ const getTemplates = async (ctx, next) => {
 };
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  create,
   getTemplates,
   list,
   listByUser,
+  update,
   updateUserPrompt
 });

package/dist/server/resource/aiWorkflowTasks.js

@@ -129,7 +129,7 @@ const aiWorkflowTasks = {
       await next();
     },
     getBySession: async (ctx, next) => {
-      var _a, _b, _c, _d, _e;
+      var _a, _b, _c, _d, _e, _f;
       const userId = (_a = ctx.auth) == null ? void 0 : _a.user.id;
       if (!userId) {
         return ctx.throw(403);
@@ -158,12 +158,20 @@ const aiWorkflowTasks = {
           id: task.nodeId
         }
       });
+      const conversation = await ctx.db.getRepository("aiConversations").findOne({
+        filter: {
+          sessionId
+        },
+        appends: ["aiEmployee"]
+      });
+      const aiEmployee = conversation == null ? void 0 : conversation.aiEmployee;
       const readonly = task.status !== "pending_approval" || task.acceptedUserId !== userId;
       ctx.body = {
         ...((_c = task == null ? void 0 : task.toJSON) == null ? void 0 : _c.call(task)) ?? task,
         read: (usersAiWorkflowTasks == null ? void 0 : usersAiWorkflowTasks.read) ?? true,
         config: (node == null ? void 0 : node.config) ?? null,
         structuredOutputSchema: ((_e = (_d = node == null ? void 0 : node.config) == null ? void 0 : _d.structuredOutput) == null ? void 0 : _e.schema) ?? null,
+        aiEmployee: ((_f = aiEmployee == null ? void 0 : aiEmployee.toJSON) == null ? void 0 : _f.call(aiEmployee)) ?? aiEmployee ?? null,
         readonly
       };
       await next();

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

@@ -0,0 +1,11 @@
+/**
+ * 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 { 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>;

package/dist/server/resource/llmServices.js

@@ -0,0 +1,85 @@
+/**
+ * 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 __create = Object.create;
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
+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 __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var llmServices_exports = {};
+__export(llmServices_exports, {
+  create: () => create,
+  update: () => update
+});
+module.exports = __toCommonJS(llmServices_exports);
+var import_actions = __toESM(require("@nocobase/actions"));
+var import_package = require("../../../package.json");
+var import_error_codes = require("../../common/error-codes");
+var import_llm_service_models = require("../../common/llm-service-models");
+const isRecord = (value) => value !== null && typeof value === "object" && !Array.isArray(value);
+const validateAndNormalizeEnabledModels = (ctx) => {
+  const values = ctx.action.params.values;
+  if (!isRecord(values) || !("enabledModels" in values)) {
+    return;
+  }
+  const enabledModels = (0, import_llm_service_models.normalizeCustomModelIds)(values.enabledModels);
+  const issue = (0, import_llm_service_models.getCustomModelIdIssues)(enabledModels)[0];
+  if (issue) {
+    const required = issue.type === "required";
+    ctx.throw(400, {
+      code: required ? import_error_codes.LLM_SERVICE_MODEL_ID_REQUIRED : import_error_codes.LLM_SERVICE_MODEL_ID_DUPLICATE,
+      message: ctx.t(required ? "Model ID is required" : "Model ID already exists", { ns: import_package.name }),
+      data: { index: issue.index }
+    });
+  }
+  ctx.action.mergeParams(
+    {
+      values: {
+        ...values,
+        enabledModels
+      }
+    },
+    { values: "overwrite" }
+  );
+};
+const create = async (ctx, next) => {
+  validateAndNormalizeEnabledModels(ctx);
+  await import_actions.default.create(ctx, next);
+};
+const update = async (ctx, next) => {
+  validateAndNormalizeEnabledModels(ctx);
+  await import_actions.default.update(ctx, next);
+};
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  create,
+  update
+});

package/dist/server/types/ai-message.type.d.ts

@@ -27,6 +27,10 @@ export type WorkContext = {
 export type AIMessageContent = {
     type: string;
     content: unknown;
+    reference?: {
+        title: string;
+        url: string;
+    }[];
 };
 export type AIToolCall = {
     id: string;

package/dist/server/types/knowledge-base.type.d.ts

@@ -19,13 +19,17 @@ export type KnowledgeBase = {
     name: string;
     description: string;
     vectorStoreProvider: string;
-    vectorStoreConfigKey?: string;
+    vectorDatabaseKey?: string;
+    llmService?: string;
+    embeddingModel?: string;
     vectorStoreProps?: VectorStoreProp[];
     enabled: boolean;
 };
 export type VectorStoreConfig = {
     vectorStoreProvider: string;
-    vectorStoreConfigKey?: string;
+    vectorDatabaseKey?: string;
+    llmService?: string;
+    embeddingModel?: string;
 };
 export type KnowledgeBaseGroup = {
     vectorStoreConfig: VectorStoreConfig;
@@ -41,8 +45,9 @@ export type DocumentSegmentedWithScore = DocumentSegmented & {
     score: number;
 };
 export type SearchOptions = {
-    knowledgeBaseKeys: string[];
+    knowledgeBaseKeys?: string[];
     query: string;
     topK?: number;
     score?: string;
+    roleNames?: string[];
 };

package/dist/server/workflow/nodes/employee/files.d.ts

@@ -11,7 +11,7 @@ import { Plugin } from '@nocobase/server';
 export declare abstract class Files {
     static resolvers(plugin: Plugin, attachmentPart: {
         attachments?: unknown[];
-    }): {
+    }, fileUrlOrigin?: string): {
         resolveAttachments: (files: AIEmployeeInstructionFiles[]) => Promise<void>;
         resolveFileIds: (files: AIEmployeeInstructionFiles[]) => Promise<void>;
         resolveUrls: (files: AIEmployeeInstructionFiles[]) => Promise<void>;

package/dist/server/workflow/nodes/employee/files.js

@@ -44,6 +44,7 @@ var import_node_os = __toESM(require("node:os"));
 var import_node_path = __toESM(require("node:path"));
 var import_lodash = __toESM(require("lodash"));
 var import_axios = __toESM(require("axios"));
+var import_plugin_file_manager = require("@nocobase/plugin-file-manager");
 var import_utils = require("../../utils");
 var import_attachments = require("../../../attachments");
 function appendSource(record, source) {
@@ -55,8 +56,45 @@ function appendSource(record, source) {
     source
   };
 }
+function toFilePlainObject(record) {
+  const value = typeof record.toJSON === "function" ? record.toJSON() : record;
+  if (!value || typeof value !== "object" || Array.isArray(value)) {
+    throw new Error("Invalid file record");
+  }
+  return value;
+}
+async function resolveInternalFileURL(plugin, url, origin) {
+  var _a;
+  const reference = (0, import_plugin_file_manager.parsePermanentFileReference)(url, origin);
+  if (!reference) {
+    return null;
+  }
+  if (reference.appName !== (plugin.app.name || "main")) {
+    throw new Error("File not found");
+  }
+  const dataSource = plugin.app.dataSourceManager.get(reference.dataSourceKey);
+  const collection = dataSource == null ? void 0 : dataSource.collectionManager.getCollection(reference.collectionName);
+  if (!dataSource || !collection || collection.name !== "attachments" && ((_a = collection.options) == null ? void 0 : _a.template) !== "file") {
+    throw new Error("File not found");
+  }
+  const record = await dataSource.collectionManager.getRepository(collection.name).findOne({
+    filter: { id: reference.id }
+  });
+  if (!record) {
+    throw new Error("File not found");
+  }
+  const file = toFilePlainObject(record);
+  if (file.storageId == null || reference.extname && reference.extname !== file.extname) {
+    throw new Error("File not found");
+  }
+  return appendSource(file, {
+    dataSourceKey: reference.dataSourceKey,
+    collectionName: collection.name,
+    trustworthy: true
+  });
+}
 class Files {
-  static resolvers(plugin, attachmentPart) {
+  static resolvers(plugin, attachmentPart, fileUrlOrigin) {
     const resolveAttachments = async (files) => {
       const attachments = files.filter((it) => it.type === "attachments").flatMap((it) => import_lodash.default.isArray(it.value) ? it.value : [it.value]).map(
         (attachment) => appendSource(attachment, {
@@ -114,6 +152,15 @@ class Files {
         const storageName = (_a = settings == null ? void 0 : settings.options) == null ? void 0 : _a.storage;
         const attachments = await Promise.all(
           urls.map(async (url) => {
+            const internalFile = await resolveInternalFileURL(plugin, url, fileUrlOrigin);
+            if (internalFile) {
+              return internalFile;
+            }
+            try {
+              new URL(url);
+            } catch (error) {
+              throw new Error(`File URL must be an absolute URL or a NocoBase permanent file URL: ${url}`);
+            }
             const response = await import_axios.default.get(url, {
               responseType: "arraybuffer"
             });

package/dist/server/workflow/nodes/employee/index.js

@@ -59,7 +59,8 @@ class AIEmployeeInstruction extends import_plugin_workflow.Instruction {
       model,
       requiresApproval = import_constants.REQUIRES_APPROVAL.NO_REQUIRED,
       userId,
-      files
+      files,
+      fileUrlOrigin
     } = processor.getParsedValue(node.config, node.id);
     const toolName = import_ai.SYSTEM_TOOLS.WORK_FLOW_TASK_OUTPUT;
     const workflowSystemPrompt = `
@@ -153,7 +154,11 @@ ${typeof message.system === "object" ? JSON.stringify(message.system) : message.
         });
         const attachmentPart = {};
         if (files == null ? void 0 : files.length) {
-          const { resolveAttachments, resolveFileIds, resolveUrls } = import_files.Files.resolvers(this.workflow, attachmentPart);
+          const { resolveAttachments, resolveFileIds, resolveUrls } = import_files.Files.resolvers(
+            this.workflow,
+            attachmentPart,
+            fileUrlOrigin
+          );
           await resolveAttachments(files);
           await resolveFileIds(files);
           await resolveUrls(files);

package/dist/server/workflow/nodes/employee/types.d.ts

@@ -29,6 +29,7 @@ export type AIEmployeeInstructionConfig = {
     assignees?: string[];
     userId: string;
     files: AIEmployeeInstructionFiles[];
+    fileUrlOrigin?: string;
 };
 export type AIEmployeeInstructionFiles = {
     type: 'attachments' | 'file_id' | 'file_url';