The end result of AI - Feudalism V Socialism

I can see only 2 outcomes for the end result of AI and they're pretty stark so bear with me. Feudalism Or Socialism The natural progression of AI, especially when combined with robotics, is that it will take away in large part the need for human labour. And that is the core deal in capitalism and the only power the majority of people have..the exchange of their labour in return for money. So we either reach a point of peak capitalism where it effectively destroys society and all of the money goes to the very few. I.e Feudalism. The only way to maintain capitalism here is to limit AI, which history tells us has never happened with technology progression. OR We have to institute some form of UBI where capitalist profits are redistributed so people can survive...and capitalism will effectively have eaten itself and we're into socialism. I'm struggling to see any other form of logical conclusion of where we're headed.