fix(vscode): clear unsent drafts on session deletion

.changeset/cleanup-drafts-on-session-delete.md

@@ -0,0 +1,5 @@
+---
+"kilo-code": patch
+---
+
+Free webview memory for deleted VS Code sessions by clearing unsent prompt text, review comments, and pending image attachments that were retained in the per-session draft cache after `sessionDeleted`.

packages/kilo-vscode/tests/unit/prompt-drafts.test.ts

@@ -1,6 +1,37 @@
-import { describe, it, expect } from "bun:test"
+import { beforeEach, describe, it, expect } from "bun:test"
+import { deleteDraftsForSession, drafts, imageDrafts, reviewDrafts } from "../../webview-ui/src/utils/draft-store"
 import { pendingDraftKey, scopeDraftKey, sessionDraftKey } from "../../webview-ui/src/utils/prompt-drafts"
 
+beforeEach(() => {
+  drafts.clear()
+  reviewDrafts.clear()
+  imageDrafts.clear()
+})
+
+describe("deleteDraftsForSession", () => {
+  it("clears deleted-session drafts without touching other sessions", () => {
+    drafts.set("prompt:default:session:a", "draft a")
+    drafts.set("prompt:default:pending:a", "pending a")
+    drafts.set("prompt:default:session:b", "draft b")
+    reviewDrafts.set("prompt:default:session:a", [])
+    imageDrafts.set("prompt:default:session:a", [])
+
+    deleteDraftsForSession("a")
+
+    expect(drafts.has("prompt:default:session:a")).toBe(false)
+    expect(drafts.has("prompt:default:pending:a")).toBe(false)
+    expect(drafts.get("prompt:default:session:b")).toBe("draft b")
+    expect(reviewDrafts.has("prompt:default:session:a")).toBe(false)
+    expect(imageDrafts.has("prompt:default:session:a")).toBe(false)
+  })
+
+  it("is a no-op when given an empty id", () => {
+    drafts.set("prompt:default:session:a", "draft a")
+    deleteDraftsForSession("")
+    expect(drafts.get("prompt:default:session:a")).toBe("draft a")
+  })
+})
+
 describe("sessionDraftKey", () => {
   it("prefixes session ids", () => {
     expect(sessionDraftKey("abc")).toBe("session:abc")

packages/kilo-vscode/webview-ui/src/components/chat/PromptInput.tsx

@@ -46,14 +46,10 @@ import {
 import type { ReviewComment, SendMessageFailedMessage, TextPart } from "../../types/messages"
 import { formatReviewCommentsMarkdown } from "../../utils/review-comment-markdown"
 import { pendingDraftKey, scopeDraftKey, sessionDraftKey } from "../../utils/prompt-drafts"
+import { drafts, imageDrafts, reviewDrafts } from "../../utils/draft-store"
 import { ReviewComments } from "./ReviewComments"
 import { partReview, reviewBody } from "../../../../src/shared/review-comments"
 
-// Per-session input text storage (module-level so it survives remounts)
-const drafts = new Map<string, string>()
-const reviewDrafts = new Map<string, ReviewComment[]>()
-const imageDrafts = new Map<string, ImageAttachment[]>()
-
 function mergeReviewComments(current: ReviewComment[], incoming: ReviewComment[]): ReviewComment[] {
   if (incoming.length === 0) return current
   const map = new Map(current.map((item) => [item.id, item]))

packages/kilo-vscode/webview-ui/src/context/session.tsx

@@ -72,6 +72,7 @@ import { getVariant, sessionVariantKeys, transferVariants, variantKey } from "./
 import { KILO_AUTO, KILO_PROVIDER_ID, parseModelString } from "../../../src/shared/provider-model"
 import { reviewMetadata, type ReviewMessageData } from "../../../src/shared/review-comments"
 import { visibleMessages as filterVisibleMessages } from "./session-queue"
+import { deleteDraftsForSession } from "../utils/draft-store"
 import { createAbortState } from "./abort-state"
 
 const RECENT_LIMIT = 5
@@ -1910,6 +1911,7 @@ export const SessionProvider: ParentComponent = (props) => {
           delete map[sessionID]
         }),
       )
+      deleteDraftsForSession(sessionID)
       if (currentSessionID() === sessionID) {
         setCurrentSessionID(undefined)
         setLoading(false)

packages/kilo-vscode/webview-ui/src/utils/draft-store.ts

@@ -0,0 +1,21 @@
+import type { ReviewComment } from "../types/messages"
+import type { ImageAttachment } from "../hooks/useImageAttachments"
+
+export const drafts = new Map<string, string>()
+export const reviewDrafts = new Map<string, ReviewComment[]>()
+export const imageDrafts = new Map<string, ImageAttachment[]>()
+
+export function deleteDraftsForSession(id: string) {
+  if (!id) return
+  const sessionSuffix = `:session:${id}`
+  const pendingSuffix = `:pending:${id}`
+  const maps = [drafts, reviewDrafts, imageDrafts]
+  for (const map of maps) {
+    for (const key of map.keys()) {
+      if (typeof key !== "string") continue
+      if (key.endsWith(sessionSuffix) || key.endsWith(pendingSuffix)) {
+        map.delete(key)
+      }
+    }
+  }
+}