2 files changed, 19 insertions, 4 deletions
diff --git a/extension/react-app/src/hooks/AbstractContinueGUIClientProtocol.ts b/extension/react-app/src/hooks/AbstractContinueGUIClientProtocol.ts
index d71186d7..998d3a6d 100644
--- a/extension/react-app/src/hooks/AbstractContinueGUIClientProtocol.ts
+++ b/extension/react-app/src/hooks/AbstractContinueGUIClientProtocol.ts
@@ -21,7 +21,7 @@ abstract class AbstractContinueGUIClientProtocol {
 
   abstract deleteAtIndex(index: number): void;
 
-  abstract deleteContextWithIds(ids: ContextItemId[]): void;
+  abstract deleteContextWithIds(ids: ContextItemId[], index?: number): void;
 
   abstract setEditingAtIds(ids: string[]): void;
 
@@ -33,6 +33,12 @@ abstract class AbstractContinueGUIClientProtocol {
 
   abstract selectContextItem(id: string, query: string): void;
 
+  abstract selectContextItemAtIndex(
+    id: string,
+    query: string,
+    index: number
+  ): void;
+
   abstract loadSession(session_id?: string): void;
 
   abstract onReconnectAtSession(session_id: string): void;
diff --git a/extension/react-app/src/hooks/ContinueGUIClientProtocol.ts b/extension/react-app/src/hooks/ContinueGUIClientProtocol.ts
index 8205a629..863b1031 100644
--- a/extension/react-app/src/hooks/ContinueGUIClientProtocol.ts
+++ b/extension/react-app/src/hooks/ContinueGUIClientProtocol.ts
@@ -101,9 +101,10 @@ class ContinueGUIClientProtocol extends AbstractContinueGUIClientProtocol {
     this.messenger?.send("delete_at_index", { index });
   }
 
-  deleteContextWithIds(ids: ContextItemId[]) {
+  deleteContextWithIds(ids: ContextItemId[], index?: number) {
     this.messenger?.send("delete_context_with_ids", {
       ids: ids.map((id) => `${id.provider_title}-${id.item_id}`),
+      index,
     });
   }
 
@@ -119,14 +120,22 @@ class ContinueGUIClientProtocol extends AbstractContinueGUIClientProtocol {
     this.messenger?.send("show_logs_at_index", { index });
   }
 
-  showContextVirtualFile(): void {
-    this.messenger?.send("show_context_virtual_file", {});
+  showContextVirtualFile(index?: number): void {
+    this.messenger?.send("show_context_virtual_file", { index });
   }
 
   selectContextItem(id: string, query: string): void {
     this.messenger?.send("select_context_item", { id, query });
   }
 
+  selectContextItemAtIndex(id: string, query: string, index: number): void {
+    this.messenger?.send("select_context_item_at_index", {
+      id,
+      query,
+      index,
+    });
+  }
+
   editStepAtIndex(userInput: string, index: number): void {
     this.messenger?.send("edit_step_at_index", {
       user_input: userInput,