📁 File Manager Pro
v10.0.3 | PHP: 8.2.31
Server: LiteSpeed
2026-06-28 12:41:58
📂
/ (Root)
/
opt
/
hc_python
/
lib
/
python3.12
/
site-packages
/
sentry_sdk
/
integrations
/
pydantic_ai
/
spans
/
__pycache__
📍 /opt/hc_python/lib/python3.12/site-packages/sentry_sdk/integrations/pydantic_ai/spans/__pycache__
🔄 Refresh
✏️
Editing: utils.cpython-312.pyc
Read Only
� |�*j� � � � d Z ddlmZ ddlZddlmZ ddlmZ ddlm Z ddl m Z ddl m Z erdd lmZmZmZ dd lmZmZ dd �Zdd �Z dd �Zy)z6Utility functions for PydanticAI span instrumentation.� )� TYPE_CHECKINGN)�BLOB_DATA_SUBSTITUTE)�DATA_URL_BASE64_REGEX)�get_modality_from_mime_type)�SPANDATA)� StreamedSpan)�Any�Dict�Union)� RequestUsage�RunUsagec �v � t | j � }t j |� }|r dt d�S d|d�S )z�Serialize an ImageUrl content item for span data. For data URLs containing base64-encoded images, the content is redacted. For regular HTTP URLs, the URL string is preserved. �image)�type�content)�str�urlr �matchr )�itemr �data_url_matchs �^/opt/hc_python/lib/python3.12/site-packages/sentry_sdk/integrations/pydantic_ai/spans/utils.py�_serialize_image_url_itemr sF � � �d�h�h�-�C�*�0�0��5�N���+� � � �� � � c �R � dt | j � | j t d�S )zFSerialize a BinaryContent item for span data, redacting the blob data.�blob)r �modality� mime_typer )r � media_typer )r s r � _serialize_binary_content_itemr ' s' � � �/����@��_�_�'� � r c � � |�yt | t � r | j n | j }t |d� r-|j �! |t j |j � t |d� r-|j �! |t j |j � t |d� r-|j �! |t j |j � t |d� r-|j �! |t j |j � t |d� r/|j �" |t j |j � yyy)a( Set token usage data on a span. This function works with both RequestUsage (single request) and RunUsage (agent run) objects from pydantic_ai. Args: span: The Sentry span to set data on. usage: RequestUsage or RunUsage object containing token usage information. N� input_tokens�cache_read_tokens�cache_write_tokens� output_tokens� total_tokens)� isinstancer � set_attribute�set_data�hasattrr! r �GEN_AI_USAGE_INPUT_TOKENSr" � GEN_AI_USAGE_INPUT_TOKENS_CACHEDr# �%GEN_AI_USAGE_INPUT_TOKENS_CACHE_WRITEr$ �GEN_AI_USAGE_OUTPUT_TOKENSr% �GEN_AI_USAGE_TOTAL_TOKENS)�span�usage� set_on_spans r �_set_usage_datar2 1 s � � �}�� )��|� <����$�-�-� � �u�n�%�%�*<�*<�*H��H�6�6��8J�8J�K� �u�)�*�u�/F�/F�/R��H�=�=�u�?V�?V�W� �u�*�+��0H�0H�0T�� � :� :� � $� $� � �u�o�&�5�+>�+>�+J��H�7�7��9L�9L�M��u�n�%�%�*<�*<�*H��H�6�6��8J�8J�K� +I�%r )r r �returnzDict[str, Any])r/ z,Union[sentry_sdk.tracing.Span, StreamedSpan]r0 z Union[RequestUsage, RunUsage]r3 N)�__doc__�typingr � sentry_sdk�sentry_sdk._typesr �sentry_sdk.ai.constsr �sentry_sdk.ai.utilsr �sentry_sdk.constsr �sentry_sdk.tracesr r r r �pydantic_ai.usager r r r r2 � r r �<module>r> sV �� <� � � 2� 6� ;� &� *��'�'�8��*�&L� 8�&L� *�&L� �&Lr
💾 Save Changes
❌ Cancel