jKj6dZddlZddlmZmZmZmZmZddlm Z m Z m Z m Z m Z mZmZddlmZdedee fdZ dd eeeefd eedee fd Zd eeefdeefd Zdee dee fdZdedefdZdedeefdZdedefdZdedefdZdedeee eeffdZdedee deefdZdedee deeefddfdZdedee deeefddfdZd eeefdefdZ dee dedee fdZ!y)z Anthropic-specific conversion utilities. This module handles the conversion of Anthropic API responses and inputs into standardized formats for PostHog tracking. N)AnyDictListOptionalTuple)FormattedContentItemFormattedFunctionCallFormattedMessageFormattedTextContentStreamingContentBlock TokenUsageToolInProgress)serialize_raw_usageresponsereturnc g}||Sg}t|dr|jD]}t|drH|jdk(r9t|dr-|jr!d|jd}|j |Wt|dsd|jdk(stt|dst|dsd|j |j t|d id d }|j ||rd |d }|j ||S)z Format an Anthropic response into standardized message format. Args: response: The response object from Anthropic API Returns: List of formatted messages with role and content contenttypetextrrtool_usenameidfunctioninputr argumentsrrr assistantroler)hasattrrrrappendrrgetattr)routputrchoice text_content function_callmessages y/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/posthog/ai/anthropic/anthropic_converter.pyformat_anthropic_responser+s&(F *,Gx#&&F'KK6)FF+KK#"KK6 |,'KK:-FF+FD)' )) & %,VWb%A!8 }-7':%   g Mmessagessystemcg}||jd|d|r=|D]8}|jdd|jddd}|j|:|S)z Format Anthropic input messages with optional system prompt. Args: messages: List of message dictionaries system: Optional system prompt to prepend Returns: List of formatted messages r.r r!userr)r#get)r-r.formatted_messagesmsg formatted_msgs r*format_anthropic_inputr6Rsq24!!8"GHC/779b1/M  % %m 4  r,kwargsc&|jddS)z Extract tool definitions from Anthropic API kwargs. Args: kwargs: Keyword arguments passed to Anthropic API Returns: Tool definitions if present, None otherwise toolsN)r2)r7s r*extract_anthropic_toolsr:ss ::gt $$r,content_blocksc(g}|D]}|jddk(r(|jd|jdxsdd?|jddk(sT|jd|jd|jdxsid|S)a Format content blocks from Anthropic streaming response. Used by streaming handlers to format accumulated content blocks. Args: content_blocks: List of content block dictionaries from streaming Returns: List of formatted content items rrr1rrrr)r2r#)r; formattedblocks r*"format_anthropic_streaming_contentr?s-/I 99V  &   "!IIf-3 YYv * ,   &))D/ % * 5 ;  $ r,c t|dsyt|jdsy|jj}t|dr tdt t |ddSy)a Extract web search count from Anthropic response. Anthropic provides exact web search counts via usage.server_tool_use.web_search_requests. Args: response: The response from Anthropic API Returns: Number of web search requests (0 if none) usagerserver_tool_useweb_search_requests)r"rArBmaxintr$)rrBs r*"extract_anthropic_web_search_countrFs\ 8W % 8>>#4 5nn44O 561c'/3H!LMNN r,ct|ddS)z,Extract stop reason from Anthropic response. stop_reasonN)r$)rs r*extract_anthropic_stop_reasonrIs 8]D 11r,ct|ds tddStt|jddt|jdd}t|jdr"|jj}|r |dkDr||d<t|jdr"|jj }|r |dkDr||d<t |}|dkDr||d<t|j}|r||d <|S) z Extract usage from a full Anthropic response (non-streaming). Args: response: The complete response from Anthropic API Returns: TokenUsage with standardized usage rAr) input_tokens output_tokensrKrLcache_read_input_tokenscache_creation_input_tokensweb_search_count raw_usage)r"r r$rArMrNrFr)rresult cache_readcache_creationrO serializeds r*%extract_anthropic_usage_from_responserUs 8W %q:: X^^^Q?hnnoqAF x~~89^^;; *q.0:F, -x~~<=!CC nq04BF0 19(C!%5!"%X^^4J({ Mr,eventct}t|dr|jdk(rt|drt|jdrt |jj dd|d<t |jj dd|d<t |jj dd|d<t |jj }|r||d <t|dr|j rt |j d d|d <t|j d rB|j j}t|d r tt |d d}|dkDr||d <t |j }|r||d <|S)z Extract usage statistics from an Anthropic streaming event. Args: event: Streaming event from Anthropic API Returns: Dictionary of usage statistics r message_startr)rArKrrNrMrPrLrBrCrO) r r"rr)r$rArrBrE)rVrArTrBrOs r*"extract_anthropic_usage_from_eventrYsc# Euf%**"? 5) $)H$+EMM,?,?QR$SE. !3: ##%BA4E/ 007 ##%>0E+ , -U]]-@-@AJ%/k"ug5;;!(oq!Io 5;; 1 2#kk99O(=>#&O-BAF$ $a'0@E,-)5 !+E+  Lr,c&t|dr|jdk(syt|dsy|j}t|dsy|jdk(r ddd}|dfS|jd k(r(d t|d dt|d did d}|dd}||fSy)z Handle content block start event from Anthropic streaming. Args: event: Content block start event Returns: Tuple of (content_block, tool_in_progress) rcontent_block_start)NN content_blockrr1rNrrrrrr)r> input_string)r"rr\r$)rVr>r\ tool_blocktool_in_progresss r*$handle_anthropic_content_block_startr`#s E6 "uzz5J'J 5/ *   E 5& ! zzV8>/K d"" z !%r*!(!;"M- 6@QS+T+++ r, current_blockct|drjt|jdrT|jjxsd}|6|jddk(r"|jd}| ||z|d<|S||d<|Sy)z Handle text delta event from Anthropic streaming. Args: event: Delta event current_block: Current text block being accumulated Returns: Text delta if present deltarr1Nr)r"rcrr2)rVra delta_texttext_vals r*handle_anthropic_text_deltarfKsug75;;#?[[%%+  $):):6)Bf)L$((0H#(0:(= f%)3 f% r,tools_in_progressct|dr|jdk(syt|dr/t|jdr|jjdk(syt|dr~|jt |kre||j}|j ddk(rA|j d|vr-||d}t |jd d }|d xx|z cc<yyyyy) z Handle tool input delta event from Anthropic streaming. Args: event: Tool delta event content_blocks: List of content blocks tools_in_progress: Dictionary tracking tools being accumulated rcontent_block_deltaNrcinput_json_deltaindexrr partial_jsonr1r])r"rrcrklenr2r$)rVr;rgr>toolrls r*handle_anthropic_tool_deltarohs E6 "uzz5J'J w EKK ( KK   2 2ug5;;^1D#Du{{+ 99V  *uyyBS/S$U4[1D"5;;CL  L 0 0T *$Er,ct|dr|jdk(syt|dr|jt|krf||j}|j ddk(rB|j d|vr.||d} t j |d|dd<||d=yyyyy#t jtf$rY&wxYw) z Finalize tool input when content block stops. Args: event: Content block stop event content_blocks: List of content blocks tools_in_progress: Dictionary tracking tools being accumulated rcontent_block_stopNrkrrr]r) r"rrkrmr2jsonloadsJSONDecodeError Exception)rVr;rgr>rns r*finalize_anthropic_tool_inputrvs E6 "uzz5I'Iug5;;^1D#Du{{+ 99V  *uyyBS/S$U4[1D 15DJ$1BCDD $%+5HIJ   r,)N)"__doc__rrtypingrrrrrposthog.ai.typesrr r r r r rrzrr+strr6r:r?rErFrIrUrYr`rfrorvr{rr,r*rs 331885E0F8x=A4S>",4SM B %DcN %x} %#./# #L   42C2HSM2 &C&J&R1c1j1h% % 8) *H^,D DE%P '(=> c]:1 1./1C/01 1B/ /.//C/0/ /@ 4T#s(^ 4 4 ./ FI   r,