feat: display media artifact creation for ui (#263)

landing-ai · Oct 10, 2024 · 1ba5438 · 1ba5438
1 parent 2fb51db
commit 1ba5438
Showing 1 changed file with 14 additions and 0 deletions.
diff --git 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]"