fix(webview): preserve existing text when selecting slash command at start

.changeset/memory-show-preserve-text.md

@@ -0,0 +1,5 @@
+---
+"kilo-code": patch
+---
+
+Typing extra text after the `/memory show` command (for example `/memory show my notes`) used to clear that text from the chat input when the command ran. The text now stays in the input so you don't lose what you typed.

.changeset/slash-command-preserve-text.md

@@ -0,0 +1,5 @@
+---
+"kilo-code": patch
+---
+
+Preserve existing text when selecting a slash command at the start of the chat input. Previously, typing `/new` or another slash command with text before the cursor would wipe the existing content entirely.

packages/kilo-memory/src/commands.ts

@@ -38,6 +38,7 @@ type Help = {
 
 type Show = {
   kind: "show"
+  rest?: string
 }
 
 type Operation =
@@ -64,6 +65,7 @@ type Operation =
   | {
       kind: "operation"
       operation: Exclude<MemoryOperation, "remember" | "correct" | "forget" | "purge" | "auto">
+      rest?: string
     }
 
 type Usage = {
@@ -93,10 +95,10 @@ function usage(reason: string): ParsedMemoryCommand {
 }
 
 function operation(verb: string, text: string): ParsedMemoryCommand | undefined {
-  if (verb === "on" || verb === "enable") return { kind: "operation", operation: "enable" }
-  if (verb === "off" || verb === "disable") return { kind: "operation", operation: "disable" }
+  if (verb === "on" || verb === "enable") return { kind: "operation", operation: "enable", rest: text }
+  if (verb === "off" || verb === "disable") return { kind: "operation", operation: "disable", rest: text }
   if (verb === "status" || verb === "inspect" || verb === "rebuild") {
-    return { kind: "operation", operation: verb }
+    return { kind: "operation", operation: verb, rest: text }
   }
   if (verb === "purge") {
     if (text.toLowerCase() === "confirm") return { kind: "operation", operation: "purge", confirm: true }
@@ -138,7 +140,7 @@ export function parseMemoryCommand(input: string): ParsedMemoryCommand | undefin
   const parts = split(picked.rest)
   const verb = parts.head
   if (!verb) return { kind: "help" }
-  if (verb === "show") return { kind: "show" }
+  if (verb === "show") return { kind: "show", rest: parts.tail }
 
   const op = operation(verb, parts.tail)
   if (op) return op

packages/kilo-memory/test/command-cases.json

@@ -19,6 +19,12 @@
     "input": "/memory project show",
     "result": "show"
   },
+  {
+    "name": "show preserves trailing text",
+    "input": "/memory show draft notes",
+    "result": "show",
+    "rest": "draft notes"
+  },
   {
     "name": "status operation",
     "input": "/memory status",
@@ -61,6 +67,27 @@
     "result": "operation",
     "operation": "rebuild"
   },
+  {
+    "name": "rebuild keeps trailing text",
+    "input": "/memory rebuild hello",
+    "result": "operation",
+    "operation": "rebuild",
+    "rest": "hello"
+  },
+  {
+    "name": "enable keeps trailing text",
+    "input": "/memory on hello",
+    "result": "operation",
+    "operation": "enable",
+    "rest": "hello"
+  },
+  {
+    "name": "project scope rebuild keeps trailing text",
+    "input": "/memory project rebuild hello",
+    "result": "operation",
+    "operation": "rebuild",
+    "rest": "hello"
+  },
   {
     "name": "purge requires confirmation",
     "input": "/memory purge",

packages/kilo-memory/test/commands.test.ts

@@ -11,14 +11,15 @@ type Case = {
   text?: string
   query?: string
   reason?: string
+  rest?: string
 }
 
 const cases = (await Bun.file(new URL("./command-cases.json", import.meta.url)).json()) as Case[]
 
 function expected(item: Case): ParsedMemoryCommand | undefined {
   if (item.result === "none") return
   if (item.result === "help") return { kind: "help" }
-  if (item.result === "show") return { kind: "show" }
+  if (item.result === "show") return { kind: "show", rest: item.rest ?? "" }
   if (item.result === "usage") return { kind: "usage", reason: item.reason ?? "" }
   if (!item.operation) throw new Error(`Missing operation for fixture: ${item.name}`)
   if (item.operation === "remember" || item.operation === "correct") {
@@ -37,7 +38,7 @@ function expected(item: Case): ParsedMemoryCommand | undefined {
     if (item.confirm !== true) throw new Error(`Missing confirmation for fixture: ${item.name}`)
     return { kind: "operation", operation: item.operation, confirm: true }
   }
-  return { kind: "operation", operation: item.operation }
+  return { kind: "operation", operation: item.operation, rest: item.rest ?? "" }
 }
 
 describe("memory commands", () => {

packages/kilo-vscode/tests/unit/memory-command.test.ts

@@ -22,6 +22,7 @@ type Case = {
   text?: string
   query?: string
   reason?: string
+  rest?: string
 }
 
 const cases = (await Bun.file(
@@ -31,7 +32,7 @@ const cases = (await Bun.file(
 function expected(item: Case): ParsedMemoryCommand | undefined {
   if (item.result === "none") return
   if (item.result === "help") return { kind: "help" }
-  if (item.result === "show") return { kind: "show" }
+  if (item.result === "show") return { kind: "show", rest: item.rest ?? "" }
   if (item.result === "usage") return { kind: "usage", reason: item.reason ?? "" }
   if (!item.operation) throw new Error(`Missing operation for fixture: ${item.name}`)
   if (item.operation === "remember" || item.operation === "correct") {
@@ -50,7 +51,7 @@ function expected(item: Case): ParsedMemoryCommand | undefined {
     if (item.confirm !== true) throw new Error(`Missing confirmation for fixture: ${item.name}`)
     return { kind: "operation", operation: item.operation, confirm: true }
   }
-  return { kind: "operation", operation: item.operation }
+  return { kind: "operation", operation: item.operation, rest: item.rest ?? "" }
 }
 
 describe("parseMemoryCommand", () => {

packages/kilo-vscode/tests/unit/prompt-input-utils.test.ts

@@ -12,7 +12,9 @@ import {
   isPathMention,
   applySandboxState,
   applySandboxStates,
+  memoryRest,
 } from "../../webview-ui/src/components/chat/prompt-input-utils"
+import { parseMemoryCommand } from "../../webview-ui/src/utils/memory-command"
 
 describe("applySandboxState", () => {
   const state = (enabled: boolean, revision: number, sessionID = "ses_1", directory = "/repo") => ({
@@ -329,3 +331,33 @@ describe("isPathMention", () => {
     expect(isPathMention("src/foo.ts")).toBe(true)
   })
 })
+
+describe("memoryRest", () => {
+  it("keeps trailing text in the input after a no-argument memory command", () => {
+    // /memory rebuild hello -> rebuild executes, "hello" stays in the input.
+    // This is the submit-path half of the trailing-text bug: handleSend sets
+    // the input to memoryRest(parsed), so a regression would drop "hello".
+    const memory = parseMemoryCommand("/memory rebuild hello")
+    expect(memory).not.toBeUndefined()
+    expect(memoryRest(memory!)).toBe("hello")
+  })
+
+  it("keeps trailing text through the project scope", () => {
+    expect(memoryRest(parseMemoryCommand("/memory project rebuild hello")!)).toBe("hello")
+  })
+
+  it("returns empty string when a no-argument command has no trailing text", () => {
+    expect(memoryRest(parseMemoryCommand("/memory rebuild")!)).toBe("")
+  })
+
+  it("keeps trailing text in the input after the show command", () => {
+    // /memory show draft notes -> show executes, "draft notes" stays in the input.
+    expect(memoryRest(parseMemoryCommand("/memory show draft notes")!)).toBe("draft notes")
+  })
+
+  it("returns empty string for argument-taking operations", () => {
+    // remember/correct/forget/auto/purge consume their text, so nothing remains.
+    expect(memoryRest(parseMemoryCommand("/memory remember hello")!)).toBe("")
+    expect(memoryRest(parseMemoryCommand("/memory auto on")!)).toBe("")
+  })
+})