From 3bfab15ef4c6d388a38571bc5c061fa3db9f965a Mon Sep 17 00:00:00 2001 From: Dillon Laird Date: Mon, 13 May 2024 14:27:02 -0700 Subject: [PATCH] fixed exit loop early --- vision_agent/agent/vision_agent_v2.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/vision_agent/agent/vision_agent_v2.py b/vision_agent/agent/vision_agent_v2.py index dc5d9626..8c4fd9cd 100644 --- 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