File tree Expand file tree Collapse file tree 1 file changed +8
-10
lines changed Expand file tree Collapse file tree 1 file changed +8
-10
lines changed Original file line number Diff line number Diff 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 )
You can’t perform that action at this time.
0 commit comments