Thanks for the link although for some reason I keep getting:

Incorrect API key provided error on jupyter for chatgpt. I'm on a paid account so not sure why...