PATH:
opt
/
hc_python
/
lib
/
python3.12
/
site-packages
/
sentry_sdk
/
integrations
/
pydantic_ai
/
spans
/
__pycache__
� o 0i� � �z � d dl Z d dlmZmZ d dlmZmZ ddlmZ ddl m Z m Z m Z m Z d dlmZ erd dlmZ d d �Zd � Zy) � N)�get_start_span_function�set_data_normalized)�OP�SPANDATA� )� SPAN_ORIGIN)�_set_agent_data�_set_available_tools�_set_model_data�_should_send_prompts)� TYPE_CHECKING)�Anyc �" � d}|rt |dd� r |j } t � t j d|� �t �� }|j t j d� t ||� t |||� t ||� t � �ryg }g }|r�t |dd� xs g } | D ]$ } t | t � s�|j | � �& t |dd� } | rmt | t � r|j | � nKt | t t" f� r5| D ]0 } t | t � r|j | � �%t% | � s�1�2 |D ] } |j | d d �gd d �� � | r}t | t � r|j | d d �gd d �� nTt | t � rDg }| D ]' }t |t � s�|j |d d �� �) |r|j |d d �� |r t' |t j( |d�� |S )z%Create a span for invoking the agent.�agent�nameNz invoke_agent )�opr �origin� invoke_agent�_system_prompts� _instructions�text)r �type�system)�content�role�userF��unpack)�getattrr r r �GEN_AI_INVOKE_AGENTr �set_datar �GEN_AI_OPERATION_NAMEr r r r � isinstance�str�append�list�tuple�callabler �GEN_AI_REQUEST_MESSAGES)� user_promptr �model�model_settings� is_streamingr �span�messages� system_texts�system_prompts�prompt� instructions�instr� system_textr �items �e/opt/hc_python/lib/python3.12/site-packages/sentry_sdk/integrations/pydantic_ai/spans/invoke_agent.py�invoke_agent_spanr8 s� � � �D� �����-��z�z�� $� "� $� � !� !� �T�F� #�� �D� �M�M�(�0�0�.�A��D�%� ��D�%� �0���u�%� � ��� � � � $�U�,=�t� D� J��N�(�� �f�c�*� �'�'��/� )� #�5�/�4�@�L�� �l�C�0� �'�'� �5� � �t�U�m�<�!-��%�e�S�1� (� /� /�� 6� %�e�_� � ".� (�K� �O�O�)4�f� E� F� $�� � (� ��+�s�+����-8�&�$I�#J� &��� �K��.� �� '�D�!�$��,� ����f�'E�F� (� � �O�O�'.�$*��� � ��h�6�6��� � �K� c �r � | r5t � r*|r't | t j t |� d�� yyyy)z'Update and close the invoke agent span.Fr N)r r r �GEN_AI_RESPONSE_TEXTr$ )r. �outputs r7 �update_invoke_agent_spanr= j s4 � � �$�&�6�� �(�/�/��V��U� � ,2�&�tr9 )F)� sentry_sdk�sentry_sdk.ai.utilsr r �sentry_sdk.constsr r �constsr �utilsr r r r �typingr r r8 r= � r9 r7 �<module>rE s2 �� � L� *� �� � !���T�n r9
[-] invoke_agent.cpython-312.pyc
[edit]
[+]
..
[-] __init__.cpython-312.pyc
[edit]
[-] ai_client.cpython-312.pyc
[edit]
[-] execute_tool.cpython-312.pyc
[edit]