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

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


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


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