fix: keep pending stream timestamp until retry

Author: ammar-agent

src/browser/components/ChatInput/index.tsx

@@ -62,7 +62,6 @@ import { setTelemetryEnabled } from "@/common/telemetry";
 import { getTokenCountPromise } from "@/browser/utils/tokenizer/rendererClient";
 import { CreationCenterContent } from "./CreationCenterContent";
 import { CreationControls } from "./CreationControls";
-import { useWorkspaceStoreRaw } from "@/browser/stores/WorkspaceStore";
 import { useCreationWorkspace } from "./useCreationWorkspace";
 
 type TokenCountReader = () => number;
@@ -137,7 +136,6 @@ export const ChatInput: React.FC<ChatInputProps> = (props) => {
   const [mode, setMode] = useMode();
   const { recentModels, addModel, evictModel } = useModelLRU();
   const commandListId = useId();
-  const workspaceStore = useWorkspaceStoreRaw();
   const telemetry = useTelemetry();
   const [vimEnabled, setVimEnabled] = usePersistedState<boolean>(VIM_ENABLED_KEY, false, {
     listener: true,
@@ -445,12 +443,6 @@ export const ChatInput: React.FC<ChatInputProps> = (props) => {
     // Workspace variant: full command handling + message send
     if (variant !== "workspace") return; // Type guard
 
-    const notifyPendingSendFailed = () => {
-      setTimeout(() => {
-        workspaceStore.markPendingStreamFailed(props.workspaceId);
-      }, 0);
-    };
-
     try {
       // Parse command
       const parsed = parseCommand(messageText);
@@ -724,7 +716,6 @@ export const ChatInput: React.FC<ChatInputProps> = (props) => {
           setToast(createErrorToast(result.error));
           // Restore input and images on error so user can try again
           setInput(messageText);
-          notifyPendingSendFailed();
           setImageAttachments(previousImageAttachments);
         } else {
           // Track telemetry for successful message send
@@ -745,7 +736,6 @@ export const ChatInput: React.FC<ChatInputProps> = (props) => {
             raw: error instanceof Error ? error.message : "Failed to send message",
           })
         );
-        notifyPendingSendFailed();
         setInput(messageText);
         setImageAttachments(previousImageAttachments);
       } finally {

src/browser/stores/WorkspaceStore.ts

@@ -365,20 +365,6 @@ export class WorkspaceStore {
     return allStates;
   }
 
-  /**
-   * Notify the aggregator that a pending stream never started (renderer send failure).
-   */
-  markPendingStreamFailed(workspaceId: string): void {
-    const aggregator = this.aggregators.get(workspaceId);
-    if (!aggregator) {
-      return;
-    }
-
-    if (aggregator.markPendingStreamStartFailed()) {
-      this.states.bump(workspaceId);
-    }
-  }
-
   /**
    * Get recency timestamps for all workspaces (for sorting in command palette).
    * Derived on-demand from individual workspace states.

src/browser/utils/messages/StreamingMessageAggregator.ts

@@ -96,6 +96,8 @@ export class StreamingMessageAggregator {
   // Track when we're waiting for stream-start after user message
   // Prevents retry barrier flash during normal send flow
   // Stores timestamp of when user message was sent (null = no pending stream)
+  // IMPORTANT: We intentionally keep this timestamp until a stream actually starts
+  // (or the user retries) so retry UI/backoff logic doesn't misfire on send failures.
   private pendingStreamStartTime: number | null = null;
 
   // Workspace creation timestamp (used for recency calculation)
@@ -267,17 +269,6 @@ export class StreamingMessageAggregator {
     return this.pendingStreamStartTime;
   }
 
-  /**
-   * Clear pending stream-start tracking when send fails before we ever receive stream-start.
-   * Returns true when the internal state changed so callers can trigger updates.
-   */
-  markPendingStreamStartFailed(): boolean {
-    if (this.pendingStreamStartTime === null) {
-      return false;
-    }
-    this.pendingStreamStartTime = null;
-    return true;
-  }
   private setPendingStreamStartTime(time: number | null): void {
     this.pendingStreamStartTime = time;
   }