error handling and step retry

2 files changed, 10 insertions, 0 deletions

diff --git a/continuedev/src/continuedev/server/gui.py b/continuedev/src/continuedev/server/gui.py
index 3d1a5a82..b873a88f 100644
--- a/continuedev/src/continuedev/server/gui.py
+++ b/continuedev/src/continuedev/server/gui.py
@@ -75,6 +75,8 @@ class GUIProtocolServer(AbstractGUIProtocolServer):
                 self.on_refinement_input(data["input"], data["index"])
             elif message_type == "reverse_to_index":
                 self.on_reverse_to_index(data["index"])
+            elif message_type == "retry_at_index":
+                self.on_retry_at_index(data["index"])
         except Exception as e:
             print(e)
 
@@ -100,6 +102,10 @@ class GUIProtocolServer(AbstractGUIProtocolServer):
         asyncio.create_task(
             self.session.autopilot.accept_refinement_input(input, index))
 
+    def on_retry_at_index(self, index: int):
+        asyncio.create_task(
+            self.session.autopilot.retry_at_index(index))
+
 
 @router.websocket("/ws")
 async def websocket_endpoint(websocket: WebSocket, session: Session = Depends(websocket_session)):
diff --git a/continuedev/src/continuedev/server/gui_protocol.py b/continuedev/src/continuedev/server/gui_protocol.py
index e32d80ef..287f9e3b 100644
--- a/continuedev/src/continuedev/server/gui_protocol.py
+++ b/continuedev/src/continuedev/server/gui_protocol.py
@@ -26,3 +26,7 @@ class AbstractGUIProtocolServer(ABC):
     @abstractmethod
     async def send_state_update(self, state: dict):
         """Send a state update to the client"""
+
+    @abstractmethod
+    def on_retry_at_index(self, index: int):
+        """Called when the user requests a retry at a previous index"""