From 0148ced33e2a015386749ac75472e755df4abef8 Mon Sep 17 00:00:00 2001
From: Yingjie He <yingjiehe@squareup.com>
Date: Wed, 16 Apr 2025 16:03:22 -0700
Subject: [PATCH] fix: expose not support tool call error (#2231)

---
 crates/goose/src/providers/utils.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/crates/goose/src/providers/utils.rs b/crates/goose/src/providers/utils.rs
index 6d089831..9697e67b 100644
--- a/crates/goose/src/providers/utils.rs
+++ b/crates/goose/src/providers/utils.rs
@@ -149,7 +149,7 @@ pub async fn handle_response_google_compat(response: Response) -> Result<Value,
             Err(ProviderError::Authentication(format!("Authentication failed. Please ensure your API keys are valid and have the required permissions. \
                 Status: {}. Response: {:?}", final_status, payload )))
         }
-        StatusCode::BAD_REQUEST => {
+        StatusCode::BAD_REQUEST | StatusCode::NOT_FOUND => {
             let mut error_msg = "Unknown error".to_string();
             if let Some(payload) = &payload {
                 if let Some(error) = payload.get("error") {