revert count

landing-ai · Jan 18, 2025 · 4039f06 · 4039f06
1 parent 6a759fd
commit 4039f06
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/vision_agent/agent/vision_agent_planner_v2.py b/vision_agent/agent/vision_agent_planner_v2.py
@@ -218,8 +218,8 @@ def execute_code_action(
             f"[bold cyan]Code Execution Output ({end - start:.2f}s):[/bold cyan] [yellow]{escape(obs)}[/yellow]"
         )
 
-    count = 0
-    while not execution.success and count < 3:
+    count = 1
+    while not execution.success and count <= 3:
         prompt = FIX_BUG.format(chat_history=get_planning(chat), code=code, error=obs)
         response = cast(str, model.chat([{"role": "user", "content": prompt}]))
         new_code = extract_tag(response, "code")