summaryrefslogtreecommitdiffstats
path: root/src
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2024-04-14 22:23:01 +0000
committersigoden <sigoden@gmail.com>2024-04-14 22:23:01 +0000
commit16e9c647d469b19f94cc90e1bfa5142175a6d75a (patch)
treef844611efcba9565d1a45ebc6358031b964807ae /src
parent38e594505bd9095a3c3618b186a29c6eb63972a5 (diff)
downloadaichat-16e9c647d469b19f94cc90e1bfa5142175a6d75a.tar.gz
refactor: adjust indicator text and render strategy
Diffstat (limited to 'src')
-rw-r--r--src/main.rs5
-rw-r--r--src/render/stream.rs2
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() {