diff options
| author | sigoden <sigoden@gmail.com> | 2024-06-06 06:08:41 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-06-06 06:08:41 +0800 |
| commit | 93ef6da6298f3a98127ed572659b6c24a3f5295e (patch) | |
| tree | 7f11cb9adfc7ab5cdfc420cf69d10dcde15179c3 /src/render | |
| parent | c26594c3c85fb99ad5b0db1c535b337585d3c103 (diff) | |
| download | aichat-93ef6da6298f3a98127ed572659b6c24a3f5295e.tar.gz | |
refactor: global IS_STDOUT_TERMINAL (#571)
Diffstat (limited to 'src/render')
| -rw-r--r-- | src/render/mod.rs | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/src/render/mod.rs b/src/render/mod.rs index 054a830..9fa6787 100644 --- a/src/render/mod.rs +++ b/src/render/mod.rs @@ -4,12 +4,10 @@ mod stream; pub use self::markdown::{MarkdownRender, RenderOptions}; use self::stream::{markdown_stream, raw_stream}; -use crate::utils::{error_text, AbortSignal}; +use crate::utils::{error_text, AbortSignal, IS_STDOUT_TERMINAL}; use crate::{client::SseEvent, config::GlobalConfig}; use anyhow::Result; -use is_terminal::IsTerminal; -use std::io::stdout; use tokio::sync::mpsc::UnboundedReceiver; pub async fn render_stream( @@ -17,7 +15,7 @@ pub async fn render_stream( config: &GlobalConfig, abort: AbortSignal, ) -> Result<()> { - if stdout().is_terminal() { + if *IS_STDOUT_TERMINAL { let render_options = config.read().get_render_options()?; let mut render = MarkdownRender::init(render_options)?; markdown_stream(rx, &mut render, &abort).await |
