summaryrefslogtreecommitdiffstats
path: root/src/main.rs
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/main.rs
parent38e594505bd9095a3c3618b186a29c6eb63972a5 (diff)
downloadaichat-16e9c647d469b19f94cc90e1bfa5142175a6d75a.tar.gz
refactor: adjust indicator text and render strategy
Diffstat (limited to 'src/main.rs')
-rw-r--r--src/main.rs5
1 files changed, 3 insertions, 2 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());