diff --git a/vision_agent/agent/vision_agent_coder.py b/vision_agent/agent/vision_agent_coder.py index 4ef6b07e..352697d7 100644 --- 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_fixed = [ + { + "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_fixed, headers='keys', tablefmt='mixed_grid', maxcolwidths=_MAX_TABULATE_COL_WIDTH)}" ) tool_infos = retrieve_tools(