address comment

landing-ai · Jul 26, 2024 · 92e7724 · 92e7724
1 parent 3549ca0
commit 92e7724
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vision_agent/agent/vision_agent.py b/vision_agent/agent/vision_agent.py
@@ -846,8 +846,8 @@ def chat_with_workflow(
  {
  "type": "final_code",
  "status": "completed" if success else "failed",
- "code": DefaultImports.prepend_imports(code),
  "payload": {
+ "code": DefaultImports.prepend_imports(code),
  "test": test,
  "result": execution_result.to_json(),
  },