Hello Everyone,
A fix has been rolled out to improve this behavior, which should significantly reflect the changes in app Preview.
https://discuss.ai.google.dev/t/code-returned-in-chat-instead-of-modifying-files/110292/59
Could you please confirm if you are still facing this issue?