summaryrefslogtreecommitdiffstats
path: root/src/client.rs
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2023-03-06 16:35:51 +0800
committersigoden <sigoden@gmail.com>2023-03-06 16:35:51 +0800
commitc12ae0275108acbbb124a41ff936b441723ae625 (patch)
tree1b4ed2a76d2c3518ff817770ee620ba1663c13ed /src/client.rs
parentf73f45ac9b9041fe9c3c1a10e3757523553a05b8 (diff)
downloadaichat-c12ae0275108acbbb124a41ff936b441723ae625.tar.gz
chore: improve code by renameing some function
Diffstat (limited to 'src/client.rs')
-rw-r--r--src/client.rs19
1 files changed, 11 insertions, 8 deletions
diff --git a/src/client.rs b/src/client.rs
index 3f74aa6..589ad1a 100644
--- a/src/client.rs
+++ b/src/client.rs
@@ -29,12 +29,15 @@ impl ChatGptClient {
Ok(s)
}
- pub fn acquire(&self, input: &str, prompt: Option<String>) -> Result<String> {
- self.runtime
- .block_on(async { self.acquire_inner(input, prompt).await })
+ pub fn send_message(&self, input: &str, prompt: Option<String>) -> Result<String> {
+ self.runtime.block_on(async {
+ self.send_message_inner(input, prompt)
+ .await
+ .with_context(|| "Failed to send message")
+ })
}
- pub fn acquire_stream(
+ pub fn send_message_streaming(
&self,
input: &str,
prompt: Option<String>,
@@ -51,9 +54,9 @@ impl ChatGptClient {
let ctrlc = handler.get_ctrlc();
self.runtime.block_on(async {
tokio::select! {
- ret = self.acquire_stream_inner(input, prompt, handler) => {
+ ret = self.send_message_streaming_inner(input, prompt, handler) => {
handler.done();
- ret
+ ret.with_context(|| "Failed to send message streaming")
}
_ = watch_ctrlc(ctrlc.clone()) => {
handler.done();
@@ -67,7 +70,7 @@ impl ChatGptClient {
})
}
- async fn acquire_inner(&self, content: &str, prompt: Option<String>) -> Result<String> {
+ async fn send_message_inner(&self, content: &str, prompt: Option<String>) -> Result<String> {
if self.config.borrow().dry_run {
return Ok(combine(content, prompt));
}
@@ -82,7 +85,7 @@ impl ChatGptClient {
Ok(output.to_string())
}
- async fn acquire_stream_inner(
+ async fn send_message_streaming_inner(
&self,
content: &str,
prompt: Option<String>,