2 files changed, 9 insertions, 0 deletions
diff --git a/extension/react-app/src/hooks/AbstractContinueGUIClientProtocol.ts b/extension/react-app/src/hooks/AbstractContinueGUIClientProtocol.ts
index e018c03c..804362aa 100644
--- a/extension/react-app/src/hooks/AbstractContinueGUIClientProtocol.ts
+++ b/extension/react-app/src/hooks/AbstractContinueGUIClientProtocol.ts
@@ -34,6 +34,8 @@ abstract class AbstractContinueGUIClientProtocol {
   abstract loadSession(session_id?: string): void;
 
   abstract onReconnectAtSession(session_id: string): void;
+
+  abstract editStepAtIndex(userInput: string, index: number): void;
 }
 
 export default AbstractContinueGUIClientProtocol;
diff --git a/extension/react-app/src/hooks/ContinueGUIClientProtocol.ts b/extension/react-app/src/hooks/ContinueGUIClientProtocol.ts
index c2285f6d..82aeee28 100644
--- a/extension/react-app/src/hooks/ContinueGUIClientProtocol.ts
+++ b/extension/react-app/src/hooks/ContinueGUIClientProtocol.ts
@@ -125,6 +125,13 @@ class ContinueGUIClientProtocol extends AbstractContinueGUIClientProtocol {
   selectContextItem(id: string, query: string): void {
     this.messenger?.send("select_context_item", { id, query });
   }
+
+  editStepAtIndex(userInput: string, index: number): void {
+    this.messenger?.send("edit_step_at_index", {
+      user_input: userInput,
+      index,
+    });
+  }
 }
 
 export default ContinueGUIClientProtocol;