@nocobase/plugin-ai 2.0.0-alpha.9 → 2.0.0-beta.1

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

@@ -15,15 +15,19 @@ export declare class AIContextDatasourceManager {
     protected plugin: PluginAIServer;
     constructor(plugin: PluginAIServer);
     preview(ctx: Context, options: PreviewOptions): Promise<QueryResult | null>;
+    query(ctx: Context, options: InnerQueryOptions): Promise<QueryResult | null>;
     provideWorkContextResolveStrategy(): WorkContextResolveStrategy;
     private innerQuery;
 }
-export type PreviewOptions = Pick<AIContextDatasource, 'datasource' | 'collectionName' | 'fields' | 'appends' | 'filter' | 'sort' | 'limit'>;
+export type PreviewOptions = Pick<AIContextDatasource, 'datasource' | 'collectionName' | 'fields' | 'appends' | 'filter' | 'sort' | 'limit'> & {
+    offset?: number;
+};
 export type QueryOptions = {
     id: string;
 };
 export type QueryResult = {
     options: InnerQueryOptions;
+    total: number;
     records: {
         name: string;
         type: string;

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

@@ -37,6 +37,9 @@ class AIContextDatasourceManager {
   async preview(ctx, options) {
     return await this.innerQuery(ctx, { ...options, filter: options.filter ? (0, import_utils.transformFilter)(options.filter) : null });
   }
+  async query(ctx, options) {
+    return await this.innerQuery(ctx, { ...options, filter: options.filter });
+  }
   provideWorkContextResolveStrategy() {
     return async (ctx, contextItem) => {
       if (!contextItem.content) {
@@ -57,6 +60,7 @@ class AIContextDatasourceManager {
       this.plugin.log.warn(`Datasource ${datasource} not found`);
       return {
         options,
+        total: 0,
         records: []
       };
     }
@@ -65,6 +69,7 @@ class AIContextDatasourceManager {
       this.plugin.log.warn(`Collection ${collectionName} not found`);
       return {
         options,
+        total: 0,
         records: []
       };
     }
@@ -78,6 +83,7 @@ class AIContextDatasourceManager {
       if (!can || typeof can !== "object") {
         return {
           options,
+          total: 0,
           records: []
         };
       }
@@ -113,8 +119,9 @@ class AIContextDatasourceManager {
         }
       });
     }
-    const { fields, filter, sort, limit } = options;
-    const result = await collection.repository.find({ fields, filter, sort, limit });
+    const { fields, filter, sort, offset, limit } = options;
+    const result = await collection.repository.find({ fields, filter, sort, offset: offset ?? 0, limit });
+    const total = await collection.repository.count({ fields, filter });
     const records = result.map(
       (x) => fields.map((field) => {
         var _a2;
@@ -129,6 +136,7 @@ class AIContextDatasourceManager {
     );
     return {
       options,
+      total,
       records
     };
   }

package/dist/server/manager/built-in-manager.js

@@ -40,13 +40,15 @@ __export(built_in_manager_exports, {
 });
 module.exports = __toCommonJS(built_in_manager_exports);
 var import_data_modeling = __toESM(require("../ai-employees/built-in/data-modeling"));
-var import_form_filler = __toESM(require("../ai-employees/built-in/form-filler"));
 var import_ai_coding = __toESM(require("../ai-employees/built-in/ai-coding"));
 var import_data_organizer = __toESM(require("../ai-employees/built-in/data-organizer"));
 var import_insights_analyst = __toESM(require("../ai-employees/built-in/insights-analyst"));
 var import_research_analyst = __toESM(require("../ai-employees/built-in/research-analyst"));
 var import_translator = __toESM(require("../ai-employees/built-in/translator"));
 var import_nocobase_assistant = __toESM(require("../ai-employees/built-in/nocobase-assistant"));
+var import_email_assistant = __toESM(require("../ai-employees/built-in/email-assistant"));
+var import_data_visualization = __toESM(require("../ai-employees/built-in/data-visualization"));
+var import_lodash = __toESM(require("lodash"));
 const DEFAULT_LANGUAGE = "en-US";
 const DEFAULT_KNOWLEDGE_BASE = {
   topK: 3,
@@ -60,16 +62,21 @@ class BuiltInManager {
   }
   builtInEmployees = [
     import_data_modeling.default,
-    import_form_filler.default,
     import_ai_coding.default,
     import_data_organizer.default,
     import_insights_analyst.default,
     import_research_analyst.default,
     import_translator.default,
-    import_nocobase_assistant.default
+    import_nocobase_assistant.default,
+    import_email_assistant.default,
+    import_data_visualization.default
   ];
   builtInEmployeeMap = Object.fromEntries(this.builtInEmployees.map((x) => [x.username, x]));
   setupBuiltInInfo(locale, aiEmployee) {
+    var _a;
+    if (!aiEmployee) {
+      return;
+    }
     const builtInEmployeeInfo = this.builtInEmployeeMap[aiEmployee.username];
     if (!builtInEmployeeInfo) {
       return;
@@ -81,8 +88,14 @@ class BuiltInManager {
     aiEmployee.position = position;
     aiEmployee.bio = bio;
     aiEmployee.greeting = greeting;
-    aiEmployee.about = about;
-    aiEmployee.skillSettings = builtInEmployeeInfo.skillSettings;
+    aiEmployee.about = aiEmployee.about ?? about;
+    const builtInSkills = ((_a = builtInEmployeeInfo.skillSettings) == null ? void 0 : _a.skills) ?? [];
+    const skillSettings = aiEmployee.skillSettings ?? {};
+    const skills = skillSettings.skills ?? [];
+    aiEmployee.skillSettings = {
+      ...builtInEmployeeInfo.skillSettings ?? {},
+      skills: import_lodash.default.uniqBy([...skills, ...builtInSkills], "name")
+    };
   }
   async createOrUpdateAIEmployee(language = DEFAULT_LANGUAGE) {
     const aiEmployeesRepo = this.plugin.db.getRepository("aiEmployees");
@@ -141,13 +154,6 @@ class BuiltInManager {
         const { nickname, avatar, position, bio, greeting, about } = p;
         await aiEmployeesRepo.update({
           values: {
-            nickname,
-            position,
-            avatar,
-            bio,
-            greeting,
-            about,
-            skillSettings,
             builtIn: true
           },
           filter: {

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

@@ -24,7 +24,7 @@ export interface ToolOptions {
     execution?: 'frontend' | 'backend';
     name: string;
     schema?: any;
-    invoke: (ctx: Context, args: Record<string, any>) => Promise<{
+    invoke: (ctx: Context, args: Record<string, any>, id: string) => Promise<{
         status: 'success' | 'error';
         content: string;
     }>;

package/dist/server/plugin.d.ts

@@ -23,6 +23,11 @@ export declare class PluginAIServer extends Plugin {
     aiCodingManager: AICodingManager;
     workContextHandler: import("./types").WorkContextHandler;
     snowflake: Snowflake;
+    /**
+     * Check if the AI employee is a builder/admin-only type (e.g., Nathan, Orin).
+     * These employees have powerful capabilities (coding, schema modification) and should be restricted to admins.
+     */
+    isBuilderAI(username: string): boolean;
     afterAdd(): Promise<void>;
     beforeLoad(): Promise<void>;
     load(): Promise<void>;
@@ -35,6 +40,7 @@ export declare class PluginAIServer extends Plugin {
     registerWorkContextResolveStrategy(): void;
     handleSyncMessage(message: any): Promise<void>;
     install(): Promise<void>;
+    upgrade(): Promise<void>;
     afterEnable(): Promise<void>;
     afterDisable(): Promise<void>;
     remove(): Promise<void>;

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,15 @@ 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");
+var import_suggestions = require("./tools/suggestions");
 class PluginAIServer extends import_server.Plugin {
   features = new import_ai_feature_manager.AIPluginFeatureManagerImpl();
   aiManager = new import_ai_manager.AIManager(this);
@@ -72,6 +77,14 @@ class PluginAIServer extends import_server.Plugin {
   aiCodingManager = new import_ai_coding_manager.AICodingManager(this);
   workContextHandler = (0, import_work_context_handler.createWorkContextHandler)(this);
   snowflake;
+  /**
+   * Check if the AI employee is a builder/admin-only type (e.g., Nathan, Orin).
+   * These employees have powerful capabilities (coding, schema modification) and should be restricted to admins.
+   */
+  isBuilderAI(username) {
+    const BUILDER_AI_USERNAMES = ["nathan", "orin", "dara"];
+    return BUILDER_AI_USERNAMES.includes(username);
+  }
   async afterAdd() {
   }
   async beforeLoad() {
@@ -94,17 +107,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 +137,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,
@@ -129,6 +156,10 @@ class PluginAIServer extends import_server.Plugin {
         groupName: dataModelingGroupName,
         tool: import_tools.dataModelingIntentRouter
       },
+      {
+        groupName: dataModelingGroupName,
+        tool: import_tools.getDataSources
+      },
       {
         groupName: dataModelingGroupName,
         tool: import_tools.getCollectionNames
@@ -141,8 +172,28 @@ class PluginAIServer extends import_server.Plugin {
         groupName: dataModelingGroupName,
         tool: import_tools.defineCollections
       },
+      {
+        groupName: dataModelingGroupName,
+        tool: import_suggestions.suggestions
+      },
       {
         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({
@@ -201,7 +252,16 @@ class PluginAIServer extends import_server.Plugin {
       workflowSnippet.actions.push("ai:listModels");
     }
     this.app.db.on("roles.beforeCreate", async (instance) => {
-      instance.set("allowNewAiEmployee", ["admin", "member"].includes(instance.name));
+      instance.set("allowNewAiEmployee", true);
+    });
+    this.app.db.on("roles.afterCreate", async (instance, { transaction }) => {
+      const allAiEmployees = await this.app.db.getRepository("aiEmployees").find({
+        transaction
+      });
+      const generalAiEmployees = allAiEmployees.filter((ai) => !this.isBuilderAI(ai.username));
+      if (generalAiEmployees.length > 0) {
+        await instance.addAiEmployees(generalAiEmployees, { transaction });
+      }
     });
     this.app.db.on("aiEmployees.afterCreate", async (instance, { transaction }) => {
       const roles = await this.app.db.getRepository("roles").find({
@@ -210,8 +270,12 @@ class PluginAIServer extends import_server.Plugin {
         },
         transaction
       });
+      let targetRoles = roles;
+      if (this.isBuilderAI(instance.username)) {
+        targetRoles = roles.filter((role) => role.name === "admin");
+      }
       await this.app.db.getRepository("aiEmployees.roles", instance.username).add({
-        tk: roles.map((role) => role.name),
+        tk: targetRoles.map((role) => role.name),
         transaction
       });
     });
@@ -247,6 +311,9 @@ class PluginAIServer extends import_server.Plugin {
     await this.db.getRepository("aiSettings").create({});
     await this.setupBuiltIn();
   }
+  async upgrade() {
+    await this.setupBuiltIn();
+  }
   async afterEnable() {
   }
   async afterDisable() {

package/dist/server/resource/ai.js

@@ -104,7 +104,7 @@ const aiResource = {
       return next();
     },
     testFlightModels: async (ctx, next) => {
-      const { provider, options } = ctx.action.params.values ?? {};
+      const { provider, options, model } = ctx.action.params.values ?? {};
       const plugin = ctx.app.pm.get("ai");
       const providerOptions = plugin.aiManager.llmProviders.get(provider);
       if (!providerOptions) {
@@ -120,7 +120,12 @@ const aiResource = {
         ctx.log.error(res.errMsg);
         ctx.throw(500, ctx.t("Get models list failed, you can enter a model name manually."));
       }
-      ctx.body = res.models || [];
+      const models = res.models || [];
+      if (model) {
+        ctx.body = models.filter((m) => m.id.toLowerCase().includes(model.toLowerCase()));
+      } else {
+        ctx.body = models;
+      }
       return next();
     },
     testFlight: async (ctx, next) => {

package/dist/server/resource/aiConversations.js

@@ -64,10 +64,7 @@ function setupSSEHeaders(ctx) {
   ctx.status = 200;
 }
 function sendErrorResponse(ctx, errorMessage) {
-  ctx.res.write(`data: ${JSON.stringify({ type: "error", body: errorMessage })} 
-
-`);
-  ctx.res.end();
+  (0, import_utils.sendSSEError)(ctx, errorMessage);
 }
 var aiConversations_default = {
   name: "aiConversations",
@@ -361,7 +358,7 @@ var aiConversations_default = {
         await aiEmployee.processMessages(messages, editingMessageId);
       } catch (err) {
         ctx.log.error(err);
-        sendErrorResponse(ctx, "Chat error warning");
+        sendErrorResponse(ctx, err.message || "Tool call error");
       }
       await next();
     },
@@ -416,7 +413,7 @@ var aiConversations_default = {
         await aiEmployee.resendMessages(messageId);
       } catch (err) {
         ctx.log.error(err);
-        sendErrorResponse(ctx, "Chat error warning");
+        sendErrorResponse(ctx, err.message || "Chat error warning");
       }
       await next();
     },
@@ -513,7 +510,7 @@ var aiConversations_default = {
         await aiEmployee.callTool(message.messageId, false);
       } catch (err) {
         ctx.log.error(err);
-        sendErrorResponse(ctx, "Tool call error");
+        sendErrorResponse(ctx, err.message || "Tool call error");
       }
       await next();
     },
@@ -568,7 +565,7 @@ var aiConversations_default = {
         await aiEmployee.confirmToolCall(message.messageId, toolCallIds);
       } catch (err) {
         ctx.log.error(err);
-        sendErrorResponse(ctx, "Tool call confirm error");
+        sendErrorResponse(ctx, err.message || "Tool call confirm error");
       }
       await next();
     }

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

@@ -8,6 +8,8 @@
  */
 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>;
+export declare const getBuiltInDefault: (ctx: Context, next: Next) => Promise<void>;

package/dist/server/resource/aiEmployees.js

@@ -36,6 +36,8 @@ 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,
+  getBuiltInDefault: () => getBuiltInDefault,
   getTemplates: () => getTemplates,
   list: () => list,
   listByUser: () => listByUser,
@@ -44,6 +46,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 +65,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 +132,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, _c, _d;
+    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,11 +161,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,
-      webSearch: ((_d = (_c = row.modelSettings) == null ? void 0 : _c.builtIn) == null ? void 0 : _d.webSearch) ?? false
+      webSearch: ((_e = (_d = row.modelSettings) == null ? void 0 : _d.builtIn) == null ? void 0 : _e.webSearch) ?? false,
+      llmProvider,
+      toolsConflict: llmProvider === "google-genai"
     };
   });
   await next();
@@ -166,8 +206,23 @@ const getTemplates = async (ctx, next) => {
   ctx.body = Object.values(templates).map((template) => template[locale]);
   await next();
 };
+const getBuiltInDefault = async (ctx, next) => {
+  const { filterByTk } = ctx.action.params || {};
+  if (!filterByTk) {
+    ctx.throw(400);
+  }
+  const plugin = ctx.app.pm.get("ai");
+  const builtInManager = plugin.builtInManager;
+  const locale = ctx.getCurrentLocale();
+  const temp = { username: filterByTk };
+  builtInManager.setupBuiltInInfo(locale, temp);
+  ctx.body = { about: temp.about };
+  await next();
+};
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  create,
+  getBuiltInDefault,
   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.d.ts

@@ -10,4 +10,5 @@ import { ToolOptions } from '../manager/tool-manager';
 export declare const dataModelingIntentRouter: ToolOptions;
 export declare const getCollectionNames: ToolOptions;
 export declare const getCollectionMetadata: ToolOptions;
+export declare const getDataSources: ToolOptions;
 export declare const defineCollections: ToolOptions;