From 16e9c647d469b19f94cc90e1bfa5142175a6d75a Mon Sep 17 00:00:00 2001 From: sigoden Date: Sun, 14 Apr 2024 22:23:01 +0000 Subject: [PATCH] refactor: adjust indicator text and render strategy --- src/main.rs | 5 +++-- src/render/stream.rs | 2 +- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/src/main.rs b/src/main.rs index 0bd5a34..dc66657 100644 --- a/src/main.rs +++ b/src/main.rs @@ -117,14 +117,15 @@ fn start_directive( let mut client = init_client(config)?; ensure_model_capabilities(client.as_mut(), input.required_capabilities())?; config.read().maybe_print_send_tokens(&input); - let output = if !stdout().is_terminal() || no_stream { + let is_terminal_stdout = stdout().is_terminal(); + let output = if !is_terminal_stdout || no_stream { let output = client.send_message(input.clone())?; let output = if code_mode && output.trim_start().starts_with("```") { extract_block(&output) } else { output.clone() }; - if no_stream { + if is_terminal_stdout { let render_options = config.read().get_render_options()?; let mut markdown_render = MarkdownRender::init(render_options)?; println!("{}", markdown_render.render(&output).trim()); diff --git a/src/render/stream.rs b/src/render/stream.rs index ee38759..f2fdbb1 100644 --- a/src/render/stream.rs +++ b/src/render/stream.rs @@ -65,7 +65,7 @@ fn markdown_stream_inner( let columns = terminal::size()?.0; - let mut spinner = Spinner::new(" Generating"); + let mut spinner = Spinner::new(" Thinking"); 'outer: loop { if abort.aborted() {