You may be right. Alternatively, Jevons paradox may hit a limit yet.
It also is not clear workers will be needed. The gain may not be in employment. Perhaps another paradox will arise as the additional gains will be consumed by AI itself. How unique are your prompts? Self prompting is a clear end state.
(Also, the US budget deficit is way too high to afford a UBI.)