Merge pull request #39 from milosgajdos/update-readme

update: update README streaming example
pepperoni21 · Apr 9, 2024 · a23110e · a23110e
2 parents 2876a80 + 5758d30
commit a23110e
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -56,9 +56,11 @@ let mut stream = ollama.generate_stream(GenerationRequest::new(model, prompt)).a
 
 let mut stdout = tokio::io::stdout();
 while let Some(res) = stream.next().await {
-    let res = res.unwrap();
-    stdout.write(res.response.as_bytes()).await.unwrap();
-    stdout.flush().await.unwrap();
+    let responses = res.unwrap();
+    for resp in responses {
+        stdout.write(resp.response.as_bytes()).await.unwrap();
+        stdout.flush().await.unwrap();
+    }
 }
 ```