Archives/aichat
2
0
Fork 0
mirror of https://github.com/sigoden/aichat synced 2024-11-18 09:28:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: recovery cursor in case of stream abort

Browse Source
This commit is contained in:
sigoden 2023-03-10 07:44:34 +08:00
parent 865fb49af9
commit 54e6edbf21
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/render/repl.rs
View File

@ -70,11 +70,16 @@ fn repl_render_stream_inner(
}
ReplyStreamEvent::Done => {
let output = markdown_render.render_line_stateless(&buffer);
queue!(writer, style::Print(output.trim_end()), style::Print("\n"))?;
if cfg!(windows) {
queue!(writer, style::Print("\n"))?;
let trimed_output = output.trim_end();
if !trimed_output.is_empty() {
queue!(writer, style::Print(output.trim_end()))?;
writer.flush()?;
}
let (_, row) = cursor::position()?;
queue!(writer, cursor::MoveTo(0, row), style::Print("\n\n"))?;
writer.flush()?;
break;
}
}