add code

landing-ai · Aug 6, 2024 · fa55a0d · fa55a0d
1 parent 07b9cdf
commit fa55a0d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/vision_agent/agent/vision_agent_coder.py b/vision_agent/agent/vision_agent_coder.py
@@ -186,6 +186,7 @@ def pick_plan(
  if tool_output.success
  else "Code execution failed"
  ),
+ "code": DefaultImports.prepend_imports(code),
  # "payload": tool_output.to_json(),
  "status": "completed" if tool_output.success else "failed",
  }