o
    ٰi                     @   sZ   d dl Z d dlmZmZ ddlmZ d dlmZ erd dlZdddd	d
e	ddfddZ
dS )    N)OPSPANDATA   )SPAN_ORIGIN)TYPE_CHECKINGcontextzagents.RunContextWrapper
from_agentzagents.Agentto_agent_namereturnc                 C   s   t jtjd|j d| td'}|tjd t	|dd }|r/|tj
| W d    d S W d    d S 1 s:w   Y  d S )Nzhandoff from z to )opnameoriginhandoff_sentry_conversation_id)
sentry_sdk
start_spanr   GEN_AI_HANDOFFr   r   set_datar   GEN_AI_OPERATION_NAMEgetattrGEN_AI_CONVERSATION_ID)r   r   r	   spanconv_id r   g/home/ubuntu/.local/lib/python3.10/site-packages/sentry_sdk/integrations/openai_agents/spans/handoff.pyhandoff_span   s   	"r   )r   sentry_sdk.constsr   r   constsr   typingr   agentsstrr   r   r   r   r   <module>   s    