Ë
    ³«Žjè  ã                   óN   — d dl mZmZmZ ddlmZ  G d„ dej                  «      Zy)é    )ÚOptionalÚDictÚAnyé   )Úbase_original_provider_usagec                   ó²   ‡ — e Zd ZU dZee   ed<   	 eed<   	 eed<   	 dZee   ed<   	 dZee   ed<   	 de	d	e
e	ef   fˆ fd
„Zede
e	ef   d	d fd„«       Zˆ xZS )ÚGoogleGeminiUsagez?Google AI / VertexAI calls token usage data. Updated 11.03.2025Úcandidates_token_countÚprompt_token_countÚtotal_token_countNÚcached_content_token_countÚthoughts_token_countÚparent_key_prefixÚreturnc                 ó"   •— t         ‰|   |«      S )N)ÚsuperÚto_backend_compatible_flat_dict)Úselfr   Ú	__class__s     €út/Users/manta/Documents/Projects/TheRoad-I1/backend/.venv/lib/python3.12/site-packages/opik/llm_usage/google_usage.pyr   z1GoogleGeminiUsage.to_backend_compatible_flat_dict   s   ø€ Ü‰wÑ6Ð7HÓIÐIó    Úusagec                 ó   —  | di |¤ŽS )N© r   )Úclsr   s     r   Úfrom_original_usage_dictz*GoogleGeminiUsage.from_original_usage_dict   s   € á‰|U‰|Ðr   )Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   ÚintÚ__annotations__r   r   Ústrr   r   Úclassmethodr   r   Ú__classcell__)r   s   @r   r	   r	      s–   ø… ÙIà$ S™MÓ)Ø.àÓð yàÓØ?à04Ð ¨¡Ó4Ø]à*.Ð˜( 3™-Ó.ØðJÀð JÈÈcÐSVÈhÉõ Jð ð¨T°#°s°(©^ð Ð@Sò ó ôr   r	   N)Útypingr   r   r   Ú r   ÚBaseOriginalProviderUsager	   r   r   r   Ú<module>r)      s#   ðß &Ñ &õ +ôÐ4×NÑNõ r   