vision-agent 0.2.93__py3-none-any.whl → 0.2.94__py3-none-any.whl

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,
             },
         }
@@ -483,12 +485,15 @@ def debug_code(
     )
     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,8 @@ def retrieve_tools(
 ) -> Dict[str, str]:
     log_progress(
         {
-            "type": "tools",
+            "type": "log",
+            "log_content": ("Retrieving tools for each plan"),
             "status": "started",
         }
     )
@@ -802,17 +808,6 @@ class VisionAgentCoder(Agent):
             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:
@@ -822,6 +817,7 @@ class VisionAgentCoder(Agent):
                         play_video(res.mp4)
 
             return {
+                "status": "completed" if success else "failed",
                 "code": DefaultImports.prepend_imports(code),
                 "test": test,
                 "test_result": execution_result,

{vision_agent-0.2.93.dist-info → vision_agent-0.2.94.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vision-agent
-Version: 0.2.93
+Version: 0.2.94
 Summary: Toolset for Vision Agent
 Author: Landing AI
 Author-email: dev@landing.ai

{vision_agent-0.2.93.dist-info → vision_agent-0.2.94.dist-info}/RECORD

@@ -3,7 +3,7 @@ vision_agent/agent/__init__.py,sha256=qpduQ9YufJQfMmG6jwKC2xmlbtR2qK8_1eQC1sGA9K
 vision_agent/agent/agent.py,sha256=Bt8yhjCFXuRdZaHxKEesG40V09nWRt45sZluri1R3AA,575
 vision_agent/agent/agent_utils.py,sha256=JXdl2xz14LKQAmScY-MIW23AD2WBFCsnI0JS6dAyj3Q,1412
 vision_agent/agent/vision_agent.py,sha256=i_rNpc7faqHTifp2c9sQE4Js3qYUKuJeiqauTp90OlE,8417
-vision_agent/agent/vision_agent_coder.py,sha256=Ouq3ws7w5zq8fw550aBrCjerNUrQ0MNvR45dkqTtReE,30321
+vision_agent/agent/vision_agent_coder.py,sha256=M8J5xE9uX8Nig1WmVmwLMeCSe0E6Bg3Mo5fPPcW_a-c,30246
 vision_agent/agent/vision_agent_coder_prompts.py,sha256=a3R_vHlT2FW3-DSn4OWgzF9zEAx-uKM4ZaTi9Kn-K54,11116
 vision_agent/agent/vision_agent_prompts.py,sha256=hjs-m4ZHR7HE1HtOeX_1rOvTQA2FMEAqEkaBbGPBYDo,6072
 vision_agent/fonts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -23,7 +23,7 @@ vision_agent/utils/image_utils.py,sha256=y69wtNla0xHZ1h1x0-vv7nOyKUq69jtjSJBiDCn
 vision_agent/utils/sim.py,sha256=1HTaiVaBiKeyXIy21IYGXlPw0TipOyw9FPOJDfyLI94,4409
 vision_agent/utils/type_defs.py,sha256=oVFJcicB-s_09lqvn61u0A5ncZsTqZArZledXWbrrg0,1384
 vision_agent/utils/video.py,sha256=rNmU9KEIkZB5-EztZNlUiKYN0mm_55A_2VGUM0QpqLA,8779
-vision_agent-0.2.93.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-vision_agent-0.2.93.dist-info/METADATA,sha256=wwrgBe5pY0y09470Yb_BrrpBZMbSMDivcZ5wC8Rt4VE,10693
-vision_agent-0.2.93.dist-info/WHEEL,sha256=7Z8_27uaHI_UZAc4Uox4PpBhQ9Y5_modZXWMxtUi4NU,88
-vision_agent-0.2.93.dist-info/RECORD,,
+vision_agent-0.2.94.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+vision_agent-0.2.94.dist-info/METADATA,sha256=Ef1XAiAKdtYv9NNwCK9HUO2QwOk-oUZPflLh2wOo9r4,10693
+vision_agent-0.2.94.dist-info/WHEEL,sha256=7Z8_27uaHI_UZAc4Uox4PpBhQ9Y5_modZXWMxtUi4NU,88
+vision_agent-0.2.94.dist-info/RECORD,,