Here is a downloadable PS Action that you can import and run before invoking Mask AI. Remember this should just be used to create a Layer Mask from the base layer and pass that to Mask AI. The Action Group is called MaskAI and the action to run is called ReUse Mask.
Just import the action into the action panel in PS.
I am not familiar with PS scripting or actions so perhaps this is a dumb question, but is there any way to incorporate the actual call to Mask AI in an action? I tried both using a keyboard shortcut and selecting the plugin using the mouse and neither worked.
I assume it can be done in a script, but then I have to take a week or so to learn how to script PS and it does not seem worth the trouble if there is a way to add it to a simple action.