added visualized images to all_tool_results

landing-ai · Mar 29, 2024 · 8e68cff · 8e68cff
1 parent 1872e13
commit 8e68cff
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/vision_agent/agent/vision_agent.py b/vision_agent/agent/vision_agent.py
@@ -501,6 +501,7 @@ def chat_with_workflow(
  )
 
  visualized_images = visualize_result(all_tool_results)
+ all_tool_results.append({"visualized_images": visualized_images})
  reflection = self_reflect(
  self.reflect_model,
  question,