From 695045c6d492f7c4605e4d442bc87a90eb5525cb Mon Sep 17 00:00:00 2001 From: Kalvin Chau Date: Wed, 19 Feb 2025 11:06:18 -0800 Subject: [PATCH] feat: propagate external_model_message upwards to user --- crates/goose/src/providers/databricks.rs | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/crates/goose/src/providers/databricks.rs b/crates/goose/src/providers/databricks.rs index 6c6569d79..f23ada7df 100644 --- a/crates/goose/src/providers/databricks.rs +++ b/crates/goose/src/providers/databricks.rs @@ -181,7 +181,16 @@ impl DatabricksProvider { let mut error_msg = "Unknown error".to_string(); if let Some(payload) = &payload { - error_msg = payload.get("message").and_then(|m| m.as_str()).unwrap_or("Unknown error").to_string(); + // try to convert message to string, if that fails use external_model_message + error_msg = payload + .get("message") + .and_then(|m| m.as_str()) + .or_else(|| { + payload.get("external_model_message") + .and_then(|ext| ext.get("message")) + .and_then(|m| m.as_str()) + }) + .unwrap_or("Unknown error").to_string(); } tracing::debug!(