@nocobase/plugin-ai 2.2.0-alpha.1 → 2.2.0-alpha.11

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

@@ -45,7 +45,6 @@ var import_utils = require("../utils");
 var import_prompts = require("./prompts");
 var import_lodash = __toESM(require("lodash"));
 var import_ai_chat_conversation = require("../manager/ai-chat-conversation");
-var import_ai_feature_manager = require("../manager/ai-feature-manager");
 var import_middleware = require("./middleware");
 var import_ai = require("@nocobase/ai");
 var import_checkpoints = require("./checkpoints");
@@ -55,6 +54,10 @@ var import_stream = require("@langchain/core/utils/stream");
 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";
@@ -141,7 +144,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,
@@ -158,7 +166,7 @@ class AIEmployee {
         historyMessages: [],
         tools,
         resolvedTools,
-        middleware: await this.getMiddleware({ tools, baseToolNames, model, providerName, llmService }),
+        middleware: await this.getMiddleware({ tools, baseToolNames, model, providerName, provider, llmService }),
         config: void 0,
         state: void 0
       };
@@ -176,6 +184,7 @@ class AIEmployee {
         baseToolNames,
         model,
         providerName,
+        provider,
         llmService,
         messageId,
         agentThread
@@ -295,7 +304,7 @@ class AIEmployee {
       const { threadId } = await this.getCurrentThread();
       const invokeConfig = {
         context: { ctx: this.ctx, decisions: chatContext.decisions, ...context },
-        recursionLimit: 100,
+        recursionLimit: 200,
         configurable: this.from === "main-agent" ? { thread_id: threadId } : void 0,
         writer,
         signal,
@@ -389,7 +398,7 @@ class AIEmployee {
           streamMode: ["updates", "messages", "custom"],
           configurable: this.from === "main-agent" ? { thread_id: threadId } : void 0,
           context: { ctx: this.ctx, decisions: chatContext.decisions },
-          recursionLimit: 100,
+          recursionLimit: 200,
           ...config
         },
         state
@@ -412,6 +421,7 @@ class AIEmployee {
           const values = (0, import_utils2.convertAIMessage)({
             aiEmployee: this,
             providerName,
+            provider,
             llmService,
             model,
             aiMessage: gathered
@@ -744,7 +754,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") {
@@ -756,98 +767,16 @@ If information is missing, clearly state it in the summary.</Important>`;
       return systemPrompt;
     }
   }
-  async retrieveKnowledgeBase(userMessage) {
-    const vectorStoreProvider = this.plugin.features.vectorStoreProvider;
-    let queryResult = [];
-    const queryString = userMessage.content.content;
-    if (!queryString || import_lodash.default.isEmpty(queryString)) {
-      return queryResult;
-    }
-    const { topK, score } = this.getAIEmployeeKnowledgeBaseConfig();
-    const knowledgeBaseGroup = await this.getKnowledgeBaseGroup();
-    for (const entry of knowledgeBaseGroup) {
-      const { vectorStoreConfig, knowledgeBaseType, knowledgeBaseList } = entry;
-      if (!knowledgeBaseList || import_lodash.default.isEmpty(knowledgeBaseList)) {
-        continue;
-      }
-      if (knowledgeBaseType === "LOCAL") {
-        const vectorStoreService = await vectorStoreProvider.createVectorStoreService(
-          vectorStoreConfig.vectorStoreProvider,
-          [
-            {
-              key: "vectorStoreConfigKey",
-              value: vectorStoreConfig.vectorStoreConfigKey ?? ""
-            }
-          ]
-        );
-        const knowledgeBaseOuterIds = knowledgeBaseList.map((x) => x.knowledgeBaseOuterId);
-        const result = await vectorStoreService.search(queryString, {
-          topK,
-          score,
-          filter: {
-            knowledgeBaseOuterId: { in: knowledgeBaseOuterIds }
-          }
-        });
-        queryResult = [...queryResult, ...result];
-      } else if (knowledgeBaseType === "READONLY") {
-        const vectorStoreService = await vectorStoreProvider.createVectorStoreService(
-          vectorStoreConfig.vectorStoreProvider,
-          [
-            {
-              key: "vectorStoreConfigKey",
-              value: vectorStoreConfig.vectorStoreConfigKey ?? ""
-            }
-          ]
-        );
-        const result = await vectorStoreService.search(queryString, {
-          topK,
-          score
-        });
-        queryResult = [...queryResult, ...result];
-      } else if (knowledgeBaseType === "EXTERNAL") {
-        for (const knowledgeBase of knowledgeBaseList) {
-          const vectorStoreService = await vectorStoreProvider.createVectorStoreService(
-            vectorStoreConfig.vectorStoreProvider,
-            knowledgeBase.vectorStoreProps
-          );
-          const result = await vectorStoreService.search(queryString, {
-            topK,
-            score
-          });
-          queryResult = [...queryResult, ...result];
-        }
-      }
-    }
-    return queryResult;
-  }
-  isEnabledKnowledgeBase() {
-    const featureEnabled = this.plugin.features.isFeaturesEnabled(Object.values(import_ai_feature_manager.EEFeatures));
-    const knowledgeBaseEnabled = this.employee.enableKnowledgeBase;
-    return featureEnabled && knowledgeBaseEnabled;
-  }
-  getAIEmployeeKnowledgeBaseConfig() {
-    const { topK, score } = this.employee.knowledgeBase ?? {};
-    return {
-      topK,
-      score
-    };
-  }
-  async getKnowledgeBaseGroup() {
-    const { knowledgeBaseKeys } = this.employee.knowledgeBase ?? {};
-    if (!knowledgeBaseKeys || import_lodash.default.isEmpty(knowledgeBaseKeys)) {
-      return [];
-    }
-    return await this.plugin.features.knowledgeBase.getKnowledgeBaseGroup(knowledgeBaseKeys);
-  }
   // === Tool calls ===
   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,
@@ -1100,15 +1029,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;
@@ -1177,7 +1134,16 @@ If information is missing, clearly state it in the summary.</Important>`;
         role: "assistant",
         content,
         tool_calls: msg.toolCalls,
-        additional_kwargs: (_b = msg.metadata) == null ? void 0 : _b.additional_kwargs
+        additional_kwargs: (0, import_tool_call_sanitizer.sanitizeAdditionalKwargsForToolCalls)((_b = msg.metadata) == null ? void 0 : _b.additional_kwargs, msg.toolCalls, {
+          onDiscard: (info) => {
+            var _a2;
+            this.logger.warn("Discard malformed raw tool calls from AI message", {
+              phase: "formatMessages",
+              messageId: (_a2 = msg.metadata) == null ? void 0 : _a2.id,
+              ...info
+            });
+          }
+        }).additionalKwargs
       });
     }
     return formattedMessages;
@@ -1227,9 +1193,10 @@ 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" });
     if (this.webSearch === true) {
-      const subAgentWebSearch = await this.toolsManager.getTools(import_ai.SYSTEM_TOOLS.WEB_SEARCH);
+      const subAgentWebSearch = await this.toolsManager.getTools(import_ai.SYSTEM_TOOLS.WEB_SEARCH, { ctx: this.ctx });
       tools.push(subAgentWebSearch);
     }
     const generalToolsNameSet = new Set(tools.map((x) => x.definition.name));
@@ -1237,7 +1204,9 @@ If information is missing, clearly state it in the summary.</Important>`;
     const settingsTools = ((_a = this.employee.skillSettings) == null ? void 0 : _a.tools) ?? [];
     const employeeTools = [...settingsTools, ...this.tools];
     if (await this.plugin.knowledgeBaseManager.isEnabledKnowledgeBase(this.employee.toJSON())) {
-      const knowledgeBaseRetrieveTool = await this.toolsManager.getTools(import_ai.SYSTEM_TOOLS.KNOWLEDGE_BASE);
+      const knowledgeBaseRetrieveTool = await this.toolsManager.getTools(import_ai.SYSTEM_TOOLS.KNOWLEDGE_BASE, {
+        ctx: this.ctx
+      });
       if (knowledgeBaseRetrieveTool) {
         employeeTools.push({ name: import_ai.SYSTEM_TOOLS.KNOWLEDGE_BASE });
       }
@@ -1252,20 +1221,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);
       }
     }
   }
@@ -1307,6 +1282,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(
@@ -1375,7 +1353,7 @@ If information is missing, clearly state it in the summary.</Important>`;
     return availableAIEmployees;
   }
   async getMiddleware(options) {
-    const { providerName, llmService, model, tools, baseToolNames, messageId, agentThread } = options;
+    const { providerName, provider, llmService, model, tools, baseToolNames, messageId, agentThread } = options;
     const inWorkflow = await this.isInWorkflow();
     return [
       (0, import_middleware.skillToolBindingMiddleware)(this, {
@@ -1384,7 +1362,8 @@ If information is missing, clearly state it in the summary.</Important>`;
       (0, import_middleware.toolInteractionMiddleware)(this, tools),
       (0, import_middleware.toolCallStatusMiddleware)(this),
       ...inWorkflow ? [(0, import_middleware.workflowHistoryMiddleware)(this, this.db)] : [],
-      (0, import_middleware.conversationMiddleware)(this, { providerName, llmService, model, messageId, agentThread })
+      (0, import_middleware.conversationMiddleware)(this, { providerName, provider, llmService, model, messageId, agentThread }),
+      (0, import_middleware.toolCallSanitizerMiddleware)({ logger: this.logger })
     ];
   }
   async getCurrentThread() {
@@ -1415,7 +1394,10 @@ If information is missing, clearly state it in the summary.</Important>`;
     return new Map(tools.map((tool) => [tool.definition.name, tool]));
   }
   listTools(filter) {
-    return this.toolsManager.listTools(filter);
+    return this.toolsManager.listTools({
+      ...filter,
+      ctx: this.ctx
+    });
   }
   withRunMetadata(config) {
     return {

package/dist/server/ai-employees/ai-knowledge-base.d.ts

@@ -7,7 +7,6 @@
  * For more information, please refer to: https://www.nocobase.com/agreement.
  */
 import { AIEmployee } from '../../collections/ai-employees';
-import { DocumentSegmentedWithScore } from '../features';
 import type PluginAIServer from '../plugin';
 export type KnowledgeBaseRetrieveOptions = {
     username?: string;
@@ -18,11 +17,8 @@ export declare class KnowledgeBaseManager {
     private readonly plugin;
     constructor(plugin: PluginAIServer);
     retrievePrompt({ username, employee, query }: KnowledgeBaseRetrieveOptions): Promise<string>;
-    retrieve({ username, employee, query }: KnowledgeBaseRetrieveOptions): Promise<DocumentSegmentedWithScore[]>;
     isEnabledKnowledgeBase(username: string): Promise<boolean>;
     isEnabledKnowledgeBase(employee: AIEmployee): Promise<boolean>;
-    private getAIEmployeeKnowledgeBaseConfig;
-    private getKnowledgeBaseGroup;
     private getEmployee;
     private get aiEmployeeRepo();
 }

package/dist/server/ai-employees/ai-knowledge-base.js

@@ -42,115 +42,51 @@ module.exports = __toCommonJS(ai_knowledge_base_exports);
 var import_prompts = require("@langchain/core/prompts");
 var import_ai_feature_manager = require("../manager/ai-feature-manager");
 var import_lodash = __toESM(require("lodash"));
+const normalizeMatchedQuestions = (value) => {
+  if (!Array.isArray(value)) {
+    return [];
+  }
+  return value.filter((item) => typeof item === "string" && item.trim() !== "");
+};
+const buildKnowledgeBaseContent = (content, metadata) => {
+  const matchedQuestions = normalizeMatchedQuestions(metadata == null ? void 0 : metadata.matchedQuestions);
+  if (!matchedQuestions.length) {
+    return content;
+  }
+  return `Related questions:
+${matchedQuestions.join("\n")}
+
+${content}`;
+};
 class KnowledgeBaseManager {
   constructor(plugin) {
     this.plugin = plugin;
   }
   async retrievePrompt({ username, employee, query }) {
-    employee = employee ?? await this.getEmployee(username);
+    employee = employee ?? (username ? await this.getEmployee(username) : null);
     if (!employee) {
       return "Specified knowledge base not existed";
     }
-    const promptTemplate = import_prompts.ChatPromptTemplate.fromTemplate(employee.knowledgeBasePrompt);
-    const docs = await this.retrieve({
-      employee,
-      query
-    });
+    const { knowledgeBaseKeys, topK, score } = employee.knowledgeBase ?? {};
+    if (!knowledgeBaseKeys || import_lodash.default.isEmpty(knowledgeBaseKeys)) {
+      return "Specified knowledge base not existed";
+    }
+    const promptTemplate = import_prompts.ChatPromptTemplate.fromTemplate(employee.knowledgeBasePrompt ?? "{knowledgeBaseData}");
+    const docs = await this.plugin.features.knowledgeBase.search({ knowledgeBaseKeys, query, topK, score });
     if (!(docs == null ? void 0 : docs.length)) {
       return "No document match in knowledge base";
     }
-    const knowledgeBaseData = docs.map((x) => x.content).join("\n");
-    return import_lodash.default.isEmpty(knowledgeBaseData) ? void 0 : await promptTemplate.format({
+    const knowledgeBaseData = docs.map((doc) => buildKnowledgeBaseContent(doc.content, doc.metadata)).join("\n");
+    return import_lodash.default.isEmpty(knowledgeBaseData) ? "No document match in knowledge base" : await promptTemplate.format({
       knowledgeBaseData
     });
   }
-  async retrieve({ username, employee, query }) {
-    const vectorStoreProvider = this.plugin.features.vectorStoreProvider;
-    let queryResult = [];
-    if (!query || import_lodash.default.isEmpty(query)) {
-      return queryResult;
-    }
-    employee = employee ?? await this.getEmployee(username);
-    if (!employee) {
-      return queryResult;
-    }
-    const { topK, score } = this.getAIEmployeeKnowledgeBaseConfig(employee);
-    const knowledgeBaseGroup = await this.getKnowledgeBaseGroup(employee);
-    for (const entry of knowledgeBaseGroup) {
-      const { vectorStoreConfig, knowledgeBaseType, knowledgeBaseList } = entry;
-      if (!knowledgeBaseList || import_lodash.default.isEmpty(knowledgeBaseList)) {
-        continue;
-      }
-      if (knowledgeBaseType === "LOCAL") {
-        const vectorStoreService = await vectorStoreProvider.createVectorStoreService(
-          vectorStoreConfig.vectorStoreProvider,
-          [
-            {
-              key: "vectorStoreConfigKey",
-              value: vectorStoreConfig.vectorStoreConfigKey ?? ""
-            }
-          ]
-        );
-        const knowledgeBaseOuterIds = knowledgeBaseList.map((x) => x.knowledgeBaseOuterId);
-        const result = await vectorStoreService.search(query, {
-          topK,
-          score,
-          filter: {
-            knowledgeBaseOuterId: { in: knowledgeBaseOuterIds }
-          }
-        });
-        queryResult = [...queryResult, ...result];
-      } else if (knowledgeBaseType === "READONLY") {
-        const vectorStoreService = await vectorStoreProvider.createVectorStoreService(
-          vectorStoreConfig.vectorStoreProvider,
-          [
-            {
-              key: "vectorStoreConfigKey",
-              value: vectorStoreConfig.vectorStoreConfigKey ?? ""
-            }
-          ]
-        );
-        const result = await vectorStoreService.search(query, {
-          topK,
-          score
-        });
-        queryResult = [...queryResult, ...result];
-      } else if (knowledgeBaseType === "EXTERNAL") {
-        for (const knowledgeBase of knowledgeBaseList) {
-          const vectorStoreService = await vectorStoreProvider.createVectorStoreService(
-            vectorStoreConfig.vectorStoreProvider,
-            knowledgeBase.vectorStoreProps
-          );
-          const result = await vectorStoreService.search(query, {
-            topK,
-            score
-          });
-          queryResult = [...queryResult, ...result];
-        }
-      }
-    }
-    return queryResult;
-  }
   async isEnabledKnowledgeBase(usernameOrEmployee) {
     const featureEnabled = this.plugin.features.isFeaturesEnabled(Object.values(import_ai_feature_manager.EEFeatures));
     const employee = typeof usernameOrEmployee === "string" ? await this.getEmployee(usernameOrEmployee) : usernameOrEmployee;
     const knowledgeBaseEnabled = employee == null ? void 0 : employee.enableKnowledgeBase;
     return featureEnabled && knowledgeBaseEnabled;
   }
-  getAIEmployeeKnowledgeBaseConfig(employee) {
-    const { topK, score } = (employee == null ? void 0 : employee.knowledgeBase) ?? {};
-    return {
-      topK,
-      score
-    };
-  }
-  async getKnowledgeBaseGroup(employee) {
-    const { knowledgeBaseKeys } = (employee == null ? void 0 : employee.knowledgeBase) ?? {};
-    if (!knowledgeBaseKeys || import_lodash.default.isEmpty(knowledgeBaseKeys)) {
-      return [];
-    }
-    return await this.plugin.features.knowledgeBase.getKnowledgeBaseGroup(knowledgeBaseKeys);
-  }
   async getEmployee(username) {
     return await this.aiEmployeeRepo.findOne({ filter: { username } });
   }

package/dist/server/ai-employees/checkpoints/cleaner.d.ts

@@ -23,6 +23,8 @@ export declare class CheckpointCleaner {
     private getThreadIds;
     private get aiConversationsModel();
     private get aiMessagesModel();
+    private get aiMessagesMessageIdColumn();
+    private get isUnderscoredDatabase();
     private get sequelize();
     private get collectionManager();
 }

package/dist/server/ai-employees/checkpoints/cleaner.js

@@ -53,7 +53,10 @@ class CheckpointCleaner {
     }
     const sessionIds = outdatedConversations.map((conversation) => conversation.sessionId);
     const latestMessageRefs = await this.aiMessagesModel.findAll({
-      attributes: ["sessionId", [this.sequelize.fn("MAX", this.sequelize.col("messageId")), "messageId"]],
+      attributes: [
+        "sessionId",
+        [this.sequelize.fn("MAX", this.sequelize.col(this.aiMessagesMessageIdColumn)), "messageId"]
+      ],
       where: {
         sessionId: {
           [import_database.Op.in]: sessionIds
@@ -127,6 +130,13 @@ class CheckpointCleaner {
   get aiMessagesModel() {
     return this.collectionManager.getCollection("aiMessages").model;
   }
+  get aiMessagesMessageIdColumn() {
+    var _a;
+    return ((_a = this.aiMessagesModel.rawAttributes.messageId) == null ? void 0 : _a.field) || (this.isUnderscoredDatabase ? "message_id" : "messageId");
+  }
+  get isUnderscoredDatabase() {
+    return this.collectionManager.db.options.underscored === true || process.env.DB_UNDERSCORED === "true";
+  }
   get sequelize() {
     return this.collectionManager.db.sequelize;
   }

package/dist/server/ai-employees/middleware/conversation.d.ts

@@ -8,8 +8,10 @@
  */
 import { AIEmployee } from '../ai-employee';
 import z from 'zod';
+import { LLMProvider } from '../../llm-providers/provider';
 export declare const conversationMiddleware: (aiEmployee: AIEmployee, options: {
     providerName: string;
+    provider: LLMProvider;
     llmService?: string;
     model: string;
     messageId?: string;

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

@@ -43,10 +43,11 @@ var import_langchain = require("langchain");
 var import_zod = __toESM(require("zod"));
 var import_utils = require("../utils");
 const conversationMiddleware = (aiEmployee, options) => {
-  const { providerName, llmService, model, messageId, agentThread } = options;
+  const { providerName, provider, llmService, model, messageId, agentThread } = options;
   const convertAIMessage = (aiMessage) => (0, import_utils.convertAIMessage)({
     aiEmployee,
     providerName,
+    provider,
     llmService,
     model,
     aiMessage

package/dist/server/ai-employees/middleware/index.d.ts

@@ -8,5 +8,6 @@
  */
 export * from './conversation';
 export * from './skill-tools';
+export * from './tool-call-sanitizer';
 export * from './tools';
 export * from './workflow-history';

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

@@ -25,12 +25,14 @@ var middleware_exports = {};
 module.exports = __toCommonJS(middleware_exports);
 __reExport(middleware_exports, require("./conversation"), module.exports);
 __reExport(middleware_exports, require("./skill-tools"), module.exports);
+__reExport(middleware_exports, require("./tool-call-sanitizer"), module.exports);
 __reExport(middleware_exports, require("./tools"), module.exports);
 __reExport(middleware_exports, require("./workflow-history"), module.exports);
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   ...require("./conversation"),
   ...require("./skill-tools"),
+  ...require("./tool-call-sanitizer"),
   ...require("./tools"),
   ...require("./workflow-history")
 });

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

@@ -0,0 +1,16 @@
+/**
+ * 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.
+ */
+type ToolCallSanitizerLogger = {
+    warn: (message: string, meta?: Record<string, unknown>) => void;
+};
+type ToolCallSanitizerMiddlewareOptions = {
+    logger?: ToolCallSanitizerLogger;
+};
+export declare const toolCallSanitizerMiddleware: (options?: ToolCallSanitizerMiddlewareOptions) => import("langchain/dist/agents/middleware/types.cjs").AgentMiddleware<never, undefined, unknown, readonly (import("@langchain/core/dist/tools").ClientTool | import("@langchain/core/dist/tools").ServerTool)[]>;
+export {};