OwlCyberSecurity - MANAGER
Edit File: monitoring.cpython-312.pyc
� G��g^ � � � d dl Z d dlmZ d dlZd dlmZ d dlmZ d dlmZ d dl m Z e r d dl m Z m Z m Z edd� � Zd � Zd � Zd � Z dd �Zy)� N)�wraps)� start_span)�Span)� ContextVar)� TYPE_CHECKING)�Optional�Callable�Any�ai_pipeline_name)�defaultc �. � 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 c � � �� � �fd�}|S )Nc � �� � �� �fd�}�� �fd�}t j � � r t � � |� S t � � |� S )Nc �P �� t j � }� j d|rdnd� }t d � |d�� ��5 }|j di � j � D ] \ }}|j ||� � |j di � j � D ] \ }}|j ||� � |r#|j d|� � | i |��cd d d � S t j � � � | i |��} t j d � |cd d d � S # t $ rc}t j j |t j � j dd d �� � \ } } t j | | � � |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.pipeline.name� ai_monitoringF)�type�handled)�client_options� mechanism)�hintr �r r r �pop�items�set_tag�set_datar � Exception� sentry_sdk�utils�event_from_exception� get_client�options� capture_event��args�kwargs� curr_pipeliner �span�k�v�res�e�eventr& � description�f� span_kwargss ���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�"4�m�D� �d�-�f�-� D�C� &�)�)�+�6� 4� �� 0�� 0�� *�-�-�d�3� �- D�C�� %� *�&0�&6�&6�&K�&K� �+5�+@�+@�+B�+J�+J�/>�5�&Q� 'L� '� ��t� #�0�0��T�B� �T�)��*�� *�-�-�d�3��+ D�C�sC �BF �F �*D�3F � E?� A E:�:E?�?F�F�F � F%c � � �K � t j � }� j d|rdnd� }t d � |d�� ��5 }|j di � j � D ] \ }}|j ||� � |j di � j � D ] \ }}|j ||� � |r+|j d|� � | i |��� d { ��� cd d d � S t j � � � | i |��� d { ��� } t j d � |cd d d � S 7 �T7 �'# t $ rc}t j j |t j � j dd d �� � \ } } t j | | � � |d �d }~ww xY w# t j d � w xY w# 1 sw Y y xY w�wr r' r3 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�"4�m�D�!"�D�!3�F�!3�3� D�C� &�)�)�+�6� 4�$%�t�$6�v�$6� 6�� *�-�-�d�3� �- D�C� 4�� 7��$� *�&0�&6�&6�&K�&K� �+5�+@�+@�+B�+J�+J�/>�5�&Q� 'L� '� ��t� #�0�0��T�B� �T�)��*�� *�-�-�d�3��+ D�C�s~ �9F>�BF2�D%�F2� F>� F2�4 D)�?D'� D)�F2� F>�%F2�'D)�) F�2A F�F�F�F/�/F2�2F;�7F>)�inspect�iscoroutinefunctionr )r> r@ rB r= r? s ` ��r � decoratorzai_track.<locals>.decorator s? �� � �: �: � &� &�q� )��5��8�M�*� *��5��8�L�)� )r r )r= r? rE s `` r �ai_trackrF s �� �?*�B �r c �� � t � }|r| j d|� |�| j d|�� |�| j d|�� |� |�|�||z }|�| j d|� y y )Nr �ai_prompt_tokens_used)�value�ai_completion_tokens_used�ai_total_tokens_used)r r+ �set_measurement)r7 � prompt_tokens�completion_tokens� total_tokensr s r �record_token_usagerP a s� � � ,�-��� � � �(�*:�;�� � � � � 4�M� �J��$� � � � 8�@Q� �R�� � � %� � )�$�'8�8� �� � � � � 3�\�B� r )NNN)rC � functoolsr �sentry_sdk.utilsr- r �sentry_sdk.tracingr r �typingr r r r r r r rF rP r r r �<module>rU sO �� � � � !� #� '� ��.�.� � 1�4�@� � � #� C�N DH�Cr