diff --git a/vision_agent/agent/vision_agent_v2.py b/vision_agent/agent/vision_agent_v2.py index 336a5ef5..04b90b19 100644 --- a/vision_agent/agent/vision_agent_v2.py +++ b/vision_agent/agent/vision_agent_v2.py @@ -63,7 +63,7 @@ def extract_json(json_str: str) -> Dict[str, Any]: # get the last ``` not one from an intermediate string json_str = json_str[: json_str.find("}```")] json_dict = json.loads(json_str) - return json_dict + return json_dict # type: ignore def write_plan( @@ -80,8 +80,8 @@ def write_plan( context = USER_REQ_CONTEXT.format(user_requirement=user_requirements) prompt = PLAN.format(context=context, plan=str(plan), tool_desc=tool_desc) chat[-1]["content"] = prompt - plan = extract_json(model.chat(chat)) - return plan["user_req"], plan["plan"] # type: ignore + new_plan = extract_json(model.chat(chat)) + return new_plan["user_req"], new_plan["plan"] def write_code(