From 33cf8e71af32c59058fc15a8ac1901fa3fab0b69 Mon Sep 17 00:00:00 2001 From: Dillon Laird Date: Thu, 29 Aug 2024 19:18:53 -0700 Subject: [PATCH] add trace for last edited artifact --- vision_agent/tools/meta_tools.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/vision_agent/tools/meta_tools.py b/vision_agent/tools/meta_tools.py index b3c3ed8c..833ad542 100644 --- a/vision_agent/tools/meta_tools.py +++ b/vision_agent/tools/meta_tools.py @@ -194,6 +194,8 @@ def create_code_artifact(artifacts: Artifacts, name: str) -> str: artifacts[name] = "" return_str = f"[Artifact {name} created]" print(return_str) + + display({MimeType.APPLICATION_JSON: {"last_artifact": name}}) return return_str @@ -270,6 +272,7 @@ def edit_code_artifact( artifacts[name] = "".join(edited_lines) + display({MimeType.APPLICATION_JSON: {"last_artifact": name}}) return open_code_artifact(artifacts, name, cur_line) @@ -313,6 +316,8 @@ def detect_dogs(image_path: str): artifacts[name] = code code_lines = code.splitlines(keepends=True) total_lines = len(code_lines) + + display({MimeType.APPLICATION_JSON: {"last_artifact": name}}) return view_lines(code_lines, 0, total_lines, name, total_lines) @@ -367,6 +372,8 @@ def detect_dogs(image_path: str): artifacts[name] = code code_lines = code.splitlines(keepends=True) total_lines = len(code_lines) + + display({MimeType.APPLICATION_JSON: {"last_artifact": name}}) return view_lines(code_lines, 0, total_lines, name, total_lines)