tPj@UddlZddlmZddlmZddlZddlZddlmZddlmZm Z ddl m Z ddl m Z mZmZddlmZd Zeed <d ZGd d e Zy)N)datetime)Final)verbose_logger)create_mock_braintrust_clientshould_use_braintrust_mock) CustomLogger) HTTPHandlerget_async_httpx_clienthttpxSpecialProvider) print_verbosez!https://api.braintrustdata.com/v1API_BASEcddl}t|drtj|jStjS)NrUTC)rhasattrnowrutcnow)dts l/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/integrations/braintrust_logging.pyget_utc_datetimers/r5||BFF##  ceZdZddedzdedzddffd ZdedzfdZdedefdZdedefd Zd Zd Z d Z d Z fdZ xZ S)BraintrustLoggerNapi_keyapi_basereturnct|t|_|jrt t j d|j||xstjdxst|_ d|_ |xstjd|_ d|jzdd|_i|_t!t"j$|_t)|_y) Nz<[BRAINTRUST MOCK] Braintrust logger initialized in mock mode)rBRAINTRUST_API_BASEBRAINTRUST_API_KEYzBearer zapplication/json) Authorizationz Content-Type) llm_provider)super__init__r is_mock_moderrinfovalidate_environmentosgetenvr rdefault_project_idrheaders_project_id_cacher r LoggingCallbackglobal_braintrust_http_handlerr #global_braintrust_sync_http_handler)selfrr __class__s rr"zBraintrustLogger.__init__%s 68    ) +    ^ _ !!'!2 PBII.C$DP "&#Fryy1E'F &5.  24.DRfRvRv.w+3>=0rcg}|'tjdd|jdt|dkDrt d|dy)zP Expects BRAINTRUST_API_KEY in the environment Nrrz Missing keys=z in environment.)r&r'appendlen Exception)r.r missing_keyss rr%z%BraintrustLogger.validate_environment7sU! ?ryy)=tDL    4 5 | q mL>9IJK K !r project_namecr||jvr|j|S |jj|jd|jd|i}|j }|d}||j|<|S#t j$r'}td|jjd}~wwxYw)zs Get project ID from name, using cache if available. If project doesn't exist, creates it. /projectnamer)jsonidFailed to register project: N) r*r-postrr)r:httpxHTTPStatusErrorr3responsetextr.r5r@ project_dict project_ides rget_project_id_syncz$BraintrustLogger.get_project_id_syncEs 411 1)),7 7 N"FFKK==/* l+LH #+--/L ,T 2J3=D " "< 0 $$ N:1::??:KLM M NsAA<<B6"B11B6cK||jvr|j|S |jj|jd|jd|id{}|j }|d}||j|<|S7*#t j$r'}td|jjd}~wwxYww)z6 Async version of get_project_id_sync z/project/registerr8r9Nr;r<) r*r,r=rr)r:r>r?r3r@rArBs rget_project_id_asyncz%BraintrustLogger.get_project_id_asyncZs 411 1)),7 7 N$($G$G$L$L==/!23 l+%M%H #+--/L ,T 2J3=D " "< 0 $$ N:1::??:KLM M Ns:C:BB)BCBC"B==CCcK|jj|jd|jddid{}|j }|d|_y7wNr7r8litellmr9r;)r,r=rr)r:r(r.projectrCs r%create_default_project_and_experimentz6BraintrustLogger.create_default_project_and_experimentnsa#BBGG}}oX & FICV H  &lln ".t"4  s;AA Ac|jj|jd|jddi}|j }|d|_yrJ)r-r=rr)r:r(rLs r*create_sync_default_project_and_experimentz;BraintrustLogger.create_sync_default_project_and_experimentwsSAAFF}}oX & FICVG &lln ".t"4rc tjd |jd}|jdi}d|jdi}d}g} |2|jdddk(st|tj rd}n|9t|tj r|ddd j}|d} nc|@t|tjr&|jdj}|j} n!|t|tjr|d }|jd ixsi} | jd ixsi} | jd } | &| jd} | r|j| nd} | (|j|j|j} g}t| tr| j!D]v\}}tj"Fttj"t$r(|tj"vr|j'|d|t|t(sm||vsr|||<x|jdd}d}t+|dd}|rt|tj,rtj.j1|||j2|j4|j6||j9|j9z |j9|j9d}| jdd}| jd}|r=|j;dDcgc]#}|j=s|j=%}}| jd| jd|d}||d||ddd}| jd||d<|j!D] \}}|s |||<| "| Dcgc]}|jc}|d<n||d<|||d< t?d |j@jB|j@jC|jDd!| d"d#|gi|jF$|jHr t?d%yycc}wcc}w#tJjL$r$}tO|jPjd}~wwxYw#tN$r}|d}~wwxYw)&NREACHES BRAINTRUST SUCCESSlitellm_call_idstandard_logging_objectmessages call_type embeddingchoicesrmessagedatalitellm_paramsmetadatarDr5: response_costusage) prompt_tokenscompletion_tokens total_tokens total_costtime_to_first_tokenstartend span_nameChat Completion span_parents,span_id root_span_idrkrlrillmr8type)r;inputr\span_attributestagsoutputmetricsz/self.global_braintrust_sync_http_handler.post: /project_logs//inserteventsurlr:r)z0[BRAINTRUST MOCK] Sync event successfully mocked))rdebugget isinstancerKEmbeddingResponse ModelResponser:TextCompletionResponserXrA ImageResponserFr(rPdictitemslangfuse_default_tagslistr1strgetattrUsageutilsget_logging_idr`rarb timestampsplitstripr r-r=rr)r#r>r?r3r@)r.kwargs response_obj start_timeend_timerSrTpromptrtrXr[dynamic_metadatarDr5rskeyvaluecostru usage_objrgrisrr request_datachoicerEs rlog_success_eventz"BraintrustLogger.log_success_events9:z %+ZZ0A%BO-3ZZ8QSU-V #'J)?@FFG' ;-< CCE&y1)jwGeGe.f%--a055&..)jwG\G\.]%f-$*JJ/?$D$JN&4&8&8R&H&NB *--l;J!&6&:&:>&J GST55lCY] !**2CCE!44 D*D1"2"8"8": =JC 55A&w'D'DdK7#@#@@ se1UG$45#5#.3>U3U7</4 =!**_d;D#'G&|WdCIZ 7==A ,,ZF%.%<%<)2)D)D$-$:$:"&+3+=+=+?*BVBVBX+X'113#--/ 033KARSI,//?L3?3E3Ec3JXaaggi X X,// : 0 4 4^ D ,&O& +3,5u#E #L ##N3;'+ V$.335 . U(-L% ."FM)NF&++-)N X&)/ X&"*1 Y' 1EdFnFnFsFsEtu88====/ |7K"\N3 LL> $$!"TU%U Y4*O$(( 1 00 1 G soH)SSC?SQ=Q=-A!S SR3SA4R= SR>R99R>>S S S  Sc Ktjd |jd}|jdi}d|jdi}d}g} |2|jdddk(st|tj rd}n|9t|tj r|ddd j}|d} nc|@t|tjr&|jdj}|j} n!|t|tjr|d }|jd i} | jd ixsi} | jd } | .| jd} | r|j| d{nd} | 0|j|jd{|j} g}t| tr| j!D]v\}}tj"Fttj"t$r(|tj"vr|j'|d|t|t(sm||vsr|||<x|jdd}d}t+|dd}|rt|tj,rtj.j1|||j2|j4|j6||j9|j9d}|jd}|jd}|&|$|j9|j9z |d<| jdd}| jd}|r=|j;dDcgc]#}|j=s|j=%}}| jd| jd|d}||d|||ddd}| jd||d <|j!D] \}}|s |||<| "| Dcgc]}|jc}|d!<n||d!<|||d"< |j>jA|jBd#| d$d%|gi|jD&d{|jFr tId'yy77cc}wcc}w7-#tJjL$r$}tO|jPjd}~wwxYw#tN$r}|d}~wwxYww)(NrRrSrTrUrVrWrXrrYrZr[r\rDr5r]r^r_)r`rarbrcrerfapi_call_start_timecompletion_start_timerdrgrhrirjrkrlrmrnro)r;rqrtr\rrrsrtrurvrwrxryz1[BRAINTRUST MOCK] Async event successfully mocked))rr{r|r}rKr~rr:rrXrArrHr(rNrrrrr1rrrrrr`rarbrrrr,r=rr)r#r r>r?r3r@)r.rrrrrSrTrrtrXr[rrDr5rsrrrrurrrrgrirrrrrrEs rasync_log_success_eventz(BraintrustLogger.async_log_success_events9:| %+ZZ0A%BO-3ZZ8QSU-V #'J)?@FFG' ;-< CCE&y1)jwGeGe.f%--a055&..)jwG\G\.]%f-$*JJ/?$DN&4&8&8R&H&NB *--l;J!&6&:&:>&J NZ4#<#<\#JJJ`d !**2DDFFF!44 D*D1"2"8"8": =JC 55A&w'D'DdK7#@#@@ se1UG$45#5#.3>U3U7</4 =!**_d;D#'G&|WdCIZ 7==A ,,ZF%.%<%<)2)D)D$-$:$:"&'113#--/ .4ZZ8M-N#/5zz:Q/R%&27L7X5J5T5T5VYlYvYvYx5xG12 033KARSI,//?L3?3E3Ec3JXaaggi X X,// : 0 4 4^ D ,&O& + 3,5u#E #L ##N3;'+ V$.335 . U(-L% ."FM)NF&++-)N X&)/ X&"*1 Y' 199>>==/ |7K"\N3 LL? $$!"UV%{KGb Y6*O(( 1 00 1 G sS;E.S(R'S(/R0BS(S(D)S(=R"R"%A"S( S(R',S(>R.>R,?R.S;S(S(" S(,R..S%S  S%%S(( S81S33S88S;c(t|||||S)N)r!log_failure_event)r.rrrrr/s rrz"BraintrustLogger.log_failure_event~sw(z8TTr)NN)__name__ __module__ __qualname__rr"r%rFrHrNrPrrr __classcell__)r/s@rrr$sAd AS4ZASWA$ LC$J LNNN*NsNsN(55||~@UUrr)r&rtypingrr>rKr+litellm.integrations.braintrust_mock_clientrr"litellm.integrations.custom_loggerr&litellm.llms.custom_httpx.http_handlerr r r litellm.utilsr r __annotations__rrrrrsQ  "< (5%5![U|[Ur