o
    h                     @   s>   d dl mZ d dlmZmZ d dlmZ eG dd dZdS )    )field)InputTokensDetailsOutputTokensDetails)	dataclassc                   @   s   e Zd ZU dZeed< 	 dZeed< 	 edd dZe	ed< 	 dZ
eed< 	 ed	d dZeed
< 	 dZeed< 	 dddZdS )Usager   requestsinput_tokensc                   C   
   t ddS )Nr   cached_tokens)r    r   r   M/var/www/html/openai_agents/venv/lib/python3.10/site-packages/agents/usage.py<lambda>      
 zUsage.<lambda>)default_factoryinput_tokens_detailsoutput_tokensc                   C   r	   )Nr   reasoning_tokens)r   r   r   r   r   r      r   output_tokens_detailstotal_tokensotherreturnNc                 C   s   |  j |j r	|j nd7  _ |  j|jr|jnd7  _|  j|jr#|jnd7  _|  j|jr0|jnd7  _t| jj|jj d| _t| jj	|jj	 d| _d S )Nr   r
   r   )
r   r   r   r   r   r   r   r   r   r   )selfr   r   r   r   add   s   z	Usage.add)r   r   r   N)__name__
__module____qualname__r   int__annotations__r   r   r   r   r   r   r   r   r   r   r   r   r   r      s$   
 r   N)dataclassesr   %openai.types.responses.response_usager   r   pydantic.dataclassesr   r   r   r   r   r   <module>   s
    