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

package/dist/server/ai-employees/ai-conversations.d.ts

@@ -9,11 +9,13 @@
 import { Model, Transaction } from '@nocobase/database';
 import PluginAIServer from '../plugin';
 import { AIMessage, SubAgentConversationMetadata, UserDecision } from '../types';
+import type { FrontendToolManifest } from '../../common/frontend-tools';
 export type AIConversationsOptions = {
     systemMessage?: unknown;
     skillSettings?: unknown;
     conversationSettings?: unknown;
     modelSettings?: unknown;
+    frontendTools?: FrontendToolManifest[];
     [key: string]: unknown;
 };
 export type AIConversationFilterParams = {
@@ -29,6 +31,7 @@ export type CreateAIConversationParams = {
     title?: string;
     options?: AIConversationsOptions;
     from?: 'main-agent' | 'sub-agent';
+    scope?: string;
     transaction?: Transaction;
     category?: 'chat' | 'task';
 };
@@ -55,7 +58,7 @@ export declare const registerAIConversationReadNotification: (plugin: PluginAISe
 export declare class AIConversationsManager {
     protected plugin: PluginAIServer;
     constructor(plugin: PluginAIServer);
-    create({ userId, aiEmployee, title, options, from, transaction, category, }: CreateAIConversationParams): Promise<any>;
+    create({ userId, aiEmployee, title, options, from, scope, transaction, category, }: CreateAIConversationParams): Promise<any>;
     update({ userId, sessionId, title, options: inputOptions }: UpdateAIConversationParams): Promise<any>;
     getConversation({ sessionId, userId }: {
         sessionId: string;

package/dist/server/ai-employees/ai-conversations.js

@@ -66,6 +66,7 @@ class AIConversationsManager {
     title,
     options = {},
     from = "main-agent",
+    scope,
     transaction,
     category = "chat"
   }) {
@@ -77,6 +78,7 @@ class AIConversationsManager {
         options,
         thread: 1,
         from,
+        ...scope !== void 0 ? { scope } : {},
         category
       },
       transaction

package/dist/server/ai-employees/ai-employee.d.ts

@@ -36,6 +36,7 @@ export declare class AIEmployee {
     employee: Model;
     aiChatConversation: AIChatConversation;
     skillSettings?: Record<string, any>;
+    userMessageCount: number;
     private plugin;
     private db;
     private ctx;
@@ -117,7 +118,7 @@ export declare class AIEmployee {
     initToolCall(transaction: Transaction, messageId: string, toolCalls: {
         id: string;
         name: string;
-        args: any;
+        args: unknown;
     }[]): Promise<Model<AIToolMessage>[]>;
     updateToolCallInterrupted(sessionId: string, messageId: string, toolCallId: string, interruptId: string, interruptAction: {
         order: number;
@@ -143,6 +144,7 @@ export declare class AIEmployee {
     removeAbortController(): void;
     shouldInterruptToolCall(tools: ToolsEntry): boolean;
     isAutoCall(tools: ToolsEntry): boolean;
+    private normalizeMessageAttachments;
     private formatMessages;
     private getToolCallMap;
     private toInterruptActions;

package/dist/server/ai-employees/ai-employee.js

@@ -55,12 +55,16 @@ var import_utils2 = require("./utils");
 var import_base = require("@langchain/core/callbacks/base");
 var import_shared = require("../../ai/tools/sub-agents/shared");
 var import_tool_call_sanitizer = require("./tool-call-sanitizer");
+var import_attachments = require("../attachments");
+var import_frontend_tools = require("../../common/frontend-tools");
+var import_frontend_tools2 = require("../frontend-tools");
 class AIEmployee {
   sessionId;
   from = "main-agent";
   employee;
   aiChatConversation;
   skillSettings;
+  userMessageCount = 0;
   plugin;
   db;
   ctx;
@@ -141,7 +145,12 @@ class AIEmployee {
   }
   // === Chat flow ===
   buildState(messages) {
+    const toolCallMessage = messages.findLast((message) => {
+      var _a;
+      return (_a = message.toolCalls) == null ? void 0 : _a.length;
+    });
     return {
+      messageId: toolCallMessage == null ? void 0 : toolCallMessage.messageId,
       lastMessageIndex: {
         lastHumanMessageIndex: messages.filter((m) => m.role === "user").length,
         lastAIMessageIndex: messages.filter((m) => m.role === this.employee.username).length,
@@ -198,6 +207,7 @@ class AIEmployee {
     const { provider, model, service } = await this.plugin.aiManager.getLLMService({
       ...this.model
     });
+    this.userMessageCount = (userMessages ?? []).filter((message) => message.role === "user").length;
     const { historyMessages, tools, resolvedTools, middleware, config, state } = await this.initSession({
       messageId,
       provider,
@@ -746,7 +756,8 @@ ${addSystemPrompt.map((it) => it.content).join("\n")}`;
       },
       knowledgeBase,
       availableSkills,
-      availableAIEmployees
+      availableAIEmployees,
+      webSearch: this.webSearch
     });
     const { important } = ((_g = (_f = this.ctx.action) == null ? void 0 : _f.params) == null ? void 0 : _g.values) || {};
     if (important === "GraphRecursionError") {
@@ -762,11 +773,12 @@ If information is missing, clearly state it in the summary.</Important>`;
   async initToolCall(transaction, messageId, toolCalls) {
     const nowTime = /* @__PURE__ */ new Date();
     const toolMap = await this.getToolsMap();
+    const currentFrontendTools = toolCalls.some((toolCall) => toolCall.name === import_frontend_tools.EXECUTE_FRONTEND_TOOL_NAME) ? await (0, import_frontend_tools2.listCurrentFrontendTools)(this.ctx, this.sessionId) : [];
     return await this.aiToolMessagesRepo.create({
       values: toolCalls.map((toolCall) => {
         const toolsExisted = toolMap.has(toolCall.name);
         const tools = toolMap.get(toolCall.name);
-        const auto = this.isAutoCall(tools);
+        const auto = toolCall.name === import_frontend_tools.EXECUTE_FRONTEND_TOOL_NAME ? toolsExisted && (0, import_frontend_tools2.shouldAutoExecuteFrontendTool)(currentFrontendTools, toolCall.args) : this.isAutoCall(tools);
         return {
           id: this.plugin.snowflake.generate(),
           sessionId: this.sessionId,
@@ -1019,15 +1031,43 @@ If information is missing, clearly state it in the summary.</Important>`;
     const presetTools = employeeTools.find((s) => s.name === tools.definition.name);
     return presetTools ? presetTools.autoCall : isAutoCall;
   }
+  async normalizeMessageAttachments(messages) {
+    const attachments = messages.filter((message) => Array.isArray(message.attachments)).flatMap((message) => message.attachments);
+    if (!attachments.length) {
+      return messages;
+    }
+    const attachmentsByLookup = await (0, import_attachments.findMessageAttachments)(this.ctx, attachments);
+    return messages.map((message) => {
+      if (!Array.isArray(message.attachments) || !message.attachments.length) {
+        return message;
+      }
+      return {
+        ...message,
+        attachments: message.attachments.flatMap((attachment) => {
+          const source = (0, import_attachments.getAttachmentSource)(attachment);
+          if (!source || (0, import_attachments.shouldSkipAttachmentSourceLookup)(source)) {
+            return [attachment];
+          }
+          const lookupKey = (0, import_attachments.getMessageAttachmentLookupKey)(attachment);
+          const verifiedAttachment = lookupKey ? attachmentsByLookup.get(lookupKey) : null;
+          if (!verifiedAttachment) {
+            return [];
+          }
+          return [{ ...verifiedAttachment, source }];
+        })
+      };
+    });
+  }
   async formatMessages({ messages, provider }) {
     var _a, _b;
     const formattedMessages = [];
     const workContextHandler = this.plugin.workContextHandler;
+    const normalizedMessages = await this.normalizeMessageAttachments(messages);
     const truncate = (text, maxLen = 5e4) => {
       if (!text || text.length <= maxLen) return text;
       return text.slice(0, maxLen) + "\n...[truncated]";
     };
-    for (const msg of messages) {
+    for (const msg of normalizedMessages) {
       const attachments = msg.attachments;
       const workContext = msg.workContext;
       const userContent = msg.content;
@@ -1155,7 +1195,12 @@ If information is missing, clearly state it in the summary.</Important>`;
     if (!this.areToolsEnabled()) {
       return [];
     }
+    const currentFrontendTools = await (0, import_frontend_tools2.listCurrentFrontendTools)(this.ctx, this.sessionId);
     const tools = await this.listTools({ scope: "GENERAL" });
+    const getSkill = await this.toolsManager.getTools(import_ai.SYSTEM_TOOLS.GET_SKILL, { ctx: this.ctx });
+    if (getSkill) {
+      tools.push(getSkill);
+    }
     if (this.webSearch === true) {
       const subAgentWebSearch = await this.toolsManager.getTools(import_ai.SYSTEM_TOOLS.WEB_SEARCH, { ctx: this.ctx });
       tools.push(subAgentWebSearch);
@@ -1182,20 +1227,26 @@ If information is missing, clearly state it in the summary.</Important>`;
       }
       tools.push(tool);
     }
-    const systemTools = (0, import_ai.listSystemTools)();
+    const systemTools = [...(0, import_ai.listSystemTools)(), import_frontend_tools.LOAD_FRONTEND_TOOL_NAME, import_frontend_tools.EXECUTE_FRONTEND_TOOL_NAME];
     if (!this.skillSettings) {
-      return tools;
+      return (0, import_frontend_tools2.prepareToolsForFrontendConversation)(tools, currentFrontendTools);
     } else if (!this.skillSettings.toolsVersion) {
       const toolFilter = this.skillSettings.tools ?? [];
-      return tools.filter(
-        (t) => toolFilter.length === 0 || systemTools.includes(t.definition.name) || toolFilter.includes(t.definition.name)
+      return (0, import_frontend_tools2.prepareToolsForFrontendConversation)(
+        tools.filter(
+          (t) => toolFilter.length === 0 || systemTools.includes(t.definition.name) || toolFilter.includes(t.definition.name)
+        ),
+        currentFrontendTools
       );
     } else {
       const toolFilter = this.skillSettings.tools;
       if (import_lodash.default.isArray(toolFilter)) {
-        return tools.filter((t) => systemTools.includes(t.definition.name) || toolFilter.includes(t.definition.name));
+        return (0, import_frontend_tools2.prepareToolsForFrontendConversation)(
+          tools.filter((t) => systemTools.includes(t.definition.name) || toolFilter.includes(t.definition.name)),
+          currentFrontendTools
+        );
       } else {
-        return tools;
+        return (0, import_frontend_tools2.prepareToolsForFrontendConversation)(tools, currentFrontendTools);
       }
     }
   }
@@ -1237,6 +1288,9 @@ If information is missing, clearly state it in the summary.</Important>`;
     }
     const baseTools = await this.getAIEmployeeTools();
     const toolMap = await this.getToolsMap();
+    for (const tool of baseTools) {
+      toolMap.set(tool.definition.name, tool);
+    }
     const availableSkills = await this.getAvailableSkills();
     const skillOwnedToolNames = new Set(availableSkills.flatMap((it) => it.tools ?? []));
     const baseToolNames = new Set(

package/dist/server/ai-employees/middleware/conversation.js

@@ -98,8 +98,9 @@ const conversationMiddleware = (aiEmployee, options) => {
       })
     }),
     beforeAgent: async (state) => {
-      const lastHumanMessageIndex = state.lastMessageIndex.lastHumanMessageIndex;
-      const userMessages = state.messages.filter((x) => x.type === "human").slice(lastHumanMessageIndex).map((x) => x).map(convertHumanMessage);
+      const humanMessages = state.messages.filter((x) => x.type === "human");
+      const currentHumanMessageIndex = humanMessages.length;
+      const userMessages = (aiEmployee.userMessageCount ? humanMessages.slice(-aiEmployee.userMessageCount) : []).map((x) => x).map(convertHumanMessage);
       await aiEmployee.aiChatConversation.withTransaction(async (conversation, transaction) => {
         if (agentThread) {
           await aiEmployee.updateThread(transaction, agentThread);
@@ -111,6 +112,12 @@ const conversationMiddleware = (aiEmployee, options) => {
           await conversation.addMessages(userMessages);
         }
       });
+      return {
+        lastMessageIndex: {
+          ...state.lastMessageIndex,
+          lastHumanMessageIndex: currentHumanMessageIndex
+        }
+      };
     },
     afterAgent: async () => {
       aiEmployee.removeAbortController();

package/dist/server/ai-employees/prompts.d.ts

@@ -6,7 +6,7 @@
  * 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 function getSystemPrompt({ aiEmployee, personal, task, environment, knowledgeBase, availableSkills, availableAIEmployees, }: {
+export declare function getSystemPrompt({ aiEmployee, personal, task, environment, knowledgeBase, availableSkills, availableAIEmployees, webSearch, }: {
     aiEmployee: {
         nickname: string;
         about: string;
@@ -23,6 +23,7 @@ export declare function getSystemPrompt({ aiEmployee, personal, task, environmen
         timezone?: string;
     };
     knowledgeBase?: string;
+    webSearch?: boolean;
     availableSkills?: {
         name: string;
         description: string;

package/dist/server/ai-employees/prompts.js

@@ -36,7 +36,8 @@ function getSystemPrompt({
   environment,
   knowledgeBase,
   availableSkills,
-  availableAIEmployees
+  availableAIEmployees,
+  webSearch
 }) {
   const getDatabaseQuotingRules = () => {
     const quotingMap = {
@@ -52,6 +53,11 @@ function getSystemPrompt({
   };
   const quotingRules = getDatabaseQuotingRules();
   const isUnderscored = process.env.DB_UNDERSCORED === "true";
+  const webSearchInstructions = webSearch === true ? `
+   - When using web search, batch all independent search queries needed for this turn into one tool call whenever possible so they can run in parallel.
+   - Do not make repeated web search calls with the same or similar queries in the same turn.
+   - After receiving usable web search results, synthesize an answer from those results instead of searching again.
+   - Search again only when a critical fact is still missing and the new query is materially different from previous queries.` : "";
   return `You are **${aiEmployee.nickname}**, an AI employee working in **NocoBase**, the leading no-code platform.
 
 You assist developers in building enterprise management systems (CRM, ERP, OA, etc.) and help end users complete business tasks.
@@ -114,7 +120,7 @@ This prompt uses a structured tag system to organize your operational framework:
 5. **Tool Integration**
    - Utilize system-provided tools to enhance response quality
    - **NEVER refer to tool names when speaking to the USER.** Instead, just say what the tool is doing in natural language.
-   - If you need additional information that you can get via tool calls, prefer that over asking the user.
+   - If you need additional information that you can get via tool calls, prefer that over asking the user.${webSearchInstructions}
 </global>
 
 <ai_employee>

package/dist/server/ai-employees/tool-call-sanitizer.d.ts

@@ -20,6 +20,7 @@ type SanitizeAdditionalKwargsOptions = {
 type SanitizeAdditionalKwargsResult = {
     changed: boolean;
     additionalKwargs?: AdditionalKwargs;
+    malformedToolCalls?: unknown[];
 };
 export declare const sanitizeAdditionalKwargsForToolCalls: (additionalKwargs: AdditionalKwargs | undefined, parsedToolCalls?: unknown[] | null, options?: SanitizeAdditionalKwargsOptions) => SanitizeAdditionalKwargsResult;
 export declare const sanitizeLangChainAIMessage: (message: AIMessage, options?: SanitizeAdditionalKwargsOptions) => AIMessage<import("@langchain/core/dist/messages/message").MessageStructure<import("@langchain/core/dist/messages/message").MessageToolSet>>;

package/dist/server/ai-employees/tool-call-sanitizer.js

@@ -65,7 +65,8 @@ const sanitizeAdditionalKwargsForToolCalls = (additionalKwargs, parsedToolCalls,
   delete sanitized.tool_calls;
   return {
     changed: true,
-    additionalKwargs: Object.keys(sanitized).length > 0 ? sanitized : void 0
+    additionalKwargs: Object.keys(sanitized).length > 0 ? sanitized : void 0,
+    malformedToolCalls: rawToolCalls
   };
 };
 const sanitizeLangChainAIMessage = (message, options = {}) => {

package/dist/server/ai-employees/utils.js

@@ -40,7 +40,7 @@ const convertAIMessage = ({
   model,
   aiMessage
 }) => {
-  var _a;
+  var _a, _b;
   const message = aiMessage.content;
   const toolCalls = aiMessage.tool_calls;
   const tools = (_a = aiEmployee.skillSettings) == null ? void 0 : _a.tools;
@@ -91,20 +91,26 @@ const convertAIMessage = ({
   if (aiMessage.response_metadata) {
     values.metadata.response_metadata = aiMessage.response_metadata;
   }
-  const additionalKwargs = (0, import_tool_call_sanitizer.sanitizeAdditionalKwargsForToolCalls)(aiMessage.additional_kwargs, toolCalls, {
+  const sanitizedToolCalls = (0, import_tool_call_sanitizer.sanitizeAdditionalKwargsForToolCalls)(aiMessage.additional_kwargs, toolCalls, {
     onDiscard: (info) => {
-      var _a2, _b;
-      (_b = aiEmployee.logger) == null ? void 0 : _b.warn("Discard malformed raw tool calls from AI message", {
+      var _a2, _b2;
+      (_b2 = aiEmployee.logger) == null ? void 0 : _b2.warn("Discard malformed raw tool calls from AI message", {
         phase: "convertAIMessage",
         messageId: aiMessage.id,
         invalidToolCallCount: ((_a2 = aiMessage.invalid_tool_calls) == null ? void 0 : _a2.length) ?? 0,
         ...info
       });
     }
-  }).additionalKwargs;
+  });
+  const additionalKwargs = sanitizedToolCalls.additionalKwargs;
   if (additionalKwargs) {
     values.metadata.additional_kwargs = additionalKwargs;
   }
+  if ((_b = sanitizedToolCalls.malformedToolCalls) == null ? void 0 : _b.length) {
+    values.metadata.diagnostics = {
+      malformedToolCalls: sanitizedToolCalls.malformedToolCalls
+    };
+  }
   providerInstance.reshapeAIMessage({ aiMessage, values });
   return values;
 };

package/dist/server/attachments.d.ts

@@ -0,0 +1,24 @@
+/**
+ * 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 type { Context } from '@nocobase/actions';
+import type { AttachmentModel } from '@nocobase/plugin-file-manager';
+export type AttachmentId = string | number;
+export type AttachmentSource = {
+    dataSourceKey?: string;
+    collectionName?: string;
+    field?: string;
+    documentCache?: boolean;
+    trustworthy?: boolean;
+};
+export declare function appendAIFileAttachmentSource(body: unknown): void;
+export declare function getAttachmentId(attachment: unknown): AttachmentId | null;
+export declare function shouldSkipAttachmentSourceLookup(source: AttachmentSource | null): boolean;
+export declare function getAttachmentSource(attachment: unknown): AttachmentSource | null;
+export declare function findMessageAttachments(ctx: Context, attachments: unknown[]): Promise<Map<string, AttachmentModel>>;
+export declare function getMessageAttachmentLookupKey(attachment: unknown): string;

package/dist/server/attachments.js

@@ -0,0 +1,204 @@
+/**
+ * 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 attachments_exports = {};
+__export(attachments_exports, {
+  appendAIFileAttachmentSource: () => appendAIFileAttachmentSource,
+  findMessageAttachments: () => findMessageAttachments,
+  getAttachmentId: () => getAttachmentId,
+  getAttachmentSource: () => getAttachmentSource,
+  getMessageAttachmentLookupKey: () => getMessageAttachmentLookupKey,
+  shouldSkipAttachmentSourceLookup: () => shouldSkipAttachmentSourceLookup
+});
+module.exports = __toCommonJS(attachments_exports);
+const AI_FILES_ATTACHMENT_SOURCE = {
+  dataSourceKey: "main",
+  collectionName: "aiFiles"
+};
+function isRecord(value) {
+  return Boolean(value && typeof value === "object" && !Array.isArray(value));
+}
+function appendSourceToAttachmentRecord(record) {
+  if (!isRecord(record)) {
+    return;
+  }
+  const meta = {
+    ...isRecord(record.meta) ? record.meta : {},
+    source: AI_FILES_ATTACHMENT_SOURCE
+  };
+  if (typeof record.set === "function") {
+    const model = record;
+    model.set("meta", meta);
+    model.dataValues.meta = meta;
+    return;
+  }
+  record.meta = meta;
+}
+function appendAIFileAttachmentSource(body) {
+  if (!isRecord(body)) {
+    return;
+  }
+  if (body.data) {
+    appendSourceToAttachmentRecord(body.data);
+    return;
+  }
+  appendSourceToAttachmentRecord(body);
+}
+function getAttachmentId(attachment) {
+  if (!isRecord(attachment)) {
+    return null;
+  }
+  const id = attachment.id ?? attachment.filterByTk;
+  if (typeof id === "string" || typeof id === "number") {
+    return id;
+  }
+  return null;
+}
+function shouldSkipAttachmentSourceLookup(source) {
+  return (source == null ? void 0 : source.trustworthy) === true;
+}
+function getAttachmentSource(attachment) {
+  if (!isRecord(attachment) || !isRecord(attachment.source)) {
+    return null;
+  }
+  const { dataSourceKey, collectionName, field, documentCache, trustworthy } = attachment.source;
+  const source = {
+    dataSourceKey: typeof dataSourceKey === "string" && dataSourceKey ? dataSourceKey : "main",
+    ...typeof collectionName === "string" && collectionName ? { collectionName } : {},
+    ...typeof field === "string" ? { field } : {},
+    ...typeof documentCache === "boolean" ? { documentCache } : {},
+    ...trustworthy === true ? { trustworthy } : {}
+  };
+  if (!source.trustworthy && !source.collectionName) {
+    return null;
+  }
+  return source;
+}
+function getSourceDatabase(ctx, dataSourceKey) {
+  var _a, _b, _c;
+  if (!dataSourceKey || dataSourceKey === "main") {
+    return ctx.db;
+  }
+  const dataSource = (_b = (_a = ctx.app.dataSourceManager) == null ? void 0 : _a.dataSources) == null ? void 0 : _b.get(dataSourceKey);
+  return ((_c = dataSource == null ? void 0 : dataSource.collectionManager) == null ? void 0 : _c.db) ?? null;
+}
+function getLookupKey(lookup) {
+  const dataSourceKey = lookup.source.dataSourceKey || "main";
+  const collectionName = lookup.source.collectionName;
+  return `${dataSourceKey}:${collectionName}:${lookup.id}`;
+}
+function isValidFileCollectionSource(db, lookup) {
+  var _a, _b, _c;
+  const collectionName = lookup.source.collectionName;
+  const collection = db.getCollection(collectionName);
+  if (((_a = collection == null ? void 0 : collection.options) == null ? void 0 : _a.template) !== "file") {
+    return false;
+  }
+  if (collectionName === "aiFiles") {
+    return true;
+  }
+  if (!lookup.source.field) {
+    return false;
+  }
+  return ((_c = (_b = db.getFieldByPath(lookup.source.field)) == null ? void 0 : _b.options) == null ? void 0 : _c.target) === collectionName;
+}
+async function findSourceAttachments(ctx, lookups) {
+  var _a;
+  const attachmentsByLookup = /* @__PURE__ */ new Map();
+  const groups = /* @__PURE__ */ new Map();
+  for (const lookup of lookups) {
+    const dataSourceKey = lookup.source.dataSourceKey || "main";
+    const collectionName = lookup.source.collectionName;
+    const groupKey = `${dataSourceKey}:${collectionName}`;
+    groups.set(groupKey, [...groups.get(groupKey) ?? [], lookup]);
+  }
+  for (const [groupKey, groupLookups] of groups.entries()) {
+    const [dataSourceKey, collectionName] = groupKey.split(":");
+    const db = getSourceDatabase(ctx, dataSourceKey);
+    if (!db) {
+      continue;
+    }
+    const validLookups = groupLookups.filter((lookup) => isValidFileCollectionSource(db, lookup));
+    if (!validLookups.length) {
+      continue;
+    }
+    const filter = {
+      id: {
+        $in: [...new Set(validLookups.map((lookup) => lookup.id))]
+      }
+    };
+    if (collectionName === "aiFiles") {
+      filter.createdById = (_a = ctx.auth) == null ? void 0 : _a.user.id;
+    }
+    const records = await db.getRepository(collectionName).find({ filter });
+    for (const record of records) {
+      const attachment = record.toJSON();
+      attachmentsByLookup.set(`${dataSourceKey}:${collectionName}:${attachment.id}`, attachment);
+    }
+  }
+  return attachmentsByLookup;
+}
+async function findMessageAttachments(ctx, attachments) {
+  const lookups = [];
+  for (const attachment of attachments) {
+    const source = getAttachmentSource(attachment);
+    if (!source || shouldSkipAttachmentSourceLookup(source)) {
+      continue;
+    }
+    const id = getAttachmentId(attachment);
+    if (id == null) {
+      continue;
+    }
+    lookups.push({
+      id,
+      source
+    });
+  }
+  return findSourceAttachments(ctx, lookups);
+}
+function getMessageAttachmentLookupKey(attachment) {
+  const source = getAttachmentSource(attachment);
+  if (!source || shouldSkipAttachmentSourceLookup(source)) {
+    return null;
+  }
+  const id = getAttachmentId(attachment);
+  if (id == null) {
+    return null;
+  }
+  return getLookupKey({
+    id,
+    source
+  });
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  appendAIFileAttachmentSource,
+  findMessageAttachments,
+  getAttachmentId,
+  getAttachmentSource,
+  getMessageAttachmentLookupKey,
+  shouldSkipAttachmentSourceLookup
+});

package/dist/server/collections/ai-conversations.js

@@ -57,6 +57,11 @@ var ai_conversations_default = (0, import_database.defineCollection)({
       defaultValue: "main-agent",
       allowNull: false
     },
+    {
+      name: "scope",
+      type: "string",
+      index: true
+    },
     {
       name: "user",
       type: "belongsTo",

package/dist/server/collections/ai-usage-events.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.
+ */
+declare const _default: import("@nocobase/database").CollectionOptions;
+export default _default;