jKj)j ddlZddlZddlmZmZmZmZddlmZddl m Z  ddl Z ddlmZddlmZmZmZmZmZddlmZmZmZmZmZdd lmZmZdd lm Z!dd l"m#Z#Gd d e jHZ$dee%dee%deee%efde&deee%efdef dZ'Gdde#Z(Gdde#Z)Gdde#Z*Gdde#Z+Gdde#Z,Gdd e#Z-Gd!d"e#Z.y#e $r e dwxYw)#N)AnyDictListOptional)AsyncStreamWrapper) TokenUsagezGPlease install the OpenAI SDK to use this feature: 'pip install openai')setup)call_llm_and_track_usage_asyncextract_available_tool_callsget_model_paramsmerge_usage_statswith_privacy_mode)extract_openai_usage_from_chunk!extract_openai_content_from_chunk$extract_openai_tool_calls_from_chunkaccumulate_openai_tool_callsformat_openai_streaming_output)sanitize_openaisanitize_openai_response)Client)_OpenAIWrapperResourcec<eZdZUdZeed<ddeeffd ZxZS) AsyncOpenAIzf An async wrapper around the OpenAI SDK that automatically sends LLM usage events to PostHog. _ph_clientposthog_clientc t|di||xs t|_t |dd|_t |dd|_t |dd|_t |dd|_|j t||j |_ |j t||j |_ |jt||j|_|jt||j|_yy)a Args: posthog_client: If provided, events will be captured via this client instead of the global ``posthog`` client. **kwargs: Arguments passed to ``openai.AsyncOpenAI`` such as ``api_key`` or ``organization``. chatN embeddingsbeta responses)super__init__r rgetattr_original_chat_original_embeddings_original_beta_original_responses WrappedChatrWrappedEmbeddingsr WrappedBetarWrappedResponsesr )selfrkwargs __class__s o/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/posthog/ai/openai/openai_async.pyr#zAsyncOpenAI.__init__*s "6"(3EG&dFD9$+D,$E!%dFD9#*4d#C     *#D$*=*=>DI  $ $ 0/d6O6OPDO    *#D$*=*=>DI  # # /-dD4L4LMDN 0)N) __name__ __module__ __qualname____doc__ PostHogClient__annotations__rr# __classcell__)r/s@r0rr#s)Nx '>NNr1rposthog_distinct_idposthog_trace_idposthog_propertiesposthog_privacy_modeposthog_groupsr.c Kt||jjd|||||jj|jj f i|d{S7w)Nopenai)r _clientrbase_url _originalparse)wrapperr9r:r;r<r=r.s r0_parse_and_trackrEJsd0""        sAAAAceZdZdZ ddeedeedeeeefdedeeeefdef d Z deedeedeeeefdedeeeefdef d Z ddeedeedeeeefdedeeeefdeeefd e d e d edee eeefdeedeefdZ ddeedeedeeeefdedeeeefdef dZy)r,z@Async wrapper for OpenAI responses that tracks usage in PostHog.Nr9r:r;r<r=r.c bK|ttj}|jddr|j|||||fi|d{St ||j jd|||||j j|jjf i|d{S7\7w)a Create an OpenAI Responses API response while tracking usage in PostHog. Args: posthog_distinct_id: Optional distinct ID to associate with the usage event. posthog_trace_id: Optional trace ID. Generated automatically when omitted. posthog_properties: Additional properties to include with the usage event. posthog_privacy_mode: Whether to redact captured input and output. posthog_groups: Optional PostHog groups to associate with the event. **kwargs: Arguments passed to OpenAI's async ``responses.create`` API. Returns: The OpenAI response, or an async streaming iterator when ``stream=True``. NstreamFr? struuiduuid4get_create_streamingr r@rrArBcreater-r9r:r;r<r=r.s r0rOzWrappedResponses.createds.  #"4::<0  ::h &///# "$    4  LL # #     LL ! ! NN ! !       s%A B/B+AB/&B-'B/-B/c  Ktj t gd d jjdid{   f d}t | S7'w)Nc  K 23d{}t|dr:|jr. ,t|jdr|jj t|d}|r t |t |d}|j |t|drB|jdk(r3t|dr'|jrt|jdd}|||76 tj}|z }}j  ||td  d{7y#tj}|z }}j  ||td  d{7wxYww) Nresponsemodelr typezresponse.completedstatusr? stop_reason) hasattrrSrTrr rappendrUr$time_capture_streaming_eventr )chunk chunk_usagecontent chunk_statusend_timelatencyoutput final_contentr.model_from_responser9r=r<r;r:rSr- start_timerX usage_statss r0async_generatorz;WrappedResponses._create_streaming..async_generators 4 #+ %uj1enn.67!NNG<38..2F2F/#B%"UK")+{C@{SG*%,,W5 v.!JJ*>>#E:6!NN'.u~~x'N '3*6KK? 8D 99;"Z/&33'$&("06B' +4     99;"Z/&33'$&("06B' +4   sWFD3C%C#C%CD3#C%%D3'AF,D/-F3AF9E<:FFrHr!r[rrBrOr)r-r9r:r;r<r=r.rhrdrerSrfrXrgs``````` @@@@@@r0rNz"WrappedResponses._create_streamingspYY[ ",,  -1%) ...888: : : x"/"3HEE}9sAA=A;(A=rgrbrcavailable_tool_callsrerXc TK|ttj}|jdxs| xsd} d| t |t |j j|t|jdt |j j|t| dd|jdd|jd d|jd d|jd d||t|j jd |xsi}|jd }|t|tr |dkDr||d<| | |d<| r| |d<|d|d<t|j jdr.|j jj|xs|d||yyw)NrTunknownr?inputr input_tokensr output_tokenscache_read_input_tokensreasoning_tokens $ai_provider $ai_modelz$ai_model_parameters $ai_inputz$ai_output_choices$ai_http_status$ai_input_tokensz$ai_output_tokensz$ai_cache_read_input_tokensz$ai_reasoning_tokens $ai_latency $ai_trace_id $ai_base_urlweb_search_count$ai_web_search_count$ai_stop_reason $ai_toolsF$process_person_profilecapture$ai_generation distinct_idevent propertiesgroups)rJrKrLrMr rr@rrrrA isinstanceintrYr)r-r9r:r;r<r=r.rgrbrcrkrerXrTevent_propertiesr}s r0r\z)WrappedResponses._capture_streaming_events  #"4::<0  7#G':Gi%$4V$<* ''$(G)<= #4 ''$.v{C# # + B!,!!D+6??)1,%0OO4F$J", 5 56/ 0"'R1 8'??+=>  (+S1 1$7G 3 4  "2= . / ,@ [ )  &:? 6 7 4<<**I 6 LL # # + +/C3C&+% ,  7sF&F(c@Kt||||||fi|d{S7w)a Parse structured output using OpenAI's 'responses.parse' method, but also track usage in PostHog. Args: posthog_distinct_id: Optional ID to associate with the usage event. posthog_trace_id: Optional trace UUID for linking events. posthog_properties: Optional dictionary of extra properties to include in the event. posthog_privacy_mode: Whether to anonymize the input and output. posthog_groups: Optional dictionary of groups to associate with the event. **kwargs: Any additional parameters for the OpenAI Responses Parse API. Returns: The response from OpenAI's responses.parse call. NrErPs r0rCzWrappedResponses.parse-;.&           NNNFN)NNN)r2r3r4r5rrJrrboolrOrNrfloatrr\rCr!r1r0r,r,as4J.2*.7;%*37 / %c]/ #3-/ %T#s(^4 / # / !c3h0 / / bLF%c]LF#3-LF%T#s(^4 LF # LF !c3h0 LFLFr@D-1%)H%c]H#3-H%T#s(^4 H # H !c3h0 HS#XH HHH'tDcN';<H&c]Hc]HX.2*.7;%*37  %c] #3- %T#s(^4  #  !c3h0   r1r,c eZdZdZedZy)r)z;Async wrapper for OpenAI chat that tracks usage in PostHog.cVt|j|jjS)z:Access async chat completions with PostHog usage tracking.)WrappedCompletionsr@rB completionsr-s r0rzWrappedChat.completionsRs"$,,0J0JKKr1Nr2r3r4r5propertyrr!r1r0r)r)OsE LLr1r)ceZdZdZ ddeedeedeeeefdedeeeefdef d Z ddeedeedeeeefdedeeeefdef d Z deedeedeeeefdedeeeefdef d Z ddeedeedeeeefdedeeeefdeeefd e d e dedeeeeefdeeeeefdeedeefdZy)rzGAsync wrapper for OpenAI chat completions that tracks usage in PostHog.Nr9r:r;r<r=r.c@Kt||||||fi|d{S7w)a Parse an OpenAI chat completion while tracking usage in PostHog. Args: posthog_distinct_id: Optional distinct ID to associate with the usage event. posthog_trace_id: Optional trace ID. Generated automatically when omitted. posthog_properties: Additional properties to include with the usage event. posthog_privacy_mode: Whether to redact captured input and output. posthog_groups: Optional PostHog groups to associate with the event. **kwargs: Arguments passed to OpenAI's async ``chat.completions.parse`` API. Returns: The parsed response from OpenAI. NrrPs r0rCzWrappedCompletions.parse[rrc fK|ttj}|jddr|j|||||fi|d{St ||j jd|||||j j|jjf i|d{}|S7^7w)a Create an OpenAI chat completion while tracking usage in PostHog. Args: posthog_distinct_id: Optional distinct ID to associate with the usage event. posthog_trace_id: Optional trace ID. Generated automatically when omitted. posthog_properties: Additional properties to include with the usage event. posthog_privacy_mode: Whether to redact captured input and output. posthog_groups: Optional PostHog groups to associate with the event. **kwargs: Arguments passed to OpenAI's async ``chat.completions.create`` API. Returns: The OpenAI chat completion, or an async streaming iterator when ``stream=True``. NrHFr?rI)r-r9r:r;r<r=r.rSs r0rOzWrappedCompletions.create|s.  #"4::<0  ::h &///# "$    8  LL # #     LL ! ! NN ! !     +  s%A B1B-AB1&B/'B1/B1c( Ktj tgi d d dvrid<ddd<jjdid{    fd}t | S7(w)Nstream_optionsT include_usagecNK 23d{} t|dr |j t|d}|r t|t |d}|j |t |}|r t|t|dr?|jr3t|jddd|jdj|76 tj}|z }rtjnd}j  ||td   d{7y#tj}|z }rtjnd}j  ||td   d{7wxYww)NrTrchoicesr finish_reasonr?rW)rYrTrr rrZrrrr$rr[listvaluesr\r )r]r^r_chunk_tool_callsrarbtool_calls_listaccumulated_contentaccumulated_tool_callsr.rer9r=r<r;r:rSr-rfrXrgs r0rhz=WrappedCompletions._create_streaming..async_generators: #+ %*2wug7N.3kk+#B%"PK")+{C@vNG*+227;(LE'R$'424D  y1!MM#EMM!$4otLX&+mmA&6&D&D K= 8B 99;"Z/ ./6689 33'$&("'#06B' +4 99;"Z/ ./6689 33'$&("'#06B' +4sWF%D9CC CB>D9 CD9A!F%2D53F%9A"F"FF""F%rir!rj)r-r9r:r;r<r=r.rhrrrerSrfrXrgs``````` @@@@@@@r0rNz$WrappedCompletions._create_streamingsYY[ ",,  <>-1%) 6 )')F# $48 1...888A A A F"/"3HEEK9sAB'B()Brgrbrc tool_callsrkrerXcVK|ttj}|jdxs| xsd}d|t |t |j j|t|jdt |j j|t| d| d|jdd|jd d|jd d|jd d||t|j jd |xsi}|jd }|t|tr |dkDr||d<| | |d<| r| |d<|d|d<t|j jdr.|j jj|xs|d||yyw)NrTrmr?messagesrrorprrqrrrsrtr}r~rrFrrrr)rJrKrLrMr rr@rrrrArrrYr)r-r9r:r;r<r=r.rgrbrcrrkrerXrTrr}s r0r\z+WrappedCompletions._capture_streaming_event s  #"4::<0  7#G':Gi%$4V$<* ''$ : 67 #4 ''$.vvzJ# # + B!,!!D+6??)1,%0OO4F$J", 5 56/ 0"'R1 8'??+=>  (+S1 1$7G 3 4  "2= . / ,@ [ )  &:? 6 7 4<<**I 6 LL # # + +/C3C&+% ,  7sF'F)r)NNNN)r2r3r4r5rrJrrrrCrOrNrrrr\r!r1r0rrXsRQ.2*.7;%*37  %c] #3- %T#s(^4  #  !c3h0   F.2*.7;%*37 1%c]1#3-1%T#s(^4 1 # 1 !c3h0 11fXF%c]XF#3-XF%T#s(^4 XF # XF !c3h0 XFXFJ6:?C-1%)J%c]J#3-J%T#s(^4 J # J !c3h0 JS#XJ JJJT$sCx.12J'tDcN';<J&c]Jc]Jr1rcheZdZdZ d deedeedeeeefdedeeeefdef d Z y) r*zAAsync wrapper for OpenAI embeddings that tracks usage in PostHog.Nr9r:r;r<r=r.c VK|ttj}tj}|jj di|d{}tj} t } t|drC|jr7t t|jddt|jdd} | |z } d|jdt|jj|t|jd d | jd d| |t|jjd |xsi} |d | d<t|jjdr-|jjj!|xs|d| ||S7Qw)a Create an embedding using OpenAI's 'embeddings.create' method, but also track usage in PostHog. Args: posthog_distinct_id: Optional ID to associate with the usage event. posthog_trace_id: Optional trace UUID for linking events. posthog_properties: Optional dictionary of extra properties to include in the event. posthog_privacy_mode: Whether to anonymize the input and output. posthog_groups: Optional dictionary of groups to associate with the event. **kwargs: Any additional parameters for the OpenAI Embeddings API. Returns: The response from OpenAI's embeddings.create call. Nusage prompt_tokensrcompletion_tokens)rprqr?rTrnrorp)rurvrwrxryrzr{r|Frrz $ai_embeddingrr!)rJrKrLr[rBrOrrYrr$rMrr@rrrAr) r-r9r:r;r<r=r.rfrSrargrbrs r0rOzWrappedEmbeddings.createYs0  #"4::<0 YY[ ...88899;#-, 8W %(..$$X^^_aH%hnn6I1MK Z'%G,* ''$(G)<= # + B", 5 56 "'R   &:? 6 7 4<<**I 6 LL # # + +/C3C%+% , W9sAF)F&EF)r) r2r3r4r5rrJrrrrOr!r1r0r*r*Vs~K.2*.7;%*37 G%c]G#3-G%T#s(^4 G # G !c3h0 GGr1r*c eZdZdZedZy)r+zDAsync wrapper for OpenAI beta features that tracks usage in PostHog.cVt|j|jjS)z8Access async beta chat APIs with PostHog usage tracking.)WrappedBetaChatr@rBrrs r0rzWrappedBeta.chatst||T^^-@-@AAr1N)r2r3r4r5rrr!r1r0r+r+sN BBr1r+c eZdZdZedZy)rz@Async wrapper for OpenAI beta chat that tracks usage in PostHog.cVt|j|jjS)z?Access async beta chat completions with PostHog usage tracking.)WrappedBetaCompletionsr@rBrrs r0rzWrappedBetaChat.completionss&dllDNN4N4NOOr1Nrr!r1r0rrsJ PPr1rcheZdZdZ d deedeedeeeefdedeeeefdef d Z y) rzLAsync wrapper for OpenAI beta chat completions that tracks usage in PostHog.Nr9r:r;r<r=r.c@Kt||||||fi|d{S7w)a Parse an OpenAI beta chat completion while tracking usage in PostHog. Args: posthog_distinct_id: Optional distinct ID to associate with the usage event. posthog_trace_id: Optional trace ID. Generated automatically when omitted. posthog_properties: Additional properties to include with the usage event. posthog_privacy_mode: Whether to redact captured input and output. posthog_groups: Optional PostHog groups to associate with the event. **kwargs: Arguments passed to OpenAI's async beta ``chat.completions.parse`` API. Returns: The parsed response from OpenAI. NrrPs r0rCzWrappedBetaCompletions.parserrr) r2r3r4r5rrJrrrrCr!r1r0rrswV.2*.7;%*37  %c] #3- %T#s(^4  #  !c3h0   r1r)/r[rKtypingrrrrposthog.ai.streamrposthog.ai.typesrr? ImportErrorModuleNotFoundErrorposthogr posthog.ai.utilsr r r r r"posthog.ai.openai.openai_converterrrrrrposthog.ai.sanitizationrrposthog.clientrr6posthog.ai.openai.wrapper_utilsrrrJrrEr,r)rr*r+rrr!r1r0rsF ,,0' N2B$N&$$$NN!#sm!c3h0    T#s(^, .k -k \L(L{/{|J.JZB(BP,P" 3" W Q s C%%C3