📁 File Manager Pro
v10.0.3 | PHP: 8.2.31
Server: LiteSpeed
2026-06-28 13:47:28
📂
/ (Root)
/
opt
/
hc_python
/
lib
/
python3.12
/
site-packages
/
sentry_sdk
/
integrations
/
pydantic_ai
/
patches
/
__pycache__
📍 /opt/hc_python/lib/python3.12/site-packages/sentry_sdk/integrations/pydantic_ai/patches/__pycache__
🔄 Refresh
✏️
Editing: tools.cpython-312.pyc
Read Only
� |�*j� � �� � d dl Z d dlmZ d dlmZ d dlZd dlmZ d dlm Z m Z ddl m Z m Z ddlmZmZ erd d lmZ d d lmZ d d lmZ dd �Zdd�Zdd�Zy# e$ r d d lmZ Y � w xY w# e$ r ed � �w xY w)� N)�wraps)� TYPE_CHECKING)� DidNotEnable)�capture_internal_exceptions�reraise� )�execute_tool_span�update_execute_tool_span)�_capture_exception�get_current_agent)�Any)� ToolManager)�ToolRetryErrorzpydantic-ai not installedc �p � t t d� r t � y t t d� r t � y y )N�execute_tool_call� _call_tool)�hasattrr �_patch_execute_tool_call�_patch_call_tool� � �`/opt/hc_python/lib/python3.12/site-packages/sentry_sdk/integrations/pydantic_ai/patches/tools.py�_patch_tool_executionr s( � ��{� /�0� �"� ��l� +��� ,r c �x �� t j �t �� d�fd� � } | t _ y )Nc � � �K � |r t |d� s �| |g|��i |��� d { ��� S |j }|j }| j r| j j |� nd }t |dd � }t � }|rn|rl |j � } t j � 5 t || ||�� 5 } �| |g|��i |��� d { ��� } t | | � | cd d d � cd d d � S �| |g|��i |��� d { ��� S 7 ��# t $ r+ t |j t � r |j ni } Y ��w xY w7 �r# t $ r�} t# j$ � } t' � 5 ddlm} t j, � j/ |� }|�|j0 r t3 | d�� d d d � n # 1 sw Y nxY wt5 | � Y d } ~ nd } ~ ww xY w d d d � n # 1 sw Y nxY wd d d � ��# 1 sw Y ��xY w7 ���w)N�call�tool_def��tool_definitionr ��PydanticAIIntegrationT��handled)r r � tool_name�tools�get�getattrr � args_as_dict� Exception� isinstance�args�dict� sentry_sdk�isolation_scoper r r �sys�exc_infor �#sentry_sdk.integrations.pydantic_air! � get_client�get_integration� handled_tool_call_exceptionsr r )�self� validatedr+ �kwargsr �name�tool�selected_tool_definition�agent� args_dict�span�result�excr0 r! � integration�original_execute_tool_calls �r �wrapped_execute_tool_callz;_patch_execute_tool_call.<locals>.wrapped_execute_tool_call& s� �� �� � �� �6� :�3�D�)�U�d�U�f�U�U� U� �~�~���~�~��'+�z�z�t�z�z�~�~�d�#�t��#*�4��T�#B� � "�#�� �T� M� � -� -� /� � �+�+�-�&�� ��$<� � �+�'A� � %�(� "�(� %� (� "�� 1��v�>� %� � � .�-�D 0��i�Q�$�Q�&�Q�Q�Q�k V�� � M�)3�D�I�I�t�)D�D�I�I�"� � M��"�� *� +�#&�<�<�>�� 8� :� � +5�*?�*?�*A�*Q�*Q� 5�+ �K� !,� 7�$/�$L�$L� 2�3�� E� ;� :� :�� ��*��!+�� +�?� � �� .�-�-��D R�s� � H�D�A H�D �H�(G5�8G �:D?� D=� D?� G5�%H�?H� H�1D:�7H�9D:�:H�=D?�? G� G �&AF4 �+ G �4F= �9G �G � G�G � G5� G( �$G5�+ H�5G?�: H) r5 r r6 r r+ r r7 r �returnr )r r r )rB rA s @r r r # sa �� �!,�!>�!>� � � %�&�9R��9R� %�9R�.3�9R�?D�9R� �9R� '�9R�v %>�K�!r c �x �� t j �t �� d�fd� � } | t _ y)a� Patch ToolManager._call_tool to create execute_tool spans. This is the single point where ALL tool calls flow through in pydantic_ai, regardless of toolset type (function, MCP, combined, wrapper, etc.). By patching here, we avoid: - Patching multiple toolset classes - Dealing with signature mismatches from instrumented MCP servers - Complex nested toolset handling c � � �K � |j }| j r| j j |� nd }t |dd � }t � }|rn|rl |j � }t j � 5 t ||||�� 5 } �| |g|��i |��� d { ��� } t | | � | cd d d � cd d d � S �| |g|��i |��� d { ��� S # t $ r+ t |j t � r |j ni }Y ��w xY w7 �p# t $ r�} t j � } t# � 5 ddlm} t j( � j+ | � }|�|j, r t/ | d�� d d d � n # 1 sw Y nxY wt1 | � Y d } ~ nd } ~ ww xY w d d d � n # 1 sw Y nxY wd d d � ��# 1 sw Y ��xY w7 ���w)Nr r r r Tr" )r$ r% r& r' r r( r) r* r+ r, r- r. r r r r/ r0 r r1 r! r2 r3 r4 r r )r5 r r+ r7 r8 r9 r: r; r< r= r> r? r0 r! r@ �original_call_tools �r �wrapped_call_toolz+_patch_call_tool.<locals>.wrapped_call_tools s� �� �� � �~�~��'+�z�z�t�z�z�~�~�d�#�t��#*�4��T�#B� � "�#�� �T� M� � -� -� /� � �+�+�-�&�� ��$<� � �+�'9� � �(� "�(� %� (� "�� 1��v�>� %� � � .�-�F (� � � � � � � � � ��Q � M�)3�D�I�I�t�)D�D�I�I�"� � M��"�� *� +�#&�<�<�>�� 8� :� � +5�*?�*?�*A�*Q�*Q� 5�+ �K� !,� 7�$/�$L�$L� 2�3�� E� ;� :� :�� ��*��!+�� +�?� � �� .�-�-��F �s� �AG�C �%G�9G� F.� D�D � D�- G�6G�G�G�1D �G�D � G� D� F!� F �5AF �: F �F �F �F.� F!�!F.�% G�.F7 �3G�: G�G� GN) r5 r r r r+ r r7 r rC r )r r r )rG rF s @r r r e s\ �� � %�/�/�� � � �; ��; � �; �).�; �:?�; � �; � �; �z /�K�r )rC N)r/ � functoolsr �typingr r- �sentry_sdk.integrationsr �sentry_sdk.utilsr r �spansr r �utilsr r r �pydantic_ai.tool_managerr � ImportError�pydantic_ai._tool_manager�pydantic_ai.exceptionsr r r r r r r �<module>rR sw �� � � � � 0� A� ?� 9���4�:�8� 6� �?>�DL/��g � :�9�:�� � 4� �2� 3�3�4�s) �A �A' � A$�!A' �#A$�$A' �'A5
💾 Save Changes
❌ Cancel