
    .Ui                     J    d dl mZ d dlmZmZ d dlmZ e G d d             Zy)    )field)InputTokensDetailsOutputTokensDetails)	dataclassc                       e Zd ZU dZeed<   	 dZeed<   	  ed       Ze	ed<   	 dZ
eed<   	  ed       Zeed	<   	 dZeed
<   	 ddZy)Usager   requestsinput_tokensc                      t        d      S )Nr   cached_tokens)r        h/home/jay/workspace/tools/ai-image-gen/jaaz-app/server/venv/lib/python3.12/site-packages/agents/usage.py<lambda>zUsage.<lambda>   s     2 C r   )default_factoryinput_tokens_detailsoutput_tokensc                      t        d      S )Nr   reasoning_tokens)r   r   r   r   r   zUsage.<lambda>   s     3Q G r   output_tokens_detailstotal_tokensNc                 \   | xj                   |j                   r|j                   ndz  c_         | xj                  |j                  r|j                  ndz  c_        | xj                  |j                  r|j                  ndz  c_        | xj                  |j                  r|j                  ndz  c_        t	        | j
                  j                  |j
                  j                  z         | _        t        | j                  j                  |j                  j                  z         | _        y )Nr   r   r   )
r	   r
   r   r   r   r   r   r   r   r   )selfothers     r   addz	Usage.add   s    5>>q@53E3EU//1LU5H5He11aO53E3EU//1L$633AA((667%
!
 &9!77HH))::;&
"r   )r   r   returnN)__name__
__module____qualname__r	   int__annotations__r
   r   r   r   r   r   r   r   r   r   r   r   r   r      ss    Hc-L#7/4C0,  PM3<16G2.  QL#>
r   r   N)dataclassesr   %openai.types.responses.response_usager   r   pydantic.dataclassesr   r   r   r   r   <module>r'      s'     Y * #
 #
 #
r   