Uvh<RddlmZddlZddlmZddlmZddlmZddlm Z ddl m Z m Z ddl mZdd lmZmZdd lmZerdd lmZmZmZmZmZmZmZdd lmZ dd lmZmZddl m!Z!m"Z"erddl#m$Z$m%Z%Gdde Z'dZ(dZ)dZ*dZ+dZ,dZ-dZ.dZ/y#e&$r e dwxYw)wrapsN)consts)record_token_usage)set_data_normalized)SPANDATA) DidNotEnable Integration)should_send_default_pii)capture_internal_exceptionsevent_from_exception) TYPE_CHECKING)AnyIterableListOptionalCallable AsyncIteratorIterator)Span) CompletionsAsyncCompletions) EmbeddingsAsyncEmbeddings)ChatCompletionMessageParamChatCompletionChunkzOpenAI not installedc8eZdZdZdeZddZedZdZy)OpenAIIntegrationopenaizauto.ai.NcZ||_d|_|ddl}|j||_yyNr)include_promptstiktoken_encodingtiktoken get_encoding)selfr"tiktoken_encoding_namer$s Y/opt/face_recognition/venv/lib/python3.12/site-packages/sentry_sdk/integrations/openai.py__init__zOpenAIIntegration.__init__#s4.!% ! - %-%:%:;Q%RD " .cttjt_ttjt_t t jt _ttjt_yN) _wrap_chat_completion_creatercreate_wrap_embeddings_creater"_wrap_async_chat_completion_creater_wrap_async_embeddings_createrr*r( setup_oncezOpenAIIntegration.setup_once-sV:+:L:LM 3J4E4EF "D  # ## "??U?U!Vr*cd|j$t|jj|Syr!)r#lenencode_ordinary)r&ss r( count_tokenszOpenAIIntegration.count_tokens8s-  ! ! -t--==a@A Ar*)TN) __name__ __module__ __qualname__ identifieroriginr) staticmethodr3r8r2r*r(rrs4J | $FSWWr*rct|tjjddd\}}tj||y)NrF)typehandled)client_options mechanism)hint)r sentry_sdk get_clientoptions capture_event)exceventrDs r(_capture_exceptionrK?s?& !,,.66#6KE4 U.r*c>d}d}d}t|drt|jdr:t|jjtr|jj}t|jdr:t|jj tr|jj }t|jdr:t|jj tr|jj }|dk(r|D]}d|vs|||dz }|dk(rT||D] }|||z }n?t|dr3|jD]$} t| ds||| jz }&|dk(rd}|dk(rd}|dk(rd}t||||y) Nrusagecompletion_tokens prompt_tokens total_tokenscontentchoicesmessage) hasattrrM isinstancerNintrOrPrRrSr) messagesresponsespanstreaming_message_responsesr8rNrOrPrSchoices r( _calculate_chat_completion_usager\IsMLx! 8>>#6 7J NN , ,c= !) @ @  8>>? 3 NN ( (#9 %NN88M 8>>> 2z NN ' '8 $>>66L BGG#gi.@!AA  BA & 26 ;!\'%::! ; Xy )"** F69-%fnn)EE% F A q t],=|Lr*c /* Ktjjt||i|Sd|vr||i|S t |dt |d|d<|d |jd}|jd}tjtjjdtj j|||f t5tr'j rt# t$j& t# t$j(|t# t$j*|t- dr|trEj r9t# t$j.t t1d j2t5 dj6 j9dddnt- dr]g j:  fd } fd }t=t? j:d k(r | _n-| _n t# d d  j9dddddd S#t $r ||i|cYSwxYw#1swY SxYww)NrWmodelstreamzChat Completion)opnamer=rRc|jSr,)rS)xs r(z-_new_chat_completion_common..s qyyr* _iteratorc3Kt5D]}t|drd}|jD]v}t|drct|jdrM|jj}t |krj g|j |xsd|dz }x|t dkDr`ttd}tr'jrt tj|t |jddd j!dddy#1swYxYww)NrRrdeltarQc$dj|SNrhjoinchunks r(rdzC_new_chat_completion_common..new_iterator..bggenr*r rTrRrgrQr5appendlistmapr r"rr AI_RESPONSESr\r8__exit__ rc choice_indexr[rQ all_responsesdata_buf integrationrW old_iteratorresrYs r( new_iteratorz1_new_chat_completion_common..new_iterators702)  "1i0+,L*+))2#*67#;$*LL)A"/5ll.B.BG'*8} 'D(0(;$,\$:$A$A'-R$P , 1 2   8}q((, .new_iterator_async..rpr*rqrws r(new_iterator_asyncz7_new_chat_completion_common..new_iterator_asyncs@02#/   a"1i0+,L*+))2#*67#;$*LL)A"/5ll.B.BG'*8} 'D(0(;$,\$:$A$A'-R$P , 1 2   <8}q((, E D-B5B3B5BD-3B55A/D-$ E -D62E zunknown_responseT) rErFget_integrationriter TypeErrorrsget start_spanrOPOPENAI_CHAT_COMPLETIONS_CREATEr= __enter__r r r"rrAI_INPUT_MESSAGES AI_MODEL_ID AI_STREAMINGrTrurtrRr\r8rvrestrr@) fargskwargsr^ streamingr~rrzr{rWr|r}rYs @@@@@@r(_new_chat_completion_commonrusU'')99:KLK$!&!!$!&!!" VJ  fZ01F:j!H JJw E 8$I  99 3 3  '' D  NN4 C $ &`, " $)D)D h&@&@( KD("6"6>D("7"7C 3 "&([-H-H#))0#++>? -#tT;+C+C  MM$d + S+ &(*H==L 0 0B 0 0B4 &'+FF 2 4 ,  &8$ ? MM$d +A`,B Jg "$!&!!"$`,B JsC>JI/BJ,E9J% J/JJJJJ Jc8dtfd}|S)Nc<t|g|i|} t|\}}} ||i|}|j |S#t$r}|jcYd}~Sd}~wwxYw#t$r}t ||dd}~wwxYw#t$r}|jcYd}~Sd}~wwxYwr,rnext StopIterationvalue ExceptionrKsendrrrgeneresults r( _execute_syncz3_wrap_chat_completion_create.._execute_syncs)!=d=f= "3iOAtV  "D+F+ 88F# # 77N   ""1%T! "  77N U;AA< A AAA A9&A44A99A<< B BBBctjjt}|d|vr|i|Sg|i|SNrWrErFrrrrr{rrs r(_sentry_patched_create_synczA_wrap_chat_completion_create.._sentry_patched_create_sync sQ!++-==>OP  *F":d%f% %Q0000r*rrrrs` @r(r-r-s(( 1X11 '&r*c8dtfd}|S)NcjKt|g|i|} t|\}}} ||i|d{}|j |S#t$r}|jd{7cYd}~Sd}~wwxYw7@#t$r}t ||dd}~wwxYw#t$r}|jcYd}~Sd}~wwxYwwr,rrs r(_execute_asyncz:_wrap_async_chat_completion_create.._execute_asyncs)!=d=f= !"3iOAtV  " $1&11 88F# # !==  ! 2 ""1%T! "  77N B3A A2A0A2BB3 A-A(A A("A-#B3(A--B30A22 B;B  BB B0 B+%B0&B3+B00B3cKtjjt}|d|vr|i|d{Sg|i|d{S77wrrrrr{rrs r(_sentry_patched_create_asynczH_wrap_async_chat_completion_create.._sentry_patched_create_async0sd!++-==>OP  *F":D+F++ +#A77777,7s!9AAAAAArrrrs` @r(r0r0s(( 1X88 ('r*c/rKtjjt}|||i|Stjt j jdtj5}d|vrtr|jrt|dtr t|tj|dgnXt|dt rEt#|ddkDr4t|ddtrt|tj|dd|vrt|tj$|d|||f}d}d}t'|drt'|j(dr:t|j(j*t,r|j(j*}t'|j(dr:t|j(j.t,r|j(j.}|dk(r|j1|dxsd }t3||d|xs||cdddS#1swYyxYww) NzOpenAI Embedding Creation)r` descriptionr=inputrr^rMrOrPrh)rErFrrrrrOPENAI_EMBEDDINGS_CREATEr=r r"rUrrrrrsr5rrTrMrOrVrPr8r)rrrr{rYrXrOrPs r(_new_embeddings_create_commonr=s'')99:KLK$!&!!    99 - -/ '' &  f  # %+*E*E&/3/#D(*D*DvgFWX6'?D1w(1,vgq137#D(*D*DfWoV f  h&:&:F7O LD&(  8W %x~~7J,,c=!) < < x~~~6:++S< (~~:: A '44VG_5JKM4l6SmTM&&&sA0H72F/H+! H7+H40H7c8dtfd}|S)Nc<t|g|i|} t|\}}} ||i|}|j |S#t$r}|jcYd}~Sd}~wwxYw#t$r}t ||dd}~wwxYw#t$r}|jcYd}~Sd}~wwxYwr,rrrrrrKrrs r(rz._wrap_embeddings_create.._execute_syncns+A??? "3iOAtV  "D+F+ 88F# # 77N   ""1%T! "  77N rc~tjjt}||i|Sg|i|Sr,rrs r(rz<_wrap_embeddings_create.._sentry_patched_create_syncsJ!++-==>OP  d%f% %Q0000r*rrs` @r(r/r/ls(( 1X11 '&r*c8dtfd}|S)NcjKt|g|i|} t|\}}} ||i|d{}|j |S#t$r}|jd{7cYd}~Sd}~wwxYw7@#t$r}t ||dd}~wwxYw#t$r}|jcYd}~Sd}~wwxYwwr,rrs r(rz5_wrap_async_embeddings_create.._execute_asyncs+A??? !"3iOAtV  " $1&11 88F# # !==  ! 2 ""1%T! "  77N rcKtjjt}||i|d{Sg|i|d{S77wr,rrs r(rzC_wrap_async_embeddings_create.._sentry_patched_create_asyncs]!++-==>OP  D+F++ +#A77777,7s!5AAA A AArrs` @r(r1r1s(( 1X88 ('r*)0 functoolsrrErsentry_sdk.ai.monitoringrsentry_sdk.ai.utilsrsentry_sdk.constsrsentry_sdk.integrationsr r sentry_sdk.scoper sentry_sdk.utilsr r typingrrrrrrrrsentry_sdk.tracingr!openai.resources.chat.completionsrropenai.resourcesrropenai.types.chatrr ImportErrorrrKr\rr-r0rr/r1r2r*r(rs73&=4 !WWW'/O<U  @/)MXD 'F (F,^'D(g / - ../s BB&