Make it easier to visualize planned tasks in logs (#31)

* Make it easier to visualize planned tasks in logs

* Add type stubs

---------

Co-authored-by: Yazhou Cao <[email protected]>

pyproject.toml

@@ -30,6 +30,7 @@ openai = "1.*"
 typing_extensions = "4.*"
 moviepy = "1.*"
 opencv-python-headless = "4.*"
+tabulate = "^0.9.0"
 
 [tool.poetry.group.dev.dependencies]
 autoflake = "1.*"
@@ -47,6 +48,7 @@ setuptools = "^68.0.0"
 mkdocs = "^1.5.3"
 mkdocstrings = {extras = ["python"], version = "^0.23.0"}
 mkdocs-material = "^9.4.2"
+types-tabulate = "^0.9.0.20240106"
 
 [tool.pytest.ini_options]
 log_cli = true

vision_agent/agent/vision_agent.py

@@ -4,6 +4,8 @@
 from pathlib import Path
 from typing import Any, Callable, Dict, List, Optional, Tuple, Union
 
+from tabulate import tabulate
+
 from vision_agent.llm import LLM, OpenAILLM
 from vision_agent.lmm import LMM, OpenAILMM
 from vision_agent.tools import TOOLS
@@ -268,6 +270,11 @@ def retrieval(
  {"task": question, "tool_name": tool_name, "parameters": parameters}
  ]
 
+ _LOGGER.info(
+ f"""Going to run the following {len(tool_results)} tool(s) in sequence:
+{tabulate(tool_results, headers="keys", tablefmt="mixed_grid")}"""
+ )
+
  def parse_tool_results(result: Dict[str, Union[Dict, List]]) -> Any:
  call_results: List[Any] = []
  if isinstance(result["parameters"], Dict):
@@ -298,8 +305,6 @@ def create_tasks(
  {k: v["description"] for k, v in tools.items()},
  reflections,
  )
-
- _LOGGER.info(f"Tasks: {tasks}")
  if tasks is not None:
  task_list = [{"task": task, "id": i + 1} for i, task in enumerate(tasks)]
  task_list = task_topology(task_model, question, task_list)
@@ -309,6 +314,10 @@ def create_tasks(
  _LOGGER.error(f"Failed topological_sort on: {task_list}")
  else:
  task_list = []
+ _LOGGER.info(
+ f"""Planned tasks:
+{tabulate(task_list, headers="keys", tablefmt="mixed_grid")}"""
+ )
  return task_list