From 399ed3337f1c4edaab39a534b63cc9d00c09dd07 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Mon, 6 Nov 2023 19:37:45 +0800
Subject: refactor: improve error message for PaLM client (#213)

---
 src/client/palm.rs | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/client/palm.rs b/src/client/palm.rs
index 87b60d1..c59b269 100644
--- a/src/client/palm.rs
+++ b/src/client/palm.rs
@@ -86,7 +86,13 @@ async fn send_message(builder: RequestBuilder) -> Result<String> {
 
     let output = data["candidates"][0]["content"]
         .as_str()
-        .ok_or_else(|| anyhow!("Unexpected response {data}"))?;
+        .ok_or_else(|| {
+            if let Some(reason) = data["filters"][0]["reason"].as_str() {
+                anyhow!("Content Filtering: {reason}")
+            } else {
+                anyhow!("Unexpected response")
+            }
+        })?;
 
     Ok(output.to_string())
 }
@@ -96,7 +102,7 @@ fn check_error(data: &Value) -> Result<()> {
         if let Some(message) = error["message"].as_str() {
             bail!("{message}");
         } else {
-            bail!("Request failed. {}", data);
+            bail!("Error {}", Value::Object(error.clone()));
         }
     }
     Ok(())
-- 
cgit v1.2.3