fixed exit loop early

landing-ai · May 13, 2024 · 3bfab15 · 3bfab15
1 parent 13df90f
commit 3bfab15
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/vision_agent/agent/vision_agent_v2.py b/vision_agent/agent/vision_agent_v2.py
@@ -91,7 +91,12 @@ def write_code(
 
 
 def write_test(
- user_req: str, subtask: str, tool_info: str, code: str, model: LLM
+ user_req: str,
+ subtask: str,
+ tool_info: str,
+ _: str,
+ code: str,
+ model: LLM
 ) -> str:
  prompt = TEST.format(
  context=USER_REQ_SUBTASK_CONTEXT.format(
@@ -233,7 +238,7 @@ def run_plan(
  tool_info,
  exec,
  retrieved_ltm,
- verbose,
+ verbose=verbose,
  )
  if task["type"] == "code":
  current_code = code