better filtering of model artifacts

continuedev · Jun 3, 2024 · c96780a · c96780a
1 parent d9f9fd0
commit c96780a
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/core/autocomplete/lineStream.ts b/core/autocomplete/lineStream.ts
@@ -109,12 +109,16 @@ export async function* skipLines(stream: LineStream): LineStream {
   }
 }
 
+const LINES_TO_REMOVE_BEFORE_START = [
+  "<COMPLETION>",
+  "[CODE]",
+  "<START EDITING HERE>",
+];
+
 function shouldRemoveLineBeforeStart(line: string): boolean {
   return (
     line.trimStart().startsWith("```") ||
-    line.trim() === "[CODE]" ||
-    line.trim() === "" ||
-    line.trim() === "<START EDITING HERE>"
+    LINES_TO_REMOVE_BEFORE_START.some((l) => line.trim() === l)
   );
 }