pi-ui-extend 0.1.35 → 0.1.36

package/dist/app/app.d.ts

@@ -66,6 +66,7 @@ export declare class PiUiExtendApp {
     start(): Promise<void>;
     private checkPixUpdateOnStartup;
     private bindCurrentSession;
+    private awaitCurrentSessionExtensions;
     private activateRuntime;
     private createExtensionEventBus;
     private handleTerminalBellAttention;

package/dist/app/app.js

@@ -172,6 +172,7 @@ export class PiUiExtendApp {
                 noSession: false,
                 sessionPath,
             }),
+            awaitCurrentSessionExtensions: (runtime) => this.awaitCurrentSessionExtensions(runtime),
             activateRuntime: (runtime, options) => this.activateRuntime(runtime, options),
             disposeRuntime: (runtime) => this.terminalController.disposeRuntime(runtime),
             isRunning: () => this.running,
@@ -317,6 +318,7 @@ export class PiUiExtendApp {
             isRunning: () => this.running,
             getInput: () => this.input,
             setInput: (value) => this.setInput(value),
+            awaitCurrentSessionExtensions: (runtime) => this.awaitCurrentSessionExtensions(runtime),
             resetSessionView: () => this.resetSessionView(),
             loadSessionHistory: () => this.loadSessionHistory(),
             afterSessionReplacement: (message) => this.afterSessionReplacement(message),
@@ -341,6 +343,7 @@ export class PiUiExtendApp {
         this.workspaceActions = new AppWorkspaceActionsController({
             entries: this.entries,
             runtime: () => this.runtime,
+            awaitCurrentSessionExtensions: (runtime) => this.awaitCurrentSessionExtensions(runtime),
             findUserEntry: (entryId) => this.findUserEntry(entryId),
             touchEntry: (entry) => this.touchEntry(entry),
             resetSessionView: () => this.resetSessionView(),
@@ -471,6 +474,7 @@ export class PiUiExtendApp {
             showMenu: (items, options) => this.popupMenus.menuController.show(items, options),
             getModelMenuItems: (query) => this.menuItems.getModelMenuItems(query),
             getThinkingMenuItems: (query) => this.menuItems.getThinkingMenuItems(query),
+            awaitCurrentSessionExtensions: (runtime) => this.awaitCurrentSessionExtensions(runtime),
             modelRef: (model) => this.menuItems.modelRef(model),
             getFavoriteScopedModels: () => this.menuItems.getFavoriteScopedModels(),
             setSessionStatus: (session) => this.setSessionStatus(session),
@@ -519,6 +523,7 @@ export class PiUiExtendApp {
             showToast: (message, kind) => this.showToast(message, kind),
             render: () => this.render(),
             resetSessionView: () => this.resetSessionView(),
+            awaitCurrentSessionExtensions: (runtime) => this.awaitCurrentSessionExtensions(runtime),
             bindCurrentSession: () => this.bindCurrentSession(),
             loadSessionHistory: () => this.loadSessionHistory(),
             scrollToConversationEntry: (entryId) => this.scrollController.scrollToConversationEntry(entryId),
@@ -824,6 +829,9 @@ export class PiUiExtendApp {
     async bindCurrentSession(options) {
         await this.sessionLifecycle.bindCurrentSession(options);
     }
+    async awaitCurrentSessionExtensions(runtime) {
+        await this.sessionLifecycle.awaitCurrentSessionExtensions(runtime);
+    }
     async activateRuntime(runtime, options) {
         this.runtime = runtime;
         runtime.setRebindSession(async () => {

package/dist/app/commands/command-host.d.ts

@@ -7,6 +7,7 @@ export type DirectPopupMenu = Exclude<ActivePopupMenu, "slash">;
 export type CommandControllerHost = {
     readonly options: AppOptions;
     runtime(): AgentSessionRuntime | undefined;
+    awaitCurrentSessionExtensions(runtime?: AgentSessionRuntime): Promise<void>;
     requestHistory(): AppRequestHistory;
     getInput(): string;
     setInput(value: string): void;

package/dist/app/commands/command-model-actions.d.ts

@@ -14,6 +14,7 @@ export declare class ModelCommandActions {
     runModelCommand(model: SessionModel): Promise<void>;
     runThinkingCommand(level: ThinkingLevel): Promise<void>;
     private addPersistentSystemEntry;
+    private reloadAfterModelChange;
     private saveDefaultModel;
     private saveDefaultThinking;
 }

package/dist/app/commands/command-model-actions.js

@@ -269,6 +269,17 @@ export class ModelCommandActions {
         this.host.render();
         await runtime.session.setModel(model);
         this.host.addEntry({ id: createId("system"), kind: "system", text: `Selected model ${ref}` });
+        if (runtime.session.isStreaming) {
+            this.host.addEntry({
+                id: createId("system"),
+                kind: "system",
+                text: "Skipped reload because the agent is still running. Run /reload when idle to refresh model-specific tools.",
+            });
+            this.host.toast.warning("Model changed; reload skipped while the agent is running");
+            this.host.setSessionStatus(runtime.session);
+            return;
+        }
+        await this.reloadAfterModelChange(runtime.session, ref);
         this.host.setSessionStatus(runtime.session);
     }
     async runThinkingCommand(level) {
@@ -285,6 +296,27 @@ export class ModelCommandActions {
         appendPixSystemDisplayEntry(session, text);
         this.host.addEntry({ id: createId("system"), kind: "system", text });
     }
+    async reloadAfterModelChange(session, ref) {
+        this.host.setStatus(`reloading resources for ${ref}`);
+        this.host.render();
+        try {
+            await session.reload();
+            this.host.addEntry({
+                id: createId("system"),
+                kind: "system",
+                text: `Reloaded resources after model change to ${ref}`,
+            });
+            this.host.toast.success("Model changed and resources reloaded");
+        }
+        catch (error) {
+            this.host.addEntry({
+                id: createId("error"),
+                kind: "error",
+                text: `Model changed to ${ref}, but reload failed: ${error instanceof Error ? error.message : String(error)}`,
+            });
+            this.host.toast.error("Model changed, but reload failed");
+        }
+    }
     saveDefaultModel(modelRef) {
         const saved = savePixDefaultModel(modelRef);
         if (!saved)

package/dist/app/commands/command-navigation-actions.js

@@ -50,6 +50,7 @@ export class NavigationCommandActions {
             throw new Error("No user messages to fork from");
         this.host.setStatus("forking session");
         this.host.render();
+        await this.host.awaitCurrentSessionExtensions(runtime);
         const result = await runtime.fork(entryId);
         if (result.cancelled) {
             this.host.addEntry({ id: createId("system"), kind: "system", text: "Fork cancelled." });
@@ -74,6 +75,7 @@ export class NavigationCommandActions {
         }
         this.host.setStatus("cloning session");
         this.host.render();
+        await this.host.awaitCurrentSessionExtensions(runtime);
         const result = await runtime.fork(leafId, { position: "at" });
         if (result.cancelled) {
             this.host.addEntry({ id: createId("system"), kind: "system", text: "Clone cancelled." });
@@ -228,6 +230,7 @@ export class NavigationCommandActions {
         const resolvedSessionPath = resolve(runtime.cwd, sessionPath);
         this.host.setStatus("switching session");
         this.host.render();
+        await this.host.awaitCurrentSessionExtensions(runtime);
         const result = await runtime.switchSession(resolvedSessionPath);
         if (result.cancelled) {
             this.host.addEntry({ id: createId("system"), kind: "system", text: "Resume cancelled." });

package/dist/app/commands/command-session-actions.js

@@ -260,6 +260,7 @@ export class SessionCommandActions {
         this.host.setStatus("reloading");
         this.host.render();
         try {
+            await this.host.awaitCurrentSessionExtensions(runtime);
             await runtime.session.reload();
             this.host.setSessionStatus(runtime.session);
             this.host.addEntry({ id: createId("system"), kind: "system", text: "Reloaded keybindings, extensions, skills, prompts, themes" });
@@ -277,6 +278,7 @@ export class SessionCommandActions {
             return;
         this.host.setStatus("starting new session");
         this.host.render();
+        await this.host.awaitCurrentSessionExtensions(runtime);
         const result = await runtime.newSession();
         if (result.cancelled) {
             this.host.addEntry({ id: createId("system"), kind: "system", text: "New session cancelled." });

package/dist/app/extensions/extension-actions-controller.d.ts

@@ -4,6 +4,7 @@ export type AppExtensionActionsHost = {
     isRunning(): boolean;
     getInput(): string;
     setInput(value: string): void;
+    awaitCurrentSessionExtensions(runtime?: AgentSessionRuntime): Promise<void>;
     resetSessionView(): void;
     loadSessionHistory(): void;
     afterSessionReplacement(message?: string): void;

package/dist/app/extensions/extension-actions-controller.js

@@ -8,12 +8,14 @@ export class AppExtensionActionsController {
         return {
             waitForIdle: () => this.waitForSessionIdle(runtime),
             newSession: async (options) => {
+                await this.host.awaitCurrentSessionExtensions(runtime);
                 const result = await runtime.newSession(options);
                 if (!result.cancelled)
                     this.host.afterSessionReplacement("Started a new session.");
                 return result;
             },
             fork: async (entryId, options) => {
+                await this.host.awaitCurrentSessionExtensions(runtime);
                 const result = await runtime.fork(entryId, options);
                 if (!result.cancelled)
                     this.host.afterSessionReplacement("Forked to a new session.");
@@ -32,12 +34,14 @@ export class AppExtensionActionsController {
                 return result;
             },
             switchSession: async (sessionPath, options) => {
+                await this.host.awaitCurrentSessionExtensions(runtime);
                 const result = await runtime.switchSession(sessionPath, options);
                 if (!result.cancelled)
                     this.host.afterSessionReplacement(`Switched session: ${sessionPath}`);
                 return result;
             },
             reload: async () => {
+                await this.host.awaitCurrentSessionExtensions(runtime);
                 await runtime.session.reload();
                 this.host.setSessionStatus(runtime.session);
                 this.host.showToast("Reloaded resources", "success");

package/dist/app/input/input-controller.d.ts

@@ -46,6 +46,7 @@ export declare class AppInputController {
     private consumeCommandArrowPageMatch;
     private consumeTerminalEditShortcutSequence;
     private consumeIgnoredModifiedKeySequence;
+    private consumeModifiedArrowKeySequence;
     private consumeClipboardImagePasteSequence;
     private consumeShiftEnterSequence;
     private consumeTerminalInterruptSequence;

package/dist/app/input/input-controller.js

@@ -1,6 +1,6 @@
 import { InputPasteHandler } from "./input-paste-handler.js";
 import { hasTerminalCommandModifier, isNativeCommandPressed, isNativeShiftPressed } from "./native-modifiers.js";
-import { parseTerminalEditShortcutSequence, parseTerminalInterruptSequence, parseTerminalModifiedKeySequence, terminalEditShortcutForControlChar, terminalKeyIsClipboardImagePaste, terminalKeyIsShiftEnter, terminalKeyShouldIgnore, } from "./terminal-edit-shortcuts.js";
+import { parseTerminalEditShortcutSequence, parseTerminalInterruptSequence, parseTerminalModifiedKeySequence, terminalKeyArrowDirection, terminalEditShortcutForControlChar, terminalKeyIsClipboardImagePaste, terminalKeyIsShiftEnter, terminalKeyShouldIgnore, } from "./terminal-edit-shortcuts.js";
 const SHIFT_ENTER_ESCAPE_SEQUENCES = ["\x1b\r", "\x1b\n"];
 export class AppInputController {
     host;
@@ -112,6 +112,11 @@ export class AppInputController {
                 continue;
             if (terminalEditShortcutSequence === "pending")
                 return;
+            const modifiedArrowKeySequence = this.consumeModifiedArrowKeySequence();
+            if (modifiedArrowKeySequence === "consumed")
+                continue;
+            if (modifiedArrowKeySequence === "pending")
+                return;
             const ignoredModifiedKeySequence = this.consumeIgnoredModifiedKeySequence();
             if (ignoredModifiedKeySequence === "consumed")
                 continue;
@@ -277,6 +282,28 @@ export class AppInputController {
         this.inputBuffer = this.inputBuffer.slice(result.key.length);
         return "consumed";
     }
+    consumeModifiedArrowKeySequence() {
+        const result = parseTerminalModifiedKeySequence(this.inputBuffer);
+        if (result.kind === "pending")
+            return "pending";
+        if (result.kind === "none")
+            return "none";
+        const direction = terminalKeyArrowDirection(result.key);
+        if (!direction)
+            return "none";
+        this.inputBuffer = this.inputBuffer.slice(result.key.length);
+        if (terminalKeyShouldIgnore(result.key))
+            return "consumed";
+        if (direction === "up")
+            this.handleArrowUp();
+        else if (direction === "down")
+            this.handleArrowDown();
+        else if (direction === "right")
+            this.handleArrowRight();
+        else
+            this.handleArrowLeft();
+        return "consumed";
+    }
     consumeClipboardImagePasteSequence() {
         const result = parseTerminalModifiedKeySequence(this.inputBuffer);
         if (result.kind === "pending")

package/dist/app/input/terminal-edit-shortcuts.d.ts

@@ -39,5 +39,6 @@ export declare function parseTerminalInterruptSequence(input: string): {
 export declare function terminalKeyIsShiftEnter(key: ParsedModifiedKey): boolean;
 export declare function terminalKeyIsClipboardImagePaste(key: ParsedModifiedKey): boolean;
 export declare function terminalKeyShouldIgnore(key: ParsedModifiedKey): boolean;
+export declare function terminalKeyArrowDirection(key: ParsedModifiedKey): "up" | "down" | "right" | "left" | undefined;
 export declare function terminalEditShortcutForControlChar(char: string, shiftPressed: boolean): TerminalEditShortcut | undefined;
 export {};

package/dist/app/input/terminal-edit-shortcuts.js

@@ -11,12 +11,22 @@ const CYRILLIC_SMALL_ES_CODE = 1089;
 const CYRILLIC_CAPITAL_ES_CODE = 1057;
 const CYRILLIC_SMALL_EM_CODE = 1084;
 const CYRILLIC_CAPITAL_EM_CODE = 1052;
+const KITTY_ARROW_CODEPOINTS = {
+    A: -1,
+    B: -2,
+    C: -3,
+    D: -4,
+};
 const KITTY_CSI_U_SEQUENCE = /^\x1b\[(\d+)(?::(\d*))?(?::(\d+))?(?:;(\d+))?(?::(\d+))?u/;
+const KITTY_ARROW_SEQUENCE = /^\x1b\[1;(\d+)(?::(\d+))?([ABCD])/;
 const XTERM_MODIFY_OTHER_KEYS_SEQUENCE = /^\x1b\[27;(\d+);(\d+)~/;
 export function parseTerminalModifiedKeySequence(input) {
     const kitty = parseKittyCsiUSequence(input);
     if (kitty)
         return { kind: "key", key: kitty };
+    const kittyArrow = parseKittyArrowSequence(input);
+    if (kittyArrow)
+        return { kind: "key", key: kittyArrow };
     const xterm = parseXtermModifyOtherKeysSequence(input);
     if (xterm)
         return { kind: "key", key: xterm };
@@ -57,6 +67,20 @@ export function terminalKeyIsClipboardImagePaste(key) {
 export function terminalKeyShouldIgnore(key) {
     return key.eventType === 3;
 }
+export function terminalKeyArrowDirection(key) {
+    const effectiveModifier = key.modifier & ~LOCK_MODIFIER_MASK;
+    if (effectiveModifier !== 0)
+        return undefined;
+    if (key.codepoint === KITTY_ARROW_CODEPOINTS.A)
+        return "up";
+    if (key.codepoint === KITTY_ARROW_CODEPOINTS.B)
+        return "down";
+    if (key.codepoint === KITTY_ARROW_CODEPOINTS.C)
+        return "right";
+    if (key.codepoint === KITTY_ARROW_CODEPOINTS.D)
+        return "left";
+    return undefined;
+}
 export function terminalEditShortcutForControlChar(char, shiftPressed) {
     if (char === "\u001a")
         return shiftPressed ? "redo" : "undo";
@@ -82,6 +106,24 @@ function parseKittyCsiUSequence(input) {
         length: match[0].length,
     };
 }
+function parseKittyArrowSequence(input) {
+    const match = KITTY_ARROW_SEQUENCE.exec(input);
+    if (!match)
+        return undefined;
+    const modifierValue = Number.parseInt(match[1] ?? "", 10);
+    const eventType = match[2] ? Number.parseInt(match[2], 10) : undefined;
+    const arrow = match[3];
+    const codepoint = arrow ? KITTY_ARROW_CODEPOINTS[arrow] : undefined;
+    if (!Number.isFinite(modifierValue) || codepoint === undefined)
+        return undefined;
+    return {
+        codepoint,
+        baseLayoutKey: undefined,
+        modifier: modifierValue - 1,
+        eventType: Number.isFinite(eventType) ? eventType : undefined,
+        length: match[0].length,
+    };
+}
 function parseXtermModifyOtherKeysSequence(input) {
     const match = XTERM_MODIFY_OTHER_KEYS_SEQUENCE.exec(input);
     if (!match)

package/dist/app/popup/popup-action-controller.d.ts

@@ -7,6 +7,7 @@ import type { Entry, SlashCommand, UserMessageJumpMenuValue } from "../types.js"
 import type { AppWorkspaceActionsController } from "../workspace/workspace-actions-controller.js";
 export type AppPopupActionControllerHost = {
     runtime(): AgentSessionRuntime | undefined;
+    awaitCurrentSessionExtensions(runtime?: AgentSessionRuntime): Promise<void>;
     getBuiltinSlashCommands(): readonly SlashCommand[];
     isRunning(): boolean;
     setInput(value: string): void;

package/dist/app/popup/popup-action-controller.js

@@ -228,6 +228,7 @@ export class AppPopupActionController {
         this.host.setStatus("switching session");
         this.host.render();
         try {
+            await this.host.awaitCurrentSessionExtensions(runtime);
             const result = await runtime.switchSession(session.path);
             if (result.cancelled) {
                 this.host.addEntry({ id: createId("system"), kind: "system", text: "Resume cancelled." });

package/dist/app/session/session-lifecycle-controller.d.ts

@@ -61,6 +61,7 @@ export declare class AppSessionLifecycleController {
     constructor(host: AppSessionLifecycleHost);
     start(): Promise<void>;
     bindCurrentSession(options?: BindCurrentSessionOptions): Promise<void>;
+    awaitCurrentSessionExtensions(runtime?: AgentSessionRuntime | undefined): Promise<void>;
     unsubscribeSession(): void;
     afterSessionReplacement(message?: string): void;
     private loadReplacementHistory;

package/dist/app/session/session-lifecycle-controller.js

@@ -85,6 +85,13 @@ export class AppSessionLifecycleController {
         }
         await bindPromise;
     }
+    async awaitCurrentSessionExtensions(runtime = this.host.runtime()) {
+        if (!runtime)
+            return;
+        if (this.extensionBindRuntime !== runtime || this.extensionBindSession !== runtime.session)
+            return;
+        await this.extensionBindPromise;
+    }
     unsubscribeSession() {
         this.unsubscribe?.();
     }

package/dist/app/session/tabs-controller.d.ts

@@ -18,6 +18,7 @@ export type AppTabsControllerHost = {
     runtime(): AgentSessionRuntime | undefined;
     createRuntimeForNewSession(): Promise<AgentSessionRuntime>;
     createRuntimeForSession(sessionPath: string): Promise<AgentSessionRuntime>;
+    awaitCurrentSessionExtensions?(runtime?: AgentSessionRuntime): Promise<void>;
     activateRuntime(runtime: AgentSessionRuntime, options?: BindCurrentSessionOptions): Promise<void>;
     disposeRuntime(runtime: AgentSessionRuntime): Promise<void>;
     isRunning(): boolean;

package/dist/app/session/tabs-controller.js

@@ -730,6 +730,7 @@ export class AppTabsController {
         this.activeTabId = tab.id;
         this.host.setStatus("starting new session");
         this.host.render();
+        await this.host.awaitCurrentSessionExtensions?.(runtime);
         const result = await runtime.newSession();
         if (result.cancelled) {
             this.host.addEntry({ id: createId("system"), kind: "system", text: "New session cancelled." });

package/dist/app/workspace/workspace-actions-controller.d.ts

@@ -4,6 +4,7 @@ import { type WorkspaceMutation, type WorkspaceMutationFromToolInput, type Works
 export type AppWorkspaceActionsControllerHost = {
     readonly entries: Entry[];
     runtime(): AgentSessionRuntime | undefined;
+    awaitCurrentSessionExtensions(runtime?: AgentSessionRuntime): Promise<void>;
     findUserEntry(entryId: string): Extract<Entry, {
         kind: "user";
     }> | undefined;

package/dist/app/workspace/workspace-actions-controller.js

@@ -94,6 +94,7 @@ export class AppWorkspaceActionsController {
             throw new Error("Session entry for this message is not available yet");
         this.host.setStatus("forking session");
         this.host.render();
+        await this.host.awaitCurrentSessionExtensions(runtime);
         const result = await runtime.fork(sessionEntryId);
         if (result.cancelled) {
             this.host.addEntry({ id: createId("system"), kind: "system", text: "Fork cancelled." });

package/external/pi-tools-suite/package.json

@@ -22,13 +22,13 @@
     "smoke:tools": "PI_OFFLINE=1 pi --no-session -p \"ping\"",
     "smoke": "npm run smoke:explicit && npm run smoke:auto && npm run smoke:tools",
     "test": "bun test test",
-    "test:async-subagents-e2e": "ASYNC_SUBAGENTS_E2E=1 ASYNC_SUBAGENTS_MODEL=zai/glm-5-turbo bun test --concurrent --max-concurrency=30 test/async-subagents",
+    "test:async-subagents-e2e": "ASYNC_SUBAGENTS_E2E=1 ASYNC_SUBAGENTS_DEBUG_LOGS=1 ASYNC_SUBAGENTS_MODEL=zai/glm-5-turbo bun test --concurrent --max-concurrency=30 test/async-subagents",
     "test:async-subagents-selection-e2e": "ASYNC_SUBAGENTS_SELECTION_E2E=1 ASYNC_SUBAGENTS_MODEL=zai/glm-5-turbo bun test --concurrent --max-concurrency=30 test/async-subagents/selection-e2e.test.ts",
     "bench:locate": "PI_LOCATE_BENCH_ITERATIONS=5 PI_LOCATE_BENCH_FAKE_IDX=0 PI_LOCATE_BENCH_MODEL=zai/glm-5-turbo PI_LOCATE_BENCH_MODES=direct-read-grep,ast-structural,repo-search-hybrid,repo-discovery,subagent-search,unrestricted-suite node test/fixtures/hard-to-find-project/benchmark/run-locate-benchmark.mjs",
     "bench:locate:analyze": "node test/fixtures/hard-to-find-project/benchmark/analyze-locate-benchmark.mjs",
     "test:locate-benchmark-e2e": "PI_LOCATE_BENCH_E2E=1 PI_LOCATE_BENCH_MODEL=zai/glm-5-turbo bun test test/locate-benchmark-e2e.test.ts",
     "test:tool-selection-e2e": "TOOL_SELECTION_E2E=1 TOOL_SELECTION_E2E_MODEL=zai/glm-5-turbo bun test --concurrent --max-concurrency=30 test/tool-selection-e2e.test.ts",
-    "test:e2e": "TOOL_SELECTION_E2E=1 TOOL_SELECTION_E2E_MODEL=zai/glm-5-turbo ASYNC_SUBAGENTS_E2E=1 ASYNC_SUBAGENTS_MODEL=zai/glm-5-turbo bun test --concurrent --max-concurrency=5 test/tool-selection-e2e.test.ts test/async-subagents/e2e.test.ts test/async-subagents/selection-e2e.test.ts test/todo-persistence-e2e.test.ts",
+    "test:e2e": "TOOL_SELECTION_E2E=1 TOOL_SELECTION_E2E_MODEL=zai/glm-5-turbo ASYNC_SUBAGENTS_E2E=1 ASYNC_SUBAGENTS_DEBUG_LOGS=1 ASYNC_SUBAGENTS_MODEL=zai/glm-5-turbo bun test --concurrent --max-concurrency=5 test/tool-selection-e2e.test.ts test/async-subagents/e2e.test.ts test/async-subagents/selection-e2e.test.ts test/todo-persistence-e2e.test.ts",
     "typecheck:async-subagents": "bash scripts/typecheck-source.sh",
     "check": "npm run smoke"
   },

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-ui-extend",
-  "version": "0.1.35",
+  "version": "0.1.36",
   "private": false,
   "type": "module",
   "bin": {