I would also do it what you do.
He has to take responsibility, wether he was psychotic or not.
Also he should have told you what’s his diagnosis and keep going with a therapist.
It would be also a good idea to visit a neurologist. It’s very important to know which area or areas of the brain were damaged.

Is he trying you to come back with him? And how are your kids now? Are they coping well?