In that situation, both are fine. B would be grammatically correct but A is used frequently in colloquial English.
Typically, “want to” is followed by a verb and “want” is followed by a noun. For example:
I want to eat some chocolate.
I want some chocolate.