feat: always use stream unless set --no-stream explicitly (#415)

sigoden · Apr 15, 2024 · 9149954 · 9149954 · johnd0e · Apr 16, 2024
1 parent 16e9c64
commit 9149954
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/main.rs b/src/main.rs
@@ -118,9 +118,10 @@ fn start_directive(
     ensure_model_capabilities(client.as_mut(), input.required_capabilities())?;
     config.read().maybe_print_send_tokens(&input);
     let is_terminal_stdout = stdout().is_terminal();
-    let output = if !is_terminal_stdout || no_stream {
+    let extract_code = !is_terminal_stdout && code_mode;
+    let output = if no_stream || extract_code {
         let output = client.send_message(input.clone())?;
-        let output = if code_mode && output.trim_start().starts_with("```") {
+        let output = if extract_code && output.trim_start().starts_with("```") {
             extract_block(&output)
         } else {
             output.clone()