enable openai to use morph

Author: alexylon

src/api/anthropic.rs

@@ -37,20 +37,6 @@ impl AnthropicClient {
         Ok(Self { client })
     }
 
-    pub(crate) async fn check_response_status(
-        response: reqwest::Response,
-    ) -> Result<reqwest::Response> {
-        if !response.status().is_success() {
-            let status = response.status();
-            let error_text = response.text().await.unwrap_or_default();
-            return Err(SofosError::Api(format!(
-                "API request failed with status {}: {}",
-                status, error_text
-            )));
-        }
-        Ok(response)
-    }
-
     /// Check if we can reach the API endpoint
     #[allow(dead_code)]
     async fn check_connectivity(&self) -> Result<()> {
@@ -115,7 +101,7 @@ impl AnthropicClient {
 
             match self.client.post(&url).json(&request).send().await {
                 Ok(response) => {
-                    let response = Self::check_response_status(response).await?;
+                    let response = super::utils::check_response_status(response).await?;
                     let result = response.json::<CreateMessageResponse>().await?;
                     return Ok(result);
                 }
@@ -156,7 +142,7 @@ impl AnthropicClient {
         request.stream = Some(true);
         let url = format!("{}/messages", API_BASE);
         let response = self.client.post(&url).json(&request).send().await?;
-        let response = Self::check_response_status(response).await?;
+        let response = super::utils::check_response_status(response).await?;
 
         let stream = response
             .bytes_stream()

src/api/mod.rs

@@ -2,6 +2,7 @@ pub mod anthropic;
 pub mod morph;
 pub mod openai;
 pub mod types;
+pub mod utils;
 
 pub use anthropic::AnthropicClient;
 pub use morph::MorphClient;

src/api/morph.rs

@@ -85,7 +85,7 @@ impl MorphClient {
 
         let url = format!("{}/chat/completions", MORPH_BASE_URL);
         let response = self.client.post(&url).json(&request).send().await?;
-        let response = crate::api::anthropic::AnthropicClient::check_response_status(response).await?;
+        let response = super::utils::check_response_status(response).await?;
         let result: MorphResponse = response.json().await?;
 
         result

src/api/openai.rs

@@ -163,7 +163,7 @@ impl OpenAIClient {
 
         let url = format!("{}/chat/completions", OPENAI_API_BASE);
         let response = self.client.post(&url).json(&body).send().await?;
-        let response = check_response_status(response).await?;
+        let response = super::utils::check_response_status(response).await?;
         let parsed: OpenAIChatResponse = response.json().await?;
 
         let choice =
@@ -268,7 +268,7 @@ impl OpenAIClient {
         }
 
         let response = self.client.post(&url).json(&body).send().await?;
-        let response = check_response_status(response).await?;
+        let response = super::utils::check_response_status(response).await?;
 
         let response_text = response.text().await?;
         if std::env::var("SOFOS_DEBUG").is_ok() {
@@ -521,15 +521,3 @@ struct OpenAIResponseUsage {
     #[serde(default)]
     output_tokens: Option<u32>,
 }
-
-async fn check_response_status(response: reqwest::Response) -> Result<reqwest::Response> {
-    if !response.status().is_success() {
-        let status = response.status();
-        let error_text = response.text().await.unwrap_or_default();
-        return Err(SofosError::Api(format!(
-            "API request failed with status {}: {}",
-            status, error_text
-        )));
-    }
-    Ok(response)
-}

src/api/utils.rs

@@ -0,0 +1,13 @@
+use crate::error::{Result, SofosError};
+
+pub async fn check_response_status(response: reqwest::Response) -> Result<reqwest::Response> {
+    if !response.status().is_success() {
+        let status = response.status();
+        let error_text = response.text().await.unwrap_or_default();
+        return Err(SofosError::Api(format!(
+            "API request failed with status {}: {}",
+            status, error_text
+        )));
+    }
+    Ok(response)
+}