diff --git a/vision_agent/agent/vision_agent.py b/vision_agent/agent/vision_agent.py
index 3b7bd2a8..cc0c04a6 100644
--- a/vision_agent/agent/vision_agent.py
+++ b/vision_agent/agent/vision_agent.py
@@ -6,7 +6,11 @@
 
 from vision_agent.agent import Agent
 from vision_agent.agent.agent_utils import extract_json
-from vision_agent.agent.vision_agent_prompts import EXAMPLES_CODE1, EXAMPLES_CODE2, VA_CODE
+from vision_agent.agent.vision_agent_prompts import (
+    EXAMPLES_CODE1,
+    EXAMPLES_CODE2,
+    VA_CODE,
+)
 from vision_agent.lmm import LMM, Message, OpenAILMM
 from vision_agent.tools import META_TOOL_DOCSTRING
 from vision_agent.utils import CodeInterpreterFactory
@@ -80,7 +84,9 @@ def run_code_action(code: str, code_interpreter: CodeInterpreter) -> str:
         # for log in result.logs.stderr:
         #     return_str += log.replace("\\n", "\n")
         if result.error:
-            return_str += "\n" + result.error.value + "\n".join(result.error.traceback_raw)
+            return_str += (
+                "\n" + result.error.value + "\n".join(result.error.traceback_raw)
+            )
 
     return return_str
 
diff --git a/vision_agent/lmm/lmm.py b/vision_agent/lmm/lmm.py
index 9e313818..8f052663 100644
--- a/vision_agent/lmm/lmm.py
+++ b/vision_agent/lmm/lmm.py
@@ -5,7 +5,7 @@
 import os
 from abc import ABC, abstractmethod
 from pathlib import Path
-from typing import Any, Callable, Dict, List, Optional, Sequence, Union, cast
+from typing import Any, Callable, Dict, List, Optional, Union, cast
 
 import requests
 from openai import AzureOpenAI, OpenAI
@@ -13,6 +13,7 @@
 
 import vision_agent.tools as T
 from vision_agent.tools.prompts import CHOOSE_PARAMS, SYSTEM_PROMPT
+
 from .types import Message
 
 _LOGGER = logging.getLogger(__name__)
diff --git a/vision_agent/tools/meta_tools.py b/vision_agent/tools/meta_tools.py
index b3600103..21ba2bb6 100644
--- a/vision_agent/tools/meta_tools.py
+++ b/vision_agent/tools/meta_tools.py
@@ -1,9 +1,9 @@
 import subprocess
 from pathlib import Path
 from typing import List
-from vision_agent.lmm.types import Message
 
 import vision_agent as va
+from vision_agent.lmm.types import Message
 from vision_agent.tools.tool_utils import get_tool_documentation
 
 CURRENT_FILE = None
diff --git a/vision_agent/tools/tool_utils.py b/vision_agent/tools/tool_utils.py
index b59151c5..4fa97f47 100644
--- a/vision_agent/tools/tool_utils.py
+++ b/vision_agent/tools/tool_utils.py
@@ -1,9 +1,9 @@
+import inspect
 import logging
 import os
-import pandas as pd
-import inspect
-from typing import Any, Dict, List, Callable
+from typing import Any, Callable, Dict, List
 
+import pandas as pd
 from requests import Session
 from requests.adapters import HTTPAdapter
 from urllib3.util.retry import Retry