jKj ` ddlZddlZddlmZmZmZmZddlmZ ddl Z ddl m Z mZmZmZddlmZmZmZmZddlmZmZddlmZdd lmZdd lmZGd d e j@Z d ee!dee!deee!efde"deee!efdef dZ#GddeZ$GddeZ%GddeZ&GddeZ'GddeZ(GddeZ)Gd d!eZ*y#e $r e dwxYw)"N)AnyDictListOptional) TokenUsagezGPlease install the OpenAI SDK to use this feature: 'pip install openai')call_llm_and_track_usageextract_available_tool_callsmerge_usage_statswith_privacy_mode)extract_openai_usage_from_chunk!extract_openai_content_from_chunk$extract_openai_tool_calls_from_chunkaccumulate_openai_tool_calls)sanitize_openaisanitize_openai_response)Client)setup)_OpenAIWrapperResourcec<eZdZUdZeed<ddeeffd ZxZS)OpenAIz_ A 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.OpenAI`` such as ``api_key`` or ``organization``. chatN embeddingsbeta responses)super__init__rrgetattr_original_chat_original_embeddings_original_beta_original_responses WrappedChatrWrappedEmbeddingsr WrappedBetarWrappedResponsesr)selfrkwargs __class__s i/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/posthog/ai/openai/openai.pyr zOpenAI.__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@r-rr s)Nx '>NNr.rposthog_distinct_idposthog_trace_idposthog_propertiesposthog_privacy_modeposthog_groupsr+c t||jjd|||||jj|jj f i|S)Nopenai)r_clientrbase_url _originalparse)wrapperr6r7r8r9r:r+s r-_parse_and_trackrBGsV $""      r.ceZdZdZ ddeedeedeeeefdedeeeefdef d Z deedeedeeeefdedeeeefdef d Z ddeedeedeeeefdedeeeefdeeefd e d e d edee eeefdeedeefdZ ddeedeedeeeefdedeeeefdef dZy)r)z:Wrapper for OpenAI responses that tracks usage in PostHog.Nr6r7r8r9r:r+c 2|ttj}|jddr|j|||||fi|St ||j jd|||||j j|jjf i|S)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 ``responses.create`` API. Returns: The OpenAI response, or a streaming iterator when ``stream=True``. streamFr< struuiduuid4get_create_streamingrr=rr>r?creater*r6r7r8r9r:r+s r-rLzWrappedResponses.createa.  #"4::<0  ::h &)4))# "$    (  LL # #     LL ! ! NN ! !    r.c  tj t gd d jjdi   f d}|S)Nc3 K D]}t|dr:|jr. ,t|jdr|jj t|d}|r t |t |d}|j |t|drB|jdk(r3t|dr'|jrt|jdd}||| tj}|z }}j  ||d  y#tj}|z }}j  ||d  wxYww)Nresponsemodelrtypezresponse.completedstatus stop_reason) hasattrrQrRr r r appendrSr!time_capture_streaming_event)chunk chunk_usagecontent chunk_statusend_timelatencyoutput final_contentr+model_from_responser6r:r9r8r7rQr* start_timerV usage_statss r- generatorz5WrappedResponses._create_streaming..generators 3 %Euj1enn.67!NNG<38..2F2F/#B%"UK")+{C@{SG*%,,W5 v.!JJ*>>#E:6!NN'.u~~x'N '3*6KK?&D 99;"Z/&--'$&("' +.  99;"Z/&--'$&("' +. sECD9E:EErrYrr?rL)r*r6r7r8r9r:r+rfrbrcrQrdrVres``````` @@@@@@r-rKz"WrappedResponses._create_streamings[YY[ ",,  -1%) (4>>((2629 9 9 v{r.rer`raavailable_tool_callsrcrVc 4ddlm} ddlm}m}ddlm}||d}t|}|jdxs| xsd}| d|t|jj|||| d|||||||| }||jj|y) NrStreamingEventDataformat_openai_streaming_inputformat_openai_streaming_outputcapture_streaming_eventrrRunknownr<providerrRr>r+formatted_inputformatted_outputrer` distinct_idtrace_id properties privacy_modegroupsrV) posthog.ai.typesrk"posthog.ai.openai.openai_converterrmrnposthog.ai.utilsrprrJrGr=r>r)r*r6r7r8r9r:r+rer`rarhrcrVrkrmrnrprtsanitized_inputrR event_datas r-rZz)WrappedResponses._capture_streaming_events 8  =8 L2?C 7#G':Gi'../+;FKP#+%)-!# $  7 7Dr.c $t||||||fi|S)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. rBrMs r-r@zWrappedResponses.parse-.         r.NNNFN)NNN)r/r0r1r2rrGrrboolrLrKrfloatrrZr@rr.r-r)r)^s-D.2*.7;%*37 / %c]/ #3-/ %T#s(^4 / # / !c3h0 / / bK%c]K#3-K%T#s(^4 K # K !c3h0 KKp@D-1%)/E%c]/E#3-/E%T#s(^4 /E # /E !c3h0 /ES#X/E /E/E/E'tDcN';</E&c]/Ec]/Ef.2*.7;%*37  %c] #3- %T#s(^4  #  !c3h0   r.r)c eZdZdZedZy)r&z5Wrapper for OpenAI chat that tracks usage in PostHog.cVt|j|jjS)z4Access chat completions with PostHog usage tracking.)WrappedCompletionsr=r? completionsr*s r-rzWrappedChat.completions5s"$,,0J0JKKr.Nr/r0r1r2propertyrrr.r-r&r&2s? LLr.r&ceZdZdZ ddeedeedeeeefdedeeeefdef d Z ddeedeedeeeefdedeeeefdef d Z deedeedeeeefdedeeeefdef d Z ddeedeedeeeefdedeeeefdeeefd e d e dedeeeeefdeeeeefdeedeefdZy)rzAWrapper for OpenAI chat completions that tracks usage in PostHog.Nr6r7r8r9r:r+c $t||||||fi|S)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 ``chat.completions.parse`` API. Returns: The parsed response from OpenAI. rrMs r-r@zWrappedCompletions.parse>rr.c 2|ttj}|jddr|j|||||fi|St ||j jd|||||j j|jjf i|S)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 ``chat.completions.create`` API. Returns: The OpenAI chat completion, or a streaming iterator when ``stream=True``. rEFr<rFrMs r-rLzWrappedCompletions.create_rNr.c  tj tgi d d dvrid<ddd<jjdi    fd}|S)Nstream_optionsT include_usagec3K D]} t|dr |j t|d}|r t|t |d}|j |t |}|r t|t|dr?|jr3t|jddd|jdj| tj}|z }rtjnd}j  ||td   y#tj}|z }rtjnd}j  ||td   wxYww)NrRrchoicesr finish_reasonr<rU)rWrRr r r rXrrrr!rrYlistvaluesrZr )r[r\r]chunk_tool_callsr_r`tool_calls_listaccumulated_contentaccumulated_tool_callsr+rcr6r:r9r8r7rQr*rdrVres r-rfz7WrappedCompletions._create_streaming..generators< %E*2wug7N.3kk+#B%"PK")+{C@vNG*+227;(LE'R$'424D  y1!MM#EMM!$4otLX&+mmA&6&D&D KA&F 99;"Z/ ./6689 --'$&("'#06B' +. 99;"Z/ ./6689 --'$&("'#06B' +.sF CD'AF 'A FF rrg)r*r6r7r8r9r:r+rfrrrcrQrdrVres``````` @@@@@@@r-rKz$WrappedCompletions._create_streamingsYY[ ",,  <>-1%) 6 )')F# $48 1(4>>((262C C C J{r.rer`ra tool_callsrhrcrVc6ddlm}ddlm}m}ddlm}||d}t|}|jdxs| xsd}|d|t|jj|||| d| |||||||| }||jj|y) NrrjrlrorrRrqr<rr) r{rkr|rmrnr}rprrJrGr=r>r)r*r6r7r8r9r:r+rer`rarrhrcrVrkrmrnrprtr~rRrs r-rZz+WrappedCompletions._capture_streaming_events 8  =8G)/: 7#G':Gi'../+;FFJW#+%)-!# $  7 7Dr.r)NNNN)r/r0r1r2rrGrrrr@rLrKrrrrZrr.r-rr;sKK.2*.7;%*37  %c] #3- %T#s(^4  #  !c3h0   F.2*.7;%*37 / %c]/ #3-/ %T#s(^4 / # / !c3h0 / / bY%c]Y#3-Y%T#s(^4 Y # Y !c3h0 YYL6:?C-1%)0E%c]0E#3-0E%T#s(^4 0E # 0E !c3h0 0ES#X0E 0E0E0ET$sCx.120E'tDcN';<0E&c]0Ec]0Er.rcheZdZdZ d deedeedeeeefdedeeeefdef d Z y) r'z;Wrapper for OpenAI embeddings that tracks usage in PostHog.Nr6r7r8r9r:r+c |ttj}tj}|jj di|}tj} i} t |dr;|jr/t|jddt|jddd} | |z } d|jdt|jj|t|jdd | jdd| |t|jjd |xsi} |d | d <t |jjd r-|jjj|xs|d| ||S)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. usage prompt_tokensr total_tokens)rrr<rRinput)z $ai_providerz $ai_modelz $ai_inputz$ai_http_statusz$ai_input_tokensz $ai_latencyz $ai_trace_idz $ai_base_urlFz$process_person_profilecapturez $ai_embedding)rveventrxrzr)rGrHrIrYr?rLrWrr!rJr r=rrr>r) r*r6r7r8r9r:r+rdrQr_rer`event_propertiess r-rLzWrappedEmbeddings.create!sp0  #"4::<0 YY[ (4>>((26299; 8W %(..!(!!L ' JK Z'%G,* ''$(G)<= # + C", 5 56 "'R   &:? 6 7 4<<**I 6 LL # # + +/C3C%+% , r.r) r/r0r1r2rrGrrrrLrr.r-r'r's~E.2*.7;%*37 F%c]F#3-F%T#s(^4 F # F !c3h0 FFr.r'c eZdZdZedZy)r(z>Wrapper for OpenAI beta features that tracks usage in PostHog.cVt|j|jjS)z2Access beta chat APIs with PostHog usage tracking.)WrappedBetaChatr=r?rrs r-rzWrappedBeta.chatmst||T^^-@-@AAr.N)r/r0r1r2rrrr.r-r(r(jsH BBr.r(c eZdZdZedZy)rz:Wrapper for OpenAI beta chat that tracks usage in PostHog.cVt|j|jjS)z9Access beta chat completions with PostHog usage tracking.)WrappedBetaCompletionsr=r?rrs r-rzWrappedBetaChat.completionsvs&dllDNN4N4NOOr.Nrrr.r-rrssD PPr.rcheZdZdZ d deedeedeeeefdedeeeefdef d Z y) rzFWrapper for OpenAI beta chat completions that tracks usage in PostHog.Nr6r7r8r9r:r+c $t||||||fi|S)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 beta ``chat.completions.parse`` API. Returns: The parsed response from OpenAI. rrMs r-r@zWrappedBetaCompletions.parserr.r) r/r0r1r2rrGrrrr@rr.r-rr|swP.2*.7;%*37  %c] #3- %T#s(^4  #  !c3h0   r.r)+rYrHtypingrrrrr{rr< ImportErrorModuleNotFoundErrorr}rr r r r|r r rrposthog.ai.sanitizationrrposthog.clientrr3posthogrposthog.ai.openai.wrapper_utilsrrrGrrBr)r&rr'r(rrrr.r-rsA ,,'   N2B$NV]]$NN!#sm!c3h0    T#s(^, .Q -Q hL(L`E/`EFI.IXB(BP,P" 3" g Q s CC)