@agent-link/agent 0.1.339 → 0.1.340

package/dist/connection.d.ts

@@ -1,3 +1,4 @@
+import WebSocket from 'ws';
 import type { AgentConfig } from './config.js';
 import { type BackendType } from './backends/types.js';
 import { type EntraToken } from './entra.js';
@@ -25,7 +26,7 @@ export declare function connect(config: AgentConfig): Promise<string>;
 /** Get the E2E key for embedding in the session URL. */
 export declare function getE2EKey(): Uint8Array | null;
 export declare function disconnect(): void;
-export declare function send(msg: Record<string, unknown>): void;
+export declare function send(msg: Record<string, unknown>, expectedSocket?: WebSocket | null): void;
 export declare function sendAndFlush(msg: Record<string, unknown>, timeoutMs?: number): Promise<void>;
 export declare function beginShutdown(): void;
 export declare function getConnectionStatus(): {

package/dist/connection.js

@@ -32,7 +32,7 @@ const require = createRequire(import.meta.url);
 const pkg = require('../package.json');
 const TEAM_SESSIONS_DIR = path.join(CONFIG_DIR, 'team-sessions');
 const LOOP_SESSIONS_DIR = path.join(CONFIG_DIR, 'loop-sessions');
-import { handleChat as claudeHandleChat, addSendFn, abort as abortClaude, abortAll as abortAllClaude, cancelExecution as claudeCancelExecution, handleUserAnswer, handleToolPermissionResponse, handleBtwQuestion, getConversation, getConversations, getIsCompacting, clearSessionId, evictByClaudeSessionId, rebindConversation, addOutputObserver, removeOutputObserver, addCloseObserver, removeCloseObserver, setOutputObserver, clearOutputObserver, setCloseObserver, clearCloseObserver, restartConversation, createPlaceholderConversation, getPendingQuestions, getPendingToolPermissions, setModel, addOnSessionStarted, getBackendType, setBackendType, isCopilotBackend, isCodexBackend, isPiBackend, isYeaftBackend, addCopilotSendFn, handleCopilotChat, executeCopilotCommand, setCopilotPlanMode, startCopilotPlanImplementation, cancelCopilotExecution, listCopilotSessionsPage, listAllCopilotSessionsPage, readCopilotSessionMessages, readCopilotSessionMode, readCopilotSessionProvider, readCopilotSessionUsage, getCopilotSessionWorkDir, deleteCopilotSession, renameCopilotSession, moveCopilotSession, getCopilotConversation, getCopilotConversations, createCopilotPlaceholder, rebindCopilotConversation, getPendingCopilotDecisions, handleCopilotUserAnswer, cleanupAllCopilotConversations, copilotCapabilities, addCodexSendFn, handleCodexChat, cancelCodexExecution, listCodexSessionsPage, listAllCodexSessionsPage, readCodexSessionMessages, deleteCodexSession, renameCodexSession, moveCodexSession, getCodexConversation, getCodexConversations, createCodexPlaceholder, restartCodexConversation, rebindCodexConversation, cleanupAllCodexConversations, evictByCodexSessionId, codexCapabilities } from './runtime.js';
+import { handleChat as claudeHandleChat, addSendFn, abort as abortClaude, abortAll as abortAllClaude, cancelExecution as claudeCancelExecution, handleUserAnswer, handleToolPermissionResponse, handleBtwQuestion, getConversation, getConversations, getIsCompacting, clearSessionId, evictByClaudeSessionId, rebindConversation, addOutputObserver, removeOutputObserver, addCloseObserver, removeCloseObserver, setOutputObserver, clearOutputObserver, setCloseObserver, clearCloseObserver, restartConversation, createPlaceholderConversation, getPendingQuestions, getPendingToolPermissions, setModel, addOnSessionStarted, getBackendType, setBackendType, isCopilotBackend, isCodexBackend, isPiBackend, isYeaftBackend, addCopilotSendFn, handleCopilotChat, executeCopilotCommand, setCopilotPlanMode, startCopilotPlanImplementation, cancelCopilotExecution, readCopilotSessionMessages, readCopilotSessionMode, readCopilotSessionProvider, readCopilotSessionUsage, getCopilotSessionWorkDir, deleteCopilotSession, renameCopilotSession, moveCopilotSession, getCopilotConversation, getCopilotConversations, createCopilotPlaceholder, rebindCopilotConversation, getPendingCopilotDecisions, handleCopilotUserAnswer, cleanupAllCopilotConversations, copilotCapabilities, addCodexSendFn, handleCodexChat, cancelCodexExecution, listCodexSessionsPage, listAllCodexSessionsPage, readCodexSessionMessages, deleteCodexSession, renameCodexSession, moveCodexSession, getCodexConversation, getCodexConversations, createCodexPlaceholder, restartCodexConversation, rebindCodexConversation, cleanupAllCodexConversations, evictByCodexSessionId, codexCapabilities } from './runtime.js';
 import { addPiSendFn, cancelPiExecution, cleanupAllPiConversations, createPiPlaceholder, deletePiSession, evictByPiSessionId, getPiConversation, getPiConversations, getPiSessionWorkDir, handlePiChat, listAllPiSessionsPage, listPiSessionsPage, movePiSession, piCapabilities, readPiSessionMessages, rebindPiConversation, renamePiSession, restartPiConversation, } from './pi.js';
 import { addYeaftSendFn, cancelYeaftExecution, cleanupAllYeaftConversations, createYeaftPlaceholder, deleteYeaftSession, evictByYeaftSessionId, getYeaftConversation, getYeaftConversations, handleYeaftChat, handleYeaftUserAnswer, listAllYeaftSessionsPage, listYeaftSessionsPage, moveYeaftSession, readYeaftSessionMessages, rebindYeaftConversation, renameYeaftSession, restartYeaftConversation, } from './yeaft.js';
 import { DEFAULT_SESSION_LIST_LIMIT, LEGACY_SESSION_LIST_CAP, listSessions, listSessionsPageAsync, readSessionMessages, deleteSession, renameSession, moveSession, listAllRecentSessionsPage, getSessionCacheStats } from './history.js';
@@ -53,6 +53,7 @@ import { initScheduler, getRunningExecutions, } from './scheduler.js';
 import { handleWebUpgradeMessage } from './web-upgrade.js';
 import { refreshRuntimeLaunchSpecDeviceName } from './upgrade-launch-spec.js';
 import { getAgentRuntimeInfo } from './runtime-info.js';
+import { listCopilotSessionsPageAsync, listAllCopilotSessionsPageAsync } from './runtime.js';
 let tunnelHandler = null;
 let terminalManager = null;
 const RECONNECT_BASE_DELAY = 1000;
@@ -647,10 +648,13 @@ function deduped(key, fn) {
     });
     inflightHandlers.set(key, run());
 }
-export function send(msg) {
+export function send(msg, expectedSocket) {
+    if (expectedSocket !== undefined && state.ws !== expectedSocket)
+        return;
     if (state.ws && state.ws.readyState === WebSocket.OPEN) {
         sendQueue = sendQueue.then(async () => {
-            if (!state.ws || state.ws.readyState !== WebSocket.OPEN)
+            if (!state.ws || state.ws.readyState !== WebSocket.OPEN
+                || (expectedSocket !== undefined && state.ws !== expectedSocket))
                 return;
             let payload = msg;
             // Don't E2E-encrypt messages the server must inspect (it only decrypts
@@ -671,6 +675,8 @@ export function send(msg) {
             if (state.e2eKey && state.e2eNegotiated && !skipE2E.includes(msg.type)) {
                 payload = await e2eEncrypt(msg, state.e2eKey);
             }
+            if (expectedSocket !== undefined && state.ws !== expectedSocket)
+                return;
             return encryptAndSend(state.ws, payload, state.sessionKey);
         }).catch((err) => {
             console.error(`[AgentLink] send() failed for ${msg.type}:`, err.message);
@@ -1241,18 +1247,7 @@ function handleServerMessage(msg) {
                 });
             }
             else if (recentBackend === 'copilot') {
-                const page = listAllCopilotSessionsPage({ limit: m.limit, cursor: m.cursor });
-                console.log(`[AgentLink] → recent_sessions_list (${page.sessions.length} copilot sessions) hasMore=${page.hasMore}`);
-                send({
-                    type: 'recent_sessions_list',
-                    sessions: page.sessions,
-                    hasMore: page.hasMore,
-                    nextCursor: page.nextCursor,
-                    indexState: page.indexState,
-                    ...(m.mode ? { mode: m.mode } : {}),
-                    ...(m.requestId ? { requestId: m.requestId } : {}),
-                    ...(m.providerId ? { providerId: m.providerId } : {}),
-                });
+                void handleListRecentCopilotSessions(m);
             }
             else {
                 handleListRecentSessions(m);
@@ -3050,18 +3045,24 @@ async function handleListSessions(dir, options = {}) {
         });
     }
 }
-function handleListCopilotSessionsFn(dir, options = {}) {
+async function handleListCopilotSessionsFn(dir, options = {}) {
+    const requestSocket = state.ws;
+    const backend = getBackendType();
+    const stillCurrent = () => state.ws === requestSocket && (!!options.providerId || getBackendType() === backend);
+    const startedAt = performance.now();
     try {
         const page = options.legacy
-            ? { ...listCopilotSessionsPage(dir, { limit: Number.POSITIVE_INFINITY }), hasMore: false }
-            : listCopilotSessionsPage(dir, {
+            ? { ...await listCopilotSessionsPageAsync(dir, { limit: Number.POSITIVE_INFINITY }), hasMore: false }
+            : await listCopilotSessionsPageAsync(dir, {
                 limit: options.limit ?? DEFAULT_SESSION_LIST_LIMIT,
                 cursor: options.cursor,
             });
         if (options.legacy && page.sessions.length > LEGACY_SESSION_LIST_CAP) {
             page.sessions = page.sessions.slice(0, LEGACY_SESSION_LIST_CAP);
         }
-        console.log(`[AgentLink] → sessions_list (${page.sessions.length} copilot sessions for ${dir}) hasMore=${page.hasMore}`);
+        if (!stillCurrent())
+            return;
+        console.log(`[AgentLink] → sessions_list (${page.sessions.length} copilot sessions for ${dir}) hasMore=${page.hasMore} durationMs=${Math.round(performance.now() - startedAt)}`);
         send({
             type: 'sessions_list',
             sessions: page.sessions,
@@ -3072,10 +3073,12 @@ function handleListCopilotSessionsFn(dir, options = {}) {
             ...(options.mode ? { mode: options.mode } : {}),
             ...(options.requestId ? { requestId: options.requestId } : {}),
             ...(options.providerId ? { providerId: options.providerId } : {}),
-        });
+        }, requestSocket);
     }
     catch (err) {
         console.error(`[AgentLink] listCopilotSessions failed:`, err);
+        if (!stillCurrent())
+            return;
         send({
             type: 'sessions_list',
             sessions: [],
@@ -3086,7 +3089,32 @@ function handleListCopilotSessionsFn(dir, options = {}) {
             ...(options.mode ? { mode: options.mode } : {}),
             ...(options.requestId ? { requestId: options.requestId } : {}),
             ...(options.providerId ? { providerId: options.providerId } : {}),
-        });
+        }, requestSocket);
+    }
+}
+async function handleListRecentCopilotSessions(options) {
+    const requestSocket = state.ws;
+    const backend = getBackendType();
+    const stillCurrent = () => state.ws === requestSocket && (!!options.providerId || getBackendType() === backend);
+    const envelope = {
+        type: 'recent_sessions_list',
+        ...(options.mode ? { mode: options.mode } : {}),
+        ...(options.requestId ? { requestId: options.requestId } : {}),
+        ...(options.providerId ? { providerId: options.providerId } : {}),
+    };
+    const startedAt = performance.now();
+    try {
+        const page = await listAllCopilotSessionsPageAsync({ limit: options.limit, cursor: options.cursor });
+        if (!stillCurrent())
+            return;
+        console.log(`[AgentLink] → recent_sessions_list (${page.sessions.length} copilot sessions) hasMore=${page.hasMore} durationMs=${Math.round(performance.now() - startedAt)}`);
+        send({ ...envelope, ...page }, requestSocket);
+    }
+    catch (error) {
+        console.error('[AgentLink] listAllCopilotSessions failed:', error);
+        if (stillCurrent())
+            send({ ...envelope, sessions: [], hasMore: false, indexState: 'ready',
+                error: error instanceof Error ? error.message : String(error) }, requestSocket);
     }
 }
 function handleListCodexSessionsFn(dir, options = {}) {