@nocobase/plugin-ai 2.0.0-alpha.59 → 2.0.0-alpha.60

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

@@ -47,17 +47,22 @@ var import_lodash = __toESM(require("lodash"));
 var import_zod = require("zod");
 const idField = {
   name: "id",
-  type: "bigInt",
-  autoIncrement: true,
+  type: "snowflakeId",
+  autoIncrement: false,
   primaryKey: true,
   allowNull: false,
   uiSchema: {
     type: "number",
     title: '{{t("ID")}}',
     "x-component": "InputNumber",
-    "x-read-pretty": true
+    "x-component-props": {
+      stringMode: true,
+      separator: "0.00",
+      step: "1"
+    },
+    "x-validator": "integer"
   },
-  interface: "id"
+  interface: "snowflakeId"
 };
 const createdAtField = {
   name: "createdAt",
@@ -95,20 +100,20 @@ const createPrompt = `You are now entering the **New Schema Creation Flow**. Fol
 
 3. **Output and Confirmation**
    - Present the full schema in **formatted natural language** (not plain JSON).
-   - Wait for user confirmation, then call the \`defineCollections\` tool with the **complete schema definition**.
+   - Wait for user confirmation, then call the \`dataModeling-defineCollections\` tool with the **complete schema definition**.
    - Until the tool responds, **assume the schema is not saved** \u2014 user may continue editing.
    - **Do not say or imply the schema has been created without tool response.**
 
-Only the \`defineCollections\` tool may be used.`;
+Only the \`dataModeling-defineCollections\` tool may be used.`;
 const editPrompt = `## Existing Schema Editing Flow
 
 1. **Clarify What Needs to Be Changed**
    - Identify which tables are affected by the requested changes.
-   - If needed, call \`getCollectionNames\` to retrieve the list of all tables (ID and title).
+   - If needed, call \`dataModeling-getCollectionNames\` to retrieve the list of all tables (ID and title).
 
 2. **Fetch Table Metadata**
    - Analyze the current structure and identify what needs to be added, removed, or updated.
-   - If needed, use the \`getCollectionMetadata\` tool to retrieve schema details of the target table(s).
+   - If needed, use the \`dataModeling-getCollectionMetadata\` tool to retrieve schema details of the target table(s).
 
 3. **Propose Changes**
    - Output your change suggestions in clear **natural language**.
@@ -116,9 +121,14 @@ const editPrompt = `## Existing Schema Editing Flow
    - Wait for user confirmation before applying any changes.
 
 4. **Apply Changes**
-   - Once confirmed, call the \`defineCollections\` tool with **only the modified parts** of the schema.
+   - Once confirmed, call the \`dataModeling-defineCollections\` tool with **only the modified parts** of the schema.
    - Until the tool responds successfully, assume changes have not been saved \u2014 the user may continue editing.
    - **Do not say or imply the schema is being or has been updated until a tool response is received.**`;
+class IntentError extends Error {
+  constructor(message) {
+    super(message);
+  }
+}
 const dataModelingIntentRouter = {
   name: "intentRouter",
   title: '{{t("Intent Router")}}',
@@ -305,14 +315,26 @@ const defineCollections = {
   title: '{{t("Define collections")}}',
   description: '{{t("Create or edit collections")}}',
   schema: import_zod.z.object({
+    intent: import_zod.z.enum(["create", "edit"]).describe(
+      `Pass the intent of the current tool invocation as an enum value. The value must be either 'create' or 'edit':
+- create: create a brand-new data table definition
+- edit: modify an existing data table definition`
+    ),
     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 }
-    } = ctx.action.params.values || {};
+  invoke: async (ctx, args) => {
+    var _a, _b, _c;
+    const { intent, collections: originalCollections } = ((_c = (_b = (_a = ctx.action) == null ? void 0 : _a.params) == null ? void 0 : _b.values) == null ? void 0 : _c.args) ?? args ?? {};
+    if (!intent || !["create", "edit"].includes(intent)) {
+      return {
+        status: "error",
+        content: `Please explicitly specify your intent. The value of the intent parameter must be either 'create' or 'edit'.`
+      };
+    }
+    const collectionsType = typeof originalCollections;
+    const collections = collectionsType === "string" ? JSON.parse(originalCollections) : originalCollections;
     if (!collections || !Array.isArray(collections)) {
       return {
         status: "error",
@@ -352,12 +374,22 @@ const defineCollections = {
           const baseFields = [...systemFields, ...normalFields];
           const collection = await repo.findOne({ filter: { name: options.name }, transaction });
           if (!collection) {
+            if (intent === "edit") {
+              throw new IntentError(
+                `You want to edit a collection definition, but there is no existing data table definition named '${options.name}'.`
+              );
+            }
             await repo.create({
               values: { ...options, fields: baseFields },
               transaction,
               context: ctx
             });
           } else {
+            if (intent === "create") {
+              throw new IntentError(
+                `You want to create a collection definition, but a collection definition named '${options.name}' already exists. Please change the name of your collection definition and then invoke this tool again.`
+              );
+            }
             await repo.update({
               filterByTk: options.name,
               values: import_lodash.default.omit(options, ["fields"]),
@@ -429,6 +461,17 @@ const defineCollections = {
         stack: e.stack,
         cause: e.cause
       });
+      if (e instanceof IntentError) {
+        return {
+          status: "error",
+          content: e.message
+        };
+      }
+      if (intent === "create") {
+        for (const options of sorted) {
+          ctx.db.removeCollection(options.name);
+        }
+      }
       return {
         status: "error",
         content: `Failed to define collections: ${e.message}`

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

@@ -31,6 +31,7 @@ __export(datasource_query_exports, {
 });
 module.exports = __toCommonJS(datasource_query_exports);
 var import_zod = require("zod");
+var import_utils = require("./utils");
 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")}}'),
@@ -93,8 +94,8 @@ 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")}}')
+  offset: import_zod.z.number().optional().describe('{{t("Offset of records to be queried")}}'),
+  limit: import_zod.z.number().optional().describe('{{t("Maximum number of records to be queried")}}')
 });
 const dataSourceQuery = {
   name: "dataSourceQuery",
@@ -103,10 +104,21 @@ const dataSourceQuery = {
   schema: ArgSchema,
   invoke: async (ctx, args) => {
     const plugin = ctx.app.pm.get("ai");
-    const content = await plugin.aiContextDatasourceManager.query(ctx, args);
+    const { limit, offset } = (0, import_utils.normalizeLimitOffset)(args, { defaultLimit: 50, maxLimit: import_utils.MAX_QUERY_LIMIT });
+    const content = await plugin.aiContextDatasourceManager.query(ctx, {
+      ...args,
+      limit,
+      offset
+    });
+    const records = (0, import_utils.truncateLongStrings)((content == null ? void 0 : content.records) || []);
+    const total = (content == null ? void 0 : content.total) || 0;
+    const result = (0, import_utils.buildPagedToolResult)({ total, offset, limit, records });
+    if (result.hasMore) {
+      result.note = `Showing ${result.returned} of ${result.total} records. To fetch more, call this tool again with offset:${result.nextOffset}`;
+    }
     return {
       status: "success",
-      content: JSON.stringify(content)
+      content: JSON.stringify(result)
     };
   }
 };
@@ -117,9 +129,7 @@ const dataSourceCounting = {
   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);
+    const content = await plugin.aiContextDatasourceManager.query(ctx, { ...args, offset: 0, limit: 1 });
     return {
       status: "success",
       content: String((content == null ? void 0 : content.total) ?? 0)

package/dist/server/tools/suggestions.d.ts

@@ -0,0 +1,10 @@
+/**
+ * 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 suggestions: ToolOptions;

package/dist/server/tools/suggestions.js

@@ -0,0 +1,74 @@
+/**
+ * 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 suggestions_exports = {};
+__export(suggestions_exports, {
+  suggestions: () => suggestions
+});
+module.exports = __toCommonJS(suggestions_exports);
+var import_zod = require("zod");
+const suggestions = {
+  name: "suggestions",
+  title: '{{t("Suggestions")}}',
+  description: '{{t("Provide a list of suggested prompts for the user to choose from.")}}',
+  schema: import_zod.z.object({
+    options: import_zod.z.array(import_zod.z.string()).describe(
+      "A list of suggested prompts that can be presented to the user as selectable options.Each option represents a possible next user message."
+    )
+  }),
+  invoke: async (ctx, _args, id) => {
+    var _a, _b;
+    const { messageId, args } = ((_b = (_a = ctx.action) == null ? void 0 : _a.params) == null ? void 0 : _b.values) || {};
+    if (messageId) {
+      const messageRepo = ctx.app.db.getRepository("aiMessages");
+      const message = await messageRepo.findOne({
+        filterByTk: messageId
+      });
+      const toolCalls = message.toolCalls || [];
+      const index = toolCalls.findIndex((toolCall) => toolCall.id === id);
+      if (index !== -1) {
+        toolCalls[index] = {
+          ...toolCalls[index],
+          selectedSuggestion: args == null ? void 0 : args.option
+        };
+        await messageRepo.update({
+          filter: { messageId },
+          values: {
+            toolCalls
+          }
+        });
+      }
+    }
+    return {
+      status: "success",
+      content: args == null ? void 0 : args.option
+    };
+  }
+};
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  suggestions
+});

package/dist/server/tools/utils.d.ts

@@ -0,0 +1,40 @@
+/**
+ * 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.
+ */
+/** 单个字符串字段最大长度 */
+export declare const MAX_STRING_LENGTH = 500;
+/** 单次查询最大记录数 */
+export declare const MAX_QUERY_LIMIT = 100;
+export declare function normalizeLimitOffset(args: {
+    limit?: unknown;
+    offset?: unknown;
+}, options?: {
+    defaultLimit?: number;
+    maxLimit?: number;
+}): {
+    limit: number;
+    offset: number;
+};
+export declare function buildPagedToolResult<T>(params: {
+    total: number;
+    offset: number;
+    limit: number;
+    records: T[];
+}): {
+    total: number;
+    offset: number;
+    limit: number;
+    returned: number;
+    hasMore: boolean;
+    nextOffset: number;
+    records: T[];
+};
+/**
+ * 递归截断对象中的长字符串，保持 JSON 结构完整
+ */
+export declare function truncateLongStrings(obj: any, maxLen?: number): any;

package/dist/server/tools/utils.js

@@ -0,0 +1,87 @@
+/**
+ * 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 utils_exports = {};
+__export(utils_exports, {
+  MAX_QUERY_LIMIT: () => MAX_QUERY_LIMIT,
+  MAX_STRING_LENGTH: () => MAX_STRING_LENGTH,
+  buildPagedToolResult: () => buildPagedToolResult,
+  normalizeLimitOffset: () => normalizeLimitOffset,
+  truncateLongStrings: () => truncateLongStrings
+});
+module.exports = __toCommonJS(utils_exports);
+const MAX_STRING_LENGTH = 500;
+const MAX_QUERY_LIMIT = 100;
+function normalizeLimitOffset(args, options) {
+  const defaultLimit = (options == null ? void 0 : options.defaultLimit) ?? 50;
+  const maxLimit = (options == null ? void 0 : options.maxLimit) ?? MAX_QUERY_LIMIT;
+  const rawLimit = typeof args.limit === "number" && Number.isFinite(args.limit) ? args.limit : void 0;
+  const rawOffset = typeof args.offset === "number" && Number.isFinite(args.offset) ? args.offset : void 0;
+  const limit = Math.min(Math.max(rawLimit ?? defaultLimit, 1), maxLimit);
+  const offset = Math.max(rawOffset ?? 0, 0);
+  return { limit, offset };
+}
+function buildPagedToolResult(params) {
+  const returned = params.records.length;
+  const nextOffset = params.offset + returned;
+  const hasMore = params.total > nextOffset;
+  return {
+    total: params.total,
+    offset: params.offset,
+    limit: params.limit,
+    returned,
+    hasMore,
+    nextOffset,
+    records: params.records
+  };
+}
+function truncateLongStrings(obj, maxLen = MAX_STRING_LENGTH) {
+  if (obj === null || obj === void 0) {
+    return obj;
+  }
+  if (typeof obj === "string") {
+    return obj.length > maxLen ? obj.slice(0, maxLen) + "...[truncated]" : obj;
+  }
+  if (Array.isArray(obj)) {
+    return obj.map((item) => truncateLongStrings(item, maxLen));
+  }
+  if (typeof obj === "object") {
+    const result = {};
+    for (const key in obj) {
+      result[key] = truncateLongStrings(obj[key], maxLen);
+    }
+    return result;
+  }
+  return obj;
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  MAX_QUERY_LIMIT,
+  MAX_STRING_LENGTH,
+  buildPagedToolResult,
+  normalizeLimitOffset,
+  truncateLongStrings
+});

package/dist/server/tools/workflow-caller.js

@@ -31,6 +31,7 @@ __export(workflow_caller_exports, {
 module.exports = __toCommonJS(workflow_caller_exports);
 var import_zod = require("zod");
 var import_plugin_workflow = require("@nocobase/plugin-workflow");
+var import_utils = require("./utils");
 const buildSchema = (config) => {
   var _a;
   const schemaProperties = {};
@@ -86,7 +87,7 @@ const invoke = async (ctx, workflow, args) => {
   if (processor.execution.status !== import_plugin_workflow.EXECUTION_STATUS.RESOLVED) {
     return { status: "error", content: "Workflow execution exceptions" };
   }
-  const lastJobResult = processor.lastSavedJob.result;
+  const lastJobResult = (0, import_utils.truncateLongStrings)(processor.lastSavedJob.result);
   return {
     status: "success",
     content: JSON.stringify(lastJobResult)

package/package.json

@@ -6,7 +6,7 @@
   "description": "Create AI employees with diverse skills to collaborate with humans, build systems, and handle business operations.",
   "description.ru-RU": "Поддержка интеграции с AI-сервисами: предоставляются AI-узлы для рабочих процессов, расширяя возможности бизнес-обработки.",
   "description.zh-CN": "创建各种技能的 AI 员工，与人类协同，搭建系统，处理业务。",
-  "version": "2.0.0-alpha.59",
+  "version": "2.0.0-alpha.60",
   "main": "dist/server/index.js",
   "homepage": "https://docs.nocobase.com/handbook/action-ai",
   "homepage.ru-RU": "https://docs-ru.nocobase.com/handbook/action-ai",
@@ -47,5 +47,5 @@
   "keywords": [
     "AI"
   ],
-  "gitHead": "ca9fca2a826b17735c6287bebf877b2da9a00e24"
+  "gitHead": "9113d61ce85b60b7ba3d0e5ca64182d92a15ece4"
 }