Fix JSON parsing in generate.

package.json

@@ -1,7 +1,7 @@
 {
   "type": "module",
   "name": "ollama",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "description": "Interface with an ollama instance over HTTP.",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

src/index.ts

@@ -42,22 +42,26 @@ export class Ollama {
 		}
 
 		for await (const chunk of response.body) {
-			const res: GenerateResponse | GenerateResponseEnd = JSON.parse(chunk.toString());
-
-			if (res.done) {
-				return {
-					model: res.model,
-					createdAt: new Date(res.created_at),
-					context: res.context,
-					totalDuration: res.total_duration,
-					loadDuration: res.load_duration,
-					promptEvalCount: res.prompt_eval_count,
-					evalCount: res.eval_count,
-					evalDuration: res.eval_duration
-				};
-			}
+			const messages = chunk.toString().split("\n").filter(s => s.length !== 0);
 
-			yield res.response;
+			for (const message of messages) {
+				const res: GenerateResponse | GenerateResponseEnd = JSON.parse(message);
+
+				if (res.done) {
+					return {
+						model: res.model,
+						createdAt: new Date(res.created_at),
+						context: res.context,
+						totalDuration: res.total_duration,
+						loadDuration: res.load_duration,
+						promptEvalCount: res.prompt_eval_count,
+						evalCount: res.eval_count,
+						evalDuration: res.eval_duration
+					};
+				}
+
+				yield res.response;
+			}
 		}
 
 		throw new Error("Did not recieve done response in stream.");