diff options
| author | sigoden <sigoden@gmail.com> | 2024-04-14 22:23:01 +0000 |
|---|---|---|
| committer | sigoden <sigoden@gmail.com> | 2024-04-14 22:23:01 +0000 |
| commit | 16e9c647d469b19f94cc90e1bfa5142175a6d75a (patch) | |
| tree | f844611efcba9565d1a45ebc6358031b964807ae /src/main.rs | |
| parent | 38e594505bd9095a3c3618b186a29c6eb63972a5 (diff) | |
| download | aichat-16e9c647d469b19f94cc90e1bfa5142175a6d75a.tar.gz | |
refactor: adjust indicator text and render strategy
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 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()); |
