o BDйiс<у@sTddlmZddlZddlmZddlmZddlmZddlm Z ddl mZmZddl mZdd lmZmZdd lmZerXddlmZmZmZmZmZmZmZddlmZzdd lmZmZddl m!Z!m"Z"ersddl#m$Z$m%Z%Wne&yedГВwGddДdeГZ'ddДZ(ddДZ)ddДZ*ddДZ+ddДZ,ddДZ-dd ДZ.d!d"ДZ/dS)#щй┌wrapsN)┌consts)┌record_token_usage)┌set_data_normalized)┌SPANDATA)┌DidNotEnable┌Integration)┌should_send_default_pii)┌capture_internal_exceptions┌event_from_exception)┌ TYPE_CHECKING)┌Any┌Iterable┌List┌Optional┌Callable┌ AsyncIterator┌Iterator)┌Span)┌Completions┌AsyncCompletions)┌ Embeddings┌AsyncEmbeddings)┌ChatCompletionMessageParam┌ChatCompletionChunkzOpenAI not installedc@s8eZdZdZdeЫЭZdddДZeddДГZd d ДZdS)┌OpenAIIntegration┌openaizauto.ai.TNcCs0||_d|_|durddl}|а|б|_dSdSйNr)┌include_prompts┌tiktoken_encoding┌tiktoken┌get_encoding)┌selfr┌tiktoken_encoding_namer!йr%·W/var/www/Datamplify/venv/lib/python3.10/site-packages/sentry_sdk/integrations/openai.py┌__init__#s¤zOpenAIIntegration.__init__cCs4ttjГt_ttjГt_ttjГt_ttjГt_dSйN) ┌_wrap_chat_completion_creater┌create┌_wrap_embeddings_creater┌"_wrap_async_chat_completion_creater┌_wrap_async_embeddings_createrr%r%r%r&┌ setup_once-s zOpenAIIntegration.setup_oncecCs|jdur t|jа|бГSdSr)r ┌len┌encode_ordinary)r#┌sr%r%r&┌count_tokens8s zOpenAIIntegration.count_tokens)TN) ┌__name__┌ __module__┌__qualname__┌ identifier┌originr'┌staticmethodr.r2r%r%r%r&rs rcCs0t|tабjdddЬdН\}}tj||dНdS)NrF)┌type┌handled)┌client_options┌ mechanism)┌hint)r┌ sentry_sdk┌ get_client┌options┌ capture_event)┌exc┌eventr=r%r%r&┌_capture_exception?s ¤rDc Cs2d}d}d}t|dГr>t|jdГrt|jjtГr|jj}t|jdГr-t|jjtГr-|jj}t|jdГr>t|jjtГr>|jj}|dkrS|D]}d|vrR|||dГ7}qD|dkr~|durg|D]}|||Г7}q]nt|dГr~|jD]} t| dГr}||| jГ7}qo|dkrДd}|dkrКd}|dkrРd}t ||||ГdS) Nr┌usage┌completion_tokens┌ prompt_tokens┌total_tokens┌content┌choices┌message) ┌hasattrrE┌ isinstancerF┌intrGrHrJrKr) ┌messages┌response┌span┌streaming_message_responsesr2rFrGrHrK┌choicer%r%r&┌ _calculate_chat_completion_usageIsL А АrTc/s.БtабаtбЙИdur||i|дОSd|vr||i|дОSzt|dГWnty6||i|дОYSwt|dГ|d<|dЙ|аdб}|аdб}tjt j jdtjdНЙИа б|||fVЙtГПзtГrsИjrstИtjИГtИtj|ГtИtj|ГtИdГrмtГrЫИjrЫtИtjttddДИjГГГtИИИdИjГИаdddбnHtИd Гr▀gЙИjЙЗЗЗЗЗЗfd dД}ЗЗЗЗЗЗfdd Д}ttИjГГdkr┌|ГИ_n*|ГИ_ntИddГИаdddбWdГИSWdГИSWdГИSWdГИS1РswYИS)NrO┌model┌streamzChat Completion)┌op┌namer7rJcSs|jSr()rK)┌xr%r%r&┌Яsz-_new_chat_completion_common..┌ _iteratorc3s№БtГПiИD]9}t|dГr=d}|jD])}t|dГr8t|jdГr8|jj}tИГ|kr/ИаgбИ|а|p6dб|d7}q|VqtИГdkrfttddДИГГ}t Гr]Иj r]tИtj |ГtИИИ|ИjГWdГn1spwYИаdddбdS) NrJr┌deltarI┌щcSє dа|бSйNr]й┌joinй┌chunkr%r%r&rZ╝є zC_new_chat_completion_common..new_iterator..йrrLrJr\rIr/┌append┌list┌mapr rrr┌AI_RESPONSESrTr2┌__exit__йrY┌choice_indexrSrI┌ all_responsesй┌data_buf┌integrationrO┌old_iterator┌resrQr%r&┌new_iteratorкsBА √Аъz1_new_chat_completion_common..new_iteratorcsБtГПnИ2z=3dHW}t|dГrAd}|jD])}t|dГr.new_iterator_async..rfrlror%r&┌new_iterator_async╦sDА Ї √Аъz7_new_chat_completion_common..new_iterator_asyncz┌unknown_responseT) r>r?┌get_integrationr┌iter┌ TypeErrorrh┌get┌ start_spanr┌OP┌OPENAI_CHAT_COMPLETIONS_CREATEr7┌ __enter__rr rrr┌AI_INPUT_MESSAGES┌AI_MODEL_ID┌AI_STREAMINGrLrjrirJrTr2rkr[┌strr9)┌f┌args┌kwargsrU┌ streamingrtrur%ror&┌_new_chat_completion_commonuszА■ ¤ ¤ !! аa░ яa√ дa∙ жaЯarЗcє"ddДЙtИГЗЗfddДГ}|S)Nc _є╞t|g|вRi|дО}z t|Г\}}}Wnty)}z |jWYd}~Sd}~wwz#z ||i|дО}WntyG}zt|Г|dВd}~ww|а|бWStyb}z |jWYd}~Sd}~wwr(йrЗ┌next┌ StopIteration┌value┌ ExceptionrD┌sendйrГrДrЕ┌gen┌e┌resultr%r%r&┌ _execute_sync∙є(А А■А z3_wrap_chat_completion_create.._execute_synccsBtабаtб}|dusd|vrИ|i|дОSИИg|вRi|дОSйNrOйr>r?rwrйrДrЕrqйrФrГr%r&┌_sentry_patched_create_sync szA_wrap_chat_completion_create.._sentry_patched_create_syncrйrГrЪr%rЩr&r)ўє r)crИ)Nc ▀є╘Бt|g|вRi|дО}z t|Г\}}}Wnty-}z|jIdHWYd}~Sd}~wwz&z||i|дОIdH}WntyN}zt|Г|dВd}~ww|а|бWStyi}z |jWYd}~Sd}~wwr(rКrРr%r%r&┌_execute_asyncє*АА А■А z:_wrap_async_chat_completion_create.._execute_asynccЯsPБtабаtб}|dusd|vrИ|i|дОIdHSИИg|вRi|дОIdHSrЦrЧrШйrЮrГr%r&┌_sentry_patched_create_async0s АzH_wrap_async_chat_completion_create.._sentry_patched_create_asyncrйrГrбr%rаr&r,rЬr,cosЖБtабаtб}|dur||i|дОStjtjjdtjdНПЫ}d|vr\t Гr\|j r\t|dtГr;t |tj|dgГn!t|dtГr\t|dГdkr\t|ddtГr\t |tj|dГd|vrit |tj|dГ|||fV}d}d}t|dГrЪt|jdГrЙt|jjtГrЙ|jj}t|jdГrЪt|jjtГrЪ|jj}|dkrз|а|dpеd б}t||d|pо|Г|WdГS1s╝wYdS) NzOpenAI Embedding Creation)rW┌descriptionr7┌inputrrUrErGrHr])r>r?rwrr{rr|┌OPENAI_EMBEDDINGS_CREATEr7r rrMrВrrrrhr/rАrLrErGrNrHr2r)rГrДrЕrqrQrPrGrHr%r%r&┌_new_embeddings_create_common=sTА¤ $┌rжcrИ)Nc _rЙr(йrжrЛrМrНrОrDrПrРr%r%r&rФnrХz._wrap_embeddings_create.._execute_synccs:tабаtб}|durИ|i|дОSИИg|вRi|дОSr(rЧrШrЩr%r&rЪВsz<_wrap_embeddings_create.._sentry_patched_create_syncrrЫr%rЩr&r+lєr+crИ)Nc ▀rЭr(rзrРr%r%r&rЮРrЯz5_wrap_async_embeddings_create.._execute_asynccЯsHБtабаtб}|durИ|i|дОIdHSИИg|вRi|дОIdHSr(rЧrШrаr%r&rбдs АzC_wrap_async_embeddings_create.._sentry_patched_create_asyncrrвr%rаr&r-Оrиr-)0┌ functoolsrr>r┌sentry_sdk.ai.monitoringr┌sentry_sdk.ai.utilsr┌sentry_sdk.constsr┌sentry_sdk.integrationsrr ┌sentry_sdk.scoper ┌sentry_sdk.utilsrr┌typingr rrrrrrr┌sentry_sdk.tracingr┌!openai.resources.chat.completionsrr┌openai.resourcesrr┌openai.types.chatrr┌ImportErrorrrDrTrЗr)r,rжr+r-r%r%r%r&┌s@$А ,##/"