← Back
Editing: monitoring.cpython-312.pyc
� ]*j� � �L � d dl Z d dlZd dlmZ d dlmZ d dlZd dlmZ d dl m Z d dl m Z d dl mZ d dlmZ d d lmZmZmZ er,d d lmZmZmZmZmZmZ ed eed ef ed ee f f � � Z edd�� Zdd�Z dd�Z de ddddfd�Z d d d�Z y) � N)�wraps)� TYPE_CHECKING)� start_span)�_set_span_data_attribute)�SPANDATA)� StreamedSpan)�Span)� ContextVar�capture_internal_exceptions�reraise)�Any� Awaitable�Callable�Optional�TypeVar�Union�F.)�bound�ai_pipeline_name)�default�returnc �. � t j | � y �N)�_ai_pipeline_name�set)�names �G/opt/hc_python/lib/python3.12/site-packages/sentry_sdk/ai/monitoring.py�set_ai_pipeline_namer s � ����$� � c �* � t j � S r )r �get� r r �get_ai_pipeline_namer# s � � � � � "�"r � description� span_kwargsr zCallable[[F], F]c � � �� d� �fd� }|S )Nc � �� � d�� �fd� }d�� �fd� }t j � � r t � � |� S t � � |� S )Nc �� �� t j � }�j d|rdnd� }t d � |d����5 }|j di � j � D ] \ }}|j ||� � |j di � j � D ] \ }}|j ||� � |r1|j t j |� � | i |��cd d d � S t j � � � | i |��}t j d � cd d d � S # t $ r�}t j � } t � 5 t j j! |t j" � j$ ddd �� � \ } } t j& | | � � d d d � n # 1 sw Y nxY wt) | � Y d }~��d }~ww xY w# t j d � w xY w# 1 sw Y y xY w� N�opzai.runz ai.pipeline)r r* � sentry_tags� sentry_data� ai_monitoringF)�type�handled)�client_options� mechanism)�hintr" �r r! �popr �items�set_tag�set_datar �GEN_AI_PIPELINE_NAMEr � Exception�sys�exc_infor � sentry_sdk�utils�event_from_exception� get_client�options� capture_eventr ��args�kwargs� curr_pipeliner* �span�k�v�res�er; �eventr2 r$ �fr% s ���r � sync_wrappedz1ai_track.<locals>.decorator.<locals>.sync_wrapped s� �� � -� 1� 1� 3�M� � � ��=�x�m�T�B��C���C�{�C�t� "�J�J�}�b� 9� ?� ?� A�D�A�q��L�L��A�&� B� "�J�J�}�b� 9� ?� ?� A�D�A�q��M�M�!�Q�'� B� ��M�M�(�"?�"?� �O� �d�-�f�-� D�C� &�)�)�+�6� 4� �� 0�� 0�� *�-�-�d�3� �1 D�C�� %� +�#&�<�<�>�� 8� :�*4�*:�*:�*O�*O� !�/9�/D�/D�/F�/N�/N�3B�u�*U� +P� + �K�E�4� '� 4� 4�U�� F� ;� :� :�� ��*�� +�� *�-�-�d�3��/ D�C�sa �B G'�"G'�8D � G'� G �) G�A F,�# G�,F5 �1G� G �G � G � G$�$G'�'G0c � � �K � t j � }�j d|rdnd� }t d � |d����5 }|j di � j � D ] \ }}|j ||� � |j di � j � D ] \ }}|j ||� � |r9|j t j |� � | i |��� d { ��� cd d d � S t j � � � | i |��� d { ��� }t j d � cd d d � S 7 �S7 �&# t $ r�}t j � } t � 5 t j j! |t j" � j$ ddd �� � \ } } t j& | | � � d d d � n # 1 sw Y nxY wt) | � Y d }~��d }~ww xY w# t j d � w xY w# 1 sw Y y xY w�wr) r3 rB s ���r � async_wrappedz2ai_track.<locals>.decorator.<locals>.async_wrapped> s� �� �� � -� 1� 1� 3�M� � � ��=�x�m�T�B��C���C�{�C�t� "�J�J�}�b� 9� ?� ?� A�D�A�q��L�L��A�&� B� "�J�J�}�b� 9� ?� ?� A�D�A�q��M�M�!�Q�'� B� ��M�M�(�"?�"?� �O�!"�D�!3�F�!3�3� D�C� &�)�)�+�6� 4�$%�t�$6�v�$6� 6�� *�-�-�d�3� �1 D�C� 4�� 7��$� +�#&�<�<�>�� 8� :�*4�*:�*:�*O�*O� !�/9�/D�/D�/F�/N�/N�3B�u�*U� +P� + �K�E�4� '� 4� 4�U�� F� ;� :� :�� ��*�� +�� *�-�-�d�3��/ D�C�s� �9H �B"G=� D2� G=�" H �,G=� D6� D4�D6�G=�( H �2G=�4D6�6 G �? G� A G�9 G�G �G�G#�G � G#�#G:�:G=�=H�H )rC r rD r r r )�inspect�iscoroutinefunctionr )rL rM rO r$ r% s ` ��r � decoratorzai_track.<locals>.decorator s? �� � �< �< � &� &�q� )��5��8�M�*� *��5��8�L�)� )r )rL r r r r" )r$ r% rR s `` r �ai_trackrS s �� �@*�D �r c �� � t � }|rt | t j |� |�t | t j |� |�t | t j |� |�t | t j |� |�t | t j |� |�t | t j |� |� |�|�||z }|� t | t j |� y y r ) r# r r r8 �GEN_AI_USAGE_INPUT_TOKENS� GEN_AI_USAGE_INPUT_TOKENS_CACHED�%GEN_AI_USAGE_INPUT_TOKENS_CACHE_WRITE�GEN_AI_USAGE_OUTPUT_TOKENS�$GEN_AI_USAGE_OUTPUT_TOKENS_REASONING�GEN_AI_USAGE_TOTAL_TOKENS)rF � input_tokens�input_tokens_cached�input_tokens_cache_write� output_tokens�output_tokens_reasoning� total_tokensr s r �record_token_usagera d s� � � ,�-��� ��x�'D�'D�FV�W�� � ��x�'I�'I�<�X��&� � � � 5� 5� � � �+� � � � :� :� $� � � � � �(�5�5�}� � �*� � � � 9� 9� #� � �� � 8�]�=V�#�m�3� �� � ��x�'I�'I�<�X� r )r � Optional[str]r N)r rb )NNNNNN)rF zUnion[Span, StreamedSpan]r[ � Optional[int]r\ rc r] rc r^ rc r_ rc r` rc r N)!rP r: � functoolsr �typingr �sentry_sdk.utilsr< r �sentry_sdk.ai.utilsr �sentry_sdk.constsr �sentry_sdk.tracesr �sentry_sdk.tracingr r r r r r r r r r r r r r# �strrS ra r" r r �<module>rl s �� � � � � � !� 8� &� *� #� M� M��I�I���5��#�s�(�!3�X�c�9�S�>�>Q�5R�!R�S�T�A� � 1�4�@� � �#�C�#� C�e� C�8J� C�P %)�+/�04�%)�/3�$(�/Y� %�/Y�!�/Y� )�/Y� .� /Y� #� /Y� -� /Y� "�/Y� �/Yr
Save File
Cancel