Skip to content

Commit 6e5d29d

Browse files
committed
do the same for transformed messages
1 parent 9b3f95e commit 6e5d29d

File tree

2 files changed

+8
-3
lines changed

2 files changed

+8
-3
lines changed

shiny/ui/_chat.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1317,14 +1317,19 @@ def as_transformed_message(message: ChatMessage) -> TransformedMessage:
13171317
transform_key = "content_client"
13181318
pre_transform_key = "content_server"
13191319

1320-
return TransformedMessage(
1320+
res = TransformedMessage(
13211321
content_client=message["content"],
13221322
content_server=message["content"],
13231323
role=message["role"],
13241324
transform_key=transform_key,
13251325
pre_transform_key=pre_transform_key,
1326-
html_deps=message.get("html_deps", []),
13271326
)
13281327

1328+
deps = message.get("html_deps", [])
1329+
if deps:
1330+
res["html_deps"] = deps
1331+
1332+
return res
1333+
13291334

13301335
CHAT_INSTANCES: WeakValueDictionary[str, Chat] = WeakValueDictionary()

shiny/ui/_chat_types.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ class TransformedMessage(TypedDict):
2424
role: Role
2525
transform_key: Literal["content_client", "content_server"]
2626
pre_transform_key: Literal["content_client", "content_server"]
27-
html_deps: list[dict[str, str]]
27+
html_deps: NotRequired[list[dict[str, str]]]
2828

2929

3030
# A message that can be sent to the client

0 commit comments

Comments
 (0)