관리-도구
편집 파일: anthropic.cpython-38.pyc
U �Mg/% � @ s& d dl mZ d dlmZ d dlZd dlmZ d dlmZm Z d dl m Z m Z d dl mZ d dlmZmZmZ z$d d lmZmZ er�d d lmZ W n ek r� e d ��Y nX er�d d lmZmZmZ d d l m Z G dd� de �Z dd� Z dd� Z dd� Z!dd� Z"dd� Z#dd� Z$d d � Z%d d � Z&dS ) � ��wraps)� TYPE_CHECKINGN)�record_token_usage)�OP�SPANDATA)� DidNotEnable� Integration)�should_send_default_pii)�capture_internal_exceptions�event_from_exception�package_version)� AsyncMessages�Messages)�MessageStreamEventzAnthropic not installed)�Any� AsyncIterator�Iterator)�Spanc @ s0 e Zd ZdZde� �Zd dd�Zedd� �ZdS ) �AnthropicIntegration� anthropiczauto.ai.Tc C s || _ d S �N)�include_prompts)�selfr � r �O/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/anthropic.py�__init__ s z AnthropicIntegration.__init__c C sD t d�} | d krtd��| dk r(td��ttj�t_ttj�t_d S )Nr z Unparsable anthropic version.)r � z!anthropic 0.16 or newer required.)r r �_wrap_message_creater �create�_wrap_message_create_asyncr )�versionr r r � setup_once$ s z AnthropicIntegration.setup_onceN)T)�__name__� __module__� __qualname__� identifier�originr � staticmethodr" r r r r r s r c C s0 t | t�� jddd�d�\}}tj||d� d S )Nr F)�typeZhandled)Zclient_optionsZ mechanism)�hint)r � sentry_sdk� get_client�optionsZ capture_event)�exc�eventr* r r r �_capture_exception3 s � r0 c C sj d}d}t | d�rP| j}t |d�r4t|jt�r4|j}t |d�rPt|jt�rP|j}|| }t||||� d S )Nr �usage� input_tokens� output_tokens)�hasattrr1 � isinstancer2 �intr3 r )�result�spanr2 r3 r1 � total_tokensr r r �_calculate_token_usage= s r: c C s0 g }| D ]"}t |d�r|�|j|jd�� q|S )z, Get JSON of a Anthropic responses. �text�r) r; )r4 �appendr) r; )�content� responses�itemr r r �_get_responsesL s ��rA c C s� t � �� t| d�r�| jdkr:| jj}||j7 }||j7 }nT| jdkrFnH| jdkrlt| jd�r�|�| jj � n"| jdkrxn| jdkr�|| jj7 }W 5 Q R X |||fS )zV Count token usage and collect content blocks from the AI streaming response. r) � message_startZcontent_block_startZcontent_block_deltar; Zcontent_block_stopZ message_delta) r r4 r) �messager1 r2 r3 �deltar= r; )r/ r2 r3 �content_blocksr1 r r r �_collect_ai_data] s rF c C sf t � �V t� r4|jr4d�|�}| �tjd|d�g� || }t| |||� | �tjd� W 5 Q R X dS )zX Add token usage and content blocks from the AI streaming response to the span. � r; r< TN) r r r �join�set_datar � AI_RESPONSESr � AI_STREAMING)r8 � integrationr2 r3 rE Zcomplete_messager9 r r r �_add_ai_data_to_spanu s �rM c / s� |� d�� � d kr | ||�S d|kr.| ||�S zt|d � W n tk r\ | ||� Y S X tjtjdtjd���� � | ||fV }t |d �}|� d�}t � �� �� tj|� �� tjd� t� rڈ jrڈ� tj|� t|d��r$t� �r � j�r �� tjt|j�� t|�� ��d d d � ntt|d��r~|j�� ��fd d �}� ��fd d �}tt |j��d k�rt|� |_n|� |_n�� dd� ��d d d � W 5 Q R X |S )NrL �messageszAnthropic messages create)�op� descriptionr' �modelFr> � _iteratorc 3 s\ d} d}g }�D ](}t || ||�\} }}|jdkr|V qt�� | ||� ��d d d � d S �Nr Z message_stop�rF r) rM �__exit__�r2 r3 rE r/ �rL Z old_iteratorr8 r r � new_iterator� s( � �z3_sentry_patched_create_common.<locals>.new_iteratorc sf d} d}g }�2 z03 d H W }t || ||�\} }}|jdkr|V q6 t�� | ||� ��d d d � d S rS rT rV rW r r �new_iterator_async� s( � �z9_sentry_patched_create_common.<locals>.new_iterator_asyncz<class 'async_generator'>Zunknown_responseT) �pop�iter� TypeErrorr+ Z start_spanr ZANTHROPIC_MESSAGES_CREATEr r' � __enter__�list�getr rI r Z AI_MODEL_IDrK r r ZAI_INPUT_MESSAGESr4 rJ rA r> r: rU rR �strr) )�f�args�kwargsr7 rN rQ rX rY r rW r � _sentry_patched_create_common� sL � rd c s" dd� � t ��� �fdd��}|S )Nc _ s� t | f|�|�}zt|�\} }}W n, tk rN } z|j W Y �S d }~X Y nX zLz| ||�}W n0 tk r� } zt|� |d �W 5 d }~X Y nX |�|�W S tk r� } z|j W Y �S d }~X Y nX d S r �rd �next� StopIteration�value� Exceptionr0 �send�ra rb rc �gen�er7 r. r r r � _execute_sync� s z+_wrap_message_create.<locals>._execute_syncc s& t �� �t�}||d< � �f| �|�S �NrL �r+ r, Zget_integrationr �rb rc rL �rn ra r r �_sentry_patched_create_sync� s z9_wrap_message_create.<locals>._sentry_patched_create_syncr )ra rs r rr r r � s r c s" dd� � t ��� �fdd��}|S )Nc � s� t | f|�|�}zt|�\} }}W n2 tk rT } z|jI d H W Y �S d }~X Y nX zRz| ||�I d H }W n0 tk r� } zt|� |d �W 5 d }~X Y nX |�|�W S tk r� } z|j W Y �S d }~X Y nX d S r re rk r r r �_execute_async s " z2_wrap_message_create_async.<locals>._execute_asyncc � s, t �� �t�}||d< � �f| �|�I d H S ro rp rq �rt ra r r � _sentry_patched_create_async s z@_wrap_message_create_async.<locals>._sentry_patched_create_asyncr )ra rv r ru r r s r )'� functoolsr �typingr r+ Zsentry_sdk.ai.monitoringr Zsentry_sdk.constsr r Zsentry_sdk.integrationsr r Zsentry_sdk.scoper Zsentry_sdk.utilsr r r Zanthropic.resourcesr r Zanthropic.typesr � ImportErrorr r r Zsentry_sdk.tracingr r r0 r: rA rF rM rd r r r r r r �<module> s2 \!