* Add type hints to diagnostics test helper * Move type alias to the top * Use Any * Move TestClientGenerator to typing helper * Use `dict[str, Any]` again * Use JsonObjectType * Add cast