PATH:
opt
/
hc_python
/
lib
/
python3.12
/
site-packages
/
sentry_sdk
/
integrations
/
pydantic_ai
/
__pycache__
� o0i� � �n � d dl mZmZ d dlZddlmZmZmZm Z G d� de� Z y# e$ r ed� �w xY w)� )�DidNotEnable�IntegrationNzpydantic-ai not installed� ��_patch_agent_run�_patch_graph_nodes�_patch_model_request�_patch_tool_executionc �2 � e Zd ZdZde� �Zdd�Zed� � Zy)�PydanticAIIntegration�pydantic_aizauto.ai.c � � || _ y)z� Initialize the Pydantic AI integration. Args: include_prompts: Whether to include prompts and messages in span data. Requires send_default_pii=True. Defaults to True. N)�include_prompts)�selfr s �[/opt/hc_python/lib/python3.12/site-packages/sentry_sdk/integrations/pydantic_ai/__init__.py�__init__zPydanticAIIntegration.__init__ s � � /��� c �T � t � t � t � t � y)z� Set up the pydantic-ai integration. This patches the key methods in pydantic-ai to create Sentry spans for: - Agent invocations (Agent.run methods) - Model requests (AI client calls) - Tool executions Nr � r r � setup_oncez PydanticAIIntegration.setup_once! s � � �������r N)T)�__name__� __module__�__qualname__� identifier�originr �staticmethodr r r r r r s, � ��J�� �|� $�F� /� � � � r r )�sentry_sdk.integrationsr r r �ImportError�patchesr r r r r r r r �<module>r sC �� =�4�� � � �K� �� � 4� �2� 3�3�4�s �&