Low capacity partner

TLDR my partner hasn't had much capacity for the last 6 months because of external circumstances. How do we handle the imbalance between caretaking as I am offering a lot of care and they are not able to reciprocate?

My partner (X) and I have been long-distance for a year but in the last 6 months things have not been going so well. They moved to a new city, had to navigate a somewhat complex living situation, started a high-pressure job where they also experienced a lot of systematic transphobia, etc. So it has been a rough time for them. Normally they are amazing at communicating and we have really constructive conversations but their capacity really plummeted and I was left feeling super alone, angry, confused, etc. I kept offering them care as they went through this difficult phase but it felt so imbalanced - I constantly felt like my cup was running empty as I was the main (emotional) caregiver in our dynamic, while I was deeply in need of care because I was in distress about our changing dynamics (e.g. less texting, less check-ins, more tension and misunderstandings).

Fast forward to now, they have a bit more capacity so we're taking baby steps to start talking about the hurt that happened during those months and be constructive about small changes we can make to bring some ease into our dynamic. But the truth is they still don't have capacity for the level of enmeshment that we used to have and also quite some damage has been done in those months (I am very sensitive to a perceived lack of care or lack of acknowledgment of my care/effort, they are very sensitive to feeling like they are failing me and being a bad partner, etc).

We don't want to lose what we do have but we do both realise that a deescalation has been and is happening. So I guess I'd like advice on how to deescalate to the point where my expectations match their capacity. One particular area of struggle is imbalanced care as I mentioned. They are not able to offer me much care at the moment but I deeply want and enjoy offering care to my loved ones. A small example: they were sick this week so I helped them figure out delegating work, encouraged them to take sick days, etc. and that felt very connecting to be part of and support them in their daily life. A few days later once they recovered, I got sick and I asked for a bit of extra texting or to do a shared activity together online and they said they were too stressed and busy and while I wasn't dysregulated (yay, a big step for me), I did notice some unkind thoughts creeping in (e.g. the thought I had was: but they were hanging out with their friend/housemate last night so why can't they make time for me today when I'm sick). I do trust that they are doing their best, but I am also quite sensitive to a lack of reciprocity at the moment after so many months of struggles.

Are there any suggestions for navigating this imbalance? I do have a decent support network already so I am not without care (although I am someone who needs a lot of care and connection so my bucket is still not full as is). Perhaps it is just a reframing that for this phase of life, I am the main caretaker but in time the tables might turn. Or perhaps I need to be less of their caretaker but that feels inauthentic and sad to me to remove care from someone I love as they struggle. Are there options I'm missing? Or practical tips for how to navigate this both internally and with my partner?