
    :i                     H    S SK Jr  S SKJr   " S S\5      r " S S\5      rg)    )Field)MistralBasec                   J    \ rS rSr% SrSr\\S'   Sr\S-  \S'   Sr	\\S'   Sr
g)		UsageInfo   aa  Usage statistics for a completion request.

Attributes:
    prompt_tokens: Number of tokens in the prompt.
    completion_tokens: Number of tokens in the generated completion.
    total_tokens: Total number of tokens used in the request (prompt + completion).

Examples:
    >>> info = UsageInfo(prompt_tokens=10, completion_tokens=20, total_tokens=30)
r   prompt_tokensNcompletion_tokenstotal_tokens )__name__
__module____qualname____firstlineno____doc__r   int__annotations__r	   r
   __static_attributes__r       V/home/james-whalen/.local/lib/python3.13/site-packages/mistral_common/protocol/base.pyr   r      s,    	 M3$%sTz%L#r   r   c                       \ rS rSr% Sr\" SSSS9r\\S'   \" SSSS9r	\\S'   \" S	S
S9r
\S	-  \S'   \" S	S
S9r\S	-  \S'   Srg	)BaseCompletionRequest   a!  Base class for completion requests.

Attributes:
    temperature: Sampling temperature to use, between 0 and 1. Higher values like 0.8 will make the output more
        random, while lower values like 0.2 will make it more focused and deterministic.
    top_p: Nucleus sampling parameter, top-p probability mass, between 0 and 1.
    max_tokens: Maximum number of tokens to generate.
    random_seed: Random seed for reproducibility.

Examples:
    >>> request = BaseCompletionRequest(temperature=0.7, top_p=0.9, max_tokens=100, random_seed=42)
gffffff?g        g      ?)defaultgeletemperaturetop_pNr   )r   r   
max_tokensrandom_seedr   )r   r   r   r   r   r   r   floatr   r   r   r   r   r   r   r   r   r   r      sY     sss;K;5E55"4A6Jd
6#DQ7Kt7r   r   N)pydanticr   mistral_common.baser   r   r   r   r   r   <module>r#      s#     + "8K 8r   