feat: propagate external_model_message upwards on errors (#1297)

2025-02-19 12:11:59 -08:00
parent 537e1dcbeb
commit a59119bc27
1 changed files with 10 additions and 1 deletions
@@ -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!(