TODO: Generate top-K completetions for the given context and prefix.