@nocobase/plugin-ai 2.1.0-alpha.20 → 2.1.0-alpha.22

package/dist/server/plugin.js

@@ -49,6 +49,7 @@ var import_deepseek = require("./llm-providers/deepseek");
 var import_ai = __toESM(require("./resource/ai"));
 var import_llm = require("./workflow/nodes/llm");
 var import_aiConversations = __toESM(require("./resource/aiConversations"));
+var import_aiWorkflowTasks = __toESM(require("./resource/aiWorkflowTasks"));
 var import_aiTools = __toESM(require("./resource/aiTools"));
 var import_aiSkills = __toESM(require("./resource/aiSkills"));
 var import_ai_employees_manager = require("./ai-employees/ai-employees-manager");
@@ -72,6 +73,8 @@ var import_kimi = require("./llm-providers/kimi");
 var import_document_loader = require("./document-loader");
 var import_checkpoints = require("./ai-employees/checkpoints");
 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");
 class PluginAIServer extends import_server.Plugin {
   features = new import_ai_feature_manager.AIPluginFeatureManagerImpl();
   aiManager = new import_ai_manager.AIManager(this);
@@ -83,6 +86,7 @@ class PluginAIServer extends import_server.Plugin {
   workContextHandler = (0, import_work_context_handler.createWorkContextHandler)(this);
   documentLoaders = new import_document_loader.DocumentLoaders(this);
   subAgentsDispatcher = new import_sub_agents.SubAgentsDispatcher(this);
+  knowledgeBaseManager = new import_ai_knowledge_base.KnowledgeBaseManager(this);
   snowflake;
   /**
    * Check if the AI employee is a builder/admin-only type (e.g., Nathan, Orin).
@@ -128,6 +132,8 @@ class PluginAIServer extends import_server.Plugin {
     this.setPermissions();
     this.registerWorkflow();
     this.registerWorkContextResolveStrategy();
+    (0, import_employee.registerAIEmployeeTaskNotification)(this);
+    (0, import_employee.registerOnJobAbortedHandler)(this);
   }
   registerLLMProviders() {
     this.aiManager.registerLLMProvider("google-genai", import_google_genai.googleGenAIProviderOptions);
@@ -143,11 +149,13 @@ class PluginAIServer extends import_server.Plugin {
     const toolsManager = this.ai.toolsManager;
     toolsManager.registerTools([(0, import_tools.createDocsSearchTool)(), (0, import_tools.createReadDocEntryTool)()]);
     toolsManager.registerDynamicTools((0, import_tools.getWorkflowCallers)(this, "workflowCaller"));
+    toolsManager.registerDynamicTools((0, import_employee.getWorkflowTasks)(this));
     toolsManager.registerDynamicTools(this.ai.mcpManager.getMCPToolsProvider());
   }
   defineResources() {
     this.app.resourceManager.define(import_ai.default);
     this.app.resourceManager.define(import_aiConversations.default);
+    this.app.resourceManager.define(import_aiWorkflowTasks.default);
     this.app.resourceManager.define(import_aiTools.default);
     this.app.resourceManager.define(import_aiSkills.default);
     this.app.resourceManager.define(import_aiSettings.default);
@@ -188,6 +196,7 @@ class PluginAIServer extends import_server.Plugin {
       actions: ["aiSettings:*"]
     });
     this.app.acl.allow("aiConversations", "*", "loggedIn");
+    this.app.acl.allow("aiWorkflowTasks", "*", "loggedIn");
     this.app.acl.allow("aiContextDatasources", "get", "loggedIn");
     this.app.acl.allow("aiContextDatasources", "list", "loggedIn");
     this.app.acl.allow("aiContextDatasources", "preview", "loggedIn");
@@ -235,6 +244,7 @@ class PluginAIServer extends import_server.Plugin {
     const workflow = this.app.pm.get("workflow");
     workflow.registerTrigger("ai-employee", import_ai_employee.AIEmployeeTrigger);
     workflow.registerInstruction("llm", import_llm.LLMInstruction);
+    workflow.registerInstruction("ai-employee", import_employee.AIEmployeeInstruction);
   }
   registerWorkContextResolveStrategy() {
     this.workContextHandler.registerStrategy("datasource", {

package/dist/server/resource/aiConversations.js

@@ -80,7 +80,8 @@ var aiConversations_default = {
         filter: {
           ...filter,
           userId,
-          from: filter.from ?? "main-agent"
+          from: filter.from ?? "main-agent",
+          category: "chat"
         }
       });
       return import_actions.default.list(ctx, next);
@@ -198,6 +199,7 @@ var aiConversations_default = {
     },
     async updateToolArgs(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);
@@ -210,11 +212,9 @@ var aiConversations_default = {
       if (!sessionId) {
         ctx.throw(400);
       }
-      const conversation = await ctx.db.getRepository("aiConversations").findOne({
-        filter: {
-          sessionId,
-          userId
-        }
+      const conversation = await plugin.aiConversationsManager.getConversation({
+        sessionId,
+        userId
       });
       if (!conversation) {
         ctx.throw(400);
@@ -241,7 +241,7 @@ var aiConversations_default = {
       await next();
     },
     async sendMessages(ctx, next) {
-      var _a, _b, _c;
+      var _a, _b, _c, _d;
       const userId = (_a = ctx.auth) == null ? void 0 : _a.user.id;
       if (!userId) {
         return ctx.throw(403);
@@ -280,11 +280,9 @@ var aiConversations_default = {
         return next();
       }
       try {
-        const conversation = await ctx.db.getRepository("aiConversations").findOne({
-          filter: {
-            sessionId,
-            userId
-          }
+        const conversation = await plugin.aiConversationsManager.getConversation({
+          sessionId,
+          userId
         });
         if (!conversation) {
           if (shouldStream) {
@@ -325,6 +323,7 @@ var aiConversations_default = {
           sessionId,
           systemMessage: (_b = conversation.options) == null ? void 0 : _b.systemMessage,
           skillSettings: (_c = conversation.options) == null ? void 0 : _c.skillSettings,
+          tools: (_d = conversation.options) == null ? void 0 : _d.tools,
           webSearch,
           model,
           legacy
@@ -382,11 +381,9 @@ var aiConversations_default = {
       }
       const { sessionId } = ctx.action.params.values || {};
       const plugin = ctx.app.pm.get("ai");
-      const conversation = await ctx.db.getRepository("aiConversations").findOne({
-        filter: {
-          sessionId,
-          userId
-        }
+      const conversation = await plugin.aiConversationsManager.getConversation({
+        sessionId,
+        userId
       });
       if (!conversation) {
         ctx.throw(404, "conversation not found");
@@ -395,7 +392,8 @@ var aiConversations_default = {
       await next();
     },
     async resendMessages(ctx, next) {
-      var _a, _b, _c;
+      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);
@@ -408,11 +406,9 @@ var aiConversations_default = {
         return next();
       }
       try {
-        const conversation = await ctx.db.getRepository("aiConversations").findOne({
-          filter: {
-            sessionId,
-            userId
-          }
+        const conversation = await plugin.aiConversationsManager.getConversation({
+          sessionId,
+          userId
         });
         if (!conversation) {
           sendErrorResponse(ctx, "conversation not found");
@@ -463,6 +459,7 @@ var aiConversations_default = {
           sessionId,
           systemMessage: (_b = conversation.options) == null ? void 0 : _b.systemMessage,
           skillSettings: (_c = conversation.options) == null ? void 0 : _c.skillSettings,
+          tools: (_d = conversation.options) == null ? void 0 : _d.tools,
           webSearch,
           model
         });
@@ -484,11 +481,9 @@ var aiConversations_default = {
       if (!sessionId) {
         ctx.throw(400);
       }
-      const conversation = await ctx.db.getRepository("aiConversations").findOne({
-        filter: {
-          sessionId,
-          userId
-        }
+      const conversation = await plugin.aiConversationsManager.getConversation({
+        sessionId,
+        userId
       });
       if (!conversation) {
         ctx.throw(400);
@@ -539,7 +534,7 @@ var aiConversations_default = {
       const toolMessageMap = new Map(
         toolMessages.map((toolMessage) => [toolMessage.toolCallId, toolMessage])
       );
-      const toolsList = await plugin.ai.toolsManager.listTools();
+      const toolsList = await plugin.ai.toolsManager.listTools({ sessionId: message.sessionId });
       const toolsMap = new Map(toolsList.map((t) => [t.definition.name, t]));
       for (const toolCall2 of toolCalls) {
         const tools = toolsMap.get(toolCall2.name);
@@ -559,7 +554,7 @@ var aiConversations_default = {
       await next();
     },
     async resumeToolCall(ctx, next) {
-      var _a, _b, _c;
+      var _a, _b, _c, _d;
       const userId = (_a = ctx.auth) == null ? void 0 : _a.user.id;
       if (!userId) {
         return ctx.throw(403);
@@ -572,11 +567,9 @@ var aiConversations_default = {
         return next();
       }
       try {
-        const conversation = await ctx.db.getRepository("aiConversations").findOne({
-          filter: {
-            sessionId,
-            userId
-          }
+        const conversation = await plugin.aiConversationsManager.getConversation({
+          sessionId,
+          userId
         });
         if (!conversation) {
           sendErrorResponse(ctx, "conversation not found");
@@ -614,6 +607,7 @@ var aiConversations_default = {
           sessionId,
           systemMessage: (_b = conversation.options) == null ? void 0 : _b.systemMessage,
           skillSettings: (_c = conversation.options) == null ? void 0 : _c.skillSettings,
+          tools: (_d = conversation.options) == null ? void 0 : _d.tools,
           webSearch,
           model
         });

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

@@ -0,0 +1,12 @@
+/**
+ * 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 { ResourceOptions } from '@nocobase/resourcer';
+export declare const parseAiWorkflowTaskListRecord: (record: any, userId: string | number, jobStatusMap: Map<string, number>) => any;
+export declare const aiWorkflowTasks: ResourceOptions;
+export default aiWorkflowTasks;

package/dist/server/resource/aiWorkflowTasks.js

@@ -0,0 +1,290 @@
+/**
+ * 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 aiWorkflowTasks_exports = {};
+__export(aiWorkflowTasks_exports, {
+  aiWorkflowTasks: () => aiWorkflowTasks,
+  default: () => aiWorkflowTasks_default,
+  parseAiWorkflowTaskListRecord: () => parseAiWorkflowTaskListRecord
+});
+module.exports = __toCommonJS(aiWorkflowTasks_exports);
+var import_actions = __toESM(require("@nocobase/actions"));
+var import_plugin_workflow = require("@nocobase/plugin-workflow");
+const parseAiWorkflowTaskListRecord = (record, userId, jobStatusMap) => {
+  var _a, _b;
+  const users = (record == null ? void 0 : record.users) || [];
+  const currentUser = users.find(
+    (user) => String(user == null ? void 0 : user.id) === String(userId)
+  );
+  return {
+    ...(record == null ? void 0 : record.toJSON()) ?? {},
+    read: ((_a = currentUser == null ? void 0 : currentUser.usersAiWorkflowTasks) == null ? void 0 : _a.read) ?? false,
+    jobStatus: jobStatusMap.get(String((record == null ? void 0 : record.jobId) ?? ((_b = record == null ? void 0 : record.get) == null ? void 0 : _b.call(record, "jobId")))) ?? import_plugin_workflow.JOB_STATUS.PENDING
+  };
+};
+const aiWorkflowTasks = {
+  name: "aiWorkflowTasks",
+  actions: {
+    list: async (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 || {};
+      const appends = Array.isArray(ctx.action.params.appends) ? ctx.action.params.appends : ctx.action.params.appends ? [ctx.action.params.appends] : [];
+      if (!appends.includes("users")) {
+        appends.push("users");
+      }
+      ctx.action.mergeParams({
+        filter: {
+          $and: [
+            filter,
+            {
+              "users.id": userId
+            }
+          ]
+        },
+        appends
+      });
+      await import_actions.default.list(ctx, async () => {
+      });
+      const jobIds = Array.from(
+        new Set(
+          (ctx.body.rows || []).map((record) => {
+            var _a2;
+            return (record == null ? void 0 : record.jobId) ?? ((_a2 = record == null ? void 0 : record.get) == null ? void 0 : _a2.call(record, "jobId"));
+          }).filter((jobId) => jobId != null)
+        )
+      );
+      const jobs = jobIds.length ? await ctx.db.getRepository("jobs").find({
+        filter: {
+          id: {
+            $in: jobIds
+          }
+        },
+        fields: ["id", "status"]
+      }) : [];
+      const jobStatusMap = new Map(
+        jobs.map((job) => {
+          var _a2, _b;
+          return [
+            String((job == null ? void 0 : job.id) ?? ((_a2 = job == null ? void 0 : job.get) == null ? void 0 : _a2.call(job, "id"))),
+            (job == null ? void 0 : job.status) ?? ((_b = job == null ? void 0 : job.get) == null ? void 0 : _b.call(job, "status")) ?? import_plugin_workflow.JOB_STATUS.PENDING
+          ];
+        })
+      );
+      ctx.body.rows = ctx.body.rows.map((record) => parseAiWorkflowTaskListRecord(record, userId, jobStatusMap));
+      await next();
+    },
+    unreadCount: async (ctx, next) => {
+      var _a;
+      const userId = (_a = ctx.auth) == null ? void 0 : _a.user.id;
+      if (!userId) {
+        return ctx.throw(403);
+      }
+      const usersAiWorkflowTasksModel = ctx.db.getModel("usersAiWorkflowTasks");
+      const count = await usersAiWorkflowTasksModel.count({
+        where: {
+          userId,
+          read: false
+        }
+      });
+      ctx.body = {
+        count
+      };
+      await next();
+    },
+    getBySession: async (ctx, next) => {
+      var _a, _b, _c, _d, _e;
+      const userId = (_a = ctx.auth) == null ? void 0 : _a.user.id;
+      if (!userId) {
+        return ctx.throw(403);
+      }
+      const sessionId = (_b = ctx.action.params.values) == null ? void 0 : _b.sessionId;
+      if (!sessionId) {
+        return ctx.throw(400, "sessionId is required");
+      }
+      const task = await ctx.db.getRepository("aiWorkflowTasks").findOne({
+        filter: {
+          sessionId,
+          "users.id": userId
+        }
+      });
+      if (!task) {
+        return ctx.throw(404, "workflow task not found");
+      }
+      const usersAiWorkflowTasks = await ctx.db.getModel("usersAiWorkflowTasks").findOne({
+        where: {
+          aiWorkflowTaskId: task.id,
+          userId
+        }
+      });
+      const node = await ctx.db.getRepository("flow_nodes").findOne({
+        filter: {
+          id: task.nodeId
+        }
+      });
+      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,
+        readonly
+      };
+      await next();
+    },
+    accept: async (ctx, next) => {
+      var _a, _b;
+      const userId = (_a = ctx.auth) == null ? void 0 : _a.user.id;
+      if (!userId) {
+        return ctx.throw(403);
+      }
+      const sessionId = (_b = ctx.action.params.values) == null ? void 0 : _b.sessionId;
+      if (!sessionId) {
+        return ctx.throw(400, "sessionId is required");
+      }
+      const task = await ctx.db.getRepository("aiWorkflowTasks").findOne({
+        filter: {
+          sessionId,
+          "users.id": userId
+        }
+      });
+      if (!task) {
+        return ctx.throw(404, "workflow task not found");
+      }
+      const aiWorkflowTasksModel = ctx.db.getModel("aiWorkflowTasks");
+      const [acceptedCount] = await aiWorkflowTasksModel.update(
+        {
+          acceptedUserId: userId,
+          status: "pending_approval"
+        },
+        {
+          where: {
+            id: task.id,
+            acceptedUserId: null,
+            status: "pending_acceptance"
+          }
+        }
+      );
+      const usersAiWorkflowTasksModel = ctx.db.getModel("usersAiWorkflowTasks");
+      const [readCount] = await usersAiWorkflowTasksModel.update(
+        {
+          read: true
+        },
+        {
+          where: {
+            aiWorkflowTaskId: task.id,
+            userId
+          }
+        }
+      );
+      const latestTask = await ctx.db.getRepository("aiWorkflowTasks").findOne({
+        filter: {
+          id: task.id
+        }
+      });
+      ctx.body = {
+        accepted: acceptedCount > 0,
+        readUpdated: readCount > 0,
+        acceptedUserId: (latestTask == null ? void 0 : latestTask.acceptedUserId) ?? null
+      };
+      await next();
+    },
+    reject: async (ctx, next) => {
+      var _a, _b, _c;
+      const userId = (_a = ctx.auth) == null ? void 0 : _a.user.id;
+      if (!userId) {
+        return ctx.throw(403);
+      }
+      const id = (_b = ctx.action.params.values) == null ? void 0 : _b.id;
+      const result = (_c = ctx.action.params.values) == null ? void 0 : _c.result;
+      if (!id) {
+        return ctx.throw(400, "id is required");
+      }
+      if (!result) {
+        return ctx.throw(400, "result is required");
+      }
+      const task = await ctx.db.getRepository("aiWorkflowTasks").findOne({
+        filter: {
+          id,
+          "users.id": userId
+        }
+      });
+      if (!task) {
+        return ctx.throw(404, "workflow task not found");
+      }
+      if (task.status !== "pending_approval" || String(task.acceptedUserId) !== String(userId)) {
+        return ctx.throw(403);
+      }
+      const job = await ctx.db.getRepository("jobs").findOne({
+        filter: {
+          id: task.jobId
+        }
+      });
+      if (!job) {
+        return ctx.throw(404, "job not found");
+      }
+      await ctx.db.getRepository("aiWorkflowTasks").update({
+        values: {
+          status: "rejected"
+        },
+        filter: {
+          id: task.id,
+          acceptedUserId: userId,
+          status: "pending_approval"
+        }
+      });
+      await job.update({
+        status: import_plugin_workflow.JOB_STATUS.REJECTED,
+        result
+      });
+      const workflowPlugin = ctx.app.pm.get("workflow");
+      await workflowPlugin.resume(job);
+      ctx.body = {
+        status: "success"
+      };
+      await next();
+    }
+  }
+};
+var aiWorkflowTasks_default = aiWorkflowTasks;
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  aiWorkflowTasks,
+  parseAiWorkflowTaskListRecord
+});

package/dist/server/utils.js

@@ -131,6 +131,7 @@ async function getUser(ctx, fields) {
   return user;
 }
 async function parseVariables(ctx, value) {
+  var _a, _b;
   const re = /\{\{\$user\.([^}]+)\}\}/g;
   const userFieldsSet = /* @__PURE__ */ new Set();
   const matches = value.matchAll(re);
@@ -144,7 +145,7 @@ async function parseVariables(ctx, value) {
   for (const [key, value2] of Object.entries(dateVariables)) {
     if (typeof value2 === "function") {
       $nDate[key] = value2({
-        timezone: ctx.get("x-timezone"),
+        timezone: (_a = ctx.get) == null ? void 0 : _a.call(ctx, "x-timezone"),
         now: (/* @__PURE__ */ new Date()).toISOString()
       });
     } else {
@@ -154,7 +155,7 @@ async function parseVariables(ctx, value) {
   return (0, import_utils.parse)(value)({
     $user,
     $nRole: ctx.state.currentRole === "__union__" ? ctx.state.currentRoles : ctx.state.currentRole,
-    $nLang: ctx.getCurrentLocale(),
+    $nLang: (_b = ctx.getCurrentLocale) == null ? void 0 : _b.call(ctx),
     $nDate
   });
 }

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

@@ -0,0 +1,16 @@
+/**
+ * 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 { AIEmployeeInstructionFiles } from './types';
+import { Plugin } from '@nocobase/server';
+export declare abstract class Files {
+    static resolvers(plugin: Plugin, attachmentPart: Record<string, any>): {
+        resolveAttachments: (files: AIEmployeeInstructionFiles[]) => Promise<void>;
+        resolveUrls: (files: AIEmployeeInstructionFiles[]) => Promise<void>;
+    };
+}