"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""

from __future__ import annotations
from friendli.types import BaseModel
from typing_extensions import TypedDict


class UsageTypedDict(TypedDict):
    prompt_tokens: int
    r"""Number of tokens in the prompt."""
    completion_tokens: int
    r"""Number of tokens in the generated completions."""
    total_tokens: int
    r"""Total number of tokens used in the request (`prompt_tokens` + `completion_tokens`)."""


class Usage(BaseModel):
    prompt_tokens: int
    r"""Number of tokens in the prompt."""

    completion_tokens: int
    r"""Number of tokens in the generated completions."""

    total_tokens: int
    r"""Total number of tokens used in the request (`prompt_tokens` + `completion_tokens`)."""
