o iR@sddlZddlZddlmZddlmZddlmZddlZddl m Z ddl m Z m Z mZmZmZmZddlmZmZmZddlmZmZmZdd lmZdd lmZdd lmZm Z m!Z!m"Z"m#Z#z 5 6 7 - 8   dUd9d:Z? ;dVd5d6d8dd$d%dd=d?d=d@d=d'dAdBe@ddfdCdDZAdWdGdHZBdXdIdJZCdXdKdLZDdMdde@fdNdOZEdS)YN)Iterablewraps) TYPE_CHECKING)record_token_usage)GEN_AI_ALLOWED_MESSAGE_ROLESset_data_normalizednormalize_message_rolestruncate_and_annotate_messagesget_start_span_function transform_anthropic_content_part)OPSPANDATA SPANSTATUS)_check_minimum_version DidNotEnable Integration)should_send_default_pii)set_span_errored)capture_internal_exceptionsevent_from_exceptionpackage_versionsafe_serializereraise)NotGiven)Omit) AsyncMessagesMessages)MessageStreamEventTextBlockParamzAnthropic not installed)Any AsyncIteratorIteratorListOptionalUnion)Span)TextPartc@s>eZdZUdZeed<dZeed<dZded<dZded<dS)_RecordedUsager output_tokens input_tokensz Optional[int]cache_write_input_tokenscache_read_input_tokensN) __name__ __module__ __qualname__r)int__annotations__r*r+r,r2r2^/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/sentry_sdk/integrations/anthropic.pyr(5s   r(c@s@eZdZdZdeZd dddeddfdd Zed d d ZdS)AnthropicIntegration anthropiczauto.ai.Tselfinclude_promptsreturnNcCs ||_dSN)r7)r6r7r2r2r3__init__@s zAnthropicIntegration.__init__cCs.td}tt|ttjt_ttjt_dS)Nr5)rrr4_wrap_message_creatercreate_wrap_message_create_asyncr)versionr2r2r3 setup_onceCs  zAnthropicIntegration.setup_once)T)r8N) r-r.r/ identifieroriginboolr: staticmethodr?r2r2r2r3r4<s  r4excr r8cCs6tt|tjdddd\}}tj||ddS)Nr5F)typehandled)client_options mechanism)hint)rr sentry_sdk get_clientoptions capture_event)rDeventrIr2r2r3_capture_exceptionLs rOresultrtuple[int, int, int, int]cCsd}d}d}d}t|drH|j}t|drt|jtr|j}t|dr,t|jtr,|j}t|dr:t|jtr:|j}t|drHt|jtrH|j}||||fS)z Get token usage from the Anthropic response. Returns: (input_tokens, output_tokens, cache_read_input_tokens, cache_write_input_tokens) rusager*r)r,cache_creation_input_tokens)hasattrrR isinstancer*r0r)r,rS)rPr*r)r,r+rRr2r2r3_get_token_usageWs.   rVrNrmodel str | NonerRcontent_blocks list[str],tuple[str | None, _RecordedUsage, list[str]]cCsvtt|dr|jdkrn'|jdkr0t|jdr"||jjnt|jdr/||jjn|jdkr6 |jdkrg|jjp@|}|jj }|j |_ |j |_ t |dd |_ t |d d |_|||fWd S|jd kr|j j |_ t |j d d }|d ur||_ t |j dd }|d ur||_ t |j d d }|d ur||_|||fWd SWd n1swY|||fS) zl Collect model information, token usage, and collect content blocks from the AI streaming response. rEcontent_block_startcontent_block_deltatext partial_jsoncontent_block_stop message_startrSNr, message_deltar*)rrTrEdeltaappendr^r_messagerWrRr)r*getattrr+r,)rNrWrRrYincoming_usager*rSr,r2r2r3_collect_ai_datawsl         # ?rh content_blockdict[str, Any]cCsZ|ddkr!|d}t|tr!|ddkr!d|dddSt|}|dur+|S|S) z Transform an Anthropic content block using the Anthropic-specific transformer, with special handling for Anthropic's text-type documents. rEdocumentsourcer^datarEr^N)getrUdictr )rirlrPr2r2r3"_transform_anthropic_content_blocks  rrsystem_instructions$Union[str, Iterable[TextBlockParam]]list[TextPart]cCs$t|tr d|dgSdd|DS)Nr^rEcontentcSs,g|]}t|trd|vrd|ddqS)r^rv)rUrq).0 instructionr2r2r3 sz2_transform_system_instructions..)rUstr)rsr2r2r3_transform_system_instructionss r|spanr&kwargs integrationcCs8t|tjd|d}|d}|durt|dkrtr|jrt|ts+t|t r7| tj t t|g}|D]j}|dtjkrd|vrt|dttfrg}|dD]}t|trg|dd krgqX|t|trrt|n|qX|r||d|d q;|} d| vr| d} t| ttfrd d | D| d<|| q;t|} t} t| || } | durt|tj| d dt|tj|dd tjtjtjtj tj!d}|"D]\}}||}|durt#|rt|||q|d}|durt#|rt|dkrt|tj$t%|dSdSdSdS)zq Set input data for the span based on the provided keyword arguments for the anthropic message creation. chatsystemmessagesNrrolerwrE tool_result)rrwcSs"g|] }t|tr t|n|qSr2)rUrqrr)rxitemr2r2r3rz)s  z#_set_input_data..Funpackstream) max_tokensrW temperaturetop_ktop_ptools)&rrGEN_AI_OPERATION_NAMErplenrr7rUr{rset_dataGEN_AI_SYSTEM_INSTRUCTIONSjsondumpsr|rUSERlisttuplerqrdrrcopyr rJget_current_scoper GEN_AI_REQUEST_MESSAGESGEN_AI_RESPONSE_STREAMINGGEN_AI_REQUEST_MAX_TOKENSGEN_AI_REQUEST_MODELGEN_AI_REQUEST_TEMPERATUREGEN_AI_REQUEST_TOP_KGEN_AI_REQUEST_TOP_Pitems _is_givenGEN_AI_REQUEST_AVAILABLE_TOOLSr)r}r~rrsrnormalized_messagesretransformed_contentrtransformed_messagerwrole_normalized_messagesscope messages_datakwargs_keys_to_attributeskey attributevaluerr2r2r3_set_input_datas            "  rFr*z int | Noner)r,r+z list[Any] finish_spanc Cs|tj|trX|jrXggd} |D]} | ddkr&| d| dq| ddkr3| d| qt| ddkrGt|tj| ddd t| ddkrXt|tj | dt |||||d |rl| d d d d Sd S) z; Set output data for the span based on the AI response.)responsetoolrEr^rtool_userrFr)r*r)input_tokens_cachedinput_tokens_cache_writeN) rrGEN_AI_RESPONSE_MODELrr7rdrrGEN_AI_RESPONSE_TOOL_CALLSGEN_AI_RESPONSE_TEXTr__exit__) r}rrWr*r)r,r+rYroutput_messagesoutputr2r2r3_set_output_dataTs@    rfargsc /s$|ddur||i|Sd|vr||i|Szt|dWnty4||i|YSw|dd}ttjd|tj d t ||||fV}t t |drt|\}}}}g} |jD]*} t | dr| | qpt | d r| | qpt | d r| d | jd qptt|dd||||| d d nBt |dr|jdfdd } dfdd } tt|jdkr| |_n*| |_ndd dddWd|SWd|SWd|SWd|S1s wY|S)NrrrWrnzchat )opnamerArwto_dict model_dumpr^roT r}rrWr*r)r,r+rYr _iteratorr8Iterator[MessageStreamEvent]c 3shd}t}g}D]}t||||\}}}|Vq t||j|j|j|jd|ddgdd dSNrnr^)r^rETrr(rhrr*r)r,r+joinrWrRrYrNr old_iteratorr}r2r3 new_iterators6 z3_sentry_patched_create_common..new_iterator!AsyncIterator[MessageStreamEvent]c srd}t}g}2z3dHW}t||||\}}}|Vq 6t||j|j|j|jd|ddgdd dSrrrrr2r3new_iterator_asyncs8  z9_sentry_patched_create_common..new_iterator_asynczunknown_response)r8r)r8r)popiter TypeErrorrpr r GEN_AI_CHATstripr4rA __enter__rrrTrVrwrdrrr^rrfrr{rErr) rrr~rWrPr*r)r,r+rYrirrr2rr3_sentry_patched_create_commons                 g g g ggrc&d ddtd fdd }|S) Nrr rr~r8c _st|g|Ri|}z t|\}}}Wnty)}z |jWYd}~Sd}~wwzAz ||i|}Wn1tye}z%t}t t|Wdn1sRwYt |WYd}~nd}~ww| |WSty}z |jWYd}~Sd}~wwr9 rnext StopIterationr Exceptionsysexc_inforrOrsendrrr~generPrDrr2r2r3 _execute_sync s.  z+_wrap_message_create.._execute_synccstt}||d<z6g|Ri|Wt}|dur@|jtjkrAt| dddWdS1s:wYSSSt}|durj|jtjkrjt| dddWdw1sewYwNr rJrKget_integrationr4get_current_spanstatusrINTERNAL_ERRORrrrr~rr}rrr2r3_sentry_patched_create_sync!sz9_wrap_message_create.._sentry_patched_create_syncrr rr r~r r8r rr r~r r8r r)rrr2rr3r;   r;cr) Nrr rr~r8c st|g|Ri|}z t|\}}}Wnty-}z |jIdHWYd}~Sd}~wwzDz ||i|IdH}Wn1tyl}z%t}t t|Wdn1sYwYt |WYd}~nd}~ww| |WSty}z |jWYd}~Sd}~wwr9rrr2r2r3_execute_async2s0  z2_wrap_message_create_async.._execute_asynccstt}||d<z9g|Ri|IdHWt}|durD|jtjkrEt| dddWdS1s>wYSSSt}|durn|jtjkrnt| dddWdw1siwYwrrrrrr2r3_sentry_patched_create_asyncGs z@_wrap_message_create_async.._sentry_patched_create_asyncrrr)rrr2rr3r=1rr=objcCs0tdur t|tr dStdurt|trdSdS)zI Check for givenness safely across different anthropic versions. NFT)rrUr)rr2r2r3rWs r)rDr r8N)rPrr8rQ) rNrrWrXrRr(rYrZr8r[)rirjr8rj)rsrtr8ru)r}r&r~rjrr4r8N)Fr)rr r8r )Frrcollections.abcr functoolsrtypingrrJsentry_sdk.ai.monitoringrsentry_sdk.ai.utilsrrr r r r sentry_sdk.constsr rrsentry_sdk.integrationsrrrsentry_sdk.scopersentry_sdk.tracing_utilsrsentry_sdk.utilsrrrrrr5r ImportErrorranthropic.resourcesrranthropic.typesrrr r!r"r#r$r%sentry_sdk.tracingr&sentry_sdk._typesr'r(r4rOrVrhrrr|rrBrrr;r=rr2r2r2r3s                 N   n  3  &&