fix: check reqwest response status code is 200 (#646)

Author: badmonster0

src/llm/anthropic.rs

@@ -76,6 +76,13 @@ impl LlmGenerationClient for Client {
             .send()
             .await
             .context("HTTP error")?;
+        if !resp.status().is_success() {
+            bail!(
+                "Anthropic API error: {:?}\n{}\n",
+                resp.status(),
+                resp.text().await?
+            );
+        }
         let mut resp_json: Value = resp.json().await.context("Invalid JSON")?;
         if let Some(error) = resp_json.get("error") {
             bail!("Anthropic API error: {:?}", error);

src/llm/ollama.rs

@@ -1,8 +1,7 @@
+use crate::prelude::*;
+
 use super::LlmGenerationClient;
-use anyhow::Result;
-use async_trait::async_trait;
 use schemars::schema::SchemaObject;
-use serde::{Deserialize, Serialize};
 
 pub struct Client {
     generate_url: String,
@@ -67,8 +66,14 @@ impl LlmGenerationClient for Client {
             .json(&req)
             .send()
             .await?;
-        let body = res.text().await?;
-        let json: OllamaResponse = serde_json::from_str(&body)?;
+        if !res.status().is_success() {
+            bail!(
+                "Ollama API error: {:?}\n{}\n",
+                res.status(),
+                res.text().await?
+            );
+        }
+        let json: OllamaResponse = res.json().await?;
         Ok(super::LlmGenerateResponse {
             text: json.response,
         })