§
    ~ŠtjŽ  ã                  ó6   — d Z ddlmZ ddlmZmZ dd„Zdd„ZdS )z&Token usage extraction for Google ADK.é    )Úannotations)ÚAnyÚOptionalÚllm_responser   Úreturnúdict[str, Any]c                óö  — i }t          | dd¦  «        }|s|S t          |dd¦  «        x}�t          |¦  «        |d<   t          |dd¦  «        x}�t          |¦  «        |d<   t          |dd¦  «        x}�t          |¦  «        |d<   t          |d	d¦  «        x}�&t          |¦  «        |                     d
i ¦  «        d<   t          |dd¦  «        x}�&t          |¦  «        |                     di ¦  «        d<   |S )z%Extract token usage from LlmResponse.Úusage_metadataNÚprompt_token_countÚinput_tokensÚcandidates_token_countÚoutput_tokensÚtotal_token_countÚtotal_tokensÚcached_content_token_countÚinput_token_detailsÚ
cache_readÚthoughts_token_countÚoutput_token_detailsÚ	reasoning)ÚgetattrÚintÚ
setdefault)r   Úusager
   Úvs       úf/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/langsmith/integrations/google_adk/_usage.pyÚextract_usage_from_responser      s  € à€EÝ˜\Ð+;¸TÑBÔB€NØð Øˆå�^Ð%9¸4Ñ@Ô@Ð@ˆÐMÝ # A¡¤ˆˆnÑÝ�^Ð%=¸tÑDÔDÐDˆÐQÝ!$ Q¡¤ˆˆoÑÝ�^Ð%8¸$Ñ?Ô?Ð?ˆÐLÝ # A¡¤ˆˆnÑÝ�^Ð%AÀ4ÑHÔHÐHˆÐUÝDGÈÁFÄFˆ×ÒÐ.°Ñ3Ô3°LÑAÝ�^Ð%;¸TÑBÔBÐBˆÐOÝDGÈÁFÄFˆ×ÒÐ/°Ñ4Ô4°[ÑAà€Ló    Úllm_requestúOptional[str]c                ó´   — t          | dd¦  «        x}r"t          |dd¦  «        x}rt          |¦  «        S t          | dd¦  «        x}rt          |¦  «        S dS )z*Extract the model name from an LlmRequest.ÚconfigNÚmodel)r   Ústr)r   r"   r#   s      r   Úextract_model_namer%      sj   € å˜ h°Ñ5Ô5Ð5€vð Ý˜F G¨TÑ2Ô2Ð2ˆ5ð 	Ý�u‘:”:ÐÝ˜ W¨dÑ3Ô3Ð3€uð Ý�5‰zŒzÐØˆ4r   N)r   r   r   r   )r   r   r   r    )Ú__doc__Ú
__future__r   Útypingr   r   r   r%   © r   r   ú<module>r*      sg   ðØ ,Ð ,à "Ð "Ð "Ð "Ð "Ð "à  Ð  Ð  Ð  Ð  Ð  Ð  Ð  ðð ð ð ð*ð ð ð ð ð r   