@averyyy/pi-coding-agent 0.84.4-piclient.1 → 0.84.4-piclient.2

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

@@ -1684,6 +1684,17 @@ export class InteractiveMode {
     showWorkingStatusIndicator() {
         this.showStatusIndicator(new WorkingStatusIndicator(this.ui, this.workingMessage ?? this.defaultWorkingMessage, this.workingIndicatorOptions));
     }
+    restoreWorkingStatusIndicatorIfStreaming() {
+        if (!this.session.isStreaming) {
+            return;
+        }
+        if (this.workingVisible && this.activeStatusIndicator?.kind !== "working") {
+            this.showWorkingStatusIndicator();
+        }
+        if (this.settingsManager.getShowTerminalProgress()) {
+            this.ui.terminal.setProgress(true);
+        }
+    }
     setWorkingVisible(visible) {
         this.workingVisible = visible;
         if (!visible) {
@@ -2739,10 +2750,6 @@ export class InteractiveMode {
                 break;
             }
             case "agent_end":
-                if (this.settingsManager.getShowTerminalProgress()) {
-                    this.ui.terminal.setProgress(false);
-                }
-                this.clearStatusIndicator("working");
                 if (this.streamingComponent) {
                     this.chatContainer.removeChild(this.streamingComponent);
                     this.streamingComponent = undefined;
@@ -2752,6 +2759,10 @@ export class InteractiveMode {
                 this.ui.requestRender();
                 break;
             case "agent_settled":
+                if (this.settingsManager.getShowTerminalProgress()) {
+                    this.ui.terminal.setProgress(false);
+                }
+                this.clearStatusIndicator("working");
                 await this.checkShutdownRequested();
                 break;
             case "compaction_start": {
@@ -2811,6 +2822,7 @@ export class InteractiveMode {
                         this.chatContainer.addChild(new Text(theme.fg("error", event.errorMessage), 1, 0));
                     }
                 }
+                this.restoreWorkingStatusIndicatorIfStreaming();
                 void this.flushCompactionQueue({ willRetry: event.willRetry });
                 this.ui.requestRender();
                 break;
@@ -2836,6 +2848,7 @@ export class InteractiveMode {
                 if (!event.success) {
                     this.showError(`Retry failed after ${event.attempt} attempts: ${event.finalError || "Unknown error"}`);
                 }
+                this.restoreWorkingStatusIndicatorIfStreaming();
                 this.ui.requestRender();
                 break;
             }