diff options
Diffstat (limited to 'src/client/openai.rs')
| -rw-r--r-- | src/client/openai.rs | 16 |
1 files changed, 3 insertions, 13 deletions
diff --git a/src/client/openai.rs b/src/client/openai.rs index 37da878..e5aae24 100644 --- a/src/client/openai.rs +++ b/src/client/openai.rs @@ -1,4 +1,6 @@ -use super::{ExtraConfig, Model, ModelConfig, OpenAIClient, PromptType, ReplyHandler, SendData}; +use super::{ + catch_error, ExtraConfig, Model, ModelConfig, OpenAIClient, PromptType, ReplyHandler, SendData, +}; use crate::utils::PromptKind; @@ -154,15 +156,3 @@ pub fn openai_build_body(data: SendData, model: &Model) -> Value { } body } - -fn catch_error(data: &Value, status: u16) -> Result<()> { - debug!("Invalid response, status: {status}, data: {data}"); - if let Some(error) = data["error"].as_object() { - if let (Some(type_), Some(message)) = (error["type"].as_str(), error["message"].as_str()) { - bail!("{message} (type: {type_})"); - } - } else if let Some(message) = data["message"].as_str() { - bail!("{message}"); - } - bail!("Invalid response, status: {status}, data: {data}"); -} |
