PATH:
opt
/
hc_python
/
lib
/
python3.12
/
site-packages
/
sentry_sdk
/
integrations
/
pydantic_ai
/
spans
/
__pycache__
� o 0i�" � � � d dl Z d dlmZ d dlmZmZ d dlmZ ddlm Z ddl m Z m Z m Z mZmZmZmZ d dlmZ erd d lmZmZmZ d d lmZ d d lmZmZm Z m Z m Z m Z d � Z!d � Z"d� Z#d� Z$d� Z%y# e $ r dZdZdZ dZ dZ dZ Y �#w xY w)� N)�set_data_normalized)�OP�SPANDATA)�safe_serialize� )� SPAN_ORIGIN)�_set_agent_data�_set_available_tools�_set_model_data�_should_send_prompts�_get_model_name�get_current_agent�get_is_streaming)� TYPE_CHECKING)�Any�List�Dict)� RequestUsage)�BaseToolCallPart�BaseToolReturnPart�SystemPromptPart�UserPromptPart�TextPart� ThinkingPartc � � |�yt |d� r6|j �*| j t j |j � t |d� r6|j �*| j t j |j � t |d� r8|j �+| j t j |j � yyy)z Set token usage data on a span.N� input_tokens� output_tokens� total_tokens) �hasattrr �set_datar �GEN_AI_USAGE_INPUT_TOKENSr �GEN_AI_USAGE_OUTPUT_TOKENSr �GEN_AI_USAGE_TOTAL_TOKENS)�span�usages �b/opt/hc_python/lib/python3.12/site-packages/sentry_sdk/integrations/pydantic_ai/spans/ai_client.py�_set_usage_datar' * s� � � �}���u�n�%�%�*<�*<�*H� � � �h�8�8�%�:L�:L�M��u�o�&�5�+>�+>�+J� � � �h�9�9�5�;N�;N�O��u�n�%�%�*<�*<�*H� � � �h�8�8�%�:L�:L�M� +I�%� c � � t � sy|sy g }d}|D ]) }t |d� s�|j s� |j } n |r|j dd|d�gd�� |D �]i }t |d� s�|j D �]I }d}t rt |t � rd}n]t rt |t � s,t rt |t � st rt |t � rd }nt rt |t � rd }g }d}d} t rYt |t � rIi } t |d � r|j | d < t |d � rt |j � | d< | �r>| g}�n9t r\t |t � rLt |d � r |j } t |d� r�|j dt |j � d�� n�t |d� r�t |j t � r |j d|j d�� n�t |j t � rQ|j D ]A } t | t � r|j d| d�� �(|j t | � � �C n'|j dt |j � d�� |s|s�� d|i} |r|| d< |r|| d< | r| | d< |j | � ��L ��l |r t# | t$ j& |d�� yy# t( $ r Y yw xY w)z"Set input messages data on a span.N� instructions�system�text)�typer, )�role�content�parts�user� assistant�tool� tool_name�name�args� argumentsr/ r. � tool_calls� tool_call_idF)�unpack)r r r* �appendr0 r � isinstancer r r r r4 r r6 �strr/ �listr r �GEN_AI_REQUEST_MESSAGES� Exception) r$ �messages�formatted_messages� system_prompt�msg�partr. r/ r8 r9 �tool_call_data�item�messages r& �_set_input_messagesrI : s� � � � !�� ��Q � �� � � �C��s�N�+��0@�0@� #� 0� 0� �� � � � %� %�!�� �/V�.W�X� � �C��s�G�$� �I�I�D�!�D�'�J�t�=M�,N� '��!�j��x�&@� (�Z��l�-K� ,��D�BR�1S� *��+� �4�AS�0T� %�� �G�!%�J�#'�L� (�J�t�=M�,N�)+��"�4��5�59�^�^�N�6� 2�"�4��0�:H����:S�N�;� 7�)�*8�)9�J�+� �4�AS�0T�"�4��5�+/�>�>�L�"�4��3� #�N�N�F�C�� � �DU�+V� W� ��y�1�%�d�l�l�C�8� #�N�N�F�D�L�L�+Q� R� '�� � �d� ;�(,� � ��#-�d�C�#8�$+�N�N�F�D�3Q�$R�$+�N�N�>�$�3G�$H� )5� $�N�N�F�C�� � �DU�+V� W� �*�#)�4�.��"�18�G�I� .�%�4>�G�L� 1�'�6B�G�N� 3�*�1�1�'�:�o &� �v � ��h�6�6�8J�SX� � �� � � � �s) �K, � K, �;K, �/H*K, �AK, �, K8�7K8c � � t � sy|syt | t j |j � t |d� �r g }g }|j D ]� }t r8t |t � r(t |d� r |j |j � �At s�Ht |t � s�Yddi}t |d� r|j |d< t |d� rt |j � |d < |j |� �� |rt | t j |� |r*| j! t j" t |� � yyy# t$ $ r Y yw xY w) zSet output data on a span.Nr0 r/ r- �functionr4 r5 r6 r7 )r r r �GEN_AI_RESPONSE_MODEL� model_namer r0 r r< r; r/ r r4 r r6 �GEN_AI_RESPONSE_TEXTr �GEN_AI_RESPONSE_TOOL_CALLSr@ )r$ �response�textsr8 rE rF s r& �_set_output_datarR � s) � � � !�� ����h� <� <�h�>Q�>Q�R� � �8�W� %��E��J� ����� �4�� :�w�t�Y�?W��L�L�� � �.�%�*�T�;K�*L� � �&�N� �t�[�1�15� � ��v�.� �t�V�,�6D�T�Y�Y�6O��{�3� �%�%�n�5� '� �#�D�(�*G�*G��O��� � � �7�7� � �9S�� �) &��2 � � � �s �A$E �E �+B E � E�Ec �� � |}|rt |d� r |j }t |� xs d}t j t j d|� �t �� }|j t j d� t ||� t |||� |j t j t � � |xs t � }t! ||� | r t# || � |S )z�Create a span for an AI client call (model request). Args: messages: Full conversation history (list of messages) agent: Agent object model: Model object model_settings: Model settings �model�unknownzchat )�opr5 �origin�chat)r rT r � sentry_sdk� start_spanr � GEN_AI_CHATr r r �GEN_AI_OPERATION_NAMEr r �GEN_AI_RESPONSE_STREAMINGr r r rI )rA �agentrT �model_settings� model_objrM r$ � agent_objs r& �ai_client_spanrb � s� � � �I� ����(��K�K� � ��+�8�y�J� � � � �>�>��Z�L� !�� �D� �M�M�(�0�0�&�9��D�%� ��D�%� �0� �M�M�(�4�4�6F�6H�I� �,�*�,�I���y�)� ��D�(�+� �Kr( c �j � | sy|r"t |d� rt | |j � t | |� y)z-Update the AI client span with response data.Nr% )r r' r% rR )r$ �model_responses r& �update_ai_client_spanre � s4 � � �� �'�.�'�:���n� 2� 2�3� �T�>�*r( )&rY �sentry_sdk.ai.utilsr �sentry_sdk.constsr r �sentry_sdk.utilsr �constsr �utilsr r r r r r r �typingr r r r �pydantic_ai.usager �pydantic_ai.messagesr r r r r r � ImportErrorr' rI rR rb re � r( r& �<module>rp s� �� � 3� *� +� �� � � !��&�&�.��� �$ N� Z �z' �T'�T +��W � ��� �����N��H��L��s � A* �*A>�=A>
[-] invoke_agent.cpython-312.pyc
[edit]
[+]
..
[-] __init__.cpython-312.pyc
[edit]
[-] ai_client.cpython-312.pyc
[edit]
[-] execute_tool.cpython-312.pyc
[edit]