summaryrefslogtreecommitdiffstats
path: root/src/client.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/client.rs')
-rw-r--r--src/client.rs16
1 files changed, 8 insertions, 8 deletions
diff --git a/src/client.rs b/src/client.rs
index 29dc592..7fc2824 100644
--- a/src/client.rs
+++ b/src/client.rs
@@ -69,8 +69,8 @@ impl ChatGptClient {
}
async fn send_message_inner(&self, content: &str) -> Result<String> {
- if self.config.lock().dry_run {
- return Ok(self.config.lock().echo_messages(content));
+ if self.config.read().dry_run {
+ return Ok(self.config.read().echo_messages(content));
}
let builder = self.request_builder(content, false)?;
let data: Value = builder.send().await?.json().await?;
@@ -90,8 +90,8 @@ impl ChatGptClient {
content: &str,
handler: &mut ReplyStreamHandler,
) -> Result<()> {
- if self.config.lock().dry_run {
- handler.text(&self.config.lock().echo_messages(content))?;
+ if self.config.read().dry_run {
+ handler.text(&self.config.read().echo_messages(content))?;
return Ok(());
}
let builder = self.request_builder(content, true)?;
@@ -125,7 +125,7 @@ impl ChatGptClient {
fn build_client(&self) -> Result<Client> {
let mut builder = Client::builder();
- if let Some(proxy) = self.config.lock().proxy.as_ref() {
+ if let Some(proxy) = self.config.read().proxy.as_ref() {
builder = builder.proxy(Proxy::all(proxy).with_context(|| "Invalid config.proxy")?);
}
let client = builder
@@ -136,13 +136,13 @@ impl ChatGptClient {
}
fn request_builder(&self, content: &str, stream: bool) -> Result<RequestBuilder> {
- let messages = self.config.lock().build_messages(content)?;
+ let messages = self.config.read().build_messages(content)?;
let mut body = json!({
"model": MODEL,
"messages": messages,
});
- if let Some(v) = self.config.lock().get_temperature() {
+ if let Some(v) = self.config.read().get_temperature() {
body.as_object_mut()
.and_then(|m| m.insert("temperature".into(), json!(v)));
}
@@ -155,7 +155,7 @@ impl ChatGptClient {
let builder = self
.build_client()?
.post(API_URL)
- .bearer_auth(self.config.lock().get_api_key())
+ .bearer_auth(self.config.read().get_api_key())
.json(&body);
Ok(builder)