Add opening to first chunk

NullDev · Jul 14, 2024 · 1fd6712 · 1fd6712
1 parent 55431e2
commit 1fd6712
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/service/chunking.ts b/src/service/chunking.ts
@@ -7,16 +7,16 @@ export function splitInChunks(
     lines: readonly string[],
     { charLimitPerChunk = 2000, chunkOpeningLine, chunkClosingLine }: SplitOptions,
 ): string[] {
-    let charsInChunk = 0;
-    let currentChunk: string[] = [];
-    const chunks = [currentChunk];
-
     const open = chunkOpeningLine ?? "";
     const close = chunkClosingLine ?? "";
     // we need + 1 for line ending if there is an opening
     const chunkOverhead =
         (open.length ? open.length + 1 : 0) + (close.length ? close.length + 1 : 0);
 
+    let currentChunk = open ? [open] : [];
+    let charsInChunk = currentChunk.length;
+    const chunks = [currentChunk];
+
     for (const line of lines) {
         const appendedChars = line.length + 1; // + 1 for line ending
         if (charsInChunk + appendedChars + chunkOverhead > charLimitPerChunk) {