refactor: check res status code

1 files changed, 2 insertions, 2 deletions

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 {