From a193710a7f88de12ef11c5a5c66a5591150fe247 Mon Sep 17 00:00:00 2001 From: sigoden Date: Thu, 25 Apr 2024 14:03:16 +0800 Subject: refactor: extract common catch_error (#437) --- src/client/cohere.rs | 14 ++------------ 1 file changed, 2 insertions(+), 12 deletions(-) (limited to 'src/client/cohere.rs') diff --git a/src/client/cohere.rs b/src/client/cohere.rs index 828b799..df86843 100644 --- a/src/client/cohere.rs +++ b/src/client/cohere.rs @@ -1,6 +1,6 @@ use super::{ - extract_sytem_message, json_stream, message::*, Client, CohereClient, ExtraConfig, Model, - ModelConfig, PromptType, ReplyHandler, SendData, + catch_error, extract_sytem_message, json_stream, message::*, Client, CohereClient, ExtraConfig, + Model, ModelConfig, PromptType, ReplyHandler, SendData, }; use crate::utils::PromptKind; @@ -184,16 +184,6 @@ fn build_body(data: SendData, model: &Model) -> Result { Ok(body) } -fn catch_error(data: &Value, status: u16) -> Result<()> { - debug!("Invalid response, status: {status}, data: {data}"); - - if let Some(message) = data["message"].as_str() { - bail!("{message}"); - } else { - bail!("Invalid response, status: {status}, data: {data}"); - } -} - fn extract_text(data: &Value) -> Result<&str> { match data["text"].as_str() { Some(text) => Ok(text), -- cgit v1.2.3