fixed type issues

vision_agent/agent/vision_agent_coder.py

@@ -5,7 +5,7 @@
 from pathlib import Path
 from typing import Any, Callable, Dict, List, Optional, Sequence, Union, cast
 
-from redbaron import RedBaron
+from redbaron import RedBaron # type: ignore
 from tabulate import tabulate
 
 import vision_agent.tools as T

vision_agent/tools/meta_tools.py

@@ -11,7 +11,7 @@
 
 import numpy as np
 from IPython.display import display
-from redbaron import RedBaron
+from redbaron import RedBaron # type: ignore
 
 import vision_agent as va
 from vision_agent.agent.agent_utils import extract_json
@@ -25,8 +25,6 @@
 from vision_agent.utils.image_utils import convert_to_b64, numpy_to_bytes
 from vision_agent.utils.video import frames_to_bytes
 
-# These tools are adapted from SWE-Agent https://github.com/princeton-nlp/SWE-agent
-
 CURRENT_FILE = None
 CURRENT_LINE = 0
 DEFAULT_WINDOW_SIZE = 100
@@ -155,6 +153,9 @@ def __contains__(self, name: str) -> bool:
  return name in self.artifacts
 
 
+# These tools are adapted from SWE-Agent https://github.com/princeton-nlp/SWE-agent
+
+
 def format_lines(lines: List[str], start_idx: int) -> str:
  output = ""
  for i, line in enumerate(lines):