pause streaming on rejection

2 files changed, 14 insertions, 3 deletions

diff --git a/continuedev/src/continuedev/server/ide.py b/continuedev/src/continuedev/server/ide.py
index 1d51758e..61e7ca78 100644
--- a/continuedev/src/continuedev/server/ide.py
+++ b/continuedev/src/continuedev/server/ide.py
@@ -152,6 +152,8 @@ class IdeProtocolServer(AbstractIdeProtocolServer):
             self.onAcceptRejectDiff(data["accepted"])
         elif message_type == "mainUserInput":
             self.onMainUserInput(data["input"])
+        elif message_type == "deleteAtIndex":
+            self.onDeleteAtIndex(data["index"])
         elif message_type in ["highlightedCode", "openFiles", "readFile", "editFile", "workspaceDirectory", "getUserSecret", "runCommand", "uniqueId"]:
             self.sub_queue.post(message_type, data)
         else:
@@ -164,10 +166,11 @@ class IdeProtocolServer(AbstractIdeProtocolServer):
             "edit": file_edit.dict()
         })
 
-    async def showDiff(self, filepath: str, replacement: str):
+    async def showDiff(self, filepath: str, replacement: str, step_index: int):
         await self._send_json("showDiff", {
             "filepath": filepath,
-            "replacement": replacement
+            "replacement": replacement,
+            "step_index": step_index
         })
 
     async def setFileOpen(self, filepath: str, open: bool = True):
@@ -245,6 +248,10 @@ class IdeProtocolServer(AbstractIdeProtocolServer):
         for _, session in self.session_manager.sessions.items():
             session.autopilot.handle_manual_edits(edits)
 
+    def onDeleteAtIndex(self, index: int):
+        for _, session in self.session_manager.sessions.items():
+            session.autopilot.delete_at_index(index)
+
     def onCommandOutput(self, output: str):
         # Send the output to ALL autopilots.
         # Maybe not ideal behavior
diff --git a/continuedev/src/continuedev/server/ide_protocol.py b/continuedev/src/continuedev/server/ide_protocol.py
index 2e1f78d7..dfdca504 100644
--- a/continuedev/src/continuedev/server/ide_protocol.py
+++ b/continuedev/src/continuedev/server/ide_protocol.py
@@ -96,7 +96,11 @@ class AbstractIdeProtocolServer(ABC):
         """Called when highlighted code is updated"""
 
     @abstractmethod
-    async def showDiff(self, filepath: str, replacement: str):
+    def onDeleteAtIndex(self, index: int):
+        """Called when a step is deleted at a given index"""
+
+    @abstractmethod
+    async def showDiff(self, filepath: str, replacement: str, step_index: int):
         """Show a diff"""
 
     @abstractproperty