SampleResponse - Tinker Documentation

tinker.types.SampleResponse

class tinker.types.SampleResponse()

Response from a sampling request.

Contains generated sequences and optional prompt-level log probabilities. Numpy fields provide direct array access without format conversion. The corresponding Python-list properties convert lazily on first access.

Fields:

Counted on the prompt itself: for num_samples > 1 the prompt is shared, so this is not multiplied across samples. Prefill on the shared prompts samples (the remaining num_samples - 1) is billed as cache hits.

prompt_logprobs()

Per-token log probabilities for the prompt as a Python list.

If prompt_logprobs was set to true in the request, logprobs are computed for every token in the prompt. Each entry is a float, or None for positions where logprobs were not computed (e.g. the first prompt token). Returns None if prompt logprobs were not requested.

Converted from prompt_logprobs_np on first access (cached afterwards).

Returns: Optional[List[Optional[float]]]

topk_prompt_logprobs()

Top-k prompt logprobs as nested Python lists.

If topk_prompt_logprobs was set to a positive integer k in the request, the top-k logprobs are computed for every token in the prompt. For each prompt position: a list of up to k (token_id, logprob) tuples, or None for positions where logprobs were not computed. Returns None if top-k was not requested.

Converted from topk_prompt_logprobs_np on first access (cached afterwards).

Returns: Optional[ list[Optional[ list[tuple[int, float]]]]