Handle typed message parts when reverting multimodal content for `Cha…

…tObject`
WolframResearch · Nov 29, 2023 · 4a6e0ae · 4a6e0ae
1 parent 733dbb6
commit 4a6e0ae
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/Source/Chatbook/Actions.wl b/Source/Chatbook/Actions.wl
@@ -666,8 +666,13 @@ revertMultimodalContent // beginDefinition;
 revertMultimodalContent[ messages_List ] :=
     revertMultimodalContent /@ messages;
 
-revertMultimodalContent[ as: KeyValuePattern[ "Content" -> content_List ] ] :=
-    <| as, "Content" -> StringJoin @ Select[ content, StringQ ] |>;
+revertMultimodalContent[ as: KeyValuePattern[ "Content" -> content_List ] ] := <|
+    as,
+    "Content" -> StringJoin @ Cases[
+        content,
+        s_String | KeyValuePattern @ { "Type" -> "Text", "Data" -> s_String } :> s
+    ]
+|>;
 
 revertMultimodalContent[ as: KeyValuePattern[ "Content" -> _String ] ] :=
     as;