📁 File Manager Pro
v10.0.3 | PHP: 8.2.31
Server: LiteSpeed
2026-06-28 04:34:30
📂
/ (Root)
/
opt
/
hc_python
/
lib
/
python3.12
/
site-packages
/
sentry_sdk
/
integrations
/
google_genai
/
__pycache__
📍 /opt/hc_python/lib/python3.12/site-packages/sentry_sdk/integrations/google_genai/__pycache__
🔄 Refresh
✏️
Editing: streaming.cpython-312.pyc
Read Only
� |�*j � �� � d dl mZmZmZmZmZmZ d dlmZ d dl m Z d dl m Z d dl mZ d dlmZ ddlmZmZmZmZmZ er d d lmZ d d lmZ G d � d e� Zdd �Z dd�Z dd�Z y)� )� TYPE_CHECKING�Any�List�Optional� TypedDict�Union)�set_data_normalized)�SPANDATA)�should_send_default_pii)� StreamedSpan)�safe_serialize� )� UsageData�extract_contents_text�extract_finish_reasons�extract_tool_calls�extract_usage_data)�GenerateContentResponse)�Spanc �J � e Zd ZU ded< ded< eed< ded< ded< d ed < y ) �AccumulatedResponsez Optional[str]�id�model�textz List[str]�finish_reasonszList[dict[str, Any]]� tool_callszOptional[UsageData]�usage_metadataN)�__name__� __module__� __qualname__�__annotations__�str� � �]/opt/hc_python/lib/python3.12/site-packages/sentry_sdk/integrations/google_genai/streaming.pyr r s% � ��� �� �I� � �&�&�)�)r$ r c �� � t t | d |d � t | d |d � t | d |d � t | d |d � t | d |d � �� S )N� input_tokens� output_tokens�input_tokens_cached�output_tokens_reasoning� total_tokens)r' r( r) r* r+ )r �max)�self�others r% �element_wise_usage_maxr/ " s� � � ���n�-�u�^�/D�E��$��/���1G�H� � �&� '��/D�)E� � !$� �*� +�U�3L�-M�! � ��n�-�u�^�/D�E� � r$ c �� � g }g }g }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 |� } |�| n t || � }�� t dj |� |||||�� } | S )z?Accumulate streaming chunks into a single response-like object.N� candidates�content�parts� )r r r r r r ) �getattr�hasattrr2 r �appendr �extendr r r/ r �join)�chunks�accumulated_textr r � usage_data� response_idr �chunk� candidate�extracted_text�extracted_finish_reasons�extracted_tool_calls�chunk_usage_data�accumulated_responses r% � accumulate_streaming_responserE 0 s$ � � ���N��J��J��K� �E��� �5�,�� -� $�U�L�"� =� ��9�i�0�W� �%�%�w��6� &;�9�;L�;L�%M�N�%�(�/�/��?� >� $:�%�#@� � #� � !� !�":� ;� 1�%� 8� � � � � � 2� 3� .�e�4�� �!� �'� �4D�E� �/ �: /� �W�W�%� &�%� �!� �� �� � r$ Nc �� � t | t � r | j n | j }t � rA|j r5|j d� r$ |t j t |d g� � |j d� r t | t j |d � |j d� r# |t j t |d � � |j d� }|� |t j |� |j d� }|� |t j |� |d �y|d d r |t j |d d � |d d r |t j |d d � |d d r |t j" |d d � |d d r |t j$ |d d � |d d r |t j& |d d � yy) z1Set span data for accumulated streaming response.r r r r Nr r r' r) r( r* r+ )� isinstancer � set_attribute�set_datar �include_prompts�getr �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� integrationrD � set_on_spanr= �response_models r% �$set_span_data_for_streaming_responserZ d s � � )��|� <����$�-�-� � �!� � '� '� � $� $�V� ,�� � )� )� � 0�� 8�9� :� � � � � 0�1�� � � 3� 3� �!1� 2� � � � � �-�� � /� /� �/� �=� >� � '�*�*�4�0�K�� ��H�/�/��=�)�-�-�g�6�N��!��H�2�2�N�C�� ,�-�5��� ,�-�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$ )r- r r. r �returnr )r: z List[GenerateContentResponse]r[ r )rV zUnion[Span, StreamedSpan]rW r rD r r[ N) �typingr r r r r r �sentry_sdk.ai.utilsr �sentry_sdk.constsr �sentry_sdk.scoper �sentry_sdk.tracesr �sentry_sdk.utilsr �utilsr r r r r �google.genai.typesr �sentry_sdk.tracingr r r/ rE rZ r# r$ r% �<module>re s� �� G� G� 3� &� 4� *��� � �:�'�*�)� *� � 1 � +�1 ��1 �hH � %�H ��H � 0�H � � H r$
💾 Save Changes
❌ Cancel