@nocobase/ai 2.1.0-beta.24 → 2.1.0-beta.25

package/lib/ai-employee-manager/index.js

@@ -94,6 +94,7 @@ const _DefaultAIEmployeeManager = class _DefaultAIEmployeeManager {
         await this.aiEmployeesModel.create(
           {
             username: employee.username,
+            category: employee.category ?? "business",
             nickname: employee.nickname,
             position: employee.position,
             avatar: employee.avatar,
@@ -122,8 +123,9 @@ const _DefaultAIEmployeeManager = class _DefaultAIEmployeeManager {
         var _a;
         return (_a = s.name) == null ? void 0 : _a.startsWith("workflowCaller-");
       }) : [];
-      const mergedTools = /* @__PURE__ */ new Set([...tools, ...employee.tools]);
+      const mergedTools = /* @__PURE__ */ new Set([...tools, ...employee.tools ?? []]);
       const values = {
+        category: employee.category ?? current.category,
         nickname: employee.nickname ?? current.nickname,
         position: employee.position ?? current.position,
         avatar: employee.avatar ?? current.avatar,
@@ -131,7 +133,7 @@ const _DefaultAIEmployeeManager = class _DefaultAIEmployeeManager {
         greeting: employee.greeting ?? current.greeting,
         defaultPrompt: employee.systemPrompt,
         skillSettings: {
-          skills: [...employee.skills],
+          skills: [...employee.skills ?? []],
           tools: [...mergedTools]
         },
         sort: employee.sort

package/lib/ai-employee-manager/types.d.ts

@@ -25,6 +25,7 @@ export type AIEmployeeToolSetting = {
 };
 export type AIEmployeeOptions = {
     username: string;
+    category?: string;
     description?: string;
     skills?: string[];
     tools?: AIEmployeeToolSetting[];

package/lib/loader/employee.js

@@ -121,12 +121,24 @@ const _AIEmployeeLoader = class _AIEmployeeLoader extends import_types.LoadAndRe
     this.employeeDescriptors = descriptors.filter((item) => Boolean(item));
   }
   async register() {
+    var _a, _b;
     if (!this.employeeDescriptors.length) {
       return;
     }
     const { employeeManager } = this.ai;
     for (const descriptor of this.employeeDescriptors) {
-      await employeeManager.registerEmployee(descriptor.options);
+      try {
+        await employeeManager.registerEmployee(descriptor.options);
+        (_a = this.log) == null ? void 0 : _a.info(
+          `ai employee [${descriptor.options.username}] registered from plugin [${this.options.pluginName}]`
+        );
+      } catch (e) {
+        (_b = this.log) == null ? void 0 : _b.error(
+          `ai employee [${descriptor.name}] register ignored: error occur when invoke registerEmployee`,
+          e
+        );
+        continue;
+      }
     }
   }
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@nocobase/ai",
-  "version": "2.1.0-beta.24",
+  "version": "2.1.0-beta.25",
   "description": "",
   "license": "Apache-2.0",
   "main": "./lib/index.js",
@@ -17,10 +17,10 @@
     "@langchain/mcp-adapters": "^1.1.3",
     "@langchain/ollama": "^1.2.2",
     "@langchain/openai": "^1.2.7",
-    "@nocobase/data-source-manager": "2.1.0-beta.24",
-    "@nocobase/logger": "2.1.0-beta.24",
-    "@nocobase/resourcer": "2.1.0-beta.24",
-    "@nocobase/utils": "2.1.0-beta.24",
+    "@nocobase/data-source-manager": "2.1.0-beta.25",
+    "@nocobase/logger": "2.1.0-beta.25",
+    "@nocobase/resourcer": "2.1.0-beta.25",
+    "@nocobase/utils": "2.1.0-beta.25",
     "d3-dsv": "2",
     "fast-glob": "^3.3.2",
     "flexsearch": "^0.8.2",
@@ -37,5 +37,5 @@
     "url": "git+https://github.com/nocobase/nocobase.git",
     "directory": "packages/ai"
   },
-  "gitHead": "f77b85530a2d127d9bfe4dca3a26fbb02c1139ba"
+  "gitHead": "824f8b8200e9fe086135768934d3ef427b212446"
 }

package/src/ai-employee-manager/index.ts

@@ -80,6 +80,7 @@ export class DefaultAIEmployeeManager implements AIEmployeeManager {
         await this.aiEmployeesModel.create(
           {
             username: employee.username,
+            category: employee.category ?? 'business',
             nickname: employee.nickname,
             position: employee.position,
             avatar: employee.avatar,
@@ -106,8 +107,9 @@ export class DefaultAIEmployeeManager implements AIEmployeeManager {
       const current = existed.toJSON() as AIEmployeeEntry;
       let { tools } = current?.skillSettings ?? { tools: [] };
       tools = tools?.length ? tools.filter((s) => s.name?.startsWith('workflowCaller-')) : [];
-      const mergedTools = new Set([...tools, ...employee.tools]);
+      const mergedTools = new Set([...tools, ...(employee.tools ?? [])]);
       const values: Record<string, unknown> = {
+        category: employee.category ?? current.category,
         nickname: employee.nickname ?? current.nickname,
         position: employee.position ?? current.position,
         avatar: employee.avatar ?? current.avatar,
@@ -115,7 +117,7 @@ export class DefaultAIEmployeeManager implements AIEmployeeManager {
         greeting: employee.greeting ?? current.greeting,
         defaultPrompt: employee.systemPrompt,
         skillSettings: {
-          skills: [...employee.skills],
+          skills: [...(employee.skills ?? [])],
           tools: [...mergedTools],
         },
         sort: employee.sort,

package/src/ai-employee-manager/types.ts

@@ -29,6 +29,7 @@ export type AIEmployeeToolSetting = {
 
 export type AIEmployeeOptions = {
   username: string;
+  category?: string;
   description?: string;
   skills?: string[];
   tools?: AIEmployeeToolSetting[];

package/src/loader/employee.ts

@@ -109,7 +109,18 @@ export class AIEmployeeLoader extends LoadAndRegister<AIEmployeeLoaderOptions> {
     }
     const { employeeManager } = this.ai;
     for (const descriptor of this.employeeDescriptors) {
-      await employeeManager.registerEmployee(descriptor.options);
+      try {
+        await employeeManager.registerEmployee(descriptor.options);
+        this.log?.info(
+          `ai employee [${descriptor.options.username}] registered from plugin [${this.options.pluginName}]`,
+        );
+      } catch (e) {
+        this.log?.error(
+          `ai employee [${descriptor.name}] register ignored: error occur when invoke registerEmployee`,
+          e,
+        );
+        continue;
+      }
     }
   }
 }