@agent-link/agent 0.1.323 → 0.1.325

package/dist/connection.js

@@ -29,7 +29,8 @@ 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, addCopilotSendFn, handleCopilotChat, cancelCopilotExecution, listCopilotSessionsPage, listAllCopilotSessionsPage, readCopilotSessionMessages, deleteCopilotSession, renameCopilotSession, moveCopilotSession, getCopilotConversation, getCopilotConversations, createCopilotPlaceholder, restartCopilotConversation, rebindCopilotConversation, 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, addCopilotSendFn, handleCopilotChat, cancelCopilotExecution, listCopilotSessionsPage, listAllCopilotSessionsPage, readCopilotSessionMessages, getCopilotSessionWorkDir, deleteCopilotSession, renameCopilotSession, moveCopilotSession, getCopilotConversation, getCopilotConversations, createCopilotPlaceholder, restartCopilotConversation, rebindCopilotConversation, 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 { DEFAULT_SESSION_LIST_LIMIT, LEGACY_SESSION_LIST_CAP, listSessions, listSessionsPageAsync, readSessionMessages, deleteSession, renameSession, moveSession, listAllRecentSessionsPage, getSessionCacheStats } from './history.js';
 import { claudeCapabilities } from './backends/types.js';
 import { searchSessions } from './search-sessions.js';
@@ -69,6 +70,7 @@ const state = {
 let unsubscribeSend = null;
 let unsubscribeCopilotSend = null;
 let unsubscribeCodexSend = null;
+let unsubscribePiSend = null;
 let unsubscribeSessionStarted = null;
 let heartbeatInterval = null;
 let heartbeatTimeout = null;
@@ -84,6 +86,8 @@ function capabilitiesForBackend(backend) {
             return copilotCapabilities();
         case 'codex':
             return codexCapabilities();
+        case 'pi':
+            return piCapabilities();
         case 'claude':
         default:
             return claudeCapabilities();
@@ -93,7 +97,7 @@ function normalizeDirForCompare(value) {
     const normalized = value.replace(/[\\/]+$/, '');
     return process.platform === 'win32' ? normalized.toLowerCase() : normalized;
 }
-function hasActiveSessionConversation(backend, sessionId) {
+function hasActiveSessionConversation(backend, sessionId, providerId, workDir) {
     if (backend === 'copilot') {
         for (const conv of getCopilotConversations().values()) {
             if (conv.copilotSessionId === sessionId || conv.lastCopilotSessionId === sessionId) {
@@ -110,6 +114,16 @@ function hasActiveSessionConversation(backend, sessionId) {
             }
         }
     }
+    else if (backend === 'pi') {
+        for (const conv of getPiConversations().values()) {
+            if ((conv.piSessionId === sessionId || conv.lastPiSessionId === sessionId)
+                && (!providerId || conv.providerId === providerId)
+                && (!workDir || normalizeDirForCompare(conv.workDir) === normalizeDirForCompare(workDir))) {
+                if (conv.turnActive)
+                    return true;
+            }
+        }
+    }
     else {
         for (const conv of getConversations().values()) {
             if (conv.claudeSessionId === sessionId || conv.lastClaudeSessionId === sessionId) {
@@ -120,7 +134,7 @@ function hasActiveSessionConversation(backend, sessionId) {
     }
     return false;
 }
-function syncMovedConversationWorkDir(backend, sessionId, toWorkDir) {
+function syncMovedConversationWorkDir(backend, sessionId, toWorkDir, providerId, fromWorkDir) {
     if (backend === 'copilot') {
         for (const conv of getCopilotConversations().values()) {
             if (conv.copilotSessionId === sessionId || conv.lastCopilotSessionId === sessionId)
@@ -133,6 +147,14 @@ function syncMovedConversationWorkDir(backend, sessionId, toWorkDir) {
                 conv.workDir = toWorkDir;
         }
     }
+    else if (backend === 'pi') {
+        for (const conv of getPiConversations().values()) {
+            if ((conv.piSessionId === sessionId || conv.lastPiSessionId === sessionId)
+                && (!providerId || conv.providerId === providerId)
+                && (!fromWorkDir || normalizeDirForCompare(conv.workDir) === normalizeDirForCompare(fromWorkDir)))
+                conv.workDir = toWorkDir;
+        }
+    }
     else {
         for (const conv of getConversations().values()) {
             if (conv.claudeSessionId === sessionId || conv.lastClaudeSessionId === sessionId)
@@ -195,6 +217,13 @@ export function initializeLocalConnection(config) {
         catch { /* ignore */ }
     }
     unsubscribeCodexSend = addCodexSendFn(send);
+    if (unsubscribePiSend) {
+        try {
+            unsubscribePiSend();
+        }
+        catch { /* ignore */ }
+    }
+    unsubscribePiSend = addPiSendFn(send);
     // Wire up action item session linkage
     if (unsubscribeSessionStarted) {
         try {
@@ -492,6 +521,7 @@ export function disconnect() {
     abortAllClaude();
     cleanupAllCopilotConversations();
     cleanupAllCodexConversations();
+    cleanupAllPiConversations();
     if (unsubscribeSend) {
         try {
             unsubscribeSend();
@@ -513,6 +543,13 @@ export function disconnect() {
         catch { /* ignore */ }
         unsubscribeCodexSend = null;
     }
+    if (unsubscribePiSend) {
+        try {
+            unsubscribePiSend();
+        }
+        catch { /* ignore */ }
+        unsubscribePiSend = null;
+    }
     if (unsubscribeSessionStarted) {
         try {
             unsubscribeSessionStarted();
@@ -625,7 +662,8 @@ export function getConnectionStatus() {
         sessionRegistration: state.sessionRegistration,
         conversations: getConversations().size
             + getCopilotConversations().size
-            + getCodexConversations().size,
+            + getCodexConversations().size
+            + getPiConversations().size,
     };
 }
 function buildWsUrl(config) {
@@ -854,7 +892,20 @@ function handleServerMessage(msg) {
                 }
             }
             // Route to the selected backend.
-            if (isCodexBackend()) {
+            if (isPiBackend()) {
+                const piMsg = msg;
+                const piState = chatConvId ? getPiConversation(chatConvId) : undefined;
+                const piResumeDir = chatOptions.resumeSessionId
+                    ? getPiSessionWorkDir(chatOptions.resumeSessionId, effectiveProviderId || 'pi', explicitWorkDir)
+                    : undefined;
+                const piWorkDir = explicitWorkDir || piState?.workDir || piResumeDir || chatDir;
+                handlePiChat(chatConvId || '', piMsg.prompt, piWorkDir, {
+                    resumeSessionId: chatOptions.resumeSessionId,
+                    providerId: effectiveProviderId,
+                    piOptions: piMsg.piOptions,
+                }, piMsg.files);
+            }
+            else if (isCodexBackend()) {
                 const codexMsg = msg;
                 handleCodexChat(chatConvId || '', codexMsg.prompt, chatDir, {
                     resumeSessionId: chatOptions.resumeSessionId,
@@ -876,14 +927,34 @@ function handleServerMessage(msg) {
             break;
         }
         case 'cancel_execution':
-            if (isCodexBackend()) {
-                cancelCodexExecution(msg.conversationId || '');
-            }
-            else if (isCopilotBackend()) {
-                cancelCopilotExecution(msg.conversationId || '');
-            }
-            else {
-                claudeCancelExecution(msg.conversationId);
+            {
+                const conversationId = msg.conversationId || '';
+                // Cancellation follows conversation ownership, not the currently selected
+                // provider. This keeps background turns cancellable after a tab switches.
+                if (getPiConversation(conversationId)) {
+                    cancelPiExecution(conversationId);
+                }
+                else if (getCodexConversation(conversationId)) {
+                    cancelCodexExecution(conversationId);
+                }
+                else if (getCopilotConversation(conversationId)) {
+                    cancelCopilotExecution(conversationId);
+                }
+                else if (getConversation(conversationId)) {
+                    claudeCancelExecution(conversationId);
+                }
+                else if (isPiBackend()) {
+                    cancelPiExecution(conversationId);
+                }
+                else if (isCodexBackend()) {
+                    cancelCodexExecution(conversationId);
+                }
+                else if (isCopilotBackend()) {
+                    cancelCopilotExecution(conversationId);
+                }
+                else {
+                    claudeCancelExecution(conversationId || undefined);
+                }
             }
             break;
         case 'list_sessions': {
@@ -893,7 +964,17 @@ function handleServerMessage(msg) {
             const sessionBackend = m.providerId
                 ? getBackendTypeForProvider(m.providerId)
                 : getBackendType();
-            if (sessionBackend === 'codex') {
+            if (sessionBackend === 'pi') {
+                handleListPiSessionsFn(dir, {
+                    limit: m.limit,
+                    cursor: m.cursor,
+                    mode: m.mode,
+                    requestId: m.requestId,
+                    providerId: m.providerId,
+                    legacy: m.limit == null && !m.cursor && !m.mode && !m.requestId,
+                });
+            }
+            else if (sessionBackend === 'codex') {
                 handleListCodexSessionsFn(dir, {
                     limit: m.limit,
                     cursor: m.cursor,
@@ -933,7 +1014,25 @@ function handleServerMessage(msg) {
             const recentBackend = m.providerId
                 ? getBackendTypeForProvider(m.providerId)
                 : getBackendType();
-            if (recentBackend === 'codex') {
+            if (recentBackend === 'pi') {
+                const page = listAllPiSessionsPage({
+                    limit: m.limit,
+                    cursor: m.cursor,
+                    providerId: m.providerId || 'pi',
+                });
+                console.log(`[AgentLink] → recent_sessions_list (${page.sessions.length} PI 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 } : {}),
+                    providerId: m.providerId || 'pi',
+                });
+            }
+            else if (recentBackend === 'codex') {
                 const page = listAllCodexSessionsPage({ limit: m.limit, cursor: m.cursor });
                 console.log(`[AgentLink] → recent_sessions_list (${page.sessions.length} codex sessions) hasMore=${page.hasMore}`);
                 send({
@@ -1080,6 +1179,51 @@ function handleServerMessage(msg) {
             const resumeBackend = m.providerId
                 ? getBackendTypeForProvider(m.providerId)
                 : getBackendType();
+            if (resumeBackend === 'pi') {
+                const persistedWorkDir = getPiSessionWorkDir(m.claudeSessionId, m.providerId || 'pi', explicitWorkDir);
+                if (convId) {
+                    rebindPiConversation(m.claudeSessionId, convId, m.providerId || 'pi', explicitWorkDir || persistedWorkDir);
+                }
+                const reboundState = getPiConversation(convId || '');
+                const authoritativeWorkDir = reboundState?.workDir || persistedWorkDir;
+                const resolvedWorkDir = explicitWorkDir || authoritativeWorkDir || state.workDir;
+                if (!authoritativeWorkDir
+                    || normalizeDirForCompare(authoritativeWorkDir) !== normalizeDirForCompare(resolvedWorkDir)) {
+                    send({
+                        type: 'error',
+                        message: `PI session ${m.claudeSessionId} was not found in ${resolvedWorkDir}.`,
+                        conversationId: convId,
+                    });
+                    break;
+                }
+                if (convId && !getPiConversation(convId)) {
+                    createPiPlaceholder(convId, { providerId: m.providerId || 'pi', workDir: resolvedWorkDir });
+                }
+                const piState = getPiConversation(convId || '');
+                if (piState) {
+                    piState.piSessionId = m.claudeSessionId;
+                    piState.lastPiSessionId = m.claudeSessionId;
+                    piState.workDir = resolvedWorkDir;
+                }
+                const history = readPiSessionMessages(resolvedWorkDir, m.claudeSessionId, m.providerId || 'pi');
+                const isPiProcessing = !!(piState?.turnActive && piState.child);
+                console.log(`[AgentLink] → conversation_resumed (${history.length} PI messages, session ${m.claudeSessionId.slice(0, 8)})`);
+                send({
+                    type: 'conversation_resumed',
+                    conversationId: convId,
+                    claudeSessionId: m.claudeSessionId,
+                    history,
+                    isCompacting: false,
+                    isProcessing: isPiProcessing,
+                    planMode: false,
+                    permissionMode: 'normal',
+                    workDir: resolvedWorkDir,
+                    providerId: m.providerId || 'pi',
+                    ...(typeof m.scrollToMessageIdx === 'number' ? { scrollToMessageIdx: m.scrollToMessageIdx } : {}),
+                    ...(m.searchQuery ? { searchQuery: m.searchQuery } : {}),
+                });
+                break;
+            }
             if (resumeBackend === 'codex') {
                 if (convId) {
                     rebindCodexConversation(m.claudeSessionId, convId);
@@ -1113,6 +1257,12 @@ function handleServerMessage(msg) {
             }
             // Route to copilot backend
             if (resumeBackend === 'copilot') {
+                // Copilot rewrites workspace.yaml using the cwd of each resume process.
+                // Recover the persisted cwd before creating a placeholder so a reconnect
+                // that omits workDir cannot silently move the session to the agent cwd.
+                const resolvedWorkDir = explicitWorkDir
+                    || getCopilotSessionWorkDir(m.claudeSessionId)
+                    || state.workDir;
                 // Rebind running copilot process to new conversationId (handles page refresh)
                 if (convId) {
                     rebindCopilotConversation(m.claudeSessionId, convId);
@@ -1123,10 +1273,9 @@ function handleServerMessage(msg) {
                     const placeholder = getCopilotConversation(convId);
                     if (placeholder) {
                         placeholder.copilotSessionId = m.claudeSessionId;
-                        placeholder.workDir = explicitWorkDir || state.workDir;
+                        placeholder.workDir = resolvedWorkDir;
                     }
                 }
-                const resolvedWorkDir = explicitWorkDir || state.workDir;
                 const history = readCopilotSessionMessages(resolvedWorkDir, m.claudeSessionId);
                 // Check if this session has an active copilot process running
                 const copilotState = getCopilotConversation(convId || '');
@@ -1141,6 +1290,7 @@ function handleServerMessage(msg) {
                     isProcessing: isCopilotProcessing,
                     planMode: false,
                     permissionMode: 'normal',
+                    workDir: resolvedWorkDir,
                     ...(typeof m.scrollToMessageIdx === 'number' ? { scrollToMessageIdx: m.scrollToMessageIdx } : {}),
                     ...(m.searchQuery ? { searchQuery: m.searchQuery } : {}),
                 });
@@ -1253,7 +1403,27 @@ function handleServerMessage(msg) {
             const dir = m.workDir?.trim()
                 ? path.resolve(m.workDir)
                 : (m.providerId === 'brain' ? BRAIN_DATA_DIR : state.workDir);
-            if (sessionBackend === 'codex') {
+            if (sessionBackend === 'pi') {
+                if (evictByPiSessionId(m.sessionId, m.providerId || 'pi', dir)) {
+                    send({ type: 'error', message: 'Cannot delete a session while it is processing.' });
+                    break;
+                }
+                const ok = deletePiSession(dir, m.sessionId, m.providerId || 'pi');
+                console.log(`[AgentLink] delete PI session ${m.sessionId}: ${ok}`);
+                if (ok) {
+                    send({
+                        type: 'session_deleted',
+                        sessionId: m.sessionId,
+                        providerId: m.providerId || 'pi',
+                        workDir: dir,
+                    });
+                }
+                else {
+                    send({ type: 'error', message: 'Session not found or could not be deleted.' });
+                }
+                handleListPiSessionsFn(dir, { providerId: m.providerId || 'pi' });
+            }
+            else if (sessionBackend === 'codex') {
                 if (evictByCodexSessionId(m.sessionId)) {
                     send({ type: 'error', message: 'Cannot delete a session while it is processing.' });
                     break;
@@ -1284,7 +1454,28 @@ function handleServerMessage(msg) {
             const dir = m.workDir?.trim()
                 ? path.resolve(m.workDir)
                 : (m.providerId === 'brain' ? BRAIN_DATA_DIR : state.workDir);
-            if (sessionBackend === 'codex') {
+            if (sessionBackend === 'pi') {
+                if (hasActiveSessionConversation('pi', m.sessionId, m.providerId || 'pi', dir)) {
+                    send({ type: 'error', message: 'Cannot rename a session while it is processing.' });
+                    break;
+                }
+                const ok = renamePiSession(dir, m.sessionId, m.newTitle, m.providerId || 'pi');
+                console.log(`[AgentLink] rename PI session ${m.sessionId}: ${ok}`);
+                if (ok) {
+                    send({
+                        type: 'session_renamed',
+                        sessionId: m.sessionId,
+                        newTitle: m.newTitle,
+                        providerId: m.providerId || 'pi',
+                        workDir: dir,
+                    });
+                }
+                else {
+                    send({ type: 'error', message: 'Session not found or could not be renamed.' });
+                }
+                handleListPiSessionsFn(dir, { providerId: m.providerId || 'pi' });
+            }
+            else if (sessionBackend === 'codex') {
                 const ok = renameCodexSession(dir, m.sessionId, m.newTitle);
                 console.log(`[AgentLink] rename codex session ${m.sessionId}: ${ok}`);
                 if (ok)
@@ -1322,12 +1513,15 @@ function handleServerMessage(msg) {
             const sessionBackend = m.providerId
                 ? getBackendTypeForProvider(m.providerId)
                 : getBackendType();
-            if (hasActiveSessionConversation(sessionBackend, m.sessionId)) {
+            if (hasActiveSessionConversation(sessionBackend, m.sessionId, m.providerId, fromDir)) {
                 send({ type: 'error', message: 'Cannot move a session while it is processing.' });
                 break;
             }
             let ok = false;
-            if (sessionBackend === 'codex') {
+            if (sessionBackend === 'pi') {
+                ok = movePiSession(fromDir, toDir, m.sessionId, m.providerId || 'pi');
+            }
+            else if (sessionBackend === 'codex') {
                 ok = moveCodexSession(fromDir, toDir, m.sessionId);
             }
             else if (sessionBackend === 'copilot') {
@@ -1337,7 +1531,7 @@ function handleServerMessage(msg) {
                 ok = moveSession(fromDir, toDir, m.sessionId);
             }
             if (ok) {
-                syncMovedConversationWorkDir(sessionBackend, m.sessionId, toDir);
+                syncMovedConversationWorkDir(sessionBackend, m.sessionId, toDir, m.providerId, fromDir);
                 if (sessionBackend === 'claude') {
                     evictByClaudeSessionId(m.sessionId);
                 }
@@ -1347,9 +1541,15 @@ function handleServerMessage(msg) {
                     sessionId: m.sessionId,
                     fromWorkDir: fromDir,
                     toWorkDir: toDir,
-                    ...(m.providerId ? { providerId: m.providerId } : {}),
+                    ...(sessionBackend === 'pi'
+                        ? { providerId: m.providerId || 'pi', workDir: toDir }
+                        : (m.providerId ? { providerId: m.providerId } : {})),
                 });
-                if (sessionBackend === 'codex') {
+                if (sessionBackend === 'pi') {
+                    handleListPiSessionsFn(fromDir, { providerId: m.providerId || 'pi' });
+                    handleListPiSessionsFn(toDir, { providerId: m.providerId || 'pi' });
+                }
+                else if (sessionBackend === 'codex') {
                     handleListCodexSessionsFn(fromDir, { providerId: m.providerId });
                     handleListCodexSessionsFn(toDir, { providerId: m.providerId });
                 }
@@ -1379,7 +1579,13 @@ function handleServerMessage(msg) {
                 break;
             }
             let readResult;
-            if (exportBackend === 'codex') {
+            if (exportBackend === 'pi') {
+                readResult = {
+                    history: readPiSessionMessages(dir, m.sessionId, m.providerId || 'pi'),
+                    workDir: dir,
+                };
+            }
+            else if (exportBackend === 'codex') {
                 readResult = { history: readCodexSessionMessages(dir, m.sessionId), workDir: dir };
             }
             else if (exportBackend === 'copilot') {
@@ -1419,14 +1625,17 @@ function handleServerMessage(msg) {
                 });
                 break;
             }
-            sendSessionExportResult(requestId, m.sessionId, result);
+            sendSessionExportResult(requestId, m.sessionId, result, {
+                providerId: m.providerId || (exportBackend === 'pi' ? 'pi' : undefined),
+                workDir: readResult.workDir,
+            });
             break;
         }
         case 'query_active_conversations': {
             const request = msg;
             const active = [];
             const pushActive = (convId, conv) => {
-                const sessionId = conv.claudeSessionId ?? conv.copilotSessionId ?? conv.codexSessionId ?? null;
+                const sessionId = conv.claudeSessionId ?? conv.copilotSessionId ?? conv.codexSessionId ?? conv.piSessionId ?? null;
                 if (conv.turnActive) {
                     active.push({
                         conversationId: convId,
@@ -1459,6 +1668,9 @@ function handleServerMessage(msg) {
             for (const [convId, conv] of getCodexConversations()) {
                 pushActive(convId, conv);
             }
+            for (const [convId, conv] of getPiConversations()) {
+                pushActive(convId, conv);
+            }
             const activeTeamState = getActiveTeam();
             // Collect running loop executions for status recovery
             const runningLoops = [];
@@ -1485,7 +1697,7 @@ function handleServerMessage(msg) {
                 return {
                     type: 'turn_status',
                     conversationId: convId,
-                    claudeSessionId: conv.claudeSessionId ?? conv.copilotSessionId ?? conv.codexSessionId ?? null,
+                    claudeSessionId: conv.claudeSessionId ?? conv.copilotSessionId ?? conv.codexSessionId ?? conv.piSessionId ?? null,
                     turnId: conv.turnId,
                     status,
                     isProcessing: isProc,
@@ -1520,6 +1732,15 @@ function handleServerMessage(msg) {
                         }
                     }
                 }
+                if (!found) {
+                    for (const [convId, conv] of getPiConversations()) {
+                        if (conv.turnId === m.turnId) {
+                            send(buildTurnStatus(convId, conv));
+                            found = true;
+                            break;
+                        }
+                    }
+                }
                 if (!found) {
                     send({ type: 'turn_status', turnId: m.turnId, status: 'unknown', isProcessing: false, isCompacting: false, stateVersion: 0 });
                 }
@@ -1540,7 +1761,13 @@ function handleServerMessage(msg) {
                             send(buildTurnStatus(m.conversationId, codexConv));
                         }
                         else {
-                            send({ type: 'turn_status', conversationId: m.conversationId, status: 'unknown', isProcessing: false, isCompacting: false, stateVersion: 0 });
+                            const piConv = getPiConversation(m.conversationId);
+                            if (piConv) {
+                                send(buildTurnStatus(m.conversationId, piConv));
+                            }
+                            else {
+                                send({ type: 'turn_status', conversationId: m.conversationId, status: 'unknown', isProcessing: false, isCompacting: false, stateVersion: 0 });
+                            }
                         }
                     }
                 }
@@ -1740,6 +1967,21 @@ function handleServerMessage(msg) {
                 setBackendType(newBackendType);
                 console.log(`[AgentLink] Backend switched: ${oldBackendType} → ${newBackendType}`);
             }
+            else if (oldBackendType === 'pi') {
+                const existingPiConversation = conversationId ? getPiConversation(conversationId) : undefined;
+                if (existingPiConversation?.turnActive) {
+                    // Keep the live process bound to the provider that launched it.
+                    // The next chat carries the new providerId and restarts PI after settlement.
+                }
+                else if (conversationId && existingPiConversation) {
+                    const result = restartPiConversation(conversationId, { providerId });
+                    if (result.wasTurnActive)
+                        send({ type: 'execution_cancelled', conversationId });
+                }
+                else if (conversationId) {
+                    createPiPlaceholder(conversationId, { providerId });
+                }
+            }
             else if (oldBackendType === 'codex') {
                 const existingCodexConv = conversationId ? getCodexConversation(conversationId) : undefined;
                 if (existingCodexConv?.turnActive) {
@@ -2428,7 +2670,53 @@ function handleListCodexSessionsFn(dir, options = {}) {
         });
     }
 }
-function sendSessionExportResult(requestId, sessionId, result) {
+function handleListPiSessionsFn(dir, options = {}) {
+    try {
+        const page = options.legacy
+            ? {
+                ...listPiSessionsPage(dir, {
+                    limit: Number.POSITIVE_INFINITY,
+                    providerId: options.providerId || 'pi',
+                }),
+                hasMore: false,
+            }
+            : listPiSessionsPage(dir, {
+                limit: options.limit ?? DEFAULT_SESSION_LIST_LIMIT,
+                cursor: options.cursor,
+                providerId: options.providerId || 'pi',
+            });
+        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} PI sessions for ${dir}) hasMore=${page.hasMore}`);
+        send({
+            type: 'sessions_list',
+            sessions: page.sessions,
+            workDir: dir,
+            hasMore: page.hasMore,
+            nextCursor: page.nextCursor,
+            indexState: 'ready',
+            ...(options.mode ? { mode: options.mode } : {}),
+            ...(options.requestId ? { requestId: options.requestId } : {}),
+            providerId: options.providerId || 'pi',
+        });
+    }
+    catch (err) {
+        console.error('[AgentLink] listPiSessions failed:', err);
+        send({
+            type: 'sessions_list',
+            sessions: [],
+            workDir: dir,
+            hasMore: false,
+            indexState: 'ready',
+            error: err.message,
+            ...(options.mode ? { mode: options.mode } : {}),
+            ...(options.requestId ? { requestId: options.requestId } : {}),
+            providerId: options.providerId || 'pi',
+        });
+    }
+}
+function sendSessionExportResult(requestId, sessionId, result, scope) {
     const totalChunks = Math.max(1, Math.ceil(result.content.length / SESSION_EXPORT_CHUNK_SIZE));
     console.log(`[AgentLink] → session_exported (${result.format}, ${Buffer.byteLength(result.content, 'utf8')} bytes, ${totalChunks} chunks)`);
     if (totalChunks === 1) {
@@ -2437,6 +2725,8 @@ function sendSessionExportResult(requestId, sessionId, result) {
             requestId,
             sessionId,
             ...result,
+            ...(scope?.providerId ? { providerId: scope.providerId } : {}),
+            ...(scope?.workDir ? { workDir: scope.workDir } : {}),
         });
         return;
     }
@@ -2452,6 +2742,8 @@ function sendSessionExportResult(requestId, sessionId, result) {
             chunkIndex,
             totalChunks,
             content: result.content.slice(start, start + SESSION_EXPORT_CHUNK_SIZE),
+            ...(scope?.providerId ? { providerId: scope.providerId } : {}),
+            ...(scope?.workDir ? { workDir: scope.workDir } : {}),
         });
     }
 }