Hey everyone 
We have released a new function to solve this problem, have a look at this thread: [Announcement] Generation: Get probabilities for generated output
Since some of the snippets at the start of this thread no longer match our API, I’d like to ask for new questions/comments to be posted on the thread I’ve linked above.
(@danielcabal you might find an answer to your question in the post I linked 
)