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

package/dist/server/llm-providers/mistral.d.ts

@@ -46,9 +46,9 @@ declare class ReasoningChatMistralAI extends ChatMistralAI {
         presencePenalty?: number;
         n?: number;
         stop?: string | string[];
+        prediction?: import("@mistralai/mistralai/models/components/prediction").Prediction;
         toolChoice?: import("@mistralai/mistralai/models/components/toolchoice").ToolChoice | import("@mistralai/mistralai/models/components/toolchoiceenum").ToolChoiceEnum;
         randomSeed?: number;
-        prediction?: import("@mistralai/mistralai/models/components/prediction").Prediction;
         parallelToolCalls?: boolean;
         promptMode?: import("@mistralai/mistralai/models/components/mistralpromptmode").MistralPromptMode;
         safePrompt?: boolean;

package/dist/server/llm-providers/provider.d.ts

@@ -98,6 +98,7 @@ export declare abstract class LLMProvider {
     } | null;
     parseResponseMetadata(output: LLMResult): any;
     parseResponseError(err: any): any;
+    prepareStoredAssistantAdditionalKwargs(additionalKwargs?: Record<string, unknown>): Record<string, unknown> | undefined;
     reshapeAIMessage(_options: {
         aiMessage: AIMessage;
         values: AIMessageInput;

package/dist/server/llm-providers/provider.js

@@ -367,6 +367,9 @@ ${parsed.text}
   parseResponseError(err) {
     return (err == null ? void 0 : err.message) ?? "Unexpected LLM service error";
   }
+  prepareStoredAssistantAdditionalKwargs(additionalKwargs) {
+    return additionalKwargs;
+  }
   reshapeAIMessage(_options) {
   }
   get documentLoader() {

package/dist/server/llm-providers/shengsuanyun.d.ts

@@ -0,0 +1,29 @@
+/**
+ * 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 { ChatOpenAI } from '@langchain/openai';
+import { LLMProviderMeta } from '../manager/ai-manager';
+import { LLMProvider } from './provider';
+export type ShengSuanYunModel = {
+    id: string;
+    support_apis?: string[];
+};
+export declare function supportsChatCompletions(model: ShengSuanYunModel): boolean;
+export declare class ShengSuanYunProvider extends LLMProvider {
+    chatModel: ChatOpenAI;
+    get baseURL(): string;
+    createModel(): ChatOpenAI<import("@langchain/openai/dist/chat_models/index.cjs").ChatOpenAICallOptions>;
+    listModels(): Promise<{
+        models?: {
+            id: string;
+        }[];
+        code?: number;
+        errMsg?: string;
+    }>;
+}
+export declare const shengsuanyunProviderOptions: LLMProviderMeta;

package/dist/server/llm-providers/shengsuanyun.js

@@ -0,0 +1,114 @@
+/**
+ * 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 shengsuanyun_exports = {};
+__export(shengsuanyun_exports, {
+  ShengSuanYunProvider: () => ShengSuanYunProvider,
+  shengsuanyunProviderOptions: () => shengsuanyunProviderOptions,
+  supportsChatCompletions: () => supportsChatCompletions
+});
+module.exports = __toCommonJS(shengsuanyun_exports);
+var import_openai = require("@langchain/openai");
+var import_utils = require("@nocobase/utils");
+var import_ai_manager = require("../manager/ai-manager");
+var import_provider = require("./provider");
+const CHAT_COMPLETIONS_API = "/v1/chat/completions";
+function supportsChatCompletions(model) {
+  return !Array.isArray(model.support_apis) || model.support_apis.includes(CHAT_COMPLETIONS_API);
+}
+class ShengSuanYunProvider extends import_provider.LLMProvider {
+  get baseURL() {
+    return "https://router.shengsuanyun.com/api/v1";
+  }
+  createModel() {
+    const { apiKey, xTitle } = this.serviceOptions || {};
+    const { responseFormat, structuredOutput } = this.modelOptions || {};
+    const { name, schema } = structuredOutput || {};
+    const responseFormatOptions = {
+      type: responseFormat ?? "text"
+    };
+    if (responseFormat === "json_schema" && schema) {
+      responseFormatOptions.json_schema = {
+        schema,
+        name: name ?? "schema"
+      };
+    }
+    return new import_openai.ChatOpenAI({
+      apiKey,
+      ...this.modelOptions,
+      modelKwargs: {
+        response_format: responseFormatOptions
+      },
+      configuration: {
+        baseURL: this.getResolvedBaseURL(),
+        ...xTitle ? { defaultHeaders: { "X-Title": xTitle } } : {}
+      }
+    });
+  }
+  async listModels() {
+    var _a;
+    const { apiKey, xTitle } = this.serviceOptions || {};
+    let url;
+    try {
+      url = this.buildRequestURL("models");
+    } catch (error) {
+      return { code: 400, errMsg: error instanceof Error ? error.message : String(error) };
+    }
+    if (!apiKey) {
+      return { code: 400, errMsg: "API Key required" };
+    }
+    try {
+      const response = await (0, import_utils.serverRequest)({
+        method: "GET",
+        url,
+        headers: {
+          Authorization: `Bearer ${apiKey}`,
+          ...xTitle ? { "X-Title": xTitle } : {}
+        }
+      });
+      const models = Array.isArray((_a = response == null ? void 0 : response.data) == null ? void 0 : _a.data) ? response.data.data : [];
+      return {
+        models: models.filter(supportsChatCompletions).map(({ id }) => ({ id }))
+      };
+    } catch (error) {
+      return {
+        code: 500,
+        errMsg: error instanceof Error ? error.message : String(error)
+      };
+    }
+  }
+}
+const shengsuanyunProviderOptions = {
+  title: "SSYCloud",
+  supportedModel: [import_ai_manager.SupportedModel.LLM],
+  provider: ShengSuanYunProvider
+};
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  ShengSuanYunProvider,
+  shengsuanyunProviderOptions,
+  supportsChatCompletions
+});

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

@@ -51,7 +51,7 @@ class AIPluginFeatureManagerImpl extends import_features.BasePluginFeatureManage
     return this.features.vectorStoreProvider;
   }
   get knowledgeBase() {
-    if (!this.features.vectorStoreProvider) {
+    if (!this.features.knowledgeBase) {
       throw this.featureNotSupportedError("knowledgeBase");
     }
     return this.features.knowledgeBase;

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

@@ -16,6 +16,7 @@ export type LLMProviderMeta = {
     provider: new (opts: LLMProviderOptions) => LLMProvider;
     embedding?: new (opts: EmbeddingProviderOptions) => EmbeddingProvider;
     supportWebSearch?: boolean;
+    webSearchModels?: string[];
 };
 export declare enum SupportedModel {
     LLM = "LLM",
@@ -38,6 +39,7 @@ export type EnabledLLMService = {
     providerTitle?: string;
     enabledModels: EnabledLLMModel[];
     supportWebSearch: boolean;
+    webSearchModels?: string[];
     isToolConflict: boolean;
 };
 export declare class AIManager {
@@ -51,6 +53,7 @@ export declare class AIManager {
         title: string;
         supportedModel: SupportedModel[];
         supportWebSearch: boolean;
+        webSearchModels: string[];
     }[];
     getSupportedProvider(model: SupportedModel): string[];
     listAllEnabledModels(): Promise<EnabledLLMService[]>;

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

@@ -48,11 +48,12 @@ class AIManager {
   }
   listLLMProviders() {
     const providers = this.llmProviders.entries();
-    return Array.from(providers).map(([name, { title, supportedModel, supportWebSearch }]) => ({
+    return Array.from(providers).map(([name, { title, supportedModel, supportWebSearch, webSearchModels }]) => ({
       name,
       title,
       supportedModel: supportedModel ?? ["LLM" /* LLM */],
-      supportWebSearch: supportWebSearch ?? false
+      supportWebSearch: supportWebSearch ?? false,
+      webSearchModels
     }));
   }
   getSupportedProvider(model) {
@@ -98,6 +99,7 @@ class AIManager {
       providerTitle: providerMeta.title,
       enabledModels,
       supportWebSearch: providerMeta.supportWebSearch ?? false,
+      webSearchModels: providerMeta.webSearchModels,
       isToolConflict: providerClient.isToolConflict()
     };
   }
@@ -149,6 +151,9 @@ class AIManager {
     if (!providerOptions) {
       throw new Error("LLM service provider not found");
     }
+    if (webSearch === true && providerOptions.webSearchModels && !providerOptions.webSearchModels.includes(model)) {
+      throw new Error(`Web search is not supported by model "${model}"`);
+    }
     const Provider = providerOptions.provider;
     const provider = new Provider({
       app: this.plugin.app,

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");
@@ -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,84 @@ 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 create = async (ctx, next) => {
+  var _a;
+  validateAndNormalizeProfileValues(ctx);
+  setDefaultKnowledgeBaseRetrievalStrategy(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);
+  await import_actions.default.update(ctx, next);
+};
 const list = async (ctx, next) => {
   var _a, _b;
   const { paginate } = ctx.action.params || {};
@@ -227,8 +295,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
 });