From 3549ca09d3df1a63026d899469610746b319e34c Mon Sep 17 00:00:00 2001
From: wuyiqunLu <wuyiqun.lu@landing.ai>
Date: Fri, 26 Jul 2024 16:59:53 +0800
Subject: [PATCH 1/2] fix: change logs to log

---
 vision_agent/agent/vision_agent.py | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

diff --git a/vision_agent/agent/vision_agent.py b/vision_agent/agent/vision_agent.py
index d525962e..43b660ca 100644
--- a/vision_agent/agent/vision_agent.py
+++ b/vision_agent/agent/vision_agent.py
@@ -184,7 +184,7 @@ def pick_plan(
         k = list(plans.keys())[0]
         log_progress(
             {
-                "type": "logs",
+                "type": "log",
                 "log_content": "Plans created",
                 "status": "completed",
                 "payload": plans[k],
@@ -194,7 +194,7 @@ def pick_plan(
 
     log_progress(
         {
-            "type": "logs",
+            "type": "log",
             "log_content": "Generating code to pick best plan",
             "status": "started",
         }
@@ -212,7 +212,7 @@ def pick_plan(
     code = extract_code(model(prompt))
     log_progress(
         {
-            "type": "logs",
+            "type": "log",
             "log_content": "Executing code to test plan",
             "code": code,
             "status": "running",
@@ -229,7 +229,7 @@ def pick_plan(
 
     log_progress(
         {
-            "type": "logs",
+            "type": "log",
             "log_content": (
                 "Code execution succeed"
                 if tool_output.success
@@ -251,7 +251,7 @@ def pick_plan(
         )
         log_progress(
             {
-                "type": "logs",
+                "type": "log",
                 "log_content": "Retry running code",
                 "code": code,
                 "status": "running",
@@ -263,7 +263,7 @@ def pick_plan(
         )
         log_progress(
             {
-                "type": "logs",
+                "type": "log",
                 "log_content": (
                     "Code execution succeed"
                     if tool_output.success
@@ -318,7 +318,7 @@ def pick_plan(
 
     log_progress(
         {
-            "type": "logs",
+            "type": "log",
             "log_content": "Picked best plan",
             "status": "complete",
             "payload": best_plans,
@@ -771,7 +771,7 @@ def chat_with_workflow(
 
             self.log_progress(
                 {
-                    "type": "logs",
+                    "type": "log",
                     "log_content": "Creating plans",
                     "status": "started",
                 }
@@ -798,7 +798,7 @@ def chat_with_workflow(
             if test_multi_plan:
                 self.log_progress(
                     {
-                        "type": "logs",
+                        "type": "log",
                         "log_content": "Creating plans",
                         "status": "completed",
                         "payload": tool_lists,

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 2/2] 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(),
                     },