I think the show kind of has to separate Mando and Grogu for a while, so that when we finally see Grogu again, he can have grown up just a little bit more. He's been a delightful character as an infant/toddler, but I think they've mined just about all they can, and the character needs to start being able to interact with other characters in a more autonomous, communicative way.
I'd keep him out of almost the entire next season, frankly.