fix: preserve images when editing messages with checkpoint

[hdcodedev]

Context

Fixes #3489

When editing an already sent message and attaching an image to it, the image was being silently dropped and not sent with the edited text. This bug only occurred when a checkpoint existed after the message being edited, causing the user to see the "Proceed" or "No, edit message only" dialog.

The issue was reported in #3489.

Implementation

The bug was in webview-ui/src/App.tsx. When sending the editMessageConfirm message to the backend, there were two code paths:

1. With checkpoint (CheckpointRestoreDialog) - The images field was missing
2. Without checkpoint (EditMessageDialog) - The images field was correctly included

The fix adds the missing images: editMessageDialogState.images field to the checkpoint dialog's onConfirm handler, ensuring images are preserved in both scenarios.

 onConfirm={(restoreCheckpoint: boolean) => {
     vscode.postMessage({
         type: "editMessageConfirm",
         messageTs: editMessageDialogState.messageTs,
         text: editMessageDialogState.text,
         restoreCheckpoint,
+        images: editMessageDialogState.images,
     })

Screenshots

How to Test

1. Start a conversation and send a text-only message
2. Wait for the AI to respond (this creates a checkpoint)
3. Click the edit icon on your original message
4. Modify the text and attach an image
5. Select "Proceed" or "No, edit message only" in the confirmation dialog
6. Before fix: Image indicator disappears, exporting task shows no [Image] tag
7. After fix: Image is preserved and visible in the conversation

Alternative verification:

• Export the task as markdown after editing with an image
• The [Image] indicator should be present in the exported .md file

[changeset-bot]

🦋 Changeset detected

Latest commit: bf66def

The changes in this PR will be included in the next version bump.

This PR includes changesets to release 1 package

Name	Type
kilo-code	Patch

Not sure what this means? Click here to learn what changesets are.

Click here if you're a maintainer who wants to add another changeset to this PR

[kevinvandijk]

Thanks @hdcodedev! Tested and works!