Breastfeeding to EP

I need some help! I was EBF at first but had to supplement some the first week due to severe jaundice then went back to mainly breastfeeding only doing bottle during 2/3x a day went well she would constantly fall asleep on boob but would nurse for long periods and gaining weight. Now (5 weeks pp) she’s falling asleep like 5 minutes in or playing with my boobs not latching well and I’m constantly feeding her every hour or less. Not sure what to do if I should completely stop BF and just pump as when I go back to work I’ll have to use bottles anyways.