From 18079a63912e8740ebbe24e5ed394f17ba734fb2 Mon Sep 17 00:00:00 2001
From: wuyiqunLu <wuyiqun.lu@landing.ai>
Date: Thu, 10 Oct 2024 12:27:20 +0800
Subject: [PATCH] feat: display media artifact creation for ui

---
 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]"