From 34c03894ae2a640e95ba1f3659b6e3aa55708f08 Mon Sep 17 00:00:00 2001 From: root Date: Fri, 27 Mar 2026 20:38:27 -0500 Subject: [PATCH] Auto-retry on ALL SQL errors, not just schema errors Previous: only retried on "Schema error" or "No field named" Now: retries on any error (type mismatches, execution errors, etc.) Model gets full error message + schema to write corrected SQL. Co-Authored-By: Claude Opus 4.6 (1M context) --- crates/ui/src/main.rs | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/crates/ui/src/main.rs b/crates/ui/src/main.rs index 4786c58..e8e719d 100644 --- a/crates/ui/src/main.rs +++ b/crates/ui/src/main.rs @@ -292,7 +292,7 @@ fn AskPanel(datasets: Vec) -> Element { // Step 3b: If schema error, retry with the error as feedback if let Err(ref err) = query_result { - if err.contains("Schema error") || err.contains("No field named") { + if err.contains("error") { step.set("fixing SQL...".into()); let retry_prompt = format!( "The SQL you wrote had an error:\n{err}\n\n\ @@ -368,7 +368,7 @@ fn AskPanel(datasets: Vec) -> Element { step.set("running query...".into()); let query_result = run_sql(&sql).await; if let Err(ref err) = query_result { - if err.contains("Schema error") || err.contains("No field named") { + if err.contains("error") { step.set("fixing SQL...".into()); let retry_prompt = format!( "The SQL you wrote had an error:\n{err}\n\n{schema_ctx}\n\nOriginal question: {q}\n\nWrite a CORRECTED SQL query using ONLY the columns listed. Output ONLY SQL."