�
_�*j� � � � d dl Z d dlZd dlZd dlZd dlmZmZ d dlmZm Z m
Z
d dl m Z d dl
mZmZmZmZmZ d dlZd dlmZ d dlmZ erd d lmZ d d
lmZmZ d d l
m Z dd �Z dd
�Z dd�Z dd�Z G d� de � Z!y# e$ r ed� �w xY w)� N)�OP�
SPANSTATUS)� DidNotEnable� Integration�_check_minimum_version)�TransactionSource)�event_from_exception�logger�package_version�qualname_from_function�reraise)�remotezRay 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 � � t j | � }t |j j � � }t j
dt j
j d �� }|r<|d j t j
j u r|j d|� n|j |� |j |�� | _ y )N�_sentry_tracing)�kind�default���)�
parameters)
�inspect� signature�listr" �values� Parameter� KEYWORD_ONLYr � VAR_KEYWORD�insert�append�replace�
__signature__)�funcr$ �params�sentry_tracing_params r �#_insert_sentry_tracing_in_signaturer1 ) s� � � �!�!�$�'�I�
�)�&�&�-�-�/�
0�F�"�,�,��
�
�
�
+�
+�� �� �&��*�/�/�W�%6�%6�%B�%B�B��
�
�b�.�/��
�
�*�+�"�*�*�f�*�=�D�r c �v �� t �t j �� d d�fd�
� } | t _ y )Nc �v �� ��� t j � � r �� g���i ���S d�� ��fd� }� � |� � S |S )Nc �* �� �� t j � � r ��i ���� � S t j � � d d� d� fd�� }t |� �r �|� }n ��i ���|� }|j
� d�� fd� }||_ |S )N)r c � �� 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 �RayIntegrationr8 r �TASK�start_transaction�
set_statusr �OK� Exception�INTERNAL_ERROR�sys�exc_info�_capture_exceptionr
)r �f_args�f_kwargs� transaction�resultrD �user_fs �r �new_funczH_patch_ray_remote.<locals>.new_remote.<locals>.wrapper.<locals>.new_funcS 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;�;Dc �& �� 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
)r6 r7 r8 r N)r �
start_spanr �QUEUE_SUBMIT_RAYr r< r8 �get_current_scope� iter_trace_propagation_headersr? r r@ rA rB rC rD rE r
)
�args�kwargs�span�k�v�tracingrI rD �old_remote_methodrJ s
��r �&_remote_method_with_header_propagationzf_patch_ray_remote.<locals>.new_remote.<locals>.wrapper.<locals>._remote_method_with_header_propagationw 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)rF r r zOptional[dict[str, Any]]rG r �returnr )rQ r rR r rY r )r# �isclass� functools�wrapsr1 r ) rJ rK �rvrX rW rQ �frR �
old_remotes ` @����r �wrapperz6_patch_ray_remote.<locals>.new_remote.<locals>.wrapperK s� �� ����v�&�
3�z�4�2�6�2�6�:�:�
�_�_�V�
$� ?C�
"� �
"�!;�
"� "�
"� �
"� %�
"�6
0�� 9�� ��)��0�Z��0��0��:�� "� � � �
"� �
"�(-�
"��
"�8 ?�B�I��Ir )rJ �Callable[..., Any]rY r )r# rZ )r^ rQ rR r` r_ s ``` �r �
new_remotez%_patch_ray_remote.<locals>.new_remote@ sK �� � �?�?�1� �
�a�1�$�1�&�1� 1�J � J �X
�=��1�:� ��Nr )N)r^ z Optional[Callable[..., Any]]rQ r rR r rY ra )r r[ r\ �ray)rb r_ s @r �_patch_ray_remoterd = sV �� ��J��_�_�Z� �,0�Y� )�Y�9>�Y�JO�Y� �Y� !�Y�v �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)rD rR �client�eventrj s r rE rE � sK � �
�
"�
"�
$�F�&�� �~�~� �"�-�-�
��K�E�4� � � �U��.r c �, � e Zd ZdZde� �Zedd�� Zy)r< rc z auto.queue.Nc �N � t d� } t t | � t � y )Nrc )r r r< rd )�versions r �
setup_oncezRayIntegration.setup_once� s � �!�%�(�� �~�w�7��r �rY N)�__name__�
__module__� __qualname__rl r8 � staticmethodrs r r r r< r< � s% � ��J��:�,�
'�F�� � � r r< rt )r. ra rY N)rD r rR r rY N)"r[ r# rC r �sentry_sdk.constsr r �sentry_sdk.integrationsr r r �sentry_sdk.tracingr �sentry_sdk.utilsr r
r r r
rc r � ImportError�typingr �collections.abcr r r r r r1 rd rE r< r r r �<module>r� s� �� � �
� � ,� U� U� 0�� �-��� !��(�$�(��>�(_ �D /� �[� ��s � -�
�+�
,�,�-�s �
A: �:B