make kafka return partition info

Let's make this a kwarg, with a default of False to maintain backward compatibility.