From fb3bef7365be95c66f11751e0c5ccc3403435529 Mon Sep 17 00:00:00 2001
From: Nate Sesti <sestinj@gmail.com>
Date: Mon, 19 Jun 2023 15:45:18 -0700
Subject: remove ticks from generation

---
 continuedev/src/continuedev/core/autopilot.py  | 1 +
 continuedev/src/continuedev/steps/core/core.py | 4 ++--
 2 files changed, 3 insertions(+), 2 deletions(-)

(limited to 'continuedev')

diff --git a/continuedev/src/continuedev/core/autopilot.py b/continuedev/src/continuedev/core/autopilot.py
index 8b0eaca3..3d9c46a5 100644
--- a/continuedev/src/continuedev/core/autopilot.py
+++ b/continuedev/src/continuedev/core/autopilot.py
@@ -105,6 +105,7 @@ class Autopilot(ContinueBaseModel):
     _step_depth: int = 0
 
     async def retry_at_index(self, index: int):
+        self.history.timeline[index].step.hide = True
         self._retry_queue.post(str(index), None)
 
     async def delete_at_index(self, index: int):
diff --git a/continuedev/src/continuedev/steps/core/core.py b/continuedev/src/continuedev/steps/core/core.py
index a780cedd..12a0b2ce 100644
--- a/continuedev/src/continuedev/steps/core/core.py
+++ b/continuedev/src/continuedev/steps/core/core.py
@@ -296,7 +296,7 @@ class DefaultModelEditCodeStep(Step):
                 for line in chunk_lines:
                     if "</modified_code_to_edit>" in line:
                         break
-                    elif "<modified_code_to_edit>" in line or "<file_prefix>" in line or "</file_prefix>" in line or "<file_suffix>" in line or "</file_suffix>" in line or "<user_request>" in line or "</user_request>" in line or "<code_to_edit>" in line or "</code_to_edit>" in line:
+                    elif "```" in line or "<modified_code_to_edit>" in line or "<file_prefix>" in line or "</file_prefix>" in line or "<file_suffix>" in line or "</file_suffix>" in line or "<user_request>" in line or "</user_request>" in line or "<code_to_edit>" in line or "</code_to_edit>" in line:
                         continue
                     elif i < len(original_lines) and line == original_lines[i]:
                         i += 1
@@ -308,7 +308,7 @@ class DefaultModelEditCodeStep(Step):
             # Add the unfinished line
             if unfinished_line != "":
                 unfinished_line = unfinished_line.removesuffix(
-                    "</modified_code_to_edit>").removesuffix("</code_to_edit>")
+                    "</modified_code_to_edit>").removesuffix("</code_to_edit>").removesuffix("```")
                 if not i < len(original_lines) or not unfinished_line == original_lines[i]:
                     await add_line(i, unfinished_line)
                 lines.append(unfinished_line)
-- 
cgit v1.2.3-70-g09d2