Skip to content

Commit 8de4d68

Browse files
committed
and the same for ChatMessage
1 parent 6e5d29d commit 8de4d68

File tree

1 file changed

+8
-10
lines changed

1 file changed

+8
-10
lines changed

shiny/ui/_chat_normalize.py

Lines changed: 8 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -64,22 +64,20 @@ def normalize(self, message: Any) -> ChatMessage:
6464
if "content" not in x:
6565
raise ValueError("Message must have 'content' key")
6666
content, deps = process_ui(cast(TagChild, x["content"]))
67-
return ChatMessage(
68-
content=content,
69-
role=x.get("role", "assistant"),
70-
html_deps=deps,
71-
)
67+
res = ChatMessage(content=content, role=x.get("role", "assistant"))
68+
if deps:
69+
res["html_deps"] = deps
70+
return res
7271

7372
def normalize_chunk(self, chunk: Any) -> ChatMessage:
7473
x = cast("dict[str, Any]", chunk)
7574
if "content" not in x:
7675
raise ValueError("Message must have 'content' key")
7776
content, deps = process_ui(cast(TagChild, x["content"]))
78-
return ChatMessage(
79-
content=content,
80-
role=x.get("role", "assistant"),
81-
html_deps=deps,
82-
)
77+
res = ChatMessage(content=content, role=x.get("role", "assistant"))
78+
if deps:
79+
res["html_deps"] = deps
80+
return res
8381

8482
def can_normalize(self, message: Any) -> bool:
8583
return isinstance(message, dict)

0 commit comments

Comments
 (0)