Officials say Rishi Sunak is still open to the idea if energy firms do not reinvest their profits in the UK.