�
h 0i� � � � d dl Z d dlZd dlZd dlZd dlmZmZ d dlmZm Z m
Z
d dl m Z d dl
mZmZmZmZmZ d dlZd dlmZ erd dlmZ d d lmZmZ d d
l
mZ d � Z d � Z d
� Z G d� de
� Z y# e$ r e d� �w xY w)� N)�OP�
SPANSTATUS)�_check_minimum_version� DidNotEnable� Integration)�TransactionSource)�event_from_exception�logger�package_version�qualname_from_function�reraisezRay not installed.)�
TYPE_CHECKING)�Callable)�Any�Optional)�ExcInfoc �t � t j � j � ry t j d� y )Nz[[Tracing] Sentry not initialized in ray cluster worker, performance data will be discarded.)�
sentry_sdk�
get_client� is_activer
�debug� � �J/opt/hc_python/lib/python3.12/site-packages/sentry_sdk/integrations/ray.py�_check_sentry_initializedr s) � �� � � �(�(�*��
�L�L�e�r c �x �� t j �t j �� d�fd� � } | t _ y )Nc �t �� ��� t j � � r �� g���i ���S �� ��fd�}� � |� � S |S )Nc �� �� �� t j � � d d�� fd�
� }t j |� }t |j
j
� � }|j t j dt j j d �� � |j |�� |_ �r �
|� }n �
�i � ��|� }|j ��� fd�}||_ |S )N)�_sentry_tracingc � �� t � t j | xs i t j t �� t j t j �� }t j |� 5 } �|i |��}|j t j � cd d d � S # t $ rI |j t j � t! j" � }t% |� t' |� Y �\w xY w# 1 sw Y y xY w)N)�op�name�origin�source)r r �continue_tracer �QUEUE_TASK_RAYr �RayIntegrationr# r �TASK�start_transaction�
set_statusr �OK� Exception�INTERNAL_ERROR�sys�exc_info�_capture_exceptionr
)r �f_args�f_kwargs� transaction�resultr/ �user_fs �r �new_funczH_patch_ray_remote.<locals>.new_remote.<locals>.wrapper.<locals>.new_func9 s� �� � *�+� (� 7� 7�#�)�r��(�(�/��7�)�0�0�,�1�1� � � �1�1�+�>�+�+�!'��!<�8�!<��#�.�.�z�}�}�=� "� ?�>�� %� +�#�.�.�z�/H�/H�I�#&�<�<�>��*�8�4� ��*� +�� ?�>�s1 �2C;�4'B&�C;�&AC8�5C;�7C8�8C;�;Dr )�kind�default)�
parametersc �& �� t j t j t � � t
j �� 5 }t j � j � D ��ci c] \ }}||��
}}} �| i |�d|i��}|j t j � cddd� S c c}}w # t $ rI |j t j � t j � }t! |� t# |� Y �bw xY w# 1 sw Y yxY w)z,
Ray Client
)r! r" r# r N)r �
start_spanr �QUEUE_SUBMIT_RAYr r' r# �get_current_scope� iter_trace_propagation_headersr* r r+ r, r- r. r/ r0 r
)
�args�kwargs�span�k�v�tracingr4 r/ �old_remote_methodr5 s
��r �&_remote_method_with_header_propagationzf_patch_ray_remote.<locals>.new_remote.<locals>.wrapper.<locals>._remote_method_with_header_propagatione s� �� �
�*�*��*�*�/��7�)�0�0�� � %/�$@�$@�$B�$a�$a�$c� �$c�D�A�q� �1��$c� � � +�!2� !�"�%+�"�=D�"�� ���
�
�
�6� "�)� ��
�� %� +� ���
�(A�(A�B�#&�<�<�>��*�8�4� ��*� +�� � �sB �&D�%
B,�2D�6+B2�!D�,D�2AD�D�D�D�D)
� functools�wraps�inspect� signature�listr9 �values�append� Parameter� KEYWORD_ONLY�replace�
__signature__�remote) r5 r6 rJ �params�rvrF rE r? �fr@ �
old_remotes ` @����r �wrapperz6_patch_ray_remote.<locals>.new_remote.<locals>.wrapper7 s� �� �
�_�_�V�
$�26�
"� %�
"�4 �)�)�(�3�I��)�.�.�5�5�7�8�F� �M�M��!�!�%� �*�*�7�7� ��
� &/�%6�%6�&�%6�%I�H� "�� ��)��0�Z��0��0��:�� "� � � �
"�6 ?�B�I��Ir )rI �isclass)rU r? r@ rW rV s ``` �r �
new_remotez%_patch_ray_remote.<locals>.new_remote, sE �� � �?�?�1� �
�a�1�$�1�&�1� 1�K �Z
�=��1�:� ��Nr )N)�rayrR rG rH )rY rV s @r �_patch_ray_remoter[ ( s3 �� ����J��_�_�Z� �Z� !�Z�x �C�Jr c � � t j � }t | |j dt j
d��� \ }}t j ||�� y )NF)�handled�type)�client_options� mechanism)�hint)r r r �optionsr' �
identifier�
capture_event)r/ r@ �client�eventra s r r0 r0 � sK � �
�
"�
"�
$�F�&�� �~�~� �"�-�-�
��K�E�4� � � �U��.r c �* � e Zd ZdZde� �Zed� � Zy)r' rZ z auto.queue.c �N � t d� } t t | � t � y )NrZ )r r r' r[ )�versions r �
setup_oncezRayIntegration.setup_once� s � � "�%�(�� �~�w�7��r N)�__name__�
__module__� __qualname__rc r# � staticmethodrj r r r r' r' � s% � ��J��:�,�
'�F�� � � r r' ) rI rG r. r �sentry_sdk.constsr r �sentry_sdk.integrationsr r r �sentry_sdk.tracingr �sentry_sdk.utilsr r
r r r
rZ � ImportError�typingr �collections.abcr r r r r r[ r0 r' r r r �<module>rv s{ �� � �
� � ,� U� U� 0�� �-�� !��(�$�(��a �H /�
�[�
��Q � -�
�+�
,�,�-�s �A- �-A;