diff options
Diffstat (limited to 'src/client/vertexai.rs')
| -rw-r--r-- | src/client/vertexai.rs | 4 |
1 files changed, 2 insertions, 2 deletions
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 { |