black and isort

vision_agent/agent/agent_coder.py

@@ -4,6 +4,7 @@
 import sys
 from pathlib import Path
 from typing import Dict, List, Optional, Union
+
 from rich.console import Console
 from rich.syntax import Syntax
 
@@ -169,10 +170,14 @@ def chat(
  for _ in range(self.max_turns):
  code = write_program(question, feedback, self.coder_agent)
  if self.verbose:
- _CONSOLE.print(Syntax(code, "python", theme="gruvbox-dark", line_numbers=True))
+ _CONSOLE.print(
+ Syntax(code, "python", theme="gruvbox-dark", line_numbers=True)
+ )
  debug = write_debug(question, code, feedback, self.tester_agent)
  if self.verbose:
- _CONSOLE.print(Syntax(debug, "python", theme="gruvbox-dark", line_numbers=True))
+ _CONSOLE.print(
+ Syntax(debug, "python", theme="gruvbox-dark", line_numbers=True)
+ )
  results = execute_tests(code, debug)
  _LOGGER.info(
  f"execution results: passed: {results['passed']}\n{results['result']}"
@@ -183,7 +188,9 @@ def chat(
  code, debug, results["result"].strip(), feedback, self.coder_agent # type: ignore
  )
  if self.verbose:
- _CONSOLE.print(Syntax(code, "python", theme="gruvbox-dark", line_numbers=True))
+ _CONSOLE.print(
+ Syntax(code, "python", theme="gruvbox-dark", line_numbers=True)
+ )
  else:
  # TODO: Sometimes it prints nothing, so we need to handle that case
  # TODO: The visual agent reflection does not work very well, needs more testing

vision_agent/agent/vision_agent_v2.py

@@ -379,7 +379,9 @@ def chat_with_workflow(
  self.long_term_memory,
  self.verbosity,
  )
- success = all(task["success"] if "success" in task else False for task in plan)
+ success = all(
+ task["success"] if "success" in task else False for task in plan
+ )
  working_memory.update(working_memory_i)
 
  if not success:

vision_agent/tools/tools_v2.py

@@ -1,8 +1,8 @@
 import inspect
 import io
+import json
 import logging
 import tempfile
-import json
 from importlib import resources
 from pathlib import Path
 from typing import Any, Callable, Dict, List, Tuple, Union, cast

vision_agent/utils/sim.py

@@ -56,7 +56,9 @@ def save(self, sim_file: Union[str, Path]) -> None:
  df = df.drop("embs", axis=1)
  df.to_csv(sim_file / "df.csv", index=False)
 
- def top_k(self, query: str, k: int = 5, thresh: Optional[float] = None) -> Sequence[Dict]:
+ def top_k(
+ self, query: str, k: int = 5, thresh: Optional[float] = None
+ ) -> Sequence[Dict]:
  """Returns the top k most similar items to the query.
 
  Parameters: