One way to do this is to turn on Extended Globs. Run this at the command line, or add it to your .bashrc to make it permanent:
shopt -s extglob
Now, your complete command can look like this:
complete -f -X '*.@(hi|o)' vim
Quoting from Extended Globs in patterns:
@(list): Matches one of the given patterns.