diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/client/palm.rs | 10 |
1 files 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(()) |