From 97f6d48c4240859a981ea994840fc68000514a43 Mon Sep 17 00:00:00 2001 From: sigoden Date: Mon, 29 Apr 2024 23:09:19 +0000 Subject: refactor: check res status code --- src/client/cohere.rs | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src/client/cohere.rs') diff --git a/src/client/cohere.rs b/src/client/cohere.rs index e2a23b0..7f07640 100644 --- a/src/client/cohere.rs +++ b/src/client/cohere.rs @@ -48,7 +48,7 @@ async fn send_message(builder: RequestBuilder) -> Result<(String, CompletionDeta let res = builder.send().await?; let status = res.status(); let data: Value = res.json().await?; - if status != 200 { + if !status.is_success() { catch_error(&data, status.as_u16())?; } @@ -58,7 +58,7 @@ async fn send_message(builder: RequestBuilder) -> Result<(String, CompletionDeta async fn send_message_streaming(builder: RequestBuilder, handler: &mut SseHandler) -> Result<()> { let res = builder.send().await?; let status = res.status(); - if status != 200 { + if !status.is_success() { let data: Value = res.json().await?; catch_error(&data, status.as_u16())?; } else { -- cgit v1.2.3