From a35f1d75019fdc6123ff87cb843e3adb3e183f24 Mon Sep 17 00:00:00 2001
From: Dillon Laird <dillonalaird@gmail.com>
Date: Fri, 23 Aug 2024 19:41:32 -0700
Subject: [PATCH] flake8 black

---
 vision_agent/agent/vision_agent_coder.py   | 7 +++++--
 vision_agent/clients/landing_public_api.py | 4 ++--
 vision_agent/tools/meta_tools.py           | 6 +++---
 vision_agent/utils/execute.py              | 1 -
 4 files changed, 10 insertions(+), 8 deletions(-)

diff --git a/vision_agent/agent/vision_agent_coder.py b/vision_agent/agent/vision_agent_coder.py
index 8134aba3..3f445d80 100644
--- a/vision_agent/agent/vision_agent_coder.py
+++ b/vision_agent/agent/vision_agent_coder.py
@@ -257,7 +257,9 @@ def pick_plan(
         try:
             best_plan = extract_json(model(chat, stream=False))  # type: ignore
         except JSONDecodeError as e:
-            _LOGGER.exception(f"Error while extracting JSON during picking best plan {str(e)}")
+            _LOGGER.exception(
+                f"Error while extracting JSON during picking best plan {str(e)}"
+            )
             pass
         count += 1
 
@@ -804,7 +806,8 @@ def chat_with_workflow(
 
             results = write_and_test_code(
                 chat=[{"role": c["role"], "content": c["content"]} for c in int_chat],
-                plan=f"\n{plan_i['thoughts']}\n-" + "\n-".join([e for e in plan_i["instructions"]]),
+                plan=f"\n{plan_i['thoughts']}\n-"
+                + "\n-".join([e for e in plan_i["instructions"]]),
                 tool_info=tool_info,
                 tool_output=tool_output_str,
                 tool_utils=T.UTILITIES_DOCSTRING,
diff --git a/vision_agent/clients/landing_public_api.py b/vision_agent/clients/landing_public_api.py
index 4c50c388..8e1158d4 100644
--- a/vision_agent/clients/landing_public_api.py
+++ b/vision_agent/clients/landing_public_api.py
@@ -1,10 +1,10 @@
 import os
-from uuid import UUID
 from typing import List
+from uuid import UUID
 
 from vision_agent.clients.http import BaseHTTP
-from vision_agent.utils.type_defs import LandingaiAPIKey
 from vision_agent.tools.meta_tools_types import BboxInputBase64, PromptTask
+from vision_agent.utils.type_defs import LandingaiAPIKey
 
 
 class LandingPublicAPI(BaseHTTP):
diff --git a/vision_agent/tools/meta_tools.py b/vision_agent/tools/meta_tools.py
index 851aab18..c38aa925 100644
--- a/vision_agent/tools/meta_tools.py
+++ b/vision_agent/tools/meta_tools.py
@@ -1,16 +1,16 @@
 import os
 import subprocess
-from uuid import UUID
 from pathlib import Path
 from typing import Any, Dict, List, Union
+from uuid import UUID
 
 import vision_agent as va
+from vision_agent.clients.landing_public_api import LandingPublicAPI
 from vision_agent.lmm.types import Message
+from vision_agent.tools.meta_tools_types import BboxInput, BboxInputBase64, PromptTask
 from vision_agent.tools.tool_utils import get_tool_documentation
 from vision_agent.tools.tools import TOOL_DESCRIPTIONS
 from vision_agent.utils.image_utils import convert_to_b64
-from vision_agent.clients.landing_public_api import LandingPublicAPI
-from vision_agent.tools.meta_tools_types import BboxInput, BboxInputBase64, PromptTask
 
 # These tools are adapted from SWE-Agent https://github.com/princeton-nlp/SWE-agent
 
diff --git a/vision_agent/utils/execute.py b/vision_agent/utils/execute.py
index b157b1df..b62308ff 100644
--- a/vision_agent/utils/execute.py
+++ b/vision_agent/utils/execute.py
@@ -416,7 +416,6 @@ def download_file(self, file_path: str) -> Path:
 
 
 class E2BCodeInterpreter(CodeInterpreter):
-
     def __init__(self, *args: Any, **kwargs: Any) -> None:
         super().__init__(*args, **kwargs)
         assert os.getenv("E2B_API_KEY"), "E2B_API_KEY environment variable must be set"