From 1ba54383f80bf778fc367c648cc031e68f527d84 Mon Sep 17 00:00:00 2001
From: wuyiqunLu <132986242+wuyiqunLu@users.noreply.github.com>
Date: Thu, 10 Oct 2024 12:43:52 +0800
Subject: [PATCH] feat: display media artifact creation for ui (#263)

---
 vision_agent/tools/meta_tools.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/vision_agent/tools/meta_tools.py b/vision_agent/tools/meta_tools.py
index 78a2ecae..024d4230 100644
--- a/vision_agent/tools/meta_tools.py
+++ b/vision_agent/tools/meta_tools.py
@@ -1,3 +1,4 @@
+import base64
 import difflib
 import json
 import os
@@ -510,6 +511,19 @@ def write_media_artifact(
         return f"[Invalid media type {type(media)}]"
     artifacts[name] = media_bytes
     print(f"[Media {name} saved]")
+    display(
+        {
+            MimeType.APPLICATION_ARTIFACT: json.dumps(
+                {
+                    "name": name,
+                    "action": "create",
+                    "content": base64.b64encode(media_bytes).decode("utf-8"),
+                    "contentType": "media_output",
+                }
+            )
+        },
+        raw=True,
+    )
     return f"[Media {name} saved]"