Looking at the source code, you need to call `Set-TransientPrompt`. However, that one's not exported.