@mariozechner/pi-coding-agent 0.37.4 → 0.37.6

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

@@ -414,6 +414,15 @@ export class InteractiveMode {
                 this.showStatus("Navigated to selected point");
                 return { cancelled: false };
             },
+            setModelHandler: async (model) => {
+                const key = await this.session.modelRegistry.getApiKey(model);
+                if (!key)
+                    return false;
+                await this.session.setModel(model);
+                return true;
+            },
+            getThinkingLevelHandler: () => this.session.thinkingLevel,
+            setThinkingLevelHandler: (level) => this.session.setThinkingLevel(level),
             isIdle: () => !this.session.isStreaming,
             waitForIdle: () => this.session.agent.waitForIdle(),
             abort: () => {
@@ -600,9 +609,9 @@ export class InteractiveMode {
      */
     createExtensionUIContext() {
         return {
-            select: (title, options) => this.showExtensionSelector(title, options),
-            confirm: (title, message) => this.showExtensionConfirm(title, message),
-            input: (title, placeholder) => this.showExtensionInput(title, placeholder),
+            select: (title, options, opts) => this.showExtensionSelector(title, options, opts),
+            confirm: (title, message, opts) => this.showExtensionConfirm(title, message, opts),
+            input: (title, placeholder, opts) => this.showExtensionInput(title, placeholder, opts),
             notify: (message, type) => this.showExtensionNotify(message, type),
             setStatus: (key, text) => this.setExtensionStatus(key, text),
             setWidget: (key, content) => this.setExtensionWidget(key, content),
@@ -621,12 +630,23 @@ export class InteractiveMode {
     /**
      * Show a selector for extensions.
      */
-    showExtensionSelector(title, options) {
+    showExtensionSelector(title, options, opts) {
         return new Promise((resolve) => {
+            if (opts?.signal?.aborted) {
+                resolve(undefined);
+                return;
+            }
+            const onAbort = () => {
+                this.hideExtensionSelector();
+                resolve(undefined);
+            };
+            opts?.signal?.addEventListener("abort", onAbort, { once: true });
             this.extensionSelector = new ExtensionSelectorComponent(title, options, (option) => {
+                opts?.signal?.removeEventListener("abort", onAbort);
                 this.hideExtensionSelector();
                 resolve(option);
             }, () => {
+                opts?.signal?.removeEventListener("abort", onAbort);
                 this.hideExtensionSelector();
                 resolve(undefined);
             });
@@ -649,19 +669,30 @@ export class InteractiveMode {
     /**
      * Show a confirmation dialog for extensions.
      */
-    async showExtensionConfirm(title, message) {
-        const result = await this.showExtensionSelector(`${title}\n${message}`, ["Yes", "No"]);
+    async showExtensionConfirm(title, message, opts) {
+        const result = await this.showExtensionSelector(`${title}\n${message}`, ["Yes", "No"], opts);
         return result === "Yes";
     }
     /**
      * Show a text input for extensions.
      */
-    showExtensionInput(title, placeholder) {
+    showExtensionInput(title, placeholder, opts) {
         return new Promise((resolve) => {
+            if (opts?.signal?.aborted) {
+                resolve(undefined);
+                return;
+            }
+            const onAbort = () => {
+                this.hideExtensionInput();
+                resolve(undefined);
+            };
+            opts?.signal?.addEventListener("abort", onAbort, { once: true });
             this.extensionInput = new ExtensionInputComponent(title, placeholder, (value) => {
+                opts?.signal?.removeEventListener("abort", onAbort);
                 this.hideExtensionInput();
                 resolve(value);
             }, () => {
+                opts?.signal?.removeEventListener("abort", onAbort);
                 this.hideExtensionInput();
                 resolve(undefined);
             });
@@ -872,7 +903,7 @@ export class InteractiveMode {
                 return;
             }
             if (text.startsWith("/export")) {
-                this.handleExportCommand(text);
+                await this.handleExportCommand(text);
                 this.editor.setText("");
                 return;
             }
@@ -2086,11 +2117,11 @@ export class InteractiveMode {
     // =========================================================================
     // Command handlers
     // =========================================================================
-    handleExportCommand(text) {
+    async handleExportCommand(text) {
         const parts = text.split(/\s+/);
         const outputPath = parts.length > 1 ? parts[1] : undefined;
         try {
-            const filePath = this.session.exportToHtml(outputPath);
+            const filePath = await this.session.exportToHtml(outputPath);
             this.showStatus(`Session exported to: ${filePath}`);
         }
         catch (error) {
@@ -2113,7 +2144,7 @@ export class InteractiveMode {
         // Export to a temp file
         const tmpFile = path.join(os.tmpdir(), "session.html");
         try {
-            this.session.exportToHtml(tmpFile);
+            await this.session.exportToHtml(tmpFile);
         }
         catch (error) {
             this.showError(`Failed to export session: ${error instanceof Error ? error.message : "Unknown error"}`);