2 files changed, 17 insertions, 2 deletions
diff --git a/extension/react-app/src/hooks/AbstractContinueGUIClientProtocol.ts b/extension/react-app/src/hooks/AbstractContinueGUIClientProtocol.ts
index 804362aa..2e8aaeef 100644
--- a/extension/react-app/src/hooks/AbstractContinueGUIClientProtocol.ts
+++ b/extension/react-app/src/hooks/AbstractContinueGUIClientProtocol.ts
@@ -1,4 +1,4 @@
-import { ContextItemId } from "../../../schema/FullState";
+import { ContextItem, ContextItemId } from "../../../schema/FullState";
 
 abstract class AbstractContinueGUIClientProtocol {
   abstract sendMainInput(input: string): void;
@@ -36,6 +36,10 @@ abstract class AbstractContinueGUIClientProtocol {
   abstract onReconnectAtSession(session_id: string): void;
 
   abstract editStepAtIndex(userInput: string, index: number): void;
+
+  abstract saveContextGroup(title: string, contextItems: ContextItem[]): void;
+
+  abstract selectContextGroup(id: string): void;
 }
 
 export default AbstractContinueGUIClientProtocol;
diff --git a/extension/react-app/src/hooks/ContinueGUIClientProtocol.ts b/extension/react-app/src/hooks/ContinueGUIClientProtocol.ts
index 82aeee28..aa558adb 100644
--- a/extension/react-app/src/hooks/ContinueGUIClientProtocol.ts
+++ b/extension/react-app/src/hooks/ContinueGUIClientProtocol.ts
@@ -1,4 +1,4 @@
-import { ContextItemId } from "../../../schema/FullState";
+import { ContextItem, ContextItemId } from "../../../schema/FullState";
 import AbstractContinueGUIClientProtocol from "./AbstractContinueGUIClientProtocol";
 import { Messenger, WebsocketMessenger } from "./messenger";
 import { VscodeMessenger } from "./vscodeMessenger";
@@ -132,6 +132,17 @@ class ContinueGUIClientProtocol extends AbstractContinueGUIClientProtocol {
       index,
     });
   }
+
+  saveContextGroup(title: string, contextItems: ContextItem[]): void {
+    this.messenger?.send("save_context_group", {
+      context_items: contextItems,
+      title,
+    });
+  }
+
+  selectContextGroup(id: string): void {
+    this.messenger?.send("select_context_group", { id });
+  }
 }
 
 export default ContinueGUIClientProtocol;