@nocobase/plugin-ai 2.0.0-alpha.4 → 2.0.0-alpha.40

package/dist/server/plugin.js

@@ -44,6 +44,7 @@ var import_server = require("@nocobase/server");
 var import_ai_manager = require("./manager/ai-manager");
 var import_ai_feature_manager = require("./manager/ai-feature-manager");
 var import_openai = require("./llm-providers/openai");
+var import_openai2 = require("./llm-providers/openai");
 var import_deepseek = require("./llm-providers/deepseek");
 var import_ai = __toESM(require("./resource/ai"));
 var import_llm = require("./workflow/nodes/llm");
@@ -58,11 +59,14 @@ var import_tools = require("./tools");
 var import_anthropic = require("./llm-providers/anthropic");
 var import_aiSettings = __toESM(require("./resource/aiSettings"));
 var import_dashscope = require("./llm-providers/dashscope");
+var import_ollama = require("./llm-providers/ollama");
 var import_built_in_manager = require("./manager/built-in-manager");
 var import_ai_context_datasource_manager = require("./manager/ai-context-datasource-manager");
 var import_aiContextDatasources = require("./resource/aiContextDatasources");
 var import_work_context_handler = require("./manager/work-context-handler");
 var import_ai_coding_manager = require("./manager/ai-coding-manager");
+var import_code_editor = require("./tools/code-editor");
+var import_datasource_query = require("./tools/datasource-query");
 class PluginAIServer extends import_server.Plugin {
   features = new import_ai_feature_manager.AIPluginFeatureManagerImpl();
   aiManager = new import_ai_manager.AIManager(this);
@@ -94,17 +98,21 @@ class PluginAIServer extends import_server.Plugin {
     await this.builtInManager.createOrUpdateAIEmployee();
   }
   registerLLMProviders() {
-    this.aiManager.registerLLMProvider("openai", import_openai.openaiProviderOptions);
+    this.aiManager.registerLLMProvider("openai", import_openai.openaiResponsesProviderOptions);
+    this.aiManager.registerLLMProvider("openai-completions", import_openai2.openaiCompletionsProviderOptions);
     this.aiManager.registerLLMProvider("deepseek", import_deepseek.deepseekProviderOptions);
     this.aiManager.registerLLMProvider("google-genai", import_google_genai.googleGenAIProviderOptions);
     this.aiManager.registerLLMProvider("anthropic", import_anthropic.anthropicProviderOptions);
     this.aiManager.registerLLMProvider("dashscope", import_dashscope.dashscopeProviderOptions);
+    this.aiManager.registerLLMProvider("ollama", import_ollama.ollamaProviderOptions);
   }
   registerTools() {
     const toolManager = this.aiManager.toolManager;
     const frontendGroupName = "frontend";
     const dataModelingGroupName = "dataModeling";
     const workflowGroupName = "workflowCaller";
+    const codeEditorGroupName = "codeEditor";
+    const dataSourceGroupName = "dataSource";
     toolManager.registerToolGroup({
       groupName: frontendGroupName,
       title: '{{t("Frontend")}}',
@@ -120,6 +128,16 @@ class PluginAIServer extends import_server.Plugin {
       title: '{{t("Workflow caller")}}',
       description: '{{t("Use workflow as a tool")}}'
     });
+    toolManager.registerToolGroup({
+      groupName: codeEditorGroupName,
+      title: '{{t("CodeEditor")}}',
+      description: '{{t("CodeEditor actions")}}'
+    });
+    toolManager.registerToolGroup({
+      groupName: dataSourceGroupName,
+      title: '{{t("DataSource")}}',
+      description: '{{t("Data source query")}}'
+    });
     this.aiManager.toolManager.registerTools([
       {
         groupName: frontendGroupName,
@@ -143,6 +161,22 @@ class PluginAIServer extends import_server.Plugin {
       },
       {
         tool: import_tools.chartGenerator
+      },
+      {
+        groupName: codeEditorGroupName,
+        tool: import_code_editor.listCodeSnippet
+      },
+      {
+        groupName: codeEditorGroupName,
+        tool: import_code_editor.getCodeSnippet
+      },
+      {
+        groupName: dataSourceGroupName,
+        tool: import_datasource_query.dataSourceCounting
+      },
+      {
+        groupName: dataSourceGroupName,
+        tool: import_datasource_query.dataSourceQuery
       }
     ]);
     toolManager.registerDynamicTool({

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

@@ -13,6 +13,7 @@ declare const _default: {
         list(ctx: Context, next: Next): Promise<void>;
         create(ctx: Context, next: Next): Promise<void>;
         update(ctx: Context, next: Next): Promise<void>;
+        updateOptions(ctx: Context, next: Next): Promise<never>;
         destroy(ctx: Context, next: Next): Promise<void>;
         getMessages(ctx: Context, next: Next): Promise<never>;
         updateToolArgs(ctx: Context, next: Next): Promise<any>;

package/dist/server/resource/aiConversations.js

@@ -88,7 +88,7 @@ var aiConversations_default = {
     async create(ctx, next) {
       var _a;
       const userId = (_a = ctx.auth) == null ? void 0 : _a.user.id;
-      const { aiEmployee, systemMessage, skillSettings } = ctx.action.params.values || {};
+      const { aiEmployee, systemMessage, skillSettings, conversationSettings } = ctx.action.params.values || {};
       const employee = await getAIEmployee(ctx, aiEmployee.username);
       if (!employee) {
         ctx.throw(400, "AI employee not found");
@@ -100,7 +100,8 @@ var aiConversations_default = {
           aiEmployee,
           options: {
             systemMessage,
-            skillSettings
+            skillSettings,
+            conversationSettings
           }
         }
       });
@@ -123,6 +124,50 @@ var aiConversations_default = {
       });
       await next();
     },
+    async updateOptions(ctx, next) {
+      var _a;
+      const userId = (_a = ctx.auth) == null ? void 0 : _a.user.id;
+      if (!userId) {
+        return ctx.throw(403);
+      }
+      const { filterByTk: sessionId } = ctx.action.params;
+      if (!sessionId) {
+        return ctx.throw(400, "invalid sessionId");
+      }
+      const { systemMessage, skillSettings, conversationSettings } = ctx.action.params.values || {};
+      if (!systemMessage && !skillSettings && !conversationSettings) {
+        return ctx.throw(400, "invalid options");
+      }
+      const conversation = await ctx.db.getRepository("aiConversations").findOne({
+        filter: {
+          sessionId,
+          userId
+        }
+      });
+      if (!conversation) {
+        ctx.throw(400, "invalid sessionId");
+      }
+      const options = conversation.options ?? {};
+      if (systemMessage) {
+        options["systemMessage"] = systemMessage;
+      }
+      if (skillSettings) {
+        options["skillSettings"] = skillSettings;
+      }
+      if (conversationSettings) {
+        options["conversationSettings"] = conversationSettings;
+      }
+      ctx.body = await ctx.db.getRepository("aiConversations").update({
+        filter: {
+          userId,
+          sessionId
+        },
+        values: {
+          options
+        }
+      });
+      await next();
+    },
     async destroy(ctx, next) {
       var _a;
       const userId = (_a = ctx.auth) == null ? void 0 : _a.user.id;
@@ -266,7 +311,7 @@ var aiConversations_default = {
       await next();
     },
     async sendMessages(ctx, next) {
-      var _a, _b;
+      var _a, _b, _c, _d;
       setupSSEHeaders(ctx);
       const { sessionId, aiEmployee: employeeName, messages, editingMessageId } = ctx.action.params.values || {};
       if (!sessionId) {
@@ -309,7 +354,8 @@ var aiConversations_default = {
           employee,
           sessionId,
           (_a = conversation.options) == null ? void 0 : _a.systemMessage,
-          (_b = conversation.options) == null ? void 0 : _b.skillSettings
+          (_b = conversation.options) == null ? void 0 : _b.skillSettings,
+          (_d = (_c = conversation.options) == null ? void 0 : _c.conversationSettings) == null ? void 0 : _d.webSearch
         );
         await aiEmployee.cancelToolCall();
         await aiEmployee.processMessages(messages, editingMessageId);
@@ -326,7 +372,7 @@ var aiConversations_default = {
       await next();
     },
     async resendMessages(ctx, next) {
-      var _a, _b;
+      var _a, _b, _c, _d;
       setupSSEHeaders(ctx);
       const { sessionId, messageId } = ctx.action.params.values || {};
       if (!sessionId) {
@@ -364,7 +410,8 @@ var aiConversations_default = {
           employee,
           sessionId,
           (_a = conversation.options) == null ? void 0 : _a.systemMessage,
-          (_b = conversation.options) == null ? void 0 : _b.skillSettings
+          (_b = conversation.options) == null ? void 0 : _b.skillSettings,
+          (_d = (_c = conversation.options) == null ? void 0 : _c.conversationSettings) == null ? void 0 : _d.webSearch
         );
         await aiEmployee.resendMessages(messageId);
       } catch (err) {
@@ -411,7 +458,7 @@ var aiConversations_default = {
       await next();
     },
     async callTool(ctx, next) {
-      var _a, _b, _c;
+      var _a, _b, _c, _d, _e;
       setupSSEHeaders(ctx);
       const { sessionId, messageId } = ctx.action.params.values || {};
       if (!sessionId) {
@@ -460,7 +507,8 @@ var aiConversations_default = {
           employee,
           sessionId,
           (_b = conversation.options) == null ? void 0 : _b.systemMessage,
-          (_c = conversation.options) == null ? void 0 : _c.skillSettings
+          (_c = conversation.options) == null ? void 0 : _c.skillSettings,
+          (_e = (_d = conversation.options) == null ? void 0 : _d.conversationSettings) == null ? void 0 : _e.webSearch
         );
         await aiEmployee.callTool(message.messageId, false);
       } catch (err) {
@@ -470,7 +518,7 @@ var aiConversations_default = {
       await next();
     },
     async confirmToolCall(ctx, next) {
-      var _a, _b, _c;
+      var _a, _b, _c, _d, _e;
       setupSSEHeaders(ctx);
       const { sessionId, messageId, toolCallIds } = ctx.action.params.values || {};
       if (!sessionId) {
@@ -514,7 +562,8 @@ var aiConversations_default = {
           employee,
           sessionId,
           (_b = conversation.options) == null ? void 0 : _b.systemMessage,
-          (_c = conversation.options) == null ? void 0 : _c.skillSettings
+          (_c = conversation.options) == null ? void 0 : _c.skillSettings,
+          (_e = (_d = conversation.options) == null ? void 0 : _d.conversationSettings) == null ? void 0 : _e.webSearch
         );
         await aiEmployee.confirmToolCall(message.messageId, toolCallIds);
       } catch (err) {

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

@@ -8,6 +8,7 @@
  */
 import { Context, Next } from '@nocobase/actions';
 export declare const list: (ctx: Context, next: Next) => Promise<void>;
+export declare const create: (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>;
 export declare const getTemplates: (ctx: Context, next: Next) => Promise<void>;

package/dist/server/resource/aiEmployees.js

@@ -36,6 +36,7 @@ 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,
@@ -44,6 +45,7 @@ __export(aiEmployees_exports, {
 module.exports = __toCommonJS(aiEmployees_exports);
 var import_actions = __toESM(require("@nocobase/actions"));
 var templates = __toESM(require("../ai-employees/templates"));
+var import_lodash = __toESM(require("lodash"));
 const list = async (ctx, next) => {
   const { paginate } = ctx.action.params || {};
   const plugin = ctx.app.pm.get("ai");
@@ -62,6 +64,22 @@ const list = async (ctx, next) => {
   });
   await next();
 };
+const create = async (ctx, next) => {
+  const { skillSettings } = ctx.action.params.values ?? {};
+  const skills = skillSettings.skills ?? [];
+  skills.push(
+    {
+      name: "dataSource-dataSourceCounting",
+      autoCall: true
+    },
+    {
+      name: "dataSource-dataSourceQuery",
+      autoCall: true
+    }
+  );
+  skillSettings.skills = import_lodash.default.uniqBy(skills, "name");
+  await import_actions.default.create(ctx, next);
+};
 const listByUser = async (ctx, next) => {
   const plugin = ctx.app.pm.get("ai");
   const user = ctx.auth.user;
@@ -113,8 +131,27 @@ const listByUser = async (ctx, next) => {
       builtInManager.setupBuiltInInfo(locale, row);
     }
   });
+  const llmServiceNameSet = new Set(
+    rows.filter((row) => {
+      var _a;
+      return (_a = row.modelSettings) == null ? void 0 : _a.llmService;
+    }).map((row) => {
+      var _a;
+      return (_a = row.modelSettings) == null ? void 0 : _a.llmService;
+    })
+  );
+  const llmProviders = llmServiceNameSet.size ? await ctx.db.getRepository("llmServices").find({
+    filter: {
+      name: {
+        $in: Array.from(llmServiceNameSet)
+      }
+    }
+  }) : [];
+  const llmProviderMap = new Map(llmProviders.map((provider) => [provider.name, provider]));
   ctx.body = rows.map((row) => {
-    var _a, _b;
+    var _a, _b, _c, _d, _e;
+    const llmServiceName = row.modelSettings.llmService;
+    const llmProvider = ((_a = llmProviderMap.get(llmServiceName)) == null ? void 0 : _a.provider) ?? "";
     return {
       username: row.username,
       nickname: row.nickname,
@@ -123,10 +160,13 @@ const listByUser = async (ctx, next) => {
       bio: row.bio,
       greeting: row.greeting,
       userConfig: {
-        prompt: (_b = (_a = row.userConfigs) == null ? void 0 : _a[0]) == null ? void 0 : _b.prompt
+        prompt: (_c = (_b = row.userConfigs) == null ? void 0 : _b[0]) == null ? void 0 : _c.prompt
       },
       skillSettings: row.skillSettings,
-      builtIn: row.builtIn
+      builtIn: row.builtIn,
+      webSearch: ((_e = (_d = row.modelSettings) == null ? void 0 : _d.builtIn) == null ? void 0 : _e.webSearch) ?? false,
+      llmProvider,
+      toolsConflict: llmProvider === "google-genai"
     };
   });
   await next();
@@ -167,6 +207,7 @@ const getTemplates = async (ctx, next) => {
 };
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  create,
   getTemplates,
   list,
   listByUser,

package/dist/server/tools/code-editor.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 { ToolOptions } from '../manager/tool-manager';
+export declare const listCodeSnippet: ToolOptions;
+export declare const getCodeSnippet: ToolOptions;

package/dist/server/tools/code-editor.js

@@ -0,0 +1,84 @@
+/**
+ * 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 code_editor_exports = {};
+__export(code_editor_exports, {
+  getCodeSnippet: () => getCodeSnippet,
+  listCodeSnippet: () => listCodeSnippet
+});
+module.exports = __toCommonJS(code_editor_exports);
+var import_zod = require("zod");
+const listCodeSnippet = {
+  name: "listCodeSnippet",
+  title: '{{t("Get code snippet list")}}',
+  description: '{{t("Get code snippet list")}}',
+  execution: "frontend",
+  schema: import_zod.z.object({}),
+  invoke: async (ctx, _args, id) => {
+    const { toolCallResults } = ctx.action.params.values || {};
+    const { result } = (toolCallResults == null ? void 0 : toolCallResults.find((item) => item.id === id)) ?? {};
+    if (toolCallResults && result) {
+      return {
+        status: "success",
+        content: JSON.stringify(result)
+      };
+    } else {
+      return {
+        status: "success",
+        content: "[]"
+      };
+    }
+  }
+};
+const getCodeSnippet = {
+  name: "getCodeSnippet",
+  title: '{{t("Get code snippet content")}}',
+  description: '{{t("Get code snippet content")}}',
+  execution: "frontend",
+  schema: import_zod.z.object({
+    ref: import_zod.z.string().describe('ref from tools "listCodeSnippet" for get code snippet content.')
+  }),
+  invoke: async (ctx, _args, id) => {
+    const { toolCallResults } = ctx.action.params.values || {};
+    const { result } = (toolCallResults == null ? void 0 : toolCallResults.find((item) => item.id === id)) ?? {};
+    if (toolCallResults && result) {
+      return {
+        status: "success",
+        content: JSON.stringify(result)
+      };
+    } else {
+      return {
+        status: "success",
+        content: "[]"
+      };
+    }
+  }
+};
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  getCodeSnippet,
+  listCodeSnippet
+});

package/dist/server/tools/data-modeling.js

@@ -43,6 +43,7 @@ __export(data_modeling_exports, {
 });
 module.exports = __toCommonJS(data_modeling_exports);
 var import_lodash = __toESM(require("lodash"));
+var import_zod = require("zod");
 const idField = {
   name: "id",
   type: "bigInt",
@@ -241,21 +242,11 @@ const defineCollections = {
   name: "defineCollections",
   title: '{{t("Define collections")}}',
   description: '{{t("Create or edit collections")}}',
-  schema: {
-    type: "object",
-    properties: {
-      collections: {
-        type: "array",
-        description: "An array of collections to be defined or edited.",
-        items: {
-          type: "object",
-          additionalProperties: true
-        }
-      }
-    },
-    required: ["collections"],
-    additionalProperties: false
-  },
+  schema: import_zod.z.object({
+    collections: import_zod.z.array(
+      import_zod.z.object({}).catchall(import_zod.z.any()).describe("Valid collection object which defined in collection_type_definition")
+    ).describe("An array of collections to be defined or edited.")
+  }),
   invoke: async (ctx) => {
     const {
       args: { collections }

package/dist/server/tools/datasource-query.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 { ToolOptions } from '../manager/tool-manager';
+export declare const dataSourceQuery: ToolOptions;
+export declare const dataSourceCounting: ToolOptions;

package/dist/server/tools/datasource-query.js

@@ -0,0 +1,133 @@
+/**
+ * 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 datasource_query_exports = {};
+__export(datasource_query_exports, {
+  dataSourceCounting: () => dataSourceCounting,
+  dataSourceQuery: () => dataSourceQuery
+});
+module.exports = __toCommonJS(datasource_query_exports);
+var import_zod = require("zod");
+const ArgSchema = import_zod.z.object({
+  datasource: import_zod.z.string().describe('{{t("Data source key")}}'),
+  collectionName: import_zod.z.string().describe('{{t("Collection name")}}'),
+  fields: import_zod.z.array(import_zod.z.string()).describe('{{t("Fields to be queried")}}'),
+  appends: import_zod.z.array(import_zod.z.string()).describe('{{t("Related collection to be queried")}}'),
+  filter: import_zod.z.object({}).catchall(import_zod.z.any()).describe(`# Parameters definition
+\`\`\`
+export type QueryCondition = {
+  [field: string]: { // \`field\` key is Field name
+    [operator: string]: any; // \`operator\` key is Query condition operator.
+  };
+};
+
+
+export type QueryObject =
+  | {
+      [logic: string]: (QueryCondition | QueryObject)[]; // \`logic\` key is the relationship between query conditions, should be one of '$and', '$or', the value is recursion object array, item in array can be QueryCondition or QueryObject
+    }
+  | QueryCondition // QueryCondition definition above;
+\`\`\`
+
+
+// QueryCondition examples
+
+\`\`\`
+const example1: QueryCondition = {
+  age: { $gt: 18 },          // age great than 18
+  name: { $like: '%John%' }, // name include John
+};
+
+const example2: QueryCondition = {
+  status: { $in: ['active', 'pending'] }, // status is active or pending
+};
+\`\`\`
+
+// QueryObject example
+\`\`\`
+const example1: QueryObject = {
+  $and: [
+    { age: { $gt: 18 } },
+    { status: { $eq: 'active' } }
+  ]
+};
+
+const example2: QueryObject = {
+  $or: [
+    { name: { $like: '%John%' } },
+    {
+      $and: [
+        { age: { $gte: 30 } },
+        { status: { $eq: 'pending' } }
+      ]
+    }
+  ]
+};
+
+const example3: QueryObject = { age: { $lt: 50 } };
+\`\`\`
+`),
+  sort: import_zod.z.array(import_zod.z.string()).describe(
+    '{{t("Sort field names. By default, they are in ascending order. A minus sign before the field name indicates descending order")}}'
+  ),
+  offset: import_zod.z.number().describe('{{t("Offset of records to be queried")}}'),
+  limit: import_zod.z.number().describe('{{t("Maximum number of records to be queried")}}')
+});
+const dataSourceQuery = {
+  name: "dataSourceQuery",
+  title: '{{t("Data source query")}}',
+  description: '{{t("Use dataSource, collectionName, and collection fields to query data from the database")}}',
+  schema: ArgSchema,
+  invoke: async (ctx, args) => {
+    const plugin = ctx.app.pm.get("ai");
+    const content = await plugin.aiContextDatasourceManager.query(ctx, args);
+    return {
+      status: "success",
+      content: JSON.stringify(content)
+    };
+  }
+};
+const dataSourceCounting = {
+  name: "dataSourceCounting",
+  title: '{{t("Data source records counting")}}',
+  description: '{{t("Use dataSource, collectionName, and collection fields to query data from the database, get total count of records")}}',
+  schema: ArgSchema,
+  invoke: async (ctx, args) => {
+    const plugin = ctx.app.pm.get("ai");
+    args.offset = 0;
+    args.limit = 1;
+    const content = await plugin.aiContextDatasourceManager.query(ctx, args);
+    return {
+      status: "success",
+      content: String((content == null ? void 0 : content.total) ?? 0)
+    };
+  }
+};
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  dataSourceCounting,
+  dataSourceQuery
+});

package/dist/server/utils.js

@@ -51,13 +51,13 @@ function stripToolCallTags(content) {
   if (typeof content !== "string") {
     return content;
   }
-  return content.replace(/<[|｜]tool▁(?:calls▁begin|calls▁end|call▁begin|call▁end|sep)[|｜]>/g, "").replace(/function/, "");
+  return content.replace(/<[|｜]tool▁(?:calls▁begin|calls▁end|call▁begin|call▁end|sep)[|｜]>/g, "");
 }
 function parseResponseMessage(row) {
   const { content: rawContent, messageId, metadata, role, toolCalls, attachments, workContext } = row;
   const content = {
-    ...rawContent,
-    content: stripToolCallTags(rawContent.content),
+    ...rawContent ?? {},
+    content: stripToolCallTags(rawContent == null ? void 0 : rawContent.content),
     messageId,
     metadata,
     attachments,

package/package.json

@@ -4,7 +4,7 @@
   "displayName.zh-CN": "AI 员工",
   "description": "Create AI employees with diverse skills to collaborate with humans, build systems, and handle business operations.",
   "description.zh-CN": "创建各种技能的 AI 员工，与人类协同，搭建系统，处理业务。",
-  "version": "2.0.0-alpha.4",
+  "version": "2.0.0-alpha.40",
   "main": "dist/server/index.js",
   "peerDependencies": {
     "@nocobase/client": "2.x",
@@ -22,6 +22,7 @@
     "@langchain/core": "^0.3.61",
     "@langchain/deepseek": "^0.0.2",
     "@langchain/google-genai": "^0.2.16",
+    "@langchain/ollama": "^0.1.1",
     "@langchain/openai": "^0.6.13",
     "@nocobase/ai-employee-avatars": "^1.0.2",
     "@xyflow/react": "^12.7.0",
@@ -41,5 +42,5 @@
   "keywords": [
     "AI"
   ],
-  "gitHead": "54f3cab47e7efbdc73377014d05f5fc66a4affbb"
+  "gitHead": "dfb9fd33f43e91e676ae7b1f1a595bc38b7c20d2"
 }