fixed tabulate issue

landing-ai · Aug 12, 2024 · fa66467 · fa66467
1 parent 38949a4
commit fa66467
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/vision_agent/agent/vision_agent_coder.py b/vision_agent/agent/vision_agent_coder.py
@@ -735,8 +735,17 @@ def chat_with_workflow(
 
  if self.verbosity >= 1:
  for p in plans:
+ # tabulate will fail if the keys are not the same for all elements
+ p = [
+ {
+ "instructions": (
+ e["instructions"] if "instructions" in e else ""
+ )
+ }
+ for e in plans[p]
+ ]
  _LOGGER.info(
- f"\n{tabulate(tabular_data=plans[p], headers='keys', tablefmt='mixed_grid', maxcolwidths=_MAX_TABULATE_COL_WIDTH)}"
+ f"\n{tabulate(tabular_data=p, headers='keys', tablefmt='mixed_grid', maxcolwidths=_MAX_TABULATE_COL_WIDTH)}"
  )
 
  tool_infos = retrieve_tools(