summaryrefslogtreecommitdiffstats
path: root/src/render
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2024-06-06 06:08:41 +0800
committerGitHub <noreply@github.com>2024-06-06 06:08:41 +0800
commit93ef6da6298f3a98127ed572659b6c24a3f5295e (patch)
tree7f11cb9adfc7ab5cdfc420cf69d10dcde15179c3 /src/render
parentc26594c3c85fb99ad5b0db1c535b337585d3c103 (diff)
downloadaichat-93ef6da6298f3a98127ed572659b6c24a3f5295e.tar.gz
refactor: global IS_STDOUT_TERMINAL (#571)
Diffstat (limited to 'src/render')
-rw-r--r--src/render/mod.rs6
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