@averyyy/pi-coding-agent 0.80.6-piclient.4 → 0.80.6-piclient.5

package/dist/modes/interactive/interactive-mode.js

@@ -20,6 +20,7 @@ import { KeybindingsManager } from "../../core/keybindings.js";
 import { createCompactionSummaryMessage } from "../../core/messages.js";
 import { defaultModelPerProvider, findExactModelReferenceMatch, resolveModelScope } from "../../core/model-resolver.js";
 import { DefaultPackageManager } from "../../core/package-manager.js";
+import { getPiClientUpdateMarkerPath, PI_CLIENT_RUNTIME_RELOAD_EXIT_CODE, readPiClientUpdateMarker, writePiClientRuntimeReloadState, } from "../../core/pi-client-runtime-update.js";
 import { BUILT_IN_PROVIDER_DISPLAY_NAMES } from "../../core/provider-display-names.js";
 import { formatMissingSessionCwdPrompt, MissingSessionCwdError } from "../../core/session-cwd.js";
 import { SessionManager, sessionEntryToContextMessages } from "../../core/session-manager.js";
@@ -2181,6 +2182,12 @@ export class InteractiveMode {
                 await this.handleReloadCommand();
                 return;
             }
+            const pendingPiClientUpdate = readPiClientUpdateMarker();
+            if (pendingPiClientUpdate) {
+                this.showWarning(`pi-client ${pendingPiClientUpdate.version} is installed. Run /reload to continue this session with the new version.`);
+                this.editor.setText(text);
+                return;
+            }
             if (text === "/debug") {
                 this.handleDebugCommand();
                 this.editor.setText("");
@@ -4368,6 +4375,25 @@ export class InteractiveMode {
             this.showWarning("Wait for compaction to finish before reloading.");
             return;
         }
+        const pendingPiClientUpdate = readPiClientUpdateMarker();
+        if (pendingPiClientUpdate) {
+            const reloadStatePath = process.env.PI_CLIENT_RELOAD_STATE_PATH;
+            if (!reloadStatePath) {
+                this.showWarning(`pi-client ${pendingPiClientUpdate.version} is installed. Restart pi-client to load the new runtime.`);
+                return;
+            }
+            writePiClientRuntimeReloadState(reloadStatePath, {
+                sessionId: this.sessionManager.getSessionId(),
+                sessionDir: this.sessionManager.usesDefaultSessionDir() ? undefined : this.sessionManager.getSessionDir(),
+                updateMarkerPath: getPiClientUpdateMarkerPath(),
+            });
+            this.isShuttingDown = true;
+            this.themeController.disableAutoSync();
+            await this.ui.terminal.drainInput(1000);
+            this.stop();
+            await this.runtimeHost.dispose();
+            process.exit(PI_CLIENT_RUNTIME_RELOAD_EXIT_CODE);
+        }
         this.resetExtensionUI();
         const reloadBox = new Container();
         const borderColor = (s) => theme.fg("border", s);