^hwFSSKrSSKrSSKrSSKrSSKrSSKrSSKJrJr SSKJ r J r J r J r SSK r SSKJr SSKrSSKJr SSKJr SSKJrJr SSK7 SS KJrJr S r"S S \5rg) N)datetimetimezone)AnyDictListOptional) BaseModel)verbose_logger)CustomBatchLogger)get_async_httpx_clienthttpxSpecialProvider)*)StandardCallbackDynamicParamsStandardLoggingPayloadc[R[R[R[4n[ X5(+$)N)types CoroutineType FunctionType GeneratorTyper isinstance)valuenon_serializable_typess X/home/james-whalen/.local/lib/python3.13/site-packages/litellm/integrations/langsmith.pyis_serializablers8     %8 88c ^\rSrSrSS\\S\\S\\4U4SjjjrSS\\S\\S\\S\4SjjrS\4S jr S r S r S r S r S\S\\4SjrS\\\44SjrS\\\4S\4SjrSrSrS\4SjrSrU=r$)LangsmithLogger$langsmith_api_keylangsmith_projectlangsmith_base_urlc >URUUUS9Ul[R"S5bV[R"S5R 5R 5(a[ [R"S55OSUl[R"SS5Ul[[RS9Ul [R"SS5=(d [RnU(a[U5Ul/Ul[$R&"UR)55 [$R*"5Ul[.TU]`"S 0UDSUR,0D6 g) Nrr r!LANGSMITH_SAMPLING_RATE?LANGSMITH_DEFAULT_RUN_NAMELLMRun) llm_providerLANGSMITH_BATCH_SIZE flush_lock)get_credentials_from_envdefault_credentialsosgetenvstripisdigitfloat sampling_ratelangsmith_default_run_namer r LoggingCallbackasync_httpx_clientlitellmlangsmith_batch_sizeint batch_size log_queueasyncio create_taskperiodic_flushLockr*super__init__)selfrr r!kwargs _batch_size __class__s rrALangsmithLogger.__init__%s"$(#@#@//1$A$  yy23? 34::<DDFF "))56 7  +-)) ((+ '#9-==#  II,d 3 Sw7S7S  !+.DO57D//12!,,. >6>doo>rreturncJU=(d [R"S5nUc [S5eU=(d [R"S5=(d SnUc [S5eU=(d [R"S5=(d SnUc [S5e[UUUS 9$) NLANGSMITH_API_KEYz;Invalid Langsmith API Key given. _credentials_api_key=None.LANGSMITH_PROJECTzlitellm-completionz;Invalid Langsmith API Key given. _credentials_project=None.LANGSMITH_BASE_URLzhttps://api.smith.langchain.comzQ4R  'M   W+>!? WCW  'M   1yy-. 10  ! (N *242  r credentialsc URS05=(d 0nURS05=(d 0nURSUS5nURSUR5n URSURSS55n URSS5n URS S5n URS S5n URS S5n[R"S US U 35 URSS5nUc [ S5eUSnU SUUSUUSUSUSUS. nUSbUSS:XaUSUS'U (aU US'U (aU US'U (aU US 'U (aU US 'U(aUUS 'URS5n SU;dUSc$[ [ R"55n U US'S U;dUS cU b[U [ 5(aU US 'S U;dUS c*U b'[U [ 5(aURU S9US '[R"SU5 U$![a ef=f)Nlitellm_paramsmetadata project_namerJrun_nameidrun_id parent_run_idtrace_id session_id dotted_orderz"Langsmith Logging - project_name: z , run_name standard_logging_objectz,Error logging request payload. Payload=none.llmresponse startTimeendTime request_tags) namerun_typeinputsoutputs session_name start_timeend_timetagsextra error_strstatusfailureerror)rYz'Langsmith Logging data on langsmith: %s) getr4r debugrLstruuiduuid4rmake_dot_order)rBrC response_objrirjrR_litellm_paramsrUrVrWrYrZr[r\r]payloaddatas r_prepare_log_data!LangsmithLogger._prepare_log_dataksU $jj)92>D"O&**:r:@bH#<< ,? @L ||J0O0OPH\\$ Xt(DEF$LL$?M||J5H!lD9J#<<=L  4\N+hZX  9? )49G NOOH !!!":. ,%k2#I./! D{#/GH4E4R ' 4W #T (5_%#+Z %/\"'3^$$(HHTNF44:#5TZZ\*#T $& #+'Jvs,C,C#)Z d*'/'Jvs,C,C'+':':&':'I^$  !JD QK   s H5H88 Ic[R"S5bV[R"S5R5R5(a[ [R"S55OSn[ R "5nXe:a&[ R"SRXV55 g[ R"SUU5 URUS9nURUUUUUS9nURR[UUS95 [ R"SURS 35 [!UR5UR":aUR%5 gg![&a [ R("S 5 gf=f) Nr$r%>Skipping Langsmith logging. Sampling rate={}, random_sample={}z;Langsmith Sync Layer Logging - kwargs: %s, response_obj: %srCrCrwrirjrRrzrRz/Langsmith, event added to queue. Will flush in z seconds...z/Langsmith Layer Error - log_success_event error)r.r/r0r1r2randomr infoformatrr#_get_credentials_to_use_for_requestr{r;appendLangsmithQueueObjectflush_intervallenr: _send_batchrL exception rBrCrwrirjr3 random_samplerRrzs rlog_success_event!LangsmithLogger.log_success_eventsq* X9967CII78>>@HHJJbii 9:;  #MMOM,##T[[%   M  BB&BQK)))%!' *D NN ! !$ +   A$BUBUAVVab 4>>"doo5  "6 X  $ $%V W XsB.E'1B4E'' F  F c# URn[R"5nXe:a&[R"SR XV55 g[R "SUU5 UR US9nURUUUUUS9nURR[UUS95 [R "S[UR5UR5 [UR5UR:aUR5IShvN ggN![a [R"S5 gf=f7f)Nr~z>"doo5&&(((6(   $ $L  sHEA D!ECD!DD!ED!! EEEEc# URn[R"5nXe:a&[R"SR XV55 g[R"S5 UR US9nUR UUUUUS9nURR[UUS95 [R"S[UR5UR5 [UR5UR:aUR5IShvN ggN![a [R"S5 gf=f7f)Nr~z Langsmith Failure Event Logging!rrrrz:Langsmith Layer Error - error logging async failure event.)r3rr rrrr{r;rrrrrr:rrLrrs rasync_log_failure_event'LangsmithLogger.async_log_failure_event!s(**    (   PWW!  >? BB&BQK)))%!' *D NN ! !$ +   CDNN#  4>>"doo5&&(((6(   $ $L  s=A#E&B0DDDED E?EEEc# UR(dgUR5nUR5H/nURURUR S9IShvN M1 gN 7f)a Handles sending batches of runs to Langsmith self.log_queue contains LangsmithQueueObjects Each LangsmithQueueObject has the following: - "credentials" - credentials to use for the request (langsmith_api_key, langsmith_project, langsmith_base_url) - "data" - data to log on to langsmith for the request This function - groups the queue objects by credentials - loops through each unique credentials and sends batches to Langsmith This was added to support key/team based logging on langsmith NrR queue_objects)r;_group_batches_by_credentialsvalues_log_batch_on_langsmithrRr)rB batch_groups batch_groups rasync_send_batch LangsmithLogger.async_send_batchGsf"~~ 99; '..0K..'33)77/  1 sAA+A) A+rc(# USnUSnUS3nSU0nUVs/sHowSPM nn[R"S[U55 URR USU0US9IS hvN n U R 5 U R S :a1[R"S U R S U R35 g [R"S [UR5S35 g s snfN![RaNn [R"SU RR S U RR35 S n A g S n A f[a/ [R"S[R "535 g f=f7f)a Logs a batch of runs to Langsmith sends runs to /batch endpoint for the given credentials Args: credentials: LangsmithCredentialsObject queue_objects: List[LangsmithQueueObject] Returns: None Raises: Does not raise an exception, will only verbose_logger.exception() rKrIz /runs/batch x-api-keyrzz%Sending batch of %s runs to Langsmithpost)urljsonheadersNi,zLangsmith Error: z - z Batch of z runs successfully createdzLangsmith HTTP Error: zLangsmith Layer Error - )r rrrr6rraise_for_status status_coderptextr;httpxHTTPStatusErrorrr`rL traceback format_exc) rBrRrlangsmith_api_baserrr queue_objectelements_to_logr`es rr'LangsmithLogger._log_batch_on_langsmithbs"))=>'(;<#$K0 12DQRML/MR   7_9M "4499o.:H  % % '##s*$$'(<(<'=SP$$DNN 344NO%S $$   $ $()?)?(@AJJOOCTU     $ $*9+?+?+A*BC  seFC/FAC6+C4,AC6F-C6.F4C66F AEF9F FFFc0nURHJnUSn[USUSUSS9nXA;a [U/S9X'XRR U5 ML U$)z@Groups queue objects by credentials using a proper key structurerRrIrJrK)api_keyprojectbase_urlr)r;CredentialsKey BatchGrouprr)rBlog_queue_by_credentialsrrRkeys rr-LangsmithLogger._group_batches_by_credentialssEG  NNL&}5K #$78#$78$%9:C 20: +21(- % ) 7 7 > >| L+('rrCcURSS5nUbCURURSS5URSS5URSS5S9nU$URnU$)z Handles key/team based logging If standard_callback_dynamic_params are provided, use those credentials. Otherwise, use the default credentials. standard_callback_dynamic_paramsNrr r!r#)rqr,r-)rBrCrrRs rr3LangsmithLogger._get_credentials_to_use_for_requests JJ94 @ ) , 777"B"F"F'##C"F"F'#$D#G#G($$8 K22KrcnUR(dg[R"5nUR5(a%[R"UR 55 gUR UR 55 g![a' [R"UR 55 gf=f)z'Calls async_send_batch in an event loopN) r;r<get_event_loop is_runningr=rrun_until_complete RuntimeErrorrun)rBloops rrLangsmithLogger._send_batchs~~  1))+D  ##D$9$9$;<''(=(=(?@ 1 KK--/ 0 1sAB#B.B43B4cURSnURSnUSU3n[RRUSU0S9nUR 5$)NrIrKz/runs/r)rr)r-r7module_level_clientrqr)rBrYrrrr`s r get_run_by_idLangsmithLogger.get_run_by_idsl 445HI!556JK#$F6(3..22 "343  }}rrYc[R"[R5nUnUR S5[ U5-$)Nz%Y%m%dT%H%M%S%fZ)rnowrutcstrftimers)rBrYstid_s rrvLangsmithLogger.make_dot_orders2 \\(,, '{{-.S99r)r6r:r-r*r4r;r3)NNN)__name__ __module__ __qualname____firstlineno__rrsrArMr,r{rrrrrrrrrrrrrrrrv__static_attributes__ __classcell__)rEs@rrr$s,0+/,0 !?#C=!?$C=!?%SM !?!?J,0+/,0 ! #C=! $C=! %SM ! $ ! F] 0 ]~+XZ(T$L61/1011f(tNJ4N/O(*38n #:1$ :S::rr)r<r.rrrrtrrtypingrrrrrpydanticr r7litellm._loggingr (litellm.integrations.custom_batch_loggerr &litellm.llms.custom_httpx.http_handlerr r $litellm.types.integrations.langsmithlitellm.types.utilsrrrrr+rrrsR  ',, +F3U9F:'F:r