OwlCyberSecurity - MANAGER
Edit File: utils.cpython-312.pyc
� ��/i= � � � d dl Z d dlmZ d dlmZ d dlmZ erd dlmZmZm Z m Z m Z m Z d dl mZ d dlZd dlmZ dZ G d � d � Zej( d gej* d d gej, ddgej. ddgiZi Zej5 � D ] \ ZZ e D ] Z eee < � � dd�Z dd�Z d� Z d� Z!d� Z"d� Z#efd�Z$efd�Z%y)� N)�deque)� TYPE_CHECKING)� getsizeof)�Any�Callable�Dict�List�Optional�Tuple)�Span)�loggeri N c � � e Zd ZdZdZdZdZy)� GEN_AI_ALLOWED_MESSAGE_ROLES�system�user� assistant�toolN)�__name__� __module__� __qualname__�SYSTEM�USER� ASSISTANT�TOOL� � �B/opt/hc_python/lib/python3.12/site-packages/sentry_sdk/ai/utils.pyr r s � � �F� �D��I� �Dr r r r �humanr �air � tool_callc �r �� t | d� r t | j � ��� S t | t � r4�r t | � dk( rt | d ��� S t �fd�| D � � S t | t � r/| j � D ��ci c] \ }}|t |��� �� c}}S t | t t t t f� r| S t | � S # t $ rM}t j d|� t | t t t t f� r| n t | � cY d }~S d }~ww xY wc c}}w )N� model_dump��unpackz+Could not convert pydantic data to JSON: %s� r c 3 �8 �K � | ] }t |�� � �� � y�w)r# N)�_normalize_data)�.0�xr$ s �r � <genexpr>z"_normalize_data.<locals>.<genexpr>2 s �� �� �D�t�!�O�A�f�5�5�t�s �)�hasattrr' r" � Exceptionr �warning� isinstance�int�float�bool�str�list�len�dict�items)�datar$ �e�k�vs ` r r' r' % s �� � �t�\�"� T�"�4�?�?�#4�V�D� D� �$�� � �c�$�i�1�n�"�4��7�6�:� :��D�t�D�D�D��$�� �CG�:�:�<�P�<��!�Q��?�1�V�4�4�<�P�P� �d�S�%��s�$;�<�4�K�#�d�)�K�� � T� �N�N�H�!� L�%�d�S�%��s�,C�D�4�#�d�)� S�� T�� Qs$ �C �D3� D0�#AD+�%D0�+D0c �� � t ||�� }t |t t t t f� r| j ||� y | j |t j |� � y )Nr# ) r' r. r/ r0 r1 r2 �set_data�json�dumps)�span�key�valuer$ � normalizeds r �set_data_normalizedrC : sF � � ��v�6�J��*�s�E�4�� 5�6� � � �c�:�&� � � �c�4�:�:�j�1�2r c �. � t j | | � S )z� Normalize a message role to one of the 4 allowed gen_ai role values. Maps "ai" -> "assistant" and keeps other standard roles unchanged. )�GEN_AI_MESSAGE_ROLE_MAPPING�get)�roles r �normalize_message_rolerH C s � � '� *� *�4�� 6�6r c �� � g }| D ]Z }t |t � s|j |� �%|j � }d|v rt |d � |d< |j |� �\ |S )z� Normalize roles in a list of messages to use standard gen_ai role values. Creates a deep copy to avoid modifying the original messages. rG )r. r5 �append�copyrH )�messages�normalized_messages�message�normalized_messages r �normalize_message_rolesrP L sn � � �����'�4�(� � &� &�w� /� � $�\�\�^�� �W� �)?����)P� �v� &��"�"�#5�6� � � r c � � t j � } | d uxr | j d u}|rt j S t j S )N)� sentry_sdk�get_current_span�containing_transaction� start_span�start_transaction)� current_span�transaction_existss r �get_start_span_functionrY _ sI � � �.�.�0�L��D� �T�\�%H�%H�PT�%T� � %7�:� � �X�J�<X�<X�Xr c �� � d}t t | � dz dd� D ]E }t t j | | d�� j d� � }||z }||kD s�@|dz c S y)z� Find the index of the first message that would exceed the max bytes limit. Compute the individual message sizes, and return the index of the first message from the back of the list that would exceed the max bytes limit. r r% �����,�:�� separators�utf-8)�ranger4 r= r> �encode)rL � max_bytes� running_sum�idx�sizes r �_find_truncation_indexrh h si � � �K��S��]�Q�&��B�/���4�:�:�h�s�m� �C�J�J�7�S�T���t�� � �� "���7�N� 0� r c � � t j | d�� }t |j d� � }||k r| dfS t | |� }| |d |fS )Nr\ r_ ra r )r= r> r4 rc rh )rL rd �serialized_json� current_size�truncation_indexs r �truncate_messages_by_sizerm y s[ � ��j�j��j�A�O���-�-�g�6�7�L��y� ���{��-�h� �B�� �$�%� &�(8� 8�8r c �x � | sy t | |� \ }}|dkD r"t | � |j |j < |S )Nr )rm r4 � _gen_ai_original_message_count�span_id)rL r? �scoperd �truncated_messages� removed_counts r � truncate_and_annotate_messagesrt � sB � � ��(A�(�I�(V�%�� ��q��=@��]��,�,�T�\�\�:� � r )T)&r= � collectionsr �typingr �sysr r r r r r r �sentry_sdk.tracingr rR �sentry_sdk.utilsr �MAX_GEN_AI_MESSAGE_BYTESr r r r r �#GEN_AI_MESSAGE_ROLE_REVERSE_MAPPINGrE r6 � target_role� source_roles� source_roler' rC rH rP rY rh rm rt r r r �<module>r s� �� � � � ��A�A�'� � #�!� �� � !�'�'�(�� �%�%���'8� �*�*�[�$�,?� �%�%�� �'<� '� #� !� �!D�!J�!J�!L� �K� �#� �3>�#�K�0� $� "M� L�*3�7� �&Y� �" 3K� 9� &>� r