@ynhcj/xiaoyi-channel 0.0.6 → 0.0.7-beta

package/dist/src/bot.js

@@ -1,10 +1,11 @@
 import { getXYRuntime } from "./runtime.js";
 import { createXYReplyDispatcher } from "./reply-dispatcher.js";
-import { parseA2AMessage, extractTextFromParts, extractFileParts } from "./parser.js";
+import { parseA2AMessage, extractTextFromParts, extractFileParts, extractPushId } from "./parser.js";
 import { downloadFilesFromParts } from "./file-download.js";
 import { resolveXYConfig } from "./config.js";
 import { sendStatusUpdate, sendClearContextResponse, sendTasksCancelResponse } from "./formatter.js";
 import { registerSession, unregisterSession } from "./tools/session-manager.js";
+import { configManager } from "./utils/config-manager.js";
 /**
  * Handle an incoming A2A message.
  * This is the main entry point for message processing.
@@ -19,7 +20,8 @@ export async function handleXYMessage(params) {
     try {
         // Check for special messages BEFORE parsing (these have different param structures)
         const messageMethod = message.method;
-        log(`[DEBUG] Received message with method: ${messageMethod}, id: ${message.id}`);
+        log(`[BOT-ENTRY] <<<<<<< Received message with method: ${messageMethod}, id: ${message.id} >>>>>>>`);
+        log(`[BOT-ENTRY] Stack trace for debugging:`, new Error().stack?.split('\n').slice(1, 4).join('\n'));
         // Handle clearContext messages (params only has sessionId)
         if (messageMethod === "clearContext" || messageMethod === "clear_context") {
             const sessionId = message.params?.sessionId;
@@ -54,6 +56,18 @@ export async function handleXYMessage(params) {
         }
         // Parse the A2A message (for regular messages)
         const parsed = parseA2AMessage(message);
+        // Extract and update push_id if present
+        const pushId = extractPushId(parsed.parts);
+        if (pushId) {
+            log(`[BOT] 📌 Extracted push_id from user message`);
+            log(`[BOT]   - Session ID: ${parsed.sessionId}`);
+            log(`[BOT]   - Push ID preview: ${pushId.substring(0, 20)}...`);
+            log(`[BOT]   - Full push_id: ${pushId}`);
+            configManager.updatePushId(parsed.sessionId, pushId);
+        }
+        else {
+            log(`[BOT] ℹ️  No push_id found in message, will use config default`);
+        }
         // Resolve configuration (needed for status updates)
         const config = resolveXYConfig(cfg);
         // ✅ Resolve agent route (following feishu pattern)
@@ -61,7 +75,7 @@ export async function handleXYMessage(params) {
         // Use sessionId as peer.id to ensure all messages in the same session share context
         let route = core.channel.routing.resolveAgentRoute({
             cfg,
-            channel: "xy",
+            channel: "xiaoyi-channel",
             accountId, // "default"
             peer: {
                 kind: "direct",
@@ -82,6 +96,18 @@ export async function handleXYMessage(params) {
             agentId: route.accountId,
         });
         log(`[BOT] ✅ Session registered for tools`);
+        // Send initial status update immediately after parsing message
+        log(`[STATUS] Sending initial status update for session ${parsed.sessionId}`);
+        void sendStatusUpdate({
+            config,
+            sessionId: parsed.sessionId,
+            taskId: parsed.taskId,
+            messageId: parsed.messageId,
+            text: "任务正在处理中，请稍后~",
+            state: "working",
+        }).catch((err) => {
+            error(`Failed to send initial status update:`, err);
+        });
         // Extract text and files from parts
         const text = extractTextFromParts(parsed.parts);
         const fileParts = extractFileParts(parsed.parts);
@@ -98,7 +124,7 @@ export async function handleXYMessage(params) {
         messageBody = `${speaker}: ${messageBody}`;
         // Format agent envelope (following feishu pattern)
         const body = core.channel.reply.formatAgentEnvelope({
-            channel: "XY",
+            channel: "xiaoyi-channel",
             from: speaker,
             timestamp: new Date(),
             envelope: envelopeOptions,
@@ -118,13 +144,13 @@ export async function handleXYMessage(params) {
             GroupSubject: undefined,
             SenderName: parsed.sessionId,
             SenderId: parsed.sessionId,
-            Provider: "xy",
-            Surface: "xy",
+            Provider: "xiaoyi-channel",
+            Surface: "xiaoyi-channel",
             MessageSid: parsed.messageId,
             Timestamp: Date.now(),
             WasMentioned: false,
             CommandAuthorized: true,
-            OriginatingChannel: "xy",
+            OriginatingChannel: "xiaoyi-channel",
             OriginatingTo: parsed.sessionId, // Original message target
             ReplyToBody: undefined, // A2A protocol doesn't support reply/quote
             ...mediaPayload,
@@ -142,6 +168,7 @@ export async function handleXYMessage(params) {
             error(`Failed to send initial status update:`, err);
         });
         // Create reply dispatcher (following feishu pattern)
+        log(`[BOT-DISPATCHER] 🎯 Creating reply dispatcher for session=${parsed.sessionId}, taskId=${parsed.taskId}, messageId=${parsed.messageId}`);
         const { dispatcher, replyOptions, markDispatchIdle, startStatusInterval } = createXYReplyDispatcher({
             cfg,
             runtime,
@@ -150,6 +177,7 @@ export async function handleXYMessage(params) {
             messageId: parsed.messageId,
             accountId: route.accountId, // ✅ Use route.accountId
         });
+        log(`[BOT-DISPATCHER] ✅ Reply dispatcher created successfully`);
         // Start status update interval (will send updates every 60 seconds)
         // Interval will be automatically stopped when onIdle/onCleanup is triggered
         startStatusInterval();
@@ -177,6 +205,7 @@ export async function handleXYMessage(params) {
         log(`xy: dispatch complete (session=${parsed.sessionId})`);
     }
     catch (err) {
+        // ✅ Only log error, don't re-throw to prevent gateway restart
         error("Failed to handle XY message:", err);
         runtime.error?.(`xy: Failed to handle message: ${String(err)}`);
         log(`[BOT] ❌ Error occurred, attempting cleanup...`);
@@ -189,7 +218,7 @@ export async function handleXYMessage(params) {
                 log(`[BOT] 🧹 Cleaning up session after error: ${sessionId}`);
                 const route = core.channel.routing.resolveAgentRoute({
                     cfg,
-                    channel: "xy",
+                    channel: "xiaoyi-channel",
                     accountId,
                     peer: {
                         kind: "direct",
@@ -205,7 +234,7 @@ export async function handleXYMessage(params) {
             log(`[BOT] ⚠️  Cleanup failed:`, cleanupErr);
             // Ignore cleanup errors
         }
-        throw err;
+        // ❌ Don't re-throw: message processing error should not affect gateway stability
     }
 }
 /**

package/dist/src/channel.js

@@ -5,6 +5,7 @@ import { xyOnboardingAdapter } from "./onboarding.js";
 import { locationTool } from "./tools/location-tool.js";
 import { noteTool } from "./tools/note-tool.js";
 import { searchNoteTool } from "./tools/search-note-tool.js";
+import { calendarTool } from "./tools/calendar-tool.js";
 /**
  * Xiaoyi Channel Plugin for OpenClaw.
  * Implements Xiaoyi A2A protocol with dual WebSocket connections.
@@ -22,6 +23,7 @@ export const xyPlugin = {
     agentPrompt: {
         messageToolHints: () => [
             "- xiaoyi targeting: omit `target` to reply to the current conversation (auto-inferred). Explicit targets: `default`",
+            "- If the user requests a file, you can call the message tool with the xiaoyi-channel channel to return it. Note: sendMedia requires a text reply."
         ],
     },
     capabilities: {
@@ -43,7 +45,7 @@ export const xyPlugin = {
     },
     outbound: xyOutbound,
     onboarding: xyOnboardingAdapter,
-    agentTools: [locationTool, noteTool, searchNoteTool],
+    agentTools: [locationTool, noteTool, searchNoteTool, calendarTool],
     messaging: {
         normalizeTarget: (raw) => {
             const trimmed = raw.trim();
@@ -79,6 +81,7 @@ export const xyPlugin = {
                 runtime: context.runtime,
                 abortSignal: context.abortSignal,
                 accountId: context.accountId,
+                setStatus: context.setStatus,
             });
         },
     },

package/dist/src/client.d.ts

@@ -10,6 +10,11 @@ export declare function setClientRuntime(rt: RuntimeEnv | undefined): void;
  * Reuses existing managers if config matches.
  */
 export declare function getXYWebSocketManager(config: XYChannelConfig): XYWebSocketManager;
+/**
+ * Remove a specific WebSocket manager from cache.
+ * Disconnects the manager and removes it from the cache.
+ */
+export declare function removeXYWebSocketManager(config: XYChannelConfig): void;
 /**
  * Clear all cached WebSocket managers.
  */
@@ -18,3 +23,13 @@ export declare function clearXYWebSocketManagers(): void;
  * Get the number of cached managers.
  */
 export declare function getCachedManagerCount(): number;
+/**
+ * Diagnose all cached WebSocket managers.
+ * Helps identify connection issues and orphan connections.
+ */
+export declare function diagnoseAllManagers(): void;
+/**
+ * Clean up orphan connections across all managers.
+ * Returns the number of managers that had orphan connections.
+ */
+export declare function cleanupOrphanConnections(): number;

package/dist/src/client.js

@@ -23,16 +23,34 @@ export function getXYWebSocketManager(config) {
     let cached = wsManagerCache.get(cacheKey);
     if (cached && cached.isConfigMatch(config)) {
         const log = runtime?.log ?? console.log;
-        log(`[DEBUG] Reusing cached WebSocket manager: ${cacheKey}`);
+        log(`[WS-MANAGER-CACHE] ✅ Reusing cached WebSocket manager: ${cacheKey}, total managers: ${wsManagerCache.size}`);
         return cached;
     }
     // Create new manager
     const log = runtime?.log ?? console.log;
-    log(`Creating new WebSocket manager: ${cacheKey}`);
+    log(`[WS-MANAGER-CACHE] 🆕 Creating new WebSocket manager: ${cacheKey}, total managers before: ${wsManagerCache.size}`);
     cached = new XYWebSocketManager(config, runtime);
     wsManagerCache.set(cacheKey, cached);
+    log(`[WS-MANAGER-CACHE] 📊 Total managers after creation: ${wsManagerCache.size}`);
     return cached;
 }
+/**
+ * Remove a specific WebSocket manager from cache.
+ * Disconnects the manager and removes it from the cache.
+ */
+export function removeXYWebSocketManager(config) {
+    const cacheKey = `${config.apiKey}-${config.agentId}`;
+    const manager = wsManagerCache.get(cacheKey);
+    if (manager) {
+        console.log(`🗑️  [WS-MANAGER-CACHE] Removing manager from cache: ${cacheKey}`);
+        manager.disconnect();
+        wsManagerCache.delete(cacheKey);
+        console.log(`🗑️  [WS-MANAGER-CACHE] Manager removed, remaining managers: ${wsManagerCache.size}`);
+    }
+    else {
+        console.log(`⚠️  [WS-MANAGER-CACHE] Manager not found in cache: ${cacheKey}`);
+    }
+}
 /**
  * Clear all cached WebSocket managers.
  */
@@ -50,3 +68,80 @@ export function clearXYWebSocketManagers() {
 export function getCachedManagerCount() {
     return wsManagerCache.size;
 }
+/**
+ * Diagnose all cached WebSocket managers.
+ * Helps identify connection issues and orphan connections.
+ */
+export function diagnoseAllManagers() {
+    console.log("========================================");
+    console.log("📊 WebSocket Manager Global Diagnostics");
+    console.log("========================================");
+    console.log(`Total cached managers: ${wsManagerCache.size}`);
+    console.log("");
+    if (wsManagerCache.size === 0) {
+        console.log("ℹ️  No managers in cache");
+        console.log("========================================");
+        return;
+    }
+    let orphanCount = 0;
+    wsManagerCache.forEach((manager, key) => {
+        const diag = manager.getConnectionDiagnostics();
+        console.log(`📌 Manager: ${key}`);
+        console.log(`   Shutting down: ${diag.isShuttingDown}`);
+        console.log(`   Total event listeners on manager: ${diag.totalEventListeners}`);
+        // Server 1
+        console.log(`   🔌 Server1:`);
+        console.log(`      - Exists: ${diag.server1.exists}`);
+        console.log(`      - ReadyState: ${diag.server1.readyState}`);
+        console.log(`      - State connected/ready: ${diag.server1.stateConnected}/${diag.server1.stateReady}`);
+        console.log(`      - Reconnect attempts: ${diag.server1.reconnectAttempts}`);
+        console.log(`      - Listeners on WebSocket: ${diag.server1.listenerCount}`);
+        console.log(`      - Heartbeat active: ${diag.server1.heartbeatActive}`);
+        console.log(`      - Has reconnect timer: ${diag.server1.hasReconnectTimer}`);
+        if (diag.server1.isOrphan) {
+            console.log(`      ⚠️  ORPHAN CONNECTION DETECTED!`);
+            orphanCount++;
+        }
+        // Server 2
+        console.log(`   🔌 Server2:`);
+        console.log(`      - Exists: ${diag.server2.exists}`);
+        console.log(`      - ReadyState: ${diag.server2.readyState}`);
+        console.log(`      - State connected/ready: ${diag.server2.stateConnected}/${diag.server2.stateReady}`);
+        console.log(`      - Reconnect attempts: ${diag.server2.reconnectAttempts}`);
+        console.log(`      - Listeners on WebSocket: ${diag.server2.listenerCount}`);
+        console.log(`      - Heartbeat active: ${diag.server2.heartbeatActive}`);
+        console.log(`      - Has reconnect timer: ${diag.server2.hasReconnectTimer}`);
+        if (diag.server2.isOrphan) {
+            console.log(`      ⚠️  ORPHAN CONNECTION DETECTED!`);
+            orphanCount++;
+        }
+        console.log("");
+    });
+    if (orphanCount > 0) {
+        console.log(`⚠️  Total orphan connections found: ${orphanCount}`);
+        console.log(`💡 Suggestion: These connections should be cleaned up`);
+    }
+    else {
+        console.log(`✅ No orphan connections found`);
+    }
+    console.log("========================================");
+}
+/**
+ * Clean up orphan connections across all managers.
+ * Returns the number of managers that had orphan connections.
+ */
+export function cleanupOrphanConnections() {
+    let cleanedCount = 0;
+    wsManagerCache.forEach((manager, key) => {
+        const diag = manager.getConnectionDiagnostics();
+        if (diag.server1.isOrphan || diag.server2.isOrphan) {
+            console.log(`🧹 Cleaning up orphan connections in manager: ${key}`);
+            manager.disconnect();
+            cleanedCount++;
+        }
+    });
+    if (cleanedCount > 0) {
+        console.log(`🧹 Cleaned up ${cleanedCount} manager(s) with orphan connections`);
+    }
+    return cleanedCount;
+}

package/dist/src/file-download.js

@@ -8,8 +8,10 @@ import { logger } from "./utils/logger.js";
  */
 export async function downloadFile(url, destPath) {
     logger.debug(`Downloading file from ${url} to ${destPath}`);
+    const controller = new AbortController();
+    const timeout = setTimeout(() => controller.abort(), 30000); // 30 seconds timeout
     try {
-        const response = await fetch(url);
+        const response = await fetch(url, { signal: controller.signal });
         if (!response.ok) {
             throw new Error(`HTTP ${response.status}: ${response.statusText}`);
         }
@@ -19,9 +21,16 @@ export async function downloadFile(url, destPath) {
         logger.debug(`File downloaded successfully: ${destPath}`);
     }
     catch (error) {
+        if (error.name === 'AbortError') {
+            logger.error(`Download timeout (30s) for ${url}`);
+            throw new Error(`Download timeout after 30 seconds`);
+        }
         logger.error(`Failed to download file from ${url}:`, error);
         throw error;
     }
+    finally {
+        clearTimeout(timeout);
+    }
 }
 /**
  * Download files from A2A file parts.

package/dist/src/file-upload.js

@@ -48,7 +48,7 @@ export class XYFileUploadService {
                         uid: this.uid,
                         teamId: this.uid,
                     },
-                    useEdge: true,
+                    useEdge: false,
                 }),
             });
             if (!prepareResp.ok) {

package/dist/src/formatter.d.ts

@@ -20,6 +20,23 @@ export interface SendA2AResponseParams {
  * Send an A2A artifact update response.
  */
 export declare function sendA2AResponse(params: SendA2AResponseParams): Promise<void>;
+/**
+ * Parameters for sending a reasoning text update (intermediate, streamed).
+ */
+export interface SendReasoningTextUpdateParams {
+    config: XYChannelConfig;
+    sessionId: string;
+    taskId: string;
+    messageId: string;
+    text: string;
+    append?: boolean;
+}
+/**
+ * Send an A2A artifact-update with reasoningText part.
+ * Used for onToolStart, onToolResult, onReasoningStream, onReasoningEnd, onPartialReply.
+ * append=true, final=false, lastChunk=true, text is suffixed with newline for markdown rendering.
+ */
+export declare function sendReasoningTextUpdate(params: SendReasoningTextUpdateParams): Promise<void>;
 /**
  * Parameters for sending a status update.
  */

package/dist/src/formatter.js

@@ -67,6 +67,50 @@ export async function sendA2AResponse(params) {
     await wsManager.sendMessage(sessionId, outboundMessage);
     log(`[A2A_RESPONSE] ✅ Message sent successfully`);
 }
+/**
+ * Send an A2A artifact-update with reasoningText part.
+ * Used for onToolStart, onToolResult, onReasoningStream, onReasoningEnd, onPartialReply.
+ * append=true, final=false, lastChunk=true, text is suffixed with newline for markdown rendering.
+ */
+export async function sendReasoningTextUpdate(params) {
+    const { config, sessionId, taskId, messageId, text, append = true } = params;
+    const runtime = getXYRuntime();
+    const log = runtime?.log ?? console.log;
+    const error = runtime?.error ?? console.error;
+    const artifact = {
+        taskId,
+        kind: "artifact-update",
+        append,
+        lastChunk: true,
+        final: false,
+        artifact: {
+            artifactId: uuidv4(),
+            parts: [
+                {
+                    kind: "reasoningText",
+                    reasoningText: text,
+                },
+            ],
+        },
+    };
+    const jsonRpcResponse = {
+        jsonrpc: "2.0",
+        id: messageId,
+        result: artifact,
+    };
+    const wsManager = getXYWebSocketManager(config);
+    const outboundMessage = {
+        msgType: "agent_response",
+        agentId: config.agentId,
+        sessionId,
+        taskId,
+        msgDetail: JSON.stringify(jsonRpcResponse),
+    };
+    log(`[REASONING_TEXT] 📤 Sending reasoningText update: sessionId=${sessionId}, taskId=${taskId}, text.length=${text.length}`);
+    log(JSON.stringify(outboundMessage, null, 2));
+    await wsManager.sendMessage(sessionId, outboundMessage);
+    log(`[REASONING_TEXT] ✅ Sent successfully`);
+}
 /**
  * Send an A2A task status update.
  * Follows A2A protocol standard format with nested status object.
@@ -132,6 +176,7 @@ export async function sendCommand(params) {
     const log = runtime?.log ?? console.log;
     const error = runtime?.error ?? console.error;
     // Build artifact update with command as data
+    // Wrap command in commands array as per protocol requirement
     const artifact = {
         taskId,
         kind: "artifact-update",
@@ -143,7 +188,9 @@ export async function sendCommand(params) {
             parts: [
                 {
                     kind: "data",
-                    data: command,
+                    data: {
+                        commands: [command],
+                    },
                 },
             ],
         },

package/dist/src/heartbeat.d.ts

@@ -13,12 +13,13 @@ export declare class HeartbeatManager {
     private config;
     private onTimeout;
     private serverName;
+    private onHeartbeatSuccess?;
     private intervalTimer;
     private timeoutTimer;
     private lastPongTime;
     private log;
     private error;
-    constructor(ws: WebSocket, config: HeartbeatConfig, onTimeout: () => void, serverName?: string, logFn?: (msg: string, ...args: any[]) => void, errorFn?: (msg: string, ...args: any[]) => void);
+    constructor(ws: WebSocket, config: HeartbeatConfig, onTimeout: () => void, serverName?: string, logFn?: (msg: string, ...args: any[]) => void, errorFn?: (msg: string, ...args: any[]) => void, onHeartbeatSuccess?: () => void);
     /**
      * Start heartbeat monitoring.
      */

package/dist/src/heartbeat.js

@@ -9,17 +9,20 @@ export class HeartbeatManager {
     config;
     onTimeout;
     serverName;
+    onHeartbeatSuccess;
     intervalTimer = null;
     timeoutTimer = null;
     lastPongTime = 0;
     // Logging functions following feishu pattern
     log;
     error;
-    constructor(ws, config, onTimeout, serverName = "unknown", logFn, errorFn) {
+    constructor(ws, config, onTimeout, serverName = "unknown", logFn, errorFn, onHeartbeatSuccess // ✅ 新增：心跳成功回调
+    ) {
         this.ws = ws;
         this.config = config;
         this.onTimeout = onTimeout;
         this.serverName = serverName;
+        this.onHeartbeatSuccess = onHeartbeatSuccess;
         this.log = logFn ?? console.log;
         this.error = errorFn ?? console.error;
     }
@@ -36,6 +39,8 @@ export class HeartbeatManager {
                 clearTimeout(this.timeoutTimer);
                 this.timeoutTimer = null;
             }
+            // ✅ Report health: heartbeat successful
+            this.onHeartbeatSuccess?.();
         });
         // Start interval timer
         this.intervalTimer = setInterval(() => {
@@ -67,9 +72,12 @@ export class HeartbeatManager {
         }
         try {
             // Send application-level heartbeat message
+            console.log(`[WS-${this.serverName}-SEND] Sending heartbeat frame:`, this.config.message);
             this.ws.send(this.config.message);
+            console.log(`[WS-${this.serverName}-SEND] Heartbeat message sent, size: ${this.config.message.length} bytes`);
             // Send protocol-level ping
             this.ws.ping();
+            console.log(`[WS-${this.serverName}-SEND] Protocol-level ping sent`);
             // Setup timeout timer
             this.timeoutTimer = setTimeout(() => {
                 this.error(`Heartbeat timeout for ${this.serverName}`);

package/dist/src/monitor.d.ts

@@ -4,6 +4,11 @@ export type MonitorXYOpts = {
     runtime?: RuntimeEnv;
     abortSignal?: AbortSignal;
     accountId?: string;
+    setStatus?: (status: {
+        lastEventAt?: number;
+        lastInboundAt?: number;
+        connected?: boolean;
+    }) => void;
 };
 /**
  * Monitor XY channel WebSocket connections.