diff options
| author | sigoden <sigoden@gmail.com> | 2023-03-06 16:35:51 +0800 |
|---|---|---|
| committer | sigoden <sigoden@gmail.com> | 2023-03-06 16:35:51 +0800 |
| commit | c12ae0275108acbbb124a41ff936b441723ae625 (patch) | |
| tree | 1b4ed2a76d2c3518ff817770ee620ba1663c13ed /src/client.rs | |
| parent | f73f45ac9b9041fe9c3c1a10e3757523553a05b8 (diff) | |
| download | aichat-c12ae0275108acbbb124a41ff936b441723ae625.tar.gz | |
chore: improve code by renameing some function
Diffstat (limited to 'src/client.rs')
| -rw-r--r-- | src/client.rs | 19 |
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>, |
