Return 'cluster_labels' when 'extra_info=True'

[miguelvalente]

It would be handy if the labels for the embeddings were returned when requesting extra info.

[cvqluu]

Hi, this is a good idea. I think I'd prefer if extra_info is growing to instead return everything inside an dict.

So instead of additionally returning embeds, segments, cluster_labels separately, a dictionary containing each, i.e. {'embeds': embeds, 'segments': segments, ...}

Do you mind adding that and also updating the docstring?

[miguelvalente]

Hey, thanks, those labels would indeed be handy. Just made the extra changes as well. :v:

[cvqluu]

Looks good to me, merged. Will update the pip package

[miguelvalente]

Awesome. Thanks! :v: