diff options
| author | sigoden <sigoden@gmail.com> | 2023-03-06 07:08:51 +0800 |
|---|---|---|
| committer | sigoden <sigoden@gmail.com> | 2023-03-06 07:08:51 +0800 |
| commit | c5a2f4603c3c3eb7995f4e77453ab8ca50ff88b8 (patch) | |
| tree | 16c394645144c2dd102327d08d8c5bdcc7ae3f7b /src/render.rs | |
| parent | 2152667fc6551129fd0bb14ca9e403728855aea3 (diff) | |
| download | aichat-c5a2f4603c3c3eb7995f4e77453ab8ca50ff88b8.tar.gz | |
chore: optimize code
Diffstat (limited to 'src/render.rs')
| -rw-r--r-- | src/render.rs | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/src/render.rs b/src/render.rs index f44283c..57633a6 100644 --- a/src/render.rs +++ b/src/render.rs @@ -1,4 +1,4 @@ -use crate::{repl::RenderStreamEvent, utils::dump}; +use crate::{repl::ReplyStreamEvent, utils::dump}; use anyhow::Result; use crossbeam::channel::Receiver; use mdcat::{ @@ -14,7 +14,7 @@ use std::sync::{ use syntect::parsing::SyntaxSet; pub fn render_stream( - rx: Receiver<RenderStreamEvent>, + rx: Receiver<ReplyStreamEvent>, ctrlc: Arc<AtomicBool>, markdown_render: Arc<MarkdownRender>, ) -> Result<()> { @@ -26,7 +26,7 @@ pub fn render_stream( } if let Ok(evt) = rx.try_recv() { match evt { - RenderStreamEvent::Text(text) => { + ReplyStreamEvent::Text(text) => { buffer.push_str(&text); if text.contains('\n') { let markdown = markdown_render.render(&buffer)?; @@ -40,7 +40,7 @@ pub fn render_stream( } } } - RenderStreamEvent::Done => { + ReplyStreamEvent::Done => { let markdown = markdown_render.render(&buffer)?; let tail = markdown .lines() |
