Follow up to #1736: correct markup generation for user role messages (#…

…1737)
posit-dev · Oct 15, 2024 · f0cc575 · f0cc575
1 parent d0354b9
commit f0cc575
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/shiny/ui/_chat.py b/shiny/ui/_chat.py
@@ -1084,9 +1084,12 @@ def chat_ui(
         else:
             raise ValueError("Each message must be a string or a dictionary.")
 
-        message_tags.append(
-            Tag("shiny-chat-message", content=msg["content"], role=msg["role"])
-        )
+        if msg["role"] == "user":
+            tag_name = "shiny-user-message"
+        else:
+            tag_name = "shiny-chat-message"
+
+        message_tags.append(Tag(tag_name, content=msg["content"]))
 
     res = Tag(
         "shiny-chat-container",