How can I get the logprobs of specific IDs?

[cinjon]

I am doing constrained decoding with judges and I need to get the logprobs of specific IDs. An ideal would be something like:

client.chat.completions.create(
                    model=model_id,
                    prompt=prompt,
                    temperature=temperature,
                    top_p=top_p,
                    return_logprob=True,
                    logprob_token_ids=[foo, bar]
                )

Is this already supported?