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/vertexai.rs | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src/client/vertexai.rs') diff --git a/src/client/vertexai.rs b/src/client/vertexai.rs index 3ac917d..e798451 100644 --- a/src/client/vertexai.rs +++ b/src/client/vertexai.rs @@ -115,7 +115,7 @@ pub async fn gemini_send_message(builder: RequestBuilder) -> Result<(String, Com 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())?; } gemini_extract_completion_text(&data) @@ -127,7 +127,7 @@ pub async fn gemini_send_message_streaming( ) -> 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