@dreb/coding-agent 2.21.0 → 2.22.0

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

@@ -1977,6 +1977,12 @@ export class InteractiveMode {
                     this.ui.requestRender();
                 }
                 else if (event.message.role === "assistant") {
+                    // Defensive: remove stale streaming component if one exists (e.g. missed stream_retry)
+                    if (this.streamingComponent) {
+                        this.chatContainer.removeChild(this.streamingComponent);
+                        this.streamingComponent = undefined;
+                        this.streamingMessage = undefined;
+                    }
                     this.streamingComponent = new AssistantMessageComponent(undefined, this.hideThinkingBlock, this.getMarkdownThemeWithSettings());
                     this.streamingMessage = event.message;
                     this.chatContainer.addChild(this.streamingComponent);
@@ -2086,11 +2092,15 @@ export class InteractiveMode {
                 break;
             }
             case "agent_end":
+                if (this.retryLoader) {
+                    this.retryLoader.stop();
+                    this.retryLoader = undefined;
+                }
                 if (this.loadingAnimation) {
                     this.loadingAnimation.stop();
                     this.loadingAnimation = undefined;
-                    this.statusContainer.clear();
                 }
+                this.statusContainer.clear();
                 if (this.streamingComponent) {
                     this.chatContainer.removeChild(this.streamingComponent);
                     this.streamingComponent = undefined;
@@ -2189,6 +2199,33 @@ export class InteractiveMode {
                 this.ui.requestRender();
                 break;
             }
+            case "stream_retry": {
+                // Remove the ghost streaming component left from the dropped stream
+                if (this.streamingComponent) {
+                    this.chatContainer.removeChild(this.streamingComponent);
+                    this.streamingComponent = undefined;
+                    this.streamingMessage = undefined;
+                }
+                // Clear any pending tool components from the failed partial
+                for (const [, component] of this.pendingTools.entries()) {
+                    this.chatContainer.removeChild(component);
+                }
+                this.pendingTools.clear();
+                // Show retry status
+                this.statusContainer.clear();
+                if (this.loadingAnimation) {
+                    this.loadingAnimation.stop();
+                    this.loadingAnimation = undefined;
+                }
+                if (this.retryLoader) {
+                    this.retryLoader.stop();
+                    this.retryLoader = undefined;
+                }
+                this.retryLoader = new Loader(this.ui, (spinner) => theme.fg("warning", spinner), (text) => theme.fg("muted", text), `Stream dropped, retrying (${event.attempt}/${event.maxAttempts})... (${keyText("app.interrupt")} to cancel)`);
+                this.statusContainer.addChild(this.retryLoader);
+                this.ui.requestRender();
+                break;
+            }
             case "background_agent_start":
             case "background_agent_end": {
                 this.updateBackgroundAgentStatus();