diff --git a/examples/chat_completion.rs b/examples/chat_completion.rs index 635add58..f82173b3 100644 --- a/examples/chat_completion.rs +++ b/examples/chat_completion.rs @@ -22,10 +22,6 @@ async fn main() -> Result<(), Box> { let result = client.chat_completion(req).await?; println!("Content: {:?}", result.choices[0].message.content); - // print response headers - for (key, value) in client.response_headers.unwrap().iter() { - println!("{}: {:?}", key, value); - } Ok(()) } diff --git a/examples/completion.rs b/examples/completion.rs index 95cbc231..138c1fe7 100644 --- a/examples/completion.rs +++ b/examples/completion.rs @@ -5,7 +5,7 @@ use std::env; #[tokio::main] async fn main() -> Result<(), Box> { let api_key = env::var("OPENAI_API_KEY").unwrap().to_string(); - let mut client = OpenAIClient::builder().with_api_key(api_key).build()?; + let client = OpenAIClient::builder().with_api_key(api_key).build()?; let req = CompletionRequest::new( completion::GPT3_TEXT_DAVINCI_003.to_string(), diff --git a/examples/openrouter.rs b/examples/openrouter.rs index 5295bf41..8bd5cdbc 100644 --- a/examples/openrouter.rs +++ b/examples/openrouter.rs @@ -24,7 +24,6 @@ async fn main() -> Result<(), Box> { let result = client.chat_completion(req).await?; println!("Content: {:?}", result.choices[0].message.content); - println!("Response Headers: {:?}", client.response_headers); Ok(()) } diff --git a/src/v1/api.rs b/src/v1/api.rs index 1c50695e..7470f49b 100644 --- a/src/v1/api.rs +++ b/src/v1/api.rs @@ -72,7 +72,6 @@ pub struct OpenAIClient { proxy: Option, timeout: Option, headers: Option, - pub response_headers: Option, } impl OpenAIClientBuilder { @@ -126,7 +125,6 @@ impl OpenAIClientBuilder { proxy: self.proxy, timeout: self.timeout, headers: self.headers, - response_headers: None, }) } } @@ -184,7 +182,7 @@ impl OpenAIClient { } async fn post( - &mut self, + & self, path: &str, body: &impl serde::ser::Serialize, ) -> Result { @@ -213,7 +211,7 @@ impl OpenAIClient { } async fn post_form( - &mut self, + &self, path: &str, form: Form, ) -> Result { @@ -231,16 +229,14 @@ impl OpenAIClient { } async fn handle_response( - &mut self, + &self, response: Response, ) -> Result { let status = response.status(); - let headers = response.headers().clone(); if status.is_success() { let text = response.text().await.unwrap_or_else(|_| "".to_string()); match serde_json::from_str::(&text) { Ok(parsed) => { - self.response_headers = Some(headers); Ok(parsed) } Err(e) => Err(APIError::CustomError { @@ -259,7 +255,7 @@ impl OpenAIClient { } pub async fn completion( - &mut self, + &self, req: CompletionRequest, ) -> Result { self.post("completions", &req).await