← Back
Editing: ray.cpython-312.pyc
� ]*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
Save File
Cancel