From b34ab23bdf546992d92460f2104e4a072927a75a Mon Sep 17 00:00:00 2001
From: Mingrui Zhang <mingruizhang50@gmail.com>
Date: Tue, 30 Jul 2024 06:16:51 +0800
Subject: [PATCH 1/3] done

---
 vision_agent/agent/vision_agent_coder.py | 37 +++++++++++-------------
 1 file changed, 17 insertions(+), 20 deletions(-)

diff --git a/vision_agent/agent/vision_agent_coder.py b/vision_agent/agent/vision_agent_coder.py
index ebf9ead4..ec1bf17e 100644
--- a/vision_agent/agent/vision_agent_coder.py
+++ b/vision_agent/agent/vision_agent_coder.py
@@ -373,7 +373,7 @@ def write_and_test_code(
             "code": DefaultImports.prepend_imports(code),
             "payload": {
                 "test": test,
-                "result": result.to_json(),
+                # "result": result.to_json(),
             },
         }
     )
@@ -426,7 +426,8 @@ def debug_code(
 ) -> tuple[str, str, Execution]:
     log_progress(
         {
-            "type": "code",
+            "type": "log",
+            "log_content": ("Debugging code"),
             "status": "started",
         }
     )
@@ -469,10 +470,11 @@ def debug_code(
     )
     log_progress(
         {
-            "type": "code",
+            "type": "log",
+            "log_content": ("Running code"),
             "status": "running",
+            "code": DefaultImports.prepend_imports(code),
             "payload": {
-                "code": DefaultImports.prepend_imports(code),
                 "test": test,
             },
         }
@@ -482,13 +484,16 @@ def debug_code(
         f"{DefaultImports.to_code_string()}\n{code}\n{test}"
     )
     log_progress(
-        {
-            "type": "code",
+                {
+            "type": "log",
+            "log_content": (
+                "Code execution succeed" if result.success else "Code execution failed"
+            ),
             "status": "completed" if result.success else "failed",
+            "code": DefaultImports.prepend_imports(code),
             "payload": {
-                "code": DefaultImports.prepend_imports(code),
                 "test": test,
-                "result": result.to_json(),
+                # "result": result.to_json(),
             },
         }
     )
@@ -525,7 +530,10 @@ def retrieve_tools(
 ) -> Dict[str, str]:
     log_progress(
         {
-            "type": "tools",
+            "type": "log",
+            "log_content": (
+                "Retrieving tools for each plan"
+            ),
             "status": "started",
         }
     )
@@ -802,17 +810,6 @@ def chat_with_workflow(
             plan.append({"code": code, "test": test, "plan": plan_i})
 
             execution_result = cast(Execution, results["test_result"])
-            self.log_progress(
-                {
-                    "type": "final_code",
-                    "status": "completed" if success else "failed",
-                    "payload": {
-                        "code": DefaultImports.prepend_imports(code),
-                        "test": test,
-                        "result": execution_result.to_json(),
-                    },
-                }
-            )
 
             if display_visualization:
                 for res in execution_result.results:

From e85b31b9a375bd39b5b5689c701de7528c1fa510 Mon Sep 17 00:00:00 2001
From: Mingrui Zhang <mingruizhang50@gmail.com>
Date: Tue, 30 Jul 2024 06:21:55 +0800
Subject: [PATCH 2/3] fix lint

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

diff --git a/vision_agent/agent/vision_agent_coder.py b/vision_agent/agent/vision_agent_coder.py
index ec1bf17e..09ebf53a 100644
--- a/vision_agent/agent/vision_agent_coder.py
+++ b/vision_agent/agent/vision_agent_coder.py
@@ -484,7 +484,7 @@ def debug_code(
         f"{DefaultImports.to_code_string()}\n{code}\n{test}"
     )
     log_progress(
-                {
+        {
             "type": "log",
             "log_content": (
                 "Code execution succeed" if result.success else "Code execution failed"
@@ -819,6 +819,7 @@ def chat_with_workflow(
                         play_video(res.mp4)
 
             return {
+                "status": "completed" if success else "failed",
                 "code": DefaultImports.prepend_imports(code),
                 "test": test,
                 "test_result": execution_result,

From 1910c093d7b251464b43bbaca3f76f6f3bedd96e Mon Sep 17 00:00:00 2001
From: Mingrui Zhang <mingruizhang50@gmail.com>
Date: Tue, 30 Jul 2024 06:23:41 +0800
Subject: [PATCH 3/3] fix lint

---
 vision_agent/agent/vision_agent_coder.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/vision_agent/agent/vision_agent_coder.py b/vision_agent/agent/vision_agent_coder.py
index 09ebf53a..199a673a 100644
--- a/vision_agent/agent/vision_agent_coder.py
+++ b/vision_agent/agent/vision_agent_coder.py
@@ -531,9 +531,7 @@ def retrieve_tools(
     log_progress(
         {
             "type": "log",
-            "log_content": (
-                "Retrieving tools for each plan"
-            ),
+            "log_content": ("Retrieving tools for each plan"),
             "status": "started",
         }
     )