I'm not sure if this is planned, and I hope this isn't too big of an ask, but I know that in the original mod there was a feature where dragons would attempt to swoop under their owner if they were falling and "catch" them.
(original_mod_link - Look under update r32; "Dragons now try to catch their owners when they're falling.")
Anyways, I think this would be a great addition, and it would make accidental dismounts less catastrophic haha.

This has been added to the list of todo's in the Brain pr. Have a look there for progress.

Yup this is a planned feature!

The dragon's AI is still under heavy construction (as per #83)

There will be an "AI overhaul" sort of speak that will incorporate this feature.

Stay tuned!

Nice! Can't wait to hear/see more about it.