I am trying to get this example app to output formated math equations. The shiny chat tutorial here suggests a custom response display, but I am unable to get the suggested @chat.transform_assistant_response to modify the formating.
I used the code below as is:

@chat.transform_assistant_response
def _(content: str) -> ui.HTML:
    return ui.markdown(content)

adding transform_assistant=True to chat.messages() but I saw no change of the formating of the response.
I need help getting formated math equations of the reponse.