🤖 fix: cancel active stream on message edit to prevent history corruption

src/node/services/agentSession.ts

@@ -271,6 +271,15 @@ export class AgentSession {
     }
 
     if (options?.editMessageId) {
+      // Interrupt an existing stream or compaction, if active
+      if (this.aiService.isStreaming(this.workspaceId)) {
+        // MUST use abandonPartial=true to prevent handleAbort from performing partial compaction
+        // with mismatched history (since we're about to truncate it)
+        const stopResult = await this.interruptStream(/* abandonPartial */ true);
+        if (!stopResult.success) {
+          return Err(createUnknownSendMessageError(stopResult.error));
+        }
+      }
       const truncateResult = await this.historyService.truncateAfterMessage(
         this.workspaceId,
         options.editMessageId
@@ -367,6 +376,14 @@ export class AgentSession {
       return Err(stopResult.error);
     }
 
+    // Delete partial when abandoning to prevent commitToHistory from reintroducing it
+    if (abandonPartial) {
+      const deleteResult = await this.partialService.deletePartial(this.workspaceId);
+      if (!deleteResult.success) {
+        return Err(deleteResult.error);
+      }
+    }
+
     return Ok(undefined);
   }
 

src/node/services/ipcMain.ts

@@ -1104,12 +1104,6 @@ export class IpcMain {
             return { success: false, error: stopResult.error };
           }
 
-          // If abandonPartial is true, delete the partial instead of committing it
-          if (options?.abandonPartial) {
-            log.debug("Abandoning partial for workspace:", workspaceId);
-            await this.partialService.deletePartial(workspaceId);
-          }
-
           return { success: true, data: undefined };
         } catch (error) {
           const errorMessage = error instanceof Error ? error.message : String(error);