Hi @Lenoq ,
As far as I know, currently there is no tool that meets your given criteria by databricks/labs. The closest that I can see is the tool you gave as an example, but as you said its not that good. what I would do maybe is an intermedial option with sqlfluff that would format the sql queries, however, it would need some extra handling to strip or ignore %sql magic lines when used on notebook content.
Best, Ilir