Merge branch 'main' of https://github.com/continuedev/continue

continuedev · Jan 16, 2024 · dbbccef · dbbccef
2 parents c2c4984 + 982c0cd
commit dbbccef
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/core/llm/llms/Ollama.ts b/core/llm/llms/Ollama.ts
@@ -123,6 +123,7 @@ class Ollama extends BaseLLM {
       buffer += value;
       // Split the buffer into individual JSON chunks
       const chunks = buffer.split("\n");
+
       for (let i = 0; i < chunks.length; i++) {
         const chunk = chunks[i];
         if (chunk.trim() !== "") {
@@ -161,6 +162,7 @@ class Ollama extends BaseLLM {
       buffer += value;
       // Split the buffer into individual JSON chunks
       const chunks = buffer.split("\n");
+      buffer = chunks.pop() ?? '';
       for (let i = 0; i < chunks.length; i++) {
         const chunk = chunks[i];
         if (chunk.trim() !== "") {
@@ -179,8 +181,6 @@ class Ollama extends BaseLLM {
           }
         }
       }
-      // Assign the last chunk to the buffer
-      buffer = chunks[chunks.length - 1];
     }
   }
 }