@ynhcj/xiaoyi-channel 0.0.103-next → 0.0.104-next

package/dist/src/bot.js

@@ -6,14 +6,13 @@ import { downloadFilesFromParts } from "./file-download.js";
 import { resolveXYConfig } from "./config.js";
 import { sendStatusUpdate, sendClearContextResponse, sendTasksCancelResponse, sendA2AResponse } from "./formatter.js";
 import { appendSelfEvolutionKeywordNudge, shouldNudgeForSelfEvolutionKeyword, } from "./self-evolution-keyword.js";
-import { registerSession, unregisterSession, runWithSessionContext } from "./tools/session-manager.js";
 import { configManager } from "./utils/config-manager.js";
 import { addPushId } from "./utils/pushid-manager.js";
 import { getPushDataById } from "./utils/pushdata-manager.js";
 import { selfEvolutionManager } from "./utils/self-evolution-manager.js";
 import { saveRuntimeInfo } from "./utils/runtime-manager.js";
 import { toolCallNudgeManager } from "./utils/tool-call-nudge-manager.js";
-import { registerTaskId, decrementTaskIdRef, lockTaskId, unlockTaskId, hasActiveTask, } from "./task-manager.js";
+import { registerSession, unregisterSession, hasActiveSession, xyAsyncLocalStorage, } from "./xy-session-store.js";
 /**
  * Handle an incoming A2A message.
  * This is the main entry point for message processing.
@@ -101,22 +100,26 @@ export async function handleXYMessage(params) {
             }
         }
         // ========================================
-        // 🔑 检测steer模式和是否是第二条消息
+        // 🔑 检测steer模式
+        // Resolve route early to determine steer state
+        const config = resolveXYConfig(cfg);
+        let route = core.channel.routing.resolveAgentRoute({
+            cfg,
+            channel: "xiaoyi-channel",
+            accountId,
+            peer: {
+                kind: "direct",
+                id: parsed.sessionId,
+            },
+        });
+        log(`xy: resolved route accountId=${route.accountId}, sessionKey=${route.sessionKey}`);
         const isSteerMode = cfg.messages?.queue?.mode === "steer";
-        const isSecondMessage = isSteerMode && hasActiveTask(parsed.sessionId);
+        const isSecondMessage = isSteerMode && hasActiveSession(route.sessionKey);
         if (isSecondMessage) {
             log(`[BOT] 🔄 STEER MODE - Second message detected (will be follower)`);
             log(`[BOT]   - Session: ${parsed.sessionId}`);
             log(`[BOT]   - New taskId: ${parsed.taskId} (will replace current)`);
         }
-        // 🔑 注册taskId（第二条消息会覆盖第一条的taskId）
-        const { isUpdate, refCount } = registerTaskId(parsed.sessionId, parsed.taskId, parsed.messageId, { incrementRef: true } // 增加引用计数
-        );
-        // 🔑 如果是第一条消息，锁定taskId防止被过早清理
-        if (!isUpdate) {
-            lockTaskId(parsed.sessionId);
-            log(`[BOT] 🔒 Locked taskId for first message`);
-        }
         // Extract and update push_id if present
         const pushId = extractPushId(parsed.parts);
         if (pushId) {
@@ -142,27 +145,13 @@ export async function handleXYMessage(params) {
         ).catch((err) => {
             error(`[BOT] Failed to save runtime info:`, err);
         });
-        // Resolve configuration (needed for status updates)
-        const config = resolveXYConfig(cfg);
-        // ✅ Resolve agent route (following feishu pattern)
-        // accountId is "default" for XY (single account mode)
-        // Use sessionId as peer.id to ensure all messages in the same session share context
-        let route = core.channel.routing.resolveAgentRoute({
-            cfg,
-            channel: "xiaoyi-channel",
-            accountId, // "default"
-            peer: {
-                kind: "direct",
-                id: parsed.sessionId, // ✅ Use sessionId to share context within the same conversation session
-            },
-        });
-        log(`xy: resolved route accountId=${route.accountId}, sessionKey=${route.sessionKey}`);
+        // 🔑 Register / update session in unified store
         registerSession(route.sessionKey, {
             config,
-            sessionId: parsed.sessionId,
+            a2aSessionId: parsed.sessionId,
             taskId: parsed.taskId,
             messageId: parsed.messageId,
-            agentId: route.accountId,
+            accountId: route.accountId,
             deviceType,
         });
         // 🔑 发送初始状态更新（第二条消息也要发，用新taskId）
@@ -263,35 +252,18 @@ export async function handleXYMessage(params) {
             startStatusInterval();
             log(`[BOT-DISPATCHER] ✅ Status interval started for first message`);
         }
-        // Build session context for AsyncLocalStorage
-        const sessionContext = {
-            config,
-            sessionId: parsed.sessionId,
-            taskId: parsed.taskId,
-            messageId: parsed.messageId,
-            agentId: route.accountId,
-            deviceType,
-        };
         log(`[BOT-DISPATCH] ⏳ withReplyDispatcher starting, sessionKey=${route.sessionKey}`);
         await core.channel.reply.withReplyDispatcher({
             dispatcher,
             onSettled: () => {
                 log(`[BOT] 🏁 onSettled called for session: ${route.sessionKey}`);
-                log(`[BOT]   - isSecondMessage: ${isSecondMessage}`);
-                // 🔑 减少引用计数
-                decrementTaskIdRef(parsed.sessionId);
-                // 🔑 如果是第一条消息完成，解锁
-                if (!isSecondMessage) {
-                    unlockTaskId(parsed.sessionId);
-                    log(`[BOT] 🔓 Unlocked taskId (first message completed)`);
-                }
-                // 减少session引用计数
+                // Cleanup session from store
                 unregisterSession(route.sessionKey);
                 log(`[BOT] ✅ Cleanup completed`);
             },
             run: () => 
-            // 🔐 Use AsyncLocalStorage to provide session context to tools
-            runWithSessionContext(sessionContext, async () => {
+            // 🔐 Run inside ALS with openclawSessionKey so agentTools factory can read it
+            xyAsyncLocalStorage.run({ openclawSessionKey: route.sessionKey }, async () => {
                 log(`[BOT-DISPATCH] ⏳ dispatchReplyFromConfig starting...`);
                 log(`[BOT-DISPATCH]   - sessionKey: ${ctxPayload.SessionKey}`);
                 log(`[BOT-DISPATCH]   - provider: ${ctxPayload.Provider}`);
@@ -326,18 +298,14 @@ export async function handleXYMessage(params) {
         error("Failed to handle XY message:", err);
         runtime.error?.(`xy: Failed to handle message: ${String(err)}`);
         log(`[BOT] ❌ Error occurred, attempting cleanup...`);
-        // 🔑 错误时也要清理taskId和session
+        // 🔑 错误时也要清理session
         try {
-            const params = message.params;
-            const sessionId = params?.sessionId;
+            const msgParams = message.params;
+            const sessionId = msgParams?.sessionId;
             if (sessionId) {
                 log(`[BOT] 🧹 Cleaning up after error: ${sessionId}`);
-                // 清理 taskId
-                decrementTaskIdRef(sessionId);
-                unlockTaskId(sessionId);
-                // 清理 session
-                const core = getXYRuntime();
-                const route = core.channel.routing.resolveAgentRoute({
+                const core2 = getXYRuntime();
+                const errorRoute = core2.channel.routing.resolveAgentRoute({
                     cfg,
                     channel: "xiaoyi-channel",
                     accountId,
@@ -346,13 +314,12 @@ export async function handleXYMessage(params) {
                         id: sessionId,
                     },
                 });
-                unregisterSession(route.sessionKey);
+                unregisterSession(errorRoute.sessionKey);
                 log(`[BOT] ✅ Cleanup completed after error`);
             }
         }
         catch (cleanupErr) {
             log(`[BOT] ⚠️  Cleanup failed:`, cleanupErr);
-            // Ignore cleanup errors
         }
         // ❌ Don't re-throw: message processing error should not affect gateway stability
     }

package/dist/src/channel.js

@@ -2,7 +2,7 @@ import { resolveXYConfig, listXYAccountIds, getDefaultXYAccountId } from "./conf
 import { xyConfigSchema } from "./config-schema.js";
 import { xyOutbound } from "./outbound.js";
 import { filterToolsByDevice } from "./tools/device-tool-map.js";
-import { getCurrentSessionContext } from "./tools/session-manager.js";
+import { getSession, xyAsyncLocalStorage } from "./xy-session-store.js";
 import { createAllTools } from "./tools/create-all-tools.js";
 import { logger } from "./utils/logger.js";
 /**
@@ -44,10 +44,18 @@ export const xyPlugin = {
     },
     outbound: xyOutbound,
     agentTools: () => {
-        const ctx = getCurrentSessionContext();
-        const allTools = createAllTools(ctx);
-        const filtered = filterToolsByDevice(allTools, ctx?.deviceType);
-        logger.log(`[DEVICE-FILTER] deviceType=${ctx?.deviceType ?? "(none)"}, tools: ${allTools.length} → ${filtered.length} (${filtered.map(t => t.name).join(", ")})`);
+        // agentTools factory is called during agent run setup.
+        // ALS is still valid at this point — read sessionKey from it.
+        const alsContext = xyAsyncLocalStorage.getStore();
+        const sessionKey = alsContext?.openclawSessionKey;
+        if (!sessionKey) {
+            logger.log("[CHANNEL-TOOLS] no sessionKey in ALS, returning empty tools");
+            return [];
+        }
+        const session = getSession(sessionKey);
+        const allTools = createAllTools(sessionKey);
+        const filtered = filterToolsByDevice(allTools, session?.deviceType);
+        logger.log(`[CHANNEL-TOOLS] sessionKey=${sessionKey} deviceType=${session?.deviceType ?? "(none)"}, tools: ${allTools.length} → ${filtered.length} (${filtered.map(t => t.name).join(", ")})`);
         return filtered;
     },
     messaging: {

package/dist/src/formatter.js

@@ -2,7 +2,7 @@
 import { v4 as uuidv4 } from "uuid";
 import { getXYWebSocketManager } from "./client.js";
 import { logger } from "./utils/logger.js";
-import { getCurrentTaskId, getCurrentMessageId } from "./task-manager.js";
+import { getSessionByA2AId } from "./xy-session-store.js";
 /**
  * Send an A2A artifact update response.
  */
@@ -114,8 +114,8 @@ export async function sendStatusUpdate(params) {
     const log = runtime?.log ?? console.log;
     // Dynamic lookup: use latest taskId/messageId from task-manager (handles steer/interrupt),
     // fall back to closure-captured values
-    const currentTaskId = getCurrentTaskId(sessionId) ?? taskId;
-    const currentMessageId = getCurrentMessageId(sessionId) ?? messageId;
+    const currentTaskId = getSessionByA2AId(sessionId)?.taskId ?? taskId;
+    const currentMessageId = getSessionByA2AId(sessionId)?.messageId ?? messageId;
     // Build status update event following A2A protocol standard
     const statusUpdate = {
         taskId: currentTaskId,
@@ -162,8 +162,8 @@ export async function sendCommand(params) {
     const { config, sessionId, taskId, messageId, command } = params;
     // Dynamic lookup: use latest taskId/messageId from task-manager (handles steer/interrupt),
     // fall back to closure-captured values
-    const currentTaskId = getCurrentTaskId(sessionId) ?? taskId;
-    const currentMessageId = getCurrentMessageId(sessionId) ?? messageId;
+    const currentTaskId = getSessionByA2AId(sessionId)?.taskId ?? taskId;
+    const currentMessageId = getSessionByA2AId(sessionId)?.messageId ?? messageId;
     // Build artifact update with command as data
     // Wrap command in commands array as per protocol requirement
     const artifact = {

package/dist/src/monitor.js

@@ -2,13 +2,13 @@ import { resolveXYConfig } from "./config.js";
 import { getXYWebSocketManager, setClientRuntime, diagnoseAllManagers, cleanupOrphanConnections, removeXYWebSocketManager } from "./client.js";
 import { handleXYMessage } from "./bot.js";
 import { parseA2AMessage } from "./parser.js";
-import { hasActiveTask, getAllActiveTaskBindings } from "./task-manager.js";
+import { getAllActiveSessions, getSessionByA2AId } from "./xy-session-store.js";
 import { sendA2AResponse } from "./formatter.js";
 import { handleTriggerEvent } from "./trigger-handler.js";
 import { handleSelfEvolutionEvent, handleSelfEvolutionStateGetEvent } from "./self-evolution-handler.js";
 import { handleLoginTokenEvent } from "./login-token-handler.js";
 import { cleanupStaleTempFiles } from "./reply-dispatcher.js";
-import { cleanupStaleSessions, getActiveSessionCount, cleanupAllSessions } from "./tools/session-manager.js";
+import { cleanupStaleSessions, getActiveSessionCount, cleanupAllSessions } from "./xy-session-store.js";
 /**
  * Per-session serial queue that ensures messages from the same session are processed
  * in arrival order while allowing different sessions to run concurrently.
@@ -106,7 +106,7 @@ export async function monitorXYProvider(opts = {}) {
             try {
                 const parsed = parseA2AMessage(message);
                 const steerMode = cfg.messages?.queue?.mode === "steer";
-                const hasActiveRun = hasActiveTask(parsed.sessionId);
+                const hasActiveRun = Boolean(getSessionByA2AId(parsed.sessionId));
                 if (steerMode && hasActiveRun) {
                     // Steer模式且有活跃任务：不入队列，直接并发执行
                     log(`[MONITOR-HANDLER] 🔄 STEER MODE: Executing concurrently for messageKey=${messageKey}`);
@@ -214,25 +214,25 @@ export async function monitorXYProvider(opts = {}) {
             log("XY gateway: abort signal received, sending notifications before stopping");
             // 📤 Send restart notification to all active sessions before disconnecting
             try {
-                const activeBindings = getAllActiveTaskBindings();
-                if (activeBindings.length > 0) {
+                const activeSessions = getAllActiveSessions();
+                if (activeSessions.length > 0) {
                     const config = resolveXYConfig(cfg);
                     const notificationText = "Gateway即将重启，重启期间可能短暂出现\u201c环境异常\u201d提示，请稍候并耐心重试~";
-                    log(`[MONITOR] 📤 Sending restart notifications to ${activeBindings.length} active session(s)`);
-                    const sendPromises = activeBindings.map(binding => sendA2AResponse({
+                    log(`[MONITOR] 📤 Sending restart notifications to ${activeSessions.length} active session(s)`);
+                    const sendPromises = activeSessions.map(({ session }) => sendA2AResponse({
                         config,
-                        sessionId: binding.sessionId,
-                        taskId: binding.currentTaskId,
-                        messageId: binding.currentMessageId,
+                        sessionId: session.a2aSessionId,
+                        taskId: session.taskId,
+                        messageId: session.messageId,
                         text: notificationText,
                         append: false,
                         final: true,
                         runtime,
                     }).catch(err => {
-                        error(`[MONITOR] Failed to send restart notification to session ${binding.sessionId}: ${String(err)}`);
+                        error(`[MONITOR] Failed to send restart notification to session ${session.a2aSessionId}: ${String(err)}`);
                     }));
                     await Promise.all(sendPromises);
-                    log(`[MONITOR] ✅ Restart notifications sent to ${activeBindings.length} session(s)`);
+                    log(`[MONITOR] ✅ Restart notifications sent to ${activeSessions.length} session(s)`);
                 }
                 else {
                     log(`[MONITOR] No active sessions, skipping restart notifications`);

package/dist/src/outbound.js

@@ -1,7 +1,7 @@
 import { resolveXYConfig } from "./config.js";
 import { XYFileUploadService } from "./file-upload.js";
 import { XYPushService } from "./push.js";
-import { getCurrentSessionContext } from "./tools/session-manager.js";
+import { getSessionByA2AId } from "./xy-session-store.js";
 import { savePushData } from "./utils/pushdata-manager.js";
 import { getAllPushIds } from "./utils/pushid-manager.js";
 import { logger } from "./utils/logger.js";
@@ -68,10 +68,10 @@ export const xyOutbound = {
         // If the target doesn't contain "::", try to enhance it with taskId from session context
         if (!trimmedTo.includes("::")) {
             logger.log(`[xyOutbound.resolveTarget] Target "${trimmedTo}" missing taskId, looking up session context`);
-            // Try to get the current session context
-            const sessionContext = getCurrentSessionContext();
-            if (sessionContext && sessionContext.sessionId === trimmedTo) {
-                const enhancedTarget = `${trimmedTo}::${sessionContext.taskId}`;
+            // Try to get the current session context via A2A sessionId
+            const session = getSessionByA2AId(trimmedTo);
+            if (session) {
+                const enhancedTarget = `${trimmedTo}::${session.taskId}`;
                 logger.log(`[xyOutbound.resolveTarget] Enhanced target: ${enhancedTarget}`);
                 return {
                     ok: true,

package/dist/src/provider.js

@@ -8,8 +8,7 @@
 //   models.providers.xiaoyiprovider.api = "openai-completions"
 //   models.providers.xiaoyiprovider.models = [...]
 import { createHash } from "crypto";
-import { getCurrentSessionContext } from "./tools/session-manager.js";
-import { getCurrentTaskId } from "./task-manager.js";
+import { getSession, getSessionByA2AId, xyAsyncLocalStorage } from "./xy-session-store.js";
 import { selfEvolutionManager } from "./utils/self-evolution-manager.js";
 // ── Retry config ──────────────────────────────────────────────
 const RETRY_DELAYS_MS = [10_000, 20_000, 40_000, 60_000, 60_000];
@@ -405,9 +404,12 @@ export const xiaoyiProvider = {
      *   3. No uid available → return undefined (no headers injected)
      */
     prepareExtraParams: (ctx) => {
-        const sessionCtx = getCurrentSessionContext();
-        if (sessionCtx) {
-            const taskId = sessionCtx.taskId;
+        // Try to get session from ALS (valid during agent run setup)
+        const alsContext = xyAsyncLocalStorage.getStore();
+        const sessionKey = alsContext?.openclawSessionKey;
+        const session = sessionKey ? getSession(sessionKey) : null;
+        if (session) {
+            const taskId = session.taskId;
             const sessionId = taskId.split("&")[0];
             const interactionId = taskId.split("&")[1] || "";
             return {
@@ -415,7 +417,7 @@ export const xiaoyiProvider = {
                 [HEADER_TRACE_ID]: taskId,
                 [HEADER_SESSION_ID]: sessionId,
                 [HEADER_INTERACTION_ID]: interactionId,
-                [DEVICE_TYPE_KEY]: sessionCtx.deviceType ?? "",
+                [DEVICE_TYPE_KEY]: session.deviceType ?? "",
             };
         }
         // Fallback: store uid prefix for lazy timestamp generation in wrapStreamFn.
@@ -444,13 +446,13 @@ export const xiaoyiProvider = {
         if (!underlying)
             return underlying;
         // Capture A2A sessionId at agent setup time for multi-session isolation.
-        // openclaw calls wrapStreamFn per-agent (per session), so this runs inside
-        // the correct runWithSessionContext() ALS scope.  When multiple sessions are
-        // active concurrently, getCurrentSessionContext() may later return the WRONG
-        // session (lastRegisteredKey fallback).  The captured sessionId lets us
-        // bypass that fallback and look up the correct taskId directly from
-        // task-manager.
-        const capturedA2ASessionId = getCurrentSessionContext()?.sessionId ?? null;
+        // wrapStreamFn is called per-agent (per session), ALS is still valid here.
+        // We capture the a2aSessionId and look up the store at each request time
+        // to always get the latest taskId (supports steer).
+        const alsContext = xyAsyncLocalStorage.getStore();
+        const capturedA2ASessionId = alsContext?.openclawSessionKey
+            ? getSession(alsContext.openclawSessionKey)?.a2aSessionId ?? null
+            : null;
         return async (model, context, options) => {
             // 每次请求时从 ctx.extraParams 动态读取 header
             const dynamicHeaders = {};
@@ -472,21 +474,11 @@ export const xiaoyiProvider = {
                     }
                 }
                 else {
-                    // Session mode: resolve taskId for the correct session.
-                    //
-                    // Priority:
-                    //   1. capturedA2ASessionId → getCurrentTaskId()  (most reliable,
-                    //      bypasses lastRegisteredKey fallback)
-                    //   2. getCurrentSessionContext()?.taskId           (works when ALS
-                    //      is intact)
-                    //   3. ctx.extraParams cached values                (last resort,
-                    //      may be stale / from wrong session)
+                    // Session mode: resolve taskId from store using captured a2aSessionId.
+                    // This always returns the latest taskId (supports steer).
                     let resolvedTaskId = null;
                     if (capturedA2ASessionId) {
-                        resolvedTaskId = getCurrentTaskId(capturedA2ASessionId);
-                    }
-                    if (!resolvedTaskId) {
-                        resolvedTaskId = getCurrentSessionContext()?.taskId ?? null;
+                        resolvedTaskId = getSessionByA2AId(capturedA2ASessionId)?.taskId ?? null;
                     }
                     const traceId = resolvedTaskId ?? ctx.extraParams[HEADER_TRACE_ID];
                     const sessionId = resolvedTaskId?.split("&")[0]
@@ -517,12 +509,15 @@ export const xiaoyiProvider = {
                 console.log(`[xiaoyiprovider] system prompt length: ${context.systemPrompt.length}`);
             }
             // Prefer deviceType from extraParams (set by prepareExtraParams).
-            // Fall back to getCurrentSessionContext() because OpenClaw caches
+            // Fall back to store lookup because OpenClaw caches
             // resolvePreparedExtraParams by provider/modelId – the cache key does
             // not include session-specific data, so deviceType may be missing
             // from the cached extraParams even when a session is active.
             const extraParamsDeviceType = ctx.extraParams?.[DEVICE_TYPE_KEY] || undefined;
-            const deviceType = extraParamsDeviceType ?? getCurrentSessionContext()?.deviceType;
+            const storeDeviceType = capturedA2ASessionId
+                ? getSessionByA2AId(capturedA2ASessionId)?.deviceType
+                : undefined;
+            const deviceType = extraParamsDeviceType ?? storeDeviceType;
             // 在发送给模型前，优化 systemPrompt 结构
             if (context.systemPrompt) {
                 let sp = context.systemPrompt;

package/dist/src/reply-dispatcher.js

@@ -1,7 +1,7 @@
 import { getXYRuntime } from "./runtime.js";
 import { sendA2AResponse, sendStatusUpdate, sendReasoningTextUpdate } from "./formatter.js";
 import { resolveXYConfig } from "./config.js";
-import { getCurrentTaskId, getCurrentMessageId } from "./task-manager.js";
+import { getSessionByA2AId } from "./xy-session-store.js";
 import fs from "fs/promises";
 import path from "path";
 import { logger } from "./utils/logger.js";
@@ -59,10 +59,12 @@ export function createXYReplyDispatcher(params) {
      * 每次需要taskId时，都从TaskManager获取最新值
      */
     const getActiveTaskId = () => {
-        return getCurrentTaskId(sessionId) ?? initialTaskId;
+        const session = getSessionByA2AId(sessionId);
+        return session?.taskId ?? initialTaskId;
     };
     const getActiveMessageId = () => {
-        return getCurrentMessageId(sessionId) ?? initialMessageId;
+        const session = getSessionByA2AId(sessionId);
+        return session?.messageId ?? initialMessageId;
     };
     const core = getXYRuntime();
     const config = resolveXYConfig(cfg);

package/dist/src/steer-injector.js

@@ -1,6 +1,5 @@
 // Steer message injector for CSPL hook integration
-import { getSessionContext } from "./tools/session-manager.js";
-import { hasActiveTask, getCurrentTaskId } from "./task-manager.js";
+import { getSession, hasActiveSession } from "./xy-session-store.js";
 import { handleXYMessage } from "./bot.js";
 import { logger } from "./utils/logger.js";
 import { randomUUID } from "node:crypto";
@@ -29,13 +28,12 @@ export async function tryInjectSteer(sessionKey, message) {
     if (!sessionKey) {
         return false;
     }
-    const sessionCtx = getSessionContext(sessionKey);
-    if (!sessionCtx) {
+    const session = getSession(sessionKey);
+    if (!session) {
         return false;
     }
-    const { sessionId } = sessionCtx;
-    const activeTaskId = getCurrentTaskId(sessionId);
-    if (!hasActiveTask(sessionId)) {
+    const { a2aSessionId: sessionId, taskId: activeTaskId } = session;
+    if (!hasActiveSession(sessionKey)) {
         return false;
     }
     if (!cachedCfg || !cachedRuntime) {

package/dist/src/tools/calendar-tool.d.ts

@@ -1,7 +1,6 @@
-import type { SessionContext } from "./session-manager.js";
 /**
  * XY calendar event tool - creates a calendar event on user's device.
  * Requires title, dtStart (start time), and dtEnd (end time) parameters.
  * Time format must be: yyyy-mm-dd hh:mm:ss
  */
-export declare function createCalendarTool(ctx: SessionContext): any;
+export declare function createCalendarTool(sessionKey: string): any;

package/dist/src/tools/calendar-tool.js

@@ -1,12 +1,12 @@
 import { getXYWebSocketManager } from "../client.js";
 import { sendCommand } from "../formatter.js";
+import { requireSession } from "./session-helper.js";
 /**
  * XY calendar event tool - creates a calendar event on user's device.
  * Requires title, dtStart (start time), and dtEnd (end time) parameters.
  * Time format must be: yyyy-mm-dd hh:mm:ss
  */
-export function createCalendarTool(ctx) {
-    const { config, sessionId, taskId, messageId } = ctx;
+export function createCalendarTool(sessionKey) {
     return {
         name: "create_calendar_event",
         label: "Create Calendar Event",
@@ -34,6 +34,7 @@ export function createCalendarTool(ctx) {
             required: ["title", "dtStart", "dtEnd"],
         },
         async execute(toolCallId, params) {
+            const session = requireSession(sessionKey);
             // Validate parameters
             if (!params.title || !params.dtStart || !params.dtEnd) {
                 throw new Error("Missing required parameters: title, dtStart, and dtEnd are required");
@@ -45,7 +46,7 @@ export function createCalendarTool(ctx) {
                 throw new Error("Invalid time format. Required format: yyyy-mm-dd hh:mm:ss (e.g., 2024-01-15 14:30:00)");
             }
             // Get WebSocket manager
-            const wsManager = getXYWebSocketManager(config);
+            const wsManager = getXYWebSocketManager(session.config);
             // Build CreateCalendarEvent command
             const command = {
                 header: {
@@ -111,10 +112,10 @@ export function createCalendarTool(ctx) {
                 wsManager.on("data-event", handler);
                 // Send the command
                 sendCommand({
-                    config,
-                    sessionId,
-                    taskId,
-                    messageId,
+                    config: session.config,
+                    sessionId: session.a2aSessionId,
+                    taskId: session.taskId,
+                    messageId: session.messageId,
                     command,
                 })
                     .then(() => {

package/dist/src/tools/call-device-tool.d.ts

@@ -1,6 +1,5 @@
-import type { SessionContext } from "./session-manager.js";
 /**
  * call_device_tool - 通用端工具调度器。
  * LLM 必须先通过 get_xxx_tool_schema 获取具体工具 schema，再用本工具执行。
  */
-export declare function createCallDeviceTool(ctx: SessionContext): any;
+export declare function createCallDeviceTool(sessionKey: string): any;

package/dist/src/tools/call-device-tool.js

@@ -23,37 +23,36 @@ import { createSaveFileToPhoneTool } from "./save-file-to-phone-tool.js";
 import { createSendEmailTool } from "./send-email-tool.js";
 import { createSearchEmailTool } from "./search-email-tool.js";
 import { sendStatusUpdate } from "../formatter.js";
-import { getCurrentTaskId, getCurrentMessageId } from "../task-manager.js";
+import { requireSession } from "./session-helper.js";
 /**
  * call_device_tool - 通用端工具调度器。
  * LLM 必须先通过 get_xxx_tool_schema 获取具体工具 schema，再用本工具执行。
  */
-export function createCallDeviceTool(ctx) {
-    const { config, sessionId, taskId, messageId } = ctx;
-    const noteTool = createNoteTool(ctx);
-    const modifyNoteTool = createModifyNoteTool(ctx);
-    const createAlarmTool = makeAlarmTool(ctx);
-    const modifyAlarmTool = createModifyAlarmTool(ctx);
-    const deleteAlarmTool = createDeleteAlarmTool(ctx);
-    const callPhoneTool = createCallPhoneTool(ctx);
-    const calendarTool = createCalendarTool(ctx);
-    const searchNoteTool = createSearchNoteTool(ctx);
-    const searchMessageTool = createSearchMessageTool(ctx);
-    const sendMessageTool = createSendMessageTool(ctx);
-    const xiaoyiAddCollectionTool = createXiaoyiAddCollectionTool(ctx);
-    const xiaoyiCollectionTool = createXiaoyiCollectionTool(ctx);
-    const xiaoyiDeleteCollectionTool = createXiaoyiDeleteCollectionTool(ctx);
-    const searchPhotoGalleryTool = createSearchPhotoGalleryTool(ctx);
-    const uploadPhotoTool = createUploadPhotoTool(ctx);
-    const uploadFileTool = createUploadFileTool(ctx);
-    const sendEmailTool = createSendEmailTool(ctx);
-    const searchAlarmTool = createSearchAlarmTool(ctx);
-    const searchContactTool = createSearchContactTool(ctx);
-    const searchCalendarTool = createSearchCalendarTool(ctx);
-    const saveMediaToGalleryTool = createSaveMediaToGalleryTool(ctx);
-    const searchFileTool = createSearchFileTool(ctx);
-    const saveFileToPhoneTool = createSaveFileToPhoneTool(ctx);
-    const searchEmailTool = createSearchEmailTool(ctx);
+export function createCallDeviceTool(sessionKey) {
+    const noteTool = createNoteTool(sessionKey);
+    const modifyNoteTool = createModifyNoteTool(sessionKey);
+    const createAlarmTool = makeAlarmTool(sessionKey);
+    const modifyAlarmTool = createModifyAlarmTool(sessionKey);
+    const deleteAlarmTool = createDeleteAlarmTool(sessionKey);
+    const callPhoneTool = createCallPhoneTool(sessionKey);
+    const calendarTool = createCalendarTool(sessionKey);
+    const searchNoteTool = createSearchNoteTool(sessionKey);
+    const searchMessageTool = createSearchMessageTool(sessionKey);
+    const sendMessageTool = createSendMessageTool(sessionKey);
+    const xiaoyiAddCollectionTool = createXiaoyiAddCollectionTool(sessionKey);
+    const xiaoyiCollectionTool = createXiaoyiCollectionTool(sessionKey);
+    const xiaoyiDeleteCollectionTool = createXiaoyiDeleteCollectionTool(sessionKey);
+    const searchPhotoGalleryTool = createSearchPhotoGalleryTool(sessionKey);
+    const uploadPhotoTool = createUploadPhotoTool(sessionKey);
+    const uploadFileTool = createUploadFileTool(sessionKey);
+    const sendEmailTool = createSendEmailTool(sessionKey);
+    const searchAlarmTool = createSearchAlarmTool(sessionKey);
+    const searchContactTool = createSearchContactTool(sessionKey);
+    const searchCalendarTool = createSearchCalendarTool(sessionKey);
+    const saveMediaToGalleryTool = createSaveMediaToGalleryTool(sessionKey);
+    const searchFileTool = createSearchFileTool(sessionKey);
+    const saveFileToPhoneTool = createSaveFileToPhoneTool(sessionKey);
+    const searchEmailTool = createSearchEmailTool(sessionKey);
     /**
      * 端工具注册表 —— 按 name 索引所有可通过 call_device_tool 调度的工具。
      */
@@ -102,14 +101,15 @@ export function createCallDeviceTool(ctx) {
             required: ["toolName", "arguments"],
         },
         async execute(toolCallId, params) {
+            const session = requireSession(sessionKey);
             const { toolName, arguments: toolArgs } = params;
             // 向用户端发送具体工具名的状态更新
-            const currentTaskId = getCurrentTaskId(sessionId) ?? taskId;
-            const currentMessageId = getCurrentMessageId(sessionId) ?? messageId;
+            const currentTaskId = session.taskId;
+            const currentMessageId = session.messageId;
             try {
                 await sendStatusUpdate({
-                    config,
-                    sessionId,
+                    config: session.config,
+                    sessionId: session.a2aSessionId,
                     taskId: currentTaskId,
                     messageId: currentMessageId,
                     text: `正在使用工具: ${toolName}...`,

package/dist/src/tools/call-phone-tool.d.ts

@@ -1,6 +1,5 @@
-import type { SessionContext } from "./session-manager.js";
 /**
  * XY call phone tool - makes a phone call on user's device.
  * Requires phoneNumber parameter and optional slotId (0 for primary SIM, 1 for secondary SIM).
  */
-export declare function createCallPhoneTool(ctx: SessionContext): any;
+export declare function createCallPhoneTool(sessionKey: string): any;