I had bipolar mania/psychosis last year at this time. It was EXTREMELY traumatizing. Not just for me but for my husband, who witnessed it.
I would say it took me about six months for me to get back to where I didn't feel like the rug was going to be pulled out from me. During that time I could barely talk about the psychosis and what I experienced because it felt very raw, and at times I would stay up at night thinking about it and would just cry. I was really fragile at that time.
On the upside, I don't have any problems taking meds because I am still terrified it would happen again. It hasn't. Nowadays I can talk about what happened but it is still a painful subject. My therapist was really helpful- she specializes in trauma.
Anyway, I'm so sorry you went through that. Have hope, because it is going to get better. It just takes some time.