Add prune-context mode for context pruning functionality

Author: robertpiosik

packages/browser/src/content-scripts/send-prompt-content-script/send-prompt-content-script.ts

@@ -196,7 +196,12 @@ const main = async () => {
       client_id: number
       raw_instructions?: string
       edit_format?: string
-      mode?: 'ask' | 'edit-context' | 'code-completions' | 'no-context'
+      mode?:
+        | 'ask'
+        | 'edit-context'
+        | 'code-completions'
+        | 'no-context'
+        | 'prune-context'
     }
 
     if (!stored_data) {
@@ -231,7 +236,8 @@ const main = async () => {
     if (
       chatbot?.inject_apply_response_button &&
       (stored_data.mode == 'edit-context' ||
-        stored_data.mode == 'code-completions')
+        stored_data.mode == 'code-completions' ||
+        stored_data.mode == 'prune-context')
     ) {
       sessionStorage.setItem(
         session_data_key,

packages/vscode/src/commands/apply-chat-response-command/response-processor.ts

@@ -139,9 +139,43 @@ export const process_chat_response = async (
 
     quick_pick_items.sort((a, b) => natural_sort(a.label, b.label))
 
-    const selected_items = await vscode.window.showQuickPick(quick_pick_items, {
-      canPickMany: true,
-      placeHolder: 'Confirm context pruning'
+    const quick_pick = vscode.window.createQuickPick()
+    quick_pick.items = quick_pick_items
+    quick_pick.selectedItems = quick_pick_items.filter((i) => i.picked)
+    quick_pick.canSelectMany = true
+    quick_pick.title = 'Context Pruning'
+    quick_pick.placeholder = 'Confirm file selection'
+    quick_pick.buttons = [
+      { iconPath: new vscode.ThemeIcon('close'), tooltip: 'Close' }
+    ]
+
+    const selected_items = await new Promise<
+      readonly vscode.QuickPickItem[] | undefined
+    >((resolve) => {
+      let is_accepted = false
+      const disposables: vscode.Disposable[] = []
+
+      disposables.push(
+        quick_pick.onDidTriggerButton((button) => {
+          if (button.tooltip == 'Close') {
+            quick_pick.hide()
+            resolve(undefined)
+          }
+        }),
+        quick_pick.onDidAccept(() => {
+          is_accepted = true
+          resolve(quick_pick.selectedItems)
+          quick_pick.hide()
+        }),
+        quick_pick.onDidHide(() => {
+          if (!is_accepted) {
+            resolve(undefined)
+          }
+          disposables.forEach((d) => d.dispose())
+          quick_pick.dispose()
+        })
+      )
+      quick_pick.show()
     })
 
     if (selected_items && selected_items.length > 0) {
@@ -160,7 +194,7 @@ export const process_chat_response = async (
 
       if (files_to_check.length > 0) {
         await workspace_provider.set_checked_files(files_to_check)
-        vscode.window.showInformationMessage(`Context updated successfully.`)
+        vscode.window.showInformationMessage(`Context pruned successfully.`)
       }
     }
 

packages/vscode/src/commands/apply-chat-response-command/utils/clipboard-parser/parsers/relevant-files-parser.ts

@@ -7,7 +7,7 @@ export const parse_relevant_files = (params: {
   let start_index = -1
 
   for (let i = 0; i < lines.length; i++) {
-    if (lines[i].trim().match(/^###\s+Relevant files/i)) {
+    if (lines[i].trim().match(/^\*\*Relevant files:\*\*/i)) {
       start_index = i
       break
     }

packages/vscode/src/commands/apply-chat-response-command/utils/clipboard-parser/test-cases/relevant-files/relevant-files.txt

@@ -1,4 +1,4 @@
-### Relevant files
+**Relevant files:**
 
 * `src/hello.ts`
 - `src/welcome.ts`

packages/vscode/src/constants/instructions.ts

@@ -31,9 +31,9 @@ export const commit_message_instructions =
 
 export const prune_context_instructions = `Find relevant files for the task.
 <system>
-Your response must begin with a markdown heading "### Relevant files", then list paths relevant for the task one under another. Don't send anything else. Example:
+Your response must begin with "**Relevant files:**", then list paths relevant for the task one under another. Don't send anything else. Example:
 
-### Relevant files
+**Relevant files:**
 
 - \`src/hello.ts\`
 - \`src/welcome.ts\`