tni<:ddlmZddlmZddlmZddlZddlmZddl m Z m Z m Z m Z mZddlmZmZmZddlmZmZmZdd lmZdd lmZdd lmZmZmZmZ dd l m!Z!n #e"$rdZ!YnwxYw dd l m#Z#n #e"$rdZ#YnwxYwddl$m%Z%m&Z&erddl'm(Z(n#e"$r edwxYwerddlm)Z)m*Z*m+Z+m,Z,m-Z-m.Z.ddl/m0Z0GddeZ1d9dZ2d:dZ3dddd d!e4d"e4d#d$dd%f d&Z5 d;d,Z6 dd5Z:d>d6Z;d7dde7fd8ZeventrDs r-_capture_exceptionrJ@sa& !,..6&599KE4 U......r/resultrtuple[int, int]cd}d}t|dri|j}t|dr!t|jtr|j}t|dr!t|jtr|j}||fS)z6 Get token usage from the Anthropic response. rusage input_tokens output_tokens)hasattrrN isinstancerOintrP)rKrOrPrNs r-_get_token_usagerTKsLMvw0  5. ) ) .j9KS.Q.Q . -L 5/ * * 0z%:Ms/S/S 0!/M  &&r/rIrmodelz str | NonerOrPcontent_blocksz list[str]z&tuple[str | None, int, int, list[str]]c<t5t|dr|jdkr/|jj}||jz }||jz }|jjp|}n|jdkrn|jdkrjt|jdr | |jj n[t|jdr| |jj n&|jdkrn|jdkr||jjz }d d d n #1swxYwY||||fS) zl Collect model information, token usage, and collect content blocks from the AI streaming response. r@ message_startcontent_block_startcontent_block_deltatext partial_jsoncontent_block_stop message_deltaN) rrQr@messagerNrOrPrUdeltaappendr[r\)rIrUrOrPrVrNs r-_collect_ai_datarb[sv % & &;; 5& ! ! ;z_,, + 22 !44  +4u4444445;//D"))%+*:;;;;U[.99D"))%+*BCCC333..!:: #;;;;;;;;;;;;;;;& , ~ ==sC2D  DDspanr$kwargsdict[str, Any] integrationc t|tjd|d}|d}|t |dkrt r|jrg}|rd}t|tr|}nt|trqg}|D]l}t|trU|ddkr<|dr'| | m|r"| tj|d|D]}|d tjkrd |vrt|d t t"frp|d D]f}|dd krK| tj|d |d d dg| |t'|} t)j} t-| || } | t|tj| dt|tj|ddtjtjtjtjtjd} | D]<\} }|| }| t?|rt|||=|d}|Lt?|r?t |dkr.t|tj tC|dSdSdSdS)zq Set input data for the span based on the provided keyword arguments for the anthropic message creation. chatsystemmessagesNrr@r[)rolecontentrkrl tool_result tool_use_id)rnoutputFunpackstream) max_tokensrU temperaturetop_ktop_ptools)"r rGEN_AI_OPERATION_NAMEgetlenrr)rRstrrdictracopyrSYSTEMUSERlisttupleTOOLr rEget_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)rcrdrf system_promptrjnormalized_messagessystem_prompt_contentitemr_role_normalized_messagesscope messages_datakwargs_keys_to_attributeskey attributevaluerws r-_set_input_datar{s h%;P;P     ..r/Fz int | Nonez list[Any] finish_spanct|tj|tr|jrggd}|D]W}|ddkr"|d|d0|ddkr|d|Xt |ddkr#t|tj|dd t |ddkr!t|tj |dt||| |r| d d d d Sd S) z; Set output data for the span based on the AI response.)responsetoolr@r[rtool_userrFrp)rOrPN) set_datarGEN_AI_RESPONSE_MODELrr)rarzr GEN_AI_RESPONSE_TOOL_CALLSGEN_AI_RESPONSE_TEXTr__exit__) rcrfrUrOrPrVroutput_messagesros r-_set_output_datars~ MM(0%888  [%@3 3  % 7 7Ff~'' +226&>BBBB:--'..v666 v& ' '! + + 3'      z* + +a / / h3_Z5P    !# ( dD$'''''((r/fargsc /. K|d ||i|Sd|vr||i|S t|dn#t$r ||i|cYSwxYw|dd}t t jd|tj  t | |||fV}t5t|drt|\}}g}|jD]}t|dr(||:t|d r(||rt|d r|d |jd t) t+|dd|||d nt|dr\|j d fd } d fd } t/t1|jdkr| |_n=| |_n- dd  ddddddn #1swxYwY|S)NrfrjrUzchat )opnamer:rlto_dict model_dumpr[)r@r[TrcrfrUrOrPrVr _iteratorr*Iterator[MessageStreamEvent]c 3Kd}d}d}g}D]}t|||||\}}}}|Vt|||d|ddgddSNrrr[)r[r@TrrbrjoinrUrOrPrVrIrf old_iteratorrcs r- new_iteratorz3_sentry_patched_create_common..new_iterator<s ! .0)  E(!5, ~GE<  KKKK  +!-"/-/WW^-D-Df$U$U#V $r/!AsyncIterator[MessageStreamEvent]c Kd}d}d}g}23d{V}t|||||\}}}}|WV%6t|||d|ddgddSrrrs r-new_iterator_asyncz9_sentry_patched_create_common..new_iterator_asyncTs ! .0#/       %(!5, ~GE<  KKKKK $0! +!-"/-/WW^-D-Df$U$U#V $s3zunknown_response)r*r)r*r)popiter TypeErrorryr r GEN_AI_CHATstripr&r: __enter__rrrQrTrlrarrr[rgetattrrr{r@rr)rrrdrUrKrOrPrV content_blockrrrfrrcs @@@r-_sentry_patched_create_commonrs**]++Kq$!&!!!q$!&!!!" VJ     """q$!&!!!!!" JJw # #E $ " $ $ > U__ " " $ $#*   D  NND&+...dF? " "F $ & &R,R, 69 % %Q ,*:6*B*B 'L-N!' X X =)44X"))-*?*?*A*ABBBB]L99X"))-*B*B*D*DEEEE]F33X"))6=CU*V*VWWW 'fgt44)+-      V[ ) ): ,!+L        0        04())**.III#5#5#7#7  #/<>>   MM,d 3 3 3 MM$d + + +eR,R,R,R,R,R,R,R,R,R,R,R,R,R,R,h Ms$AAA3F J  JJcJddtd fd }|S) Nrrrrdr*cLt|g|Ri|} t|\}}}n#t$r}|jcYd}~Sd}~wwxYw ||i|}n$#t$r}t ||dd}~wwxYw||S#t$r}|jcYd}~Sd}~wwxYwr,rnext StopIterationr ExceptionrJsendrrrdgenerKr>s r- _execute_syncz+_wrap_message_create.._execute_syncys+A?????? "3iiOAtVV   7NNNNNN   $D+F++ $ $ $"3'''t# $88F## #   7NNNNNN sS& A<AAAB A0A++A00B B#BB#B#c,tjt}||d< g|Ri|tj}|S|jt jkr?t5| ddddddS#1swxYwYSSS#tj}|S|jt jkr?t5| ddddddw#1swxYwYwwwxYwNrf rErFget_integrationr&get_current_spanstatusrINTERNAL_ERRORrr)rrdrfrcrrs r-_sentry_patched_create_syncz9_wrap_message_create.._sentry_patched_create_syncs +--==>RSS +} 4 =4T444V44.00DDK:3L$L$L02244MM$d3334444444444444444 $L.00DDK:3L$L$L02244MM$d3334444444444444444 $LsA B(6BB !B (9D!D9 DD D D Drrrrrdrr*rrrrdrr*rr)rrrs` @r-r1r1xsU& 1XX 4 4 4 4 4 4X 4 '&r/cJddtd fd }|S) Nrrrrdr*chKt|g|Ri|} t|\}}}n$#t$r}|jd{VcYd}~Sd}~wwxYw ||i|d{V}n$#t$r}t ||dd}~wwxYw||S#t$r}|jcYd}~Sd}~wwxYwr,rrs r-_execute_asyncz2_wrap_message_create_async.._execute_asyncs(+A?????? !"3iiOAtVV ! ! !======  !  $ q$1&11111111 $ $ $"3'''t# $88F## #   7NNNNNN sU( A  AA A AB A>'A99A>>B B1 B,&B1,B1c>Ktjt}||d< g|Ri|d{V tj}|S|jt jkr?t5| ddddddS#1swxYwYSSS#tj}|S|jt jkr?t5| ddddddw#1swxYwYwwwxYwrr)rrdrfrcrrs r-_sentry_patched_create_asyncz@_wrap_message_create_async.._sentry_patched_create_asyncs +--==>RSS +} 4';D;;;F;;;;;;;; ;.00DDK:3L$L$L02244MM$d3334444444444444444 $L.00DDK:3L$L$L02244MM$d3334444444444444444 $LsAB1?B##B' *B' 19D*D DD DD Drrr)rrrs` @r-r3r3sU& 1XX 4 4 4 4 4 4X 4 ('r/objc~tt|trdStt|trdSdS)zI Check for givenness safely across different anthropic versions. NFT)rrRr)rs r-rrs= 3 9 9u JsD11u 4r/)r>rr*N)rKrr*rL)rcr$rdrerfr&r*N)Fr)rrr*r)=collections.abcr functoolsrtypingrrEsentry_sdk.ai.monitoringrsentry_sdk.ai.utilsrr r r r sentry_sdk.constsr rrsentry_sdk.integrationsrrrsentry_sdk.scopersentry_sdk.tracing_utilsrsentry_sdk.utilsrrrrr'r ImportErrorranthropic.resourcesrranthropic.typesrrrr r!r"r#sentry_sdk.tracingr$r&rJrTrSrbrr;rrr1r3rr=r/r-rs)$$$$$$ 7777777666666666UUUUUUUUUU4444445555552&&&&&&& """"""" <;;;;;;;7666666222 ,0 1 112(JJJJJJJJJJJJJJJJ'''''' P P P P P; P P P //// ' ' ' ' > > >> >  > . >>>>@Z Z *Z 9OZ  Z Z Z Z H,(,( ,(',( ,( ,(  ,(  ,(,( ,(,(,(,(^nnnnb!'!'!'!'H!(!(!(!(H5TsNA B A*'B)A**B.A54B5A?<B>A??BB$