From 92e7724f1dccbe8316bd9ff0222fd408ae6627bd Mon Sep 17 00:00:00 2001
From: wuyiqunLu <wuyiqun.lu@landing.ai>
Date: Fri, 26 Jul 2024 17:09:04 +0800
Subject: [PATCH] address comment

---
 vision_agent/agent/vision_agent.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/vision_agent/agent/vision_agent.py b/vision_agent/agent/vision_agent.py
index 43b660ca..3c8246e3 100644
--- 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(),
                     },