From 9149954c7bbc5cce541793e2e7cdcd50b6af2a3c Mon Sep 17 00:00:00 2001 From: sigoden Date: Mon, 15 Apr 2024 11:28:25 +0800 Subject: feat: always use stream unless set `--no-stream` explicitly (#415) --- src/main.rs | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'src/main.rs') 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() -- cgit v1.2.3