From acef8138280632ddd249c60abe1d2461c83410a2 Mon Sep 17 00:00:00 2001 From: shankar-landing-ai Date: Tue, 28 May 2024 21:53:54 -0700 Subject: [PATCH] fix black errors --- vision_agent/agent/agent_coder.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/vision_agent/agent/agent_coder.py b/vision_agent/agent/agent_coder.py index 02aea97a..bb1dd761 100644 --- a/vision_agent/agent/agent_coder.py +++ b/vision_agent/agent/agent_coder.py @@ -67,7 +67,9 @@ def parse_file_name(s: str) -> str: return "".join([p for p in s.split(" ") if p.endswith(".png")]) -def write_program(question: str, feedback: str, model: LLM, image: Optional[Union[str, Path]] = None) -> str: +def write_program( + question: str, feedback: str, model: LLM, image: Optional[Union[str, Path]] = None +) -> str: prompt = PROGRAM.format( docstring=TOOL_DOCSTRING, question=question, feedback=feedback )