PATH:
opt
/
hc_python
/
lib
/
python3.12
/
site-packages
/
sentry_sdk
/
integrations
/
google_genai
/
__pycache__
� o 0iG � � � d dl mZmZmZmZmZ d dlmZ d dlm Z d dl m Z d dl m Z ddlmZmZmZmZmZ er d dlmZ d d lmZ G d � d e� Zd � Zd � Zy)� )� TYPE_CHECKING�Any�List� TypedDict�Optional)�set_data_normalized)�SPANDATA)�should_send_default_pii)�safe_serialize� )�extract_tool_calls�extract_finish_reasons�extract_contents_text�extract_usage_data� UsageData)�Span)�GenerateContentResponsec �l � e Zd ZU ee ed< ee ed< eed< ee ed< eeeef ed< e ed< y)�AccumulatedResponse�id�model�text�finish_reasons� tool_calls�usage_metadataN) �__name__� __module__� __qualname__r �str�__annotations__r �dictr r � � �]/opt/hc_python/lib/python3.12/site-packages/sentry_sdk/integrations/google_genai/streaming.pyr r s> � ��� �� �C�=�� �I���I� ��T�#�s�(�^�$�$� � r# r c �P � g }g }g }d}d}d}d}d}d} d} | D ]� } t | dd� r`t | dg � D ]P } t | d� s�t | j dg � s�(t | j � } | s�@|j | � �R t | � }|r|j |� t | � }|r|j |� t | � }||d z }||d z }||d z }||d z }||d z }�� t d j |� ||t |||||� � | | � � }|S )z?Accumulate streaming chunks into a single response-like object.r N� candidates�content�parts� input_tokens� output_tokens�input_tokens_cached�output_tokens_reasoning� total_tokens� )r) r* r+ r, r- )r r r r r r ) �getattr�hasattrr' r �appendr �extendr r r �joinr )�chunks�accumulated_textr r �total_input_tokens�total_output_tokensr- �total_cached_tokens�total_reasoning_tokens� response_idr �chunk� candidate�extracted_text�extracted_finish_reasons�extracted_tool_calls�extracted_usage_data�accumulated_responses r$ � accumulate_streaming_responserB % s� � � ���N��J������L��� ���K� �E��� �5�,�� -� $�U�L�"� =� ��9�i�0�W� �%�%�w��6� &;�9�;L�;L�%M�N�%�(�/�/��?� >� $:�%�#@� � #� � !� !�":� ;� 1�%� 8� � � � � � 2� 3� 2�%� 8� �� 2�>� B�B��� 3�O� D�D��� 3�4I� J�J�� �"6�7P�"Q�Q� ��,�^�<�<� �3 �6 /� �W�W�%� &�%� � �+�-� 3�$:�%� � �� �� � r# c �" � t � rJ|j r>|j d� r-| j t j t |d g� � |j d� r t | t j |d � |j d� r,| j t j t |d � � |j d� r#| j t j |d � |j d� r#| j t j |d � |d d r&| j t j |d d � |d d r&| j t j |d d � |d d r&| j t j |d d � |d d r&| j t j |d d � |d d r'| j t j |d d � y y ) z1Set span data for accumulated streaming response.r r r r r r r) r+ r* r, r- N)r �include_prompts�get�set_datar �GEN_AI_RESPONSE_TEXTr r � GEN_AI_RESPONSE_FINISH_REASONS�GEN_AI_RESPONSE_TOOL_CALLS�GEN_AI_RESPONSE_ID�GEN_AI_RESPONSE_MODEL�GEN_AI_USAGE_INPUT_TOKENS� GEN_AI_USAGE_INPUT_TOKENS_CACHED�GEN_AI_USAGE_OUTPUT_TOKENS�$GEN_AI_USAGE_OUTPUT_TOKENS_REASONING�GEN_AI_USAGE_TOTAL_TOKENS)�span� integrationrA s r$ �$set_span_data_for_streaming_responserS ` s � � �!� � '� '� � $� $�V� ,� � � � � )� )� � 0�� 8�9� :� � � � � 0�1�� � � 3� 3� �!1� 2� � � � � �-� � � � � /� /� �/� �=� >� � � � ��%� � � �h�1�1�3G��3M�N�� � ��(� � � �h�4�4�6J�7�6S�T�� ,�-�n�=� � � � � .� .� �!1� 2�>� B� � � ,�-�.C�D� � � � � 5� 5� �!1� 2�3H� I� � � ,�-�o�>� � � � � /� /� �!1� 2�?� C� � � ,�-�.G�H� � � � � 9� 9� �!1� 2�3L� M� � � ,�-�n�=� � � � � .� .� �!1� 2�>� B� � >r# N)�typingr r r r r �sentry_sdk.ai.utilsr �sentry_sdk.constsr �sentry_sdk.scoper �sentry_sdk.utilsr �utilsr r r r r �sentry_sdk.tracingr �google.genai.typesr r rB rS r" r# r$ �<module>r\ sK ��� � 4� &� 4��� � �'�:� �)� �8 �v; r#
[-] streaming.cpython-312.pyc
[edit]
[+]
..
[-] consts.cpython-312.pyc
[edit]
[-] __init__.cpython-312.pyc
[edit]
[-] utils.cpython-312.pyc
[edit]