diff options
| author | sigoden <sigoden@gmail.com> | 2024-10-29 14:15:02 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-10-29 14:15:02 +0800 |
| commit | bf07cdf256e1e3008578972f5916a4d6763554f5 (patch) | |
| tree | 3f48ead051403a7f270198578e86a92672bb4208 /src/render/mod.rs | |
| parent | bb542f6e92fd782b70c44fddb21d013c289d0a3c (diff) | |
| download | aichat-bf07cdf256e1e3008578972f5916a4d6763554f5.tar.gz | |
refactor: improve code quality (#956)
Diffstat (limited to 'src/render/mod.rs')
| -rw-r--r-- | src/render/mod.rs | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/render/mod.rs b/src/render/mod.rs index 97161f8..9d93203 100644 --- a/src/render/mod.rs +++ b/src/render/mod.rs @@ -13,14 +13,14 @@ use tokio::sync::mpsc::UnboundedReceiver; pub async fn render_stream( rx: UnboundedReceiver<SseEvent>, config: &GlobalConfig, - abort: AbortSignal, + abort_signal: AbortSignal, ) -> Result<()> { let ret = if *IS_STDOUT_TERMINAL { let render_options = config.read().render_options()?; let mut render = MarkdownRender::init(render_options)?; - markdown_stream(rx, &mut render, &abort).await + markdown_stream(rx, &mut render, &abort_signal).await } else { - raw_stream(rx, &abort).await + raw_stream(rx, &abort_signal).await }; ret.map_err(|err| err.context("Failed to reader stream")) } |
