summaryrefslogtreecommitdiffstats
path: root/src/main.rs
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2023-03-03 00:40:38 +0800
committersigoden <sigoden@gmail.com>2023-03-03 00:51:23 +0800
commit1099edcb1a529e5920f2af2ec86da3f9301ec812 (patch)
treedd7079266f53c0fbfe40f6ab09c896c86f4799ae /src/main.rs
parent48ebf1895df934b900e12d4c9955829e99babe75 (diff)
downloadaichat-1099edcb1a529e5920f2af2ec86da3f9301ec812.tar.gz
refactor: optimize line sperator
Diffstat (limited to 'src/main.rs')
-rw-r--r--src/main.rs11
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)?;