jKjyj dZddlmZmZmZmZddlmZmZm Z m Z m Z m Z ddl mZdedee fdZ dd eeeeefd eedee fd Zd eeefdeefd Z d dedeeeeefdeefdZdedefdZdedeefdZdede fdZ d!dedede fdZ d!dededeefdZdedeeeeeffdZdeeeeeffdeeeefddfdZ d"dededeeeeefdee fdZ d!d eeefdedefdZy)#z OpenAI-specific conversion utilities. This module handles the conversion of OpenAI API responses and inputs into standardized formats for PostHog tracking. It supports both Chat Completions API and Responses API formats. )AnyDictListOptional)FormattedContentItemFormattedFunctionCallFormattedImageContentFormattedMessageFormattedTextContent TokenUsage)serialize_raw_usageresponsereturnc Lg}||St|drg}d}|jD]}t|ds|js|jjr|jj}|jjr(|j d|jjdt|jdr{|jj re|jj D]L}|j d|j|jj|jjdd Nt|jd s1|jjsI|jjj}|j d d i||r|j ||d t|d rg}d}|jD]}|jdk(r9|j}t|drt|jt r|jD]}t|d r:|jdk(r+t|dr|j d|j"dIt|dr|j d|j"dtt|d s|jdk(st|dsd|j$d} |j | t|ds"dt'|jd} |j | Mt|d s[|jdk(sl|j dt)|dt)|dd|jt)|didd |r|j ||d |S)a Format an OpenAI response into standardized message format. Handles both Chat Completions API and Responses API formats. Args: response: The response object from OpenAI API Returns: List of formatted messages with role and content choices assistantmessagetexttyper tool_callsfunctionname argumentsridraudiorrolecontentoutputr! output_text input_image image_urlimage)rr& function_callcall_idrr)hasattrrrr r!appendrrrrrr model_dumpr"r isinstancelistrr%strgetattr) rr"r!r choice tool_call audio_dictitem content_item image_content text_contents s/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/posthog/ai/openai/openai_converter.pyformat_openai_responser9s&(F x#.0&&Fvy)fnn>>&&!>>..D>>))NN$*$*NN$:$:6>><8V^^=V=V%+^^%>%> (2&/ll,5,>,>,C,C1:1C1C1M1M-" &?6>>738L8L!'!5!5!@!@!BJNNFG#Bz#BC='@  MM & x"OODyyI%yy4+ 4<<0N(, #L&9 , 1 1] B ' f =#NN,2,8,=,=!"%\6:#NNFLDUDU+VW$L&9 , 1 1] B ' k B)0)5)?)?DM$NN=93)56T9-,2C S&T >> - -x#&&Fvy)gfnnm.T$nn88 &1  **d;'NN62!(VT!B( +n<#$'2vw'GFLL-,P$ll66 &1  **d;'NN62!(VT!B( +n<#$'2-'Dx"OODtY'Jt||T,J$(LLL|];&2&>&> &.9 (2*d'C%/NN6$:)0VT)J!0 $3n#D+,/: %1$$ r:ct|dr&|jrt|jdddSt|dr t|ddSy)z)Extract stop reason from OpenAI response.rr finish_reasonNstatus)r*rr0)rs r8extract_openai_stop_reasonratsLx#(8(8x''*OTBBx"x400 r:ct|ds tddSd}d}d}d}t|jdr|jj}t|jdr|jj}t|jdr@t|jj dr |jj j }t|jdr@t|jjd r |jjj}t|jd r|jj}t|jd r|jj}t|jd r@t|jjdr |jjj }t|jd r@t|jjd r |jjj}t||}| |dkDr||d<| |dkDr||d <t|}|dkDr||d<t|j}|r||d<|S)a  Extract usage statistics from a full OpenAI response (non-streaming). Handles both Chat Completions and Responses API. Args: response: The complete response from OpenAI API Returns: TokenUsage with standardized usage statistics rSr) input_tokens output_tokensrcrdinput_tokens_details cached_tokensoutput_tokens_detailsreasoning_tokens prompt_tokenscompletion_tokensprompt_tokens_detailscompletion_tokens_detailscache_read_input_tokensrZ raw_usage)r*r rSrcrdrerfrgrhrirjrkrlr]r )rrfrcrdrhresultrZ serializeds r8"extract_openai_usage_from_responserqs! 8W %q::MLMx~~~.~~22 x~~/ 44 x~~567++_<!;;II x~~67G,,.@=$>>??PPx~~/~~33 x~~23 88 x~~67G,,o=!<<JJ x~~:;002DA$>>CCTT !#F  ]Q%6,9()#(81(<%5!"6x@!%5!"%X^^4J({ Mr:chunk provider_typec8t}|dk(r;t|}|dkDr||d<t|dr |js|St |jdd|d<t |jdd|d<t|jd rGt|jj d r'|jj j }|||d <t|jd rGt|jjd r'|jjj}|||d <t|j}|r||d<|S|dk(rHt|dr;|jdk(r+t|drt|jdr|jjr|jj}t |dd|d<t |dd|d<t|dr3t|jd r|jj }|||d <t|dr3t|jd r|jj}|||d <t|drt|j}|dkDr||d<t|}|r||d<|S)aC Extract usage statistics from an OpenAI streaming chunk. Handles both Chat Completions and Responses API formats. Args: chunk: Streaming chunk from OpenAI API provider_type: Either "chat" or "responses" to handle different API formats Returns: Dictionary of usage statistics chatrrZrSrircrjrdrkrfrmrlrhrn responsesrresponse.completedrrerg)r r]r*rSr0rkrfrlrhr rrrerg)rrrsrSrZcached reasoningrpresponse_usages r8extract_openai_usage_from_chunkr{s # E;5A a (8E$ %ug&ekkL!( _a Hn!(6I1!Mo 5;; 7 8W KK - -> [[66DDF!39/0 5;; ; < KK 1 13EB  ==NNI$,5())5 !+E+ T LQ + % 5& !ejj4H&Hz*ENNG4NN((!&!5!5(/PQ(Rn%)0RS)To&>+ABw"77H,@@NNF);A78>+BC"88:LI!/ D D U UI ,4=015*-'Fu~~'V$'!+4D011@ )3E+& Lr:c|dk(rt|dr|jrwt|jdkDr_|jdjrF|jdjjr#|jdjjSy|dk(rrt|drf|j dk(rWt|drK|j r?|j }|jr't|jdkDr|jdSy) aC Extract content from an OpenAI streaming chunk. Handles both Chat Completions and Responses API formats. Args: chunk: Streaming chunk from OpenAI API provider_type: Either "chat" or "responses" to handle different API formats Returns: Text content if present, None otherwise rurrrvrrwrN)r*rrYrWr!rrr")rrrsress r8!extract_openai_content_from_chunkr~(s  E9 % EMM"Q& a && a &&..==#))11 1  + % 5& !ejj4H&Huj)ennnn::#cjj/A"5::a=( r:ct|dr|jrt|jdkDr|jdjrt|jdjdri|jdjjrEg}|jdjjD]}dt |ddi}t|dr|j r|j |d<t|dr|jr|j|d<t|dr|jri}t|jd r/|jjr|jj|d <t|jd r/|jjr|jj|d <||d<|j||Sy) z Extract tool calls from an OpenAI streaming chunk. Args: chunk: Streaming chunk from OpenAI API Returns: List of tool call deltas if present, None otherwise rrrindexNrrrrr) r*rrYrWrr0rrrrrr+)rrrr2tc_dict function_dicts r8$extract_openai_tool_calls_from_chunkrOs y! MM   " MM!  " " EMM!$**L 9 MM!  " " - - q)//::IGT:Gy$'ILL )  y&)inn"+..y*-)2D2D " 9--v69;M;M;R;R,5,>,>,C,CM&)I.. <!**441:1C1C1M1MM+.&3 #   g &-;. r:accumulated_tool_callschunk_tool_callsc |D]}|jd}|||vr dddddd||<||}d|vr |dr|d|d<d|vr |dr|d|d<d|vsS|d}d |vr|d r |d |dd <d |vsq|d sw|dd xx|d z cc<y) a Accumulate tool calls from streaming chunks. OpenAI sends tool calls incrementally: - First chunk has id, type, function.name and partial function.arguments - Subsequent chunks have more function.arguments Args: accumulated_tool_calls: Dictionary mapping index to accumulated tool call data chunk_tool_calls: List of tool call deltas from current chunk rNr)rr)rrrrrrr)r?)rrtool_call_deltartc func_deltas r8accumulate_openai_tool_callsr~s,##G, =  . ."!#- "5 )$E * ? "t'<&t,BtH _ $)@(0BvJ  ((4J# 6(:)3F);:v&j(Z -D:{+z+/FF+?,r:c|dk(rg}t|tr|r|jd|dn>t|tr.dj d|D}|r|jd|d|r6|D]1}d|vsd|j dd|dd}|j|3|rd |d gSd gd gS|d k(r.t|tr|r|St|tr d d|dgd gSd dt|dgd gS) aj Format the final output from OpenAI streaming. Args: accumulated_content: Accumulated content from streaming (string for chat, list for responses) provider_type: Either "chat" or "responses" to handle different API formats tool_calls: Optional list of accumulated tool calls Returns: List of formatted messages rurrr)c38K|]}|st|ywN)r/).0r4s r8 z1format_openai_streaming_output..sM1D3t91Dsrrrrrrv)r-r/r+r.joinr?)rJrsr content_itemsrr2r's r8format_openai_streaming_outputrsA"46  )3 /4G  &:M!N O +T 277M1DMMD$$fd%CD ' * *'mmD"5$-j$9<M "((7( (]CD D)R89 9 + % )4 05H& & +S 1()/9L MN  !'5H1IJK  r:api_typec ddlm}||dS)z Format OpenAI streaming input based on API type. Args: kwargs: Keyword arguments passed to OpenAI API api_type: Either "chat" or "responses" Returns: Formatted input ready for PostHog tracking r)merge_system_promptopenai)posthog.ai.utilsr)rDrrs r8format_openai_streaming_inputrs5 vx 00r:)NNr)ru)ruN)__doc__typingrrrrposthog.ai.typesrrr r r r rr r9r/rCrIrMintr]rarqr{r~rrrrrHr:r8rsQ-,1|S|T2B-C|@RV5tDcN+,5AI#5 5pc3hHSM.LP##*24S#X3G*H# #Lh ch ch V#CCCN&,` `"``H&,$ $"$ c]$N,,d3PS8nAU8V,^.G d38n!45.G4S>*.G .Gf 15BBBd38n-.B  BL-31 cN1&)11r: