@cloudbase/agent-adapter-wx 1.0.1-alpha.25 → 1.0.1-alpha.25-2

package/README.md

@@ -151,7 +151,7 @@ new WeChatSender(config: WeChatConfig)
 ```typescript
 enum WeChatPlatform {
   CUSTOM_SERVICE = "WXCustomerService",
-  MINI_APP = "WXMiniApp",
+  MINI_APP = "WXMiniapp",
   SERVICE = "WXService",
   SUBSCRIPTION = "WXSubscription",
 }

package/dist/index.d.mts

@@ -1,7 +1,7 @@
 import { Request, Response } from 'express';
 import { AbstractAgent, AgentConfig, BaseEvent, RunAgentInput } from '@ag-ui/client';
 import { Observable } from 'rxjs';
-import { WxSendMessageInput } from '@cloudbase/aiagent-framework';
+import { WxSendMessageInput, aitools } from '@cloudbase/aiagent-framework';
 import { Logger } from '@cloudbase/agent-shared';
 
 /**
@@ -11,7 +11,7 @@ declare enum WeChatPlatform {
     /** 企业微信客服 */
     CUSTOM_SERVICE = "WXCustomerService",
     /** 微信小程序 */
-    MINI_APP = "WXMiniApp",
+    MINI_APP = "WXMiniapp",
     /** 微信服务号 */
     SERVICE = "WXService",
     /** 微信订阅号 */
@@ -127,7 +127,7 @@ interface WeChatWorkCommonInput {
 /**
  * Supported trigger sources - 支持的触发源
  */
-type WeChatTriggerSrc = "WXSubscription" | "WXService" | "WXMiniApp" | "WXCustomerService";
+type WeChatTriggerSrc = WeChatPlatform.SUBSCRIPTION | WeChatPlatform.SERVICE | WeChatPlatform.MINI_APP | WeChatPlatform.CUSTOM_SERVICE;
 /**
  * Supported message types - 支持的消息类型
  */
@@ -236,8 +236,8 @@ interface WeChatReplyMessage {
     createTime: number;
     msgType: string;
     content: string;
+    msgId: string;
     openKfId?: string;
-    msgId?: string;
 }
 /**
  * Format reply message for WeChat
@@ -465,13 +465,13 @@ interface IWxRunAgentInput extends RunAgentInput {
 declare class WeChatAgent extends AbstractAgent {
     private wrappedAgent;
     private wechatSender?;
-    private aitools?;
     private wechatConfig;
     private _historyManager;
     private messageFormatter;
     private eventFilter;
     private messageBuffer;
     private recommendQuestions?;
+    aitools?: aitools.AITools;
     constructor(config: WeChatAgentConfig);
     /**
      * Last assembled reply content after run() completes
@@ -568,15 +568,15 @@ interface AgentCreatorResult {
         toAGUIAgent(): AbstractAgent;
     };
     cleanup?: () => void;
-    aitools?: {
-        getWxMediaContent: (botId: string, triggerSrc: string, mediaId: string) => Promise<string | null>;
-    };
 }
 /**
  * Agent creator function type
  */
 type AgentCreator = (params: {
     request: Request;
+    options?: {
+        agentId: string;
+    };
 }) => AgentCreatorResult | Promise<AgentCreatorResult>;
 /**
  * Handler options
@@ -589,7 +589,7 @@ interface WxMessageHandlerOptions {
  * Directly using sendMessageAGUI.handler from @cloudbase/agent-server
  * Aligned with chat_wx_v2.service.ts logic
  *
- * @param createAgent - Function to create agent instance (returns { agent, cleanup?, aitools? })
+ * @param createAgent - Function to create agent instance (returns { agent, cleanup?})
  * @param options - Handler options
  * @returns Express route handler
  */

package/dist/index.d.ts

@@ -1,7 +1,7 @@
 import { Request, Response } from 'express';
 import { AbstractAgent, AgentConfig, BaseEvent, RunAgentInput } from '@ag-ui/client';
 import { Observable } from 'rxjs';
-import { WxSendMessageInput } from '@cloudbase/aiagent-framework';
+import { WxSendMessageInput, aitools } from '@cloudbase/aiagent-framework';
 import { Logger } from '@cloudbase/agent-shared';
 
 /**
@@ -11,7 +11,7 @@ declare enum WeChatPlatform {
     /** 企业微信客服 */
     CUSTOM_SERVICE = "WXCustomerService",
     /** 微信小程序 */
-    MINI_APP = "WXMiniApp",
+    MINI_APP = "WXMiniapp",
     /** 微信服务号 */
     SERVICE = "WXService",
     /** 微信订阅号 */
@@ -127,7 +127,7 @@ interface WeChatWorkCommonInput {
 /**
  * Supported trigger sources - 支持的触发源
  */
-type WeChatTriggerSrc = "WXSubscription" | "WXService" | "WXMiniApp" | "WXCustomerService";
+type WeChatTriggerSrc = WeChatPlatform.SUBSCRIPTION | WeChatPlatform.SERVICE | WeChatPlatform.MINI_APP | WeChatPlatform.CUSTOM_SERVICE;
 /**
  * Supported message types - 支持的消息类型
  */
@@ -236,8 +236,8 @@ interface WeChatReplyMessage {
     createTime: number;
     msgType: string;
     content: string;
+    msgId: string;
     openKfId?: string;
-    msgId?: string;
 }
 /**
  * Format reply message for WeChat
@@ -465,13 +465,13 @@ interface IWxRunAgentInput extends RunAgentInput {
 declare class WeChatAgent extends AbstractAgent {
     private wrappedAgent;
     private wechatSender?;
-    private aitools?;
     private wechatConfig;
     private _historyManager;
     private messageFormatter;
     private eventFilter;
     private messageBuffer;
     private recommendQuestions?;
+    aitools?: aitools.AITools;
     constructor(config: WeChatAgentConfig);
     /**
      * Last assembled reply content after run() completes
@@ -568,15 +568,15 @@ interface AgentCreatorResult {
         toAGUIAgent(): AbstractAgent;
     };
     cleanup?: () => void;
-    aitools?: {
-        getWxMediaContent: (botId: string, triggerSrc: string, mediaId: string) => Promise<string | null>;
-    };
 }
 /**
  * Agent creator function type
  */
 type AgentCreator = (params: {
     request: Request;
+    options?: {
+        agentId: string;
+    };
 }) => AgentCreatorResult | Promise<AgentCreatorResult>;
 /**
  * Handler options
@@ -589,7 +589,7 @@ interface WxMessageHandlerOptions {
  * Directly using sendMessageAGUI.handler from @cloudbase/agent-server
  * Aligned with chat_wx_v2.service.ts logic
  *
- * @param createAgent - Function to create agent instance (returns { agent, cleanup?, aitools? })
+ * @param createAgent - Function to create agent instance (returns { agent, cleanup?})
  * @param options - Handler options
  * @returns Express route handler
  */

package/dist/index.js

@@ -61,6 +61,27 @@ __export(index_exports, {
 });
 module.exports = __toCommonJS(index_exports);
 
+// src/types.ts
+var WeChatPlatform = /* @__PURE__ */ ((WeChatPlatform2) => {
+  WeChatPlatform2["CUSTOM_SERVICE"] = "WXCustomerService";
+  WeChatPlatform2["MINI_APP"] = "WXMiniapp";
+  WeChatPlatform2["SERVICE"] = "WXService";
+  WeChatPlatform2["SUBSCRIPTION"] = "WXSubscription";
+  return WeChatPlatform2;
+})(WeChatPlatform || {});
+var MessageType = /* @__PURE__ */ ((MessageType2) => {
+  MessageType2["TEXT"] = "text";
+  MessageType2["IMAGE"] = "image";
+  MessageType2["EVENT"] = "event";
+  MessageType2["VOICE"] = "voice";
+  return MessageType2;
+})(MessageType || {});
+var WeChatSendMode = /* @__PURE__ */ ((WeChatSendMode2) => {
+  WeChatSendMode2["LOCAL"] = "local";
+  WeChatSendMode2["AITOOLS"] = "aitools";
+  return WeChatSendMode2;
+})(WeChatSendMode || {});
+
 // src/wechat-message-handler.ts
 var SUPPORTED_MSG_TYPES = ["text", "voice"];
 function validateMessageType(msgType) {
@@ -77,21 +98,25 @@ function validateMessageType(msgType) {
 function extractMsgType(originMsg) {
   return originMsg.msgType || originMsg.MsgType || "text";
 }
-function extractTriggerSrc(originMsg, defaultSrc = "WXCustomerService") {
+function extractTriggerSrc(originMsg, defaultSrc = "WXService" /* SERVICE */) {
   return originMsg.triggerSrc || defaultSrc;
 }
 function extractTextContent(originMsg, triggerSrc) {
-  if (["WXSubscription", "WXService", "WXMiniApp"].includes(triggerSrc)) {
+  if ([
+    "WXSubscription" /* SUBSCRIPTION */,
+    "WXService" /* SERVICE */,
+    "WXMiniapp" /* MINI_APP */
+  ].includes(triggerSrc)) {
     return originMsg.Content || originMsg.content || "";
-  } else if (triggerSrc === "WXCustomerService") {
+  } else if (triggerSrc === "WXCustomerService" /* CUSTOM_SERVICE */) {
     return originMsg.text?.content || "";
   }
   return "";
 }
 function extractVoiceMediaId(originMsg, triggerSrc) {
-  if (["WXSubscription", "WXService"].includes(triggerSrc)) {
+  if (["WXSubscription" /* SUBSCRIPTION */, "WXService" /* SERVICE */].includes(triggerSrc)) {
     return originMsg.MediaId || originMsg.mediaId || "";
-  } else if (triggerSrc === "WXCustomerService") {
+  } else if (triggerSrc === "WXCustomerService" /* CUSTOM_SERVICE */) {
     return originMsg.voice?.mediaId || "";
   }
   return "";
@@ -111,17 +136,25 @@ async function getWxChatContent(originMsg, triggerSrc, voiceMessageHandler, botI
   return "";
 }
 function extractSender(originMsg, triggerSrc) {
-  if (["WXSubscription", "WXService", "WXMiniApp"].includes(triggerSrc)) {
+  if ([
+    "WXSubscription" /* SUBSCRIPTION */,
+    "WXService" /* SERVICE */,
+    "WXMiniapp" /* MINI_APP */
+  ].includes(triggerSrc)) {
     return originMsg.FromUserName || originMsg.fromUserName || "";
-  } else if (triggerSrc === "WXCustomerService") {
+  } else if (triggerSrc === "WXCustomerService" /* CUSTOM_SERVICE */) {
     return originMsg.externalUserId || "";
   }
   return "";
 }
 function extractConversation(originMsg, triggerSrc) {
-  if (["WXSubscription", "WXService", "WXMiniApp"].includes(triggerSrc)) {
+  if ([
+    "WXSubscription" /* SUBSCRIPTION */,
+    "WXService" /* SERVICE */,
+    "WXMiniapp" /* MINI_APP */
+  ].includes(triggerSrc)) {
     return originMsg.FromUserName || originMsg.fromUserName || "";
-  } else if (triggerSrc === "WXCustomerService") {
+  } else if (triggerSrc === "WXCustomerService" /* CUSTOM_SERVICE */) {
     return originMsg.externalUserId || "";
   }
   return "";
@@ -133,13 +166,15 @@ function extractMsgId(originMsg) {
   return originMsg.msgId || originMsg.MsgId || Date.now().toString();
 }
 function extractTimestamp(originMsg, triggerSrc) {
-  if (triggerSrc === "WXCustomerService") {
+  if (triggerSrc === "WXCustomerService" /* CUSTOM_SERVICE */) {
     return originMsg.sendTime || Math.floor(Date.now() / 1e3);
   }
   return originMsg.createTime || originMsg.CreateTime || Math.floor(Date.now() / 1e3);
 }
 function needAsyncReply(triggerSrc, wxVerify) {
-  if (["WXMiniApp", "WXCustomerService"].includes(triggerSrc)) {
+  if (["WXMiniapp" /* MINI_APP */, "WXCustomerService" /* CUSTOM_SERVICE */].includes(
+    triggerSrc
+  )) {
     return true;
   }
   return wxVerify;
@@ -147,14 +182,15 @@ function needAsyncReply(triggerSrc, wxVerify) {
 function formatWeChatReplyMessage(callbackData, triggerSrc, content) {
   const createTime = Math.floor(Date.now() / 1e3);
   const finalContent = content || "\u62B1\u6B49\u6682\u65F6\u65E0\u6CD5\u5904\u7406\u8FD9\u4E2A\u7C7B\u578B\u7684\u6D88\u606F";
-  if (triggerSrc === "WXCustomerService") {
+  if (triggerSrc === "WXCustomerService" /* CUSTOM_SERVICE */) {
     return {
       toUserName: callbackData.externalUserId || callbackData.fromUserName,
       fromUserName: callbackData.openKfId,
       openKfId: callbackData.openKfId,
       createTime,
       msgType: "text",
-      content: finalContent
+      content: finalContent,
+      msgId: callbackData.msgId
     };
   }
   return {
@@ -162,7 +198,8 @@ function formatWeChatReplyMessage(callbackData, triggerSrc, content) {
     fromUserName: callbackData.toUserName || callbackData.ToUserName,
     createTime,
     msgType: "text",
-    content: finalContent
+    content: finalContent,
+    msgId: callbackData.msgId
   };
 }
 function dealMsgData(originMsg, triggerSrc, wxVerify, botId) {
@@ -242,20 +279,57 @@ var import_agent_server = require("@cloudbase/agent-server");
 var import_agent_shared = require("@cloudbase/agent-shared");
 var import_client = require("@ag-ui/client");
 var import_agent_shared2 = require("@cloudbase/agent-shared");
+var import_aiagent_framework = require("@cloudbase/aiagent-framework");
+function getServiceNameFromHostname(hostname) {
+  const parts = hostname.split("-");
+  let versionIdx = -1;
+  for (let i = parts.length - 1; i >= 0; i--) {
+    if (/^\d+$/.test(parts[i])) {
+      versionIdx = i;
+      break;
+    }
+  }
+  return versionIdx > 0 ? parts.slice(0, versionIdx).join("-") : hostname;
+}
+function getBotId(url, fallback = "agent-id") {
+  if (process.env.SCF_FUNCTIONNAME) {
+    return process.env.SCF_FUNCTIONNAME;
+  }
+  if (url) {
+    const parsedBotId = import_aiagent_framework.utils.parseBotId(url);
+    if (parsedBotId) {
+      return parsedBotId;
+    }
+  }
+  if (process.env.HOSTNAME) {
+    return getServiceNameFromHostname(process.env.HOSTNAME);
+  }
+  return fallback;
+}
 function createWxMessageHandler(createAgent, options) {
   const { logger: parentLogger = import_agent_shared.noopLogger } = options ?? {};
   const adapterLogger = parentLogger.child?.({ component: "sendWXMessageAGUI" }) ?? parentLogger;
   return async (req, res) => {
     let cleanup = null;
+    let agent = void 0;
+    const { callbackData } = import_aiagent_framework.utils.transformKeysToCamelCase(req.body);
+    const { triggerSrc, wxVerify } = import_aiagent_framework.utils.parseWxEnvParam({
+      httpContext: {
+        headers: req.headers
+      }
+    });
+    const wxSendmessageInput = {
+      callbackData,
+      triggerSrc,
+      wxVerify
+    };
+    const requestId = (0, import_uuid.v4)();
+    const logger = adapterLogger.child?.({ requestId }) ?? adapterLogger;
+    const botId = getBotId(
+      `${req.protocol}://${req.get("host")}${req.originalUrl}`,
+      "agent-wx-send-message"
+    );
     try {
-      const {
-        callbackData,
-        triggerSrc,
-        wxVerify = false,
-        botId = "default"
-      } = req.body;
-      const requestId = (0, import_uuid.v4)();
-      const logger = adapterLogger.child?.({ requestId }) ?? adapterLogger;
       if (!callbackData || !triggerSrc) {
         return res.status(400).json({
           error: "Bad Request",
@@ -263,9 +337,11 @@ function createWxMessageHandler(createAgent, options) {
         });
       }
       logger.info?.("[WX] Received message:", { triggerSrc, wxVerify, botId });
-      const { agent: unknownAgent, cleanup: agentCleanup } = await Promise.resolve(createAgent({ request: req }));
+      const { agent: unknownAgent, cleanup: agentCleanup } = await Promise.resolve(
+        createAgent({ request: req, options: { agentId: botId } })
+      );
       cleanup = agentCleanup ?? null;
-      const agent = "toAGUIAgent" in unknownAgent ? unknownAgent.toAGUIAgent() : unknownAgent;
+      agent = "toAGUIAgent" in unknownAgent ? unknownAgent.toAGUIAgent() : unknownAgent;
       const { msgData, replyMsgData } = dealMsgData(
         callbackData,
         triggerSrc,
@@ -286,17 +362,12 @@ function createWxMessageHandler(createAgent, options) {
       );
       const voiceHandler = async (vBotId, vTriggerSrc, mediaId) => {
         try {
-          const content2 = await agent?.aitools?.getWxMediaContent(
+          const result = await agent?.aitools?.getWxMediaContent(
             vBotId,
             vTriggerSrc,
             mediaId
           );
-          if (content2) {
-            return {
-              content: content2
-            };
-          }
-          return null;
+          return result || null;
         } catch {
           return null;
         }
@@ -353,7 +424,7 @@ function createWxMessageHandler(createAgent, options) {
         tools: [],
         context: [],
         forwardedProps: {
-          wxSendmessageInput: req.body,
+          wxSendmessageInput,
           // Pass full msgData for history storage
           msgData,
           // Pass replay info with replyMsgData for assistant history
@@ -381,10 +452,34 @@ function createWxMessageHandler(createAgent, options) {
         formatWeChatReplyMessage(callbackData, triggerSrc, finalContent)
       );
     } catch (error) {
-      console.error("[WX] Error processing message:", error);
+      logger.error?.("[WX] Error processing message:", error);
+      const message = error instanceof Error ? error.message : String(error);
+      try {
+        if (agent?.aitools?.sendWxClientMessage) {
+          const data = formatWeChatReplyMessage(
+            callbackData,
+            triggerSrc,
+            message
+          );
+          await agent.aitools.sendWxClientMessage(
+            agent.agentId || botId,
+            triggerSrc,
+            {
+              msgType: data.msgType,
+              touser: data.toUserName,
+              text: {
+                content: data.content
+              },
+              openKfId: data.openKfId,
+              msgId: data.msgId
+            }
+          );
+        }
+      } catch (e) {
+      }
       return res.status(500).json({
         error: "Internal Server Error",
-        message: error instanceof Error ? error.message : String(error)
+        message
       });
     } finally {
       if (cleanup) cleanup();
@@ -392,27 +487,6 @@ function createWxMessageHandler(createAgent, options) {
   };
 }
 
-// src/types.ts
-var WeChatPlatform = /* @__PURE__ */ ((WeChatPlatform2) => {
-  WeChatPlatform2["CUSTOM_SERVICE"] = "WXCustomerService";
-  WeChatPlatform2["MINI_APP"] = "WXMiniApp";
-  WeChatPlatform2["SERVICE"] = "WXService";
-  WeChatPlatform2["SUBSCRIPTION"] = "WXSubscription";
-  return WeChatPlatform2;
-})(WeChatPlatform || {});
-var MessageType = /* @__PURE__ */ ((MessageType2) => {
-  MessageType2["TEXT"] = "text";
-  MessageType2["IMAGE"] = "image";
-  MessageType2["EVENT"] = "event";
-  MessageType2["VOICE"] = "voice";
-  return MessageType2;
-})(MessageType || {});
-var WeChatSendMode = /* @__PURE__ */ ((WeChatSendMode2) => {
-  WeChatSendMode2["LOCAL"] = "local";
-  WeChatSendMode2["AITOOLS"] = "aitools";
-  return WeChatSendMode2;
-})(WeChatSendMode || {});
-
 // src/token-manager.ts
 var TokenManager = class {
   constructor() {
@@ -560,7 +634,7 @@ var PlatformRouter = class {
     switch (config.platform) {
       case "WXCustomerService" /* CUSTOM_SERVICE */:
         return this.sendToCustomService(message, accessToken, config);
-      case "WXMiniApp" /* MINI_APP */:
+      case "WXMiniapp" /* MINI_APP */:
         return this.sendToMiniApp(message, accessToken);
       case "WXService" /* SERVICE */:
       case "WXSubscription" /* SUBSCRIPTION */:
@@ -744,7 +818,7 @@ var WeChatHistoryManager = class _WeChatHistoryManager {
       await db.createCollection(this.collectionName);
       this.collectionReady = true;
     } catch (error) {
-      if (error?.code === -502005 || error?.message?.includes("already exists")) {
+      if (error?.code === "DATABASE_COLLECTION_ALREADY_EXIST") {
         this.collectionReady = true;
       } else {
         console.error(
@@ -808,9 +882,9 @@ var WeChatHistoryManager = class _WeChatHistoryManager {
 var getHistoryManager = WeChatHistoryManager.getInstance;
 
 // src/agent.ts
-var import_aiagent_framework = require("@cloudbase/aiagent-framework");
+var import_aiagent_framework2 = require("@cloudbase/aiagent-framework");
 var import_crypto = require("crypto");
-var AITOOLS = import_aiagent_framework.aitools.AITools;
+var AITOOLS = import_aiagent_framework2.aitools.AITools;
 var WeChatAgent = class extends import_client2.AbstractAgent {
   constructor(config) {
     super({
@@ -1039,7 +1113,7 @@ var WeChatAgent = class extends import_client2.AbstractAgent {
     try {
       const needAsyncReply2 = input.forwardedProps?.replay.needAsyncReply ?? true;
       if (needAsyncReply2) {
-        const sendMode = this.wechatConfig.sendMode || "local" /* LOCAL */;
+        const sendMode = this.wechatConfig.sendMode || "aitools" /* AITOOLS */;
         if (sendMode === "aitools" /* AITOOLS */) {
           await this.sendViaAITools(content, input);
         } else {
@@ -1104,15 +1178,19 @@ var WeChatAgent = class extends import_client2.AbstractAgent {
       return;
     }
     const triggerSrc = wxSendmessageInput.triggerSrc || this.wechatConfig.platform;
-    await this.aitools.sendMessageToClient(triggerSrc, {
-      msgType: toWxMsgData.msgType,
-      touser: toWxMsgData.toUserName,
-      text: {
-        content: toWxMsgData.content
-      },
-      openKfId: toWxMsgData.openKfId,
-      msgId: toWxMsgData.msgId
-    });
+    await this.aitools.sendWxClientMessage(
+      this.agentId || "agent",
+      triggerSrc,
+      {
+        msgType: toWxMsgData.msgType,
+        touser: toWxMsgData.toUserName,
+        text: {
+          content: toWxMsgData.content
+        },
+        openKfId: toWxMsgData.openKfId,
+        msgId: toWxMsgData.msgId || ""
+      }
+    );
   }
   /**
    * Format reply message for WeChat HTTP response (sync mode)