How can you disconnect if your partner has a different sleep pattern to you, and works in bed while you are sleeping?

Talk to them about setting up a workspace other than the bedroom. If you make it mandatory that the bedroom is for rest, like the kitchen is for cooking, then there won't be an issue with the partner using a desk or dining table to work. There has to be a mutual respect for each others sleep schedule, especially the all nighter.