diff options
Diffstat (limited to 'src/main.rs')
| -rw-r--r-- | src/main.rs | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/main.rs b/src/main.rs index dc66657..be46bf1 100644 --- 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() |