
    *`i                     V    d dl mZ d dlmZ  G d de          Z G d de          ZdS )    )Field)MistralBasec                   D    e Zd ZU dZdZeed<   dZedz  ed<   dZeed<   dS )	UsageInfoa}  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____doc__r   int__annotations__r   r	        p/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/mistral_common/protocol/base.pyr   r      sS         	 	 M3$%sTz%%%L#r   r   c                       e Zd ZU dZ eddd          Zeed<    eddd          Zeed<    edd	
          Z	e
dz  ed<    edd	
          Ze
dz  ed<   dS )BaseCompletionRequestaE  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_seed)r
   r   r   r   r   r   floatr   r   r   r   r   r   r   r   r   r      s           sss;;;K;;;5555E5555"U4A666Jd
666#eDQ777Kt77777r   r   N)pydanticr   mistral_common.baser   r   r   r   r   r   <module>r      s          + + + + + +       "8 8 8 8 8K 8 8 8 8 8r   