Attempt plan design a couple times before giving up

hecrj · Feb 15, 2025 · 49cf60e · 49cf60e
1 parent a4ee831
commit 49cf60e
Showing 1 changed file with 17 additions and 1 deletion.
diff --git a/core/src/plan.rs b/core/src/plan.rs
@@ -86,7 +86,23 @@ impl Plan {
                 return Ok(());
             };
 
-            let plan = design(assistant, history).run(&progress).await?;
+            let plan = {
+                let mut attempt = 0;
+
+                loop {
+                    log::info!("Designing plan ({attempt})...");
+
+                    match design(assistant, history).run(&progress).await {
+                        Err(error) if attempt < 3 => {
+                            log::warn!("Plan design failed: {error}");
+                        }
+                        result => break result?,
+                    }
+
+                    attempt += 1;
+                }
+            };
+
             progress.send(Event::Designed(plan.clone())).await;
 
             execute(assistant, history, query, &plan)