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/ollama.rs | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src/client/ollama.rs') diff --git a/src/client/ollama.rs b/src/client/ollama.rs index 158bbbe..eebf301 100644 --- a/src/client/ollama.rs +++ b/src/client/ollama.rs @@ -65,7 +65,7 @@ async fn send_message(builder: RequestBuilder) -> Result<(String, CompletionDeta let res = builder.send().await?; let status = res.status(); let data = res.json().await?; - if status != 200 { + if !status.is_success() { catch_error(&data, status.as_u16())?; } let text = data["message"]["content"] @@ -77,7 +77,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 = res.json().await?; catch_error(&data, status.as_u16())?; } else { -- cgit v1.2.3