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