diff options
| author | sigoden <sigoden@gmail.com> | 2023-03-03 00:40:38 +0800 |
|---|---|---|
| committer | sigoden <sigoden@gmail.com> | 2023-03-03 00:51:23 +0800 |
| commit | 1099edcb1a529e5920f2af2ec86da3f9301ec812 (patch) | |
| tree | dd7079266f53c0fbfe40f6ab09c896c86f4799ae | |
| parent | 48ebf1895df934b900e12d4c9955829e99babe75 (diff) | |
| download | aichat-1099edcb1a529e5920f2af2ec86da3f9301ec812.tar.gz | |
refactor: optimize line sperator
| -rw-r--r-- | src/main.rs | 11 |
1 files changed, 5 insertions, 6 deletions
diff --git a/src/main.rs b/src/main.rs index b7e3f23..c1535d7 100644 --- a/src/main.rs +++ b/src/main.rs @@ -99,7 +99,7 @@ fn start() -> Result<()> { match text { Some(text) => { let output = runtime.block_on(async move { acquire(&client, &config, &text).await })?; - println!("{output}"); + println!("{}", output.trim()); } None => run_repl(runtime, client, config, role_name)?, } @@ -159,7 +159,7 @@ fn run_repl( } _ = tokio::signal::ctrl_c() => { *trigged_ctrlc = true; - dump(" Abort current session") + dump(" Abort current session.") } } }); @@ -200,9 +200,9 @@ fn run_repl( Some(role_) => { *role = Some(role_.clone()); } - None => dump("Unknown role"), + None => dump("Unknown role."), }, - None => dump("Usage: .role <name>"), + None => dump("Usage: .role <name>."), }, _ => { dump("Unknown command. Type \".help\" for more information."); @@ -265,8 +265,7 @@ async fn handle_input(client: &Client, config: &Config, text: &str) -> Result<() while let Some(part) = stream.next().await { let chunk = part?.data; if chunk == "[DONE]" { - println!(); - stdout().flush().unwrap(); + dump("\n"); break; } else { let data: Value = serde_json::from_str(&chunk)?; |
