Cj#]SSKrSSKrSSKrSSKrSSKrSSKJrJr SSKJrJ r J r J r SSK r SSK Jr SSKrSSKJr SSKJr SSKJr SSKJrJr SS KJr SS KJrJr SS K7 SS KJ r J!r! S r""SS\5r#g)N)datetimetimezone)AnyDictListOptional) BaseModel)verbose_logger)uuid)CustomBatchLogger)create_mock_langsmith_clientshould_use_langsmith_mock)redact_user_api_key_info)get_async_httpx_clienthttpxSpecialProvider)*)StandardCallbackDynamicParamsStandardLoggingPayloadc[R[R[R[4n[ X5(+$N)types CoroutineType FunctionType GeneratorTyper isinstance)valuenon_serializable_typess p/home/rurouni/.local/share/pipx/venvs/strix-agent/lib/python3.13/site-packages/litellm/integrations/langsmith.pyis_serializablers8     %8 88c ^\rSrSrS)S\\S\\S\\S\\S\\4 U4SjjjrS\\R\ 4S jr S*S jr S+S\\S\\S\\S\\S \ S\4 S jjrS \S\4SjrS \4SjrS\S\\\ 44SjrS\S\\4SjrS\4SjrSrSrSrSrS,S\S\S\S\4SjjrS\S\\4S jrS\\ \!44S!jr"S"\\\ 4S\4S#jr#S"\\\ 4S\4S$jr$S%r%S&r&S\4S'jr'S(r(U=r)$)-LangsmithLogger)langsmith_api_keylangsmith_projectlangsmith_base_urllangsmith_sampling_ratelangsmith_tenant_idc P>[R"5Ul[TU]"S 0UDSUR0D6 [ 5UlUR (a [5 [R"S5 URUUUUS9Ul [R"S5b_[R"S5R5R5(a(U=(d [![R"S55OSUl[R"SS5Ul['[(R*S9Ul[R"S S5=(d [.R0nU(a[3U5Ul/UlUR95Ulg) N flush_lockz:[LANGSMITH MOCK] LangSmith logger initialized in mock mode)r$r%r&r(LANGSMITH_SAMPLING_RATEg?LANGSMITH_DEFAULT_RUN_NAMELLMRun) llm_providerLANGSMITH_BATCH_SIZE)asyncioLockr*super__init__r is_mock_moder r debugget_credentials_from_envdefault_credentialsosgetenvstripisdigitfloat sampling_ratelangsmith_default_run_namerrLoggingCallbackasync_httpx_clientlitellmlangsmith_batch_sizeint batch_size log_queue_start_periodic_flush_task _flush_task) selfr$r%r&r'r(kwargs _batch_size __class__s rr4LangsmithLogger.__init__*sX",,. >6>doo>57    ( *  L $(#@#@//1 3 $A$  yy23? 34::<DDFF $ ;RYY89: +-)) ((+ '#9-==#  II,d 3 Sw7S7S  !+.DO57  + + - r returnc[R"5nUR UR 55$![a [R"S5 gf=f)zIStart the periodic flush task only when an event loop is already running.zRLangsmith logger init: no running event loop, skipping periodic flush task startupN)r1get_running_loop RuntimeErrorr r6 create_taskperiodic_flushrIloops rrG*LangsmithLogger._start_periodic_flush_task[sX ++-D 3 3 566    d   s6 AAcURbURR5(aUR5Ulggr)rHdonerG)rIs r_ensure_periodic_flush_task+LangsmithLogger._ensure_periodic_flush_taskgs:    #t'7'7'<'<'>'>#>>@D (?r allow_env_credentialscjUSLaUbUnU=(d SnUnUn OU=(d [R"S5nU=(d [R"S5=(d SnU=(d [R"S5=(d SnU=(d [R"S5n [UUUU S9$) NFzlitellm-completionLANGSMITH_API_KEYLANGSMITH_PROJECTLANGSMITH_BASE_URLzhttps://api.smith.langchain.comLANGSMITH_TENANT_ID)r]r_r^r`)r9r:LangsmithCredentialsObject) rIr$r%r&r(r[_credentials_api_key_credentials_project_credentials_base_url_credentials_tenant_ids rr7(LangsmithLogger.get_credentials_from_envns !E ).@.L#4 #4#L8L $6 !%8 "#4#V BU8V !(9901(' ! #5991254 " &9&BII%= "*242 6   r metadata credentialsc 0URSUS5URSUR5URSURSS55URSS5URSS5URSS5URS S5S .$) N project_namer^run_nameidrun_id parent_run_idtrace_id session_id dotted_order)rjrkrmrnrorprq)getr?)rIrgrhs r_extract_metadata_fields(LangsmithLogger._extract_metadata_fieldss%LL ,? @! Z1P1PQll4h)EF%\\/4@ Z6",,|T:$LL>  r c&[U5nURS5nU(a2[U[5(aSHnXC;dM XB;dMX4X$'M [US9nURS5n[U[5(a [US9US'U$)Nrequester_metadata)rp thread_idconversation_id)rg)dictrrrr)rIrgextra_metadatarvkeynesteds r_build_extra_metadata%LangsmithLogger._build_extra_metadatash+//0DE *-?"F"FE,1J*<*AN'F 2>J##$89 fd # #3K4N/ 0r payloadcUSn[U[5(a0UEnOSU0nURSS5URSS5URSS5URSS5S.US 'U$) Nresponseoutput prompt_tokensrcompletion_tokens total_tokens response_cost) input_tokens output_tokensr total_costusage_metadata)rryrr)rIrroutputss r_build_outputs_with_usage)LangsmithLogger._build_outputs_with_usages{:& h % %"lG*G#KK;$[[)!*Fr c URS05=(d 0nURS05=(d 0nURXu5n[R"SUSSUS35 URSS5n U c [ S5eU SnUR [ U55n URU 5n USS U U USU S U S U S U S . n U SbU SS:XaU SU S'SHn U S:XaSOU nX(dMXX'M URXS5 [R"SU 5 U $![a ef=f)Nlitellm_paramsrgz"Langsmith Logging - project_name: rjz , run_name rkstandard_logging_objectz,Error logging request payload. Payload=none.llm startTimeendTime request_tags) namerun_typeinputsr session_name start_timeend_timetagsextra error_strstatusfailureerror)rlrnrorprqrlrmz'Langsmith Logging data on langsmith: %s) rrrsr r6 Exceptionr}ryrr)rIrJ response_objrrrh_litellm_paramsrgfieldsrrzrrr{ field_keys r_prepare_log_data!LangsmithLogger._prepare_log_datas1 $jj)92>D"O&**:r:@bH228IF  4VN5K4LKX^_iXjWkl 9? )49G NOOz*H!77XGN44W=Gz*!!" &~ 6%k2#I./' D{#/GH4E4R ' 4W ),t H $$ & 1DI  % %d8,< =  !JD QK   sDD>6D>> E cZURUS9n[R"5nXe:a&[R"SR XV55 g[R "SUU5 UR US9nURUUUUUS9nURR[UUS95 [R "SURS35 [UR5UR:aUR5 gg![a [R "S5 gf=f) NrJ>Skipping Langsmith logging. Sampling rate={}, random_sample={}z;Langsmith Sync Layer Logging - kwargs: %s, response_obj: %srJrrrrhrrhz/Langsmith, event added to queue. Will flush in z seconds...z/Langsmith Layer Error - log_success_event error)%_get_sampling_rate_to_use_for_requestrandomr infoformatr6#_get_credentials_to_use_for_requestrrFappendLangsmithQueueObjectflush_intervallenrE _send_batchr exception rIrJrrrr> random_samplerhrs rlog_success_event!LangsmithLogger.log_success_event s0& X FFfFUM"MMOM,##T[[%   M  BB&BQK)))%!' *D NN ! !$ +   A$BUBUAVVab 4>>"doo5  "6 X  $ $%V W XsADB4D D*)D*c# UR5 URUS9n[R"5nXe:a&[R"SR XV55 g[R "SUU5 URUS9nURUUUUUS9nURR[UUS95 [R "S[UR5UR5 [UR5UR:aUR5IShvN ggN![a [R "S5 gf=f7f)Nrrz>"doo5&&(((6(   $ $L  sHEAD4"E#CD4+D2,D40E2D44 EEEEc# UR5 URUS9n[R"5nXe:a&[R"SR XV55 g[R"S5 UR US9nURUUUUUS9nURR[UUS95 [R"S[UR5UR5 [UR5UR:aUR5IShvN ggN![a [R "S5 gf=f7f)Nrrz Langsmith Failure Event Logging!rrrz:Langsmith Layer Error - error logging async failure event.)rYrrr rrrrrFrrr6rrErrrrs rasync_log_failure_event'LangsmithLogger.async_log_failure_event`s<$   , , . FFfFUM"MMOM,##T[[%     B CBB&BQK)))%!' *D NN ! !$ +   CDNN#  4>>"doo5&&(((6(   $ $L  sHEAD2"E#CD2)D0*D2.E0D22 EEEEc# 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 Nrh queue_objects)rF_group_batches_by_credentialsvalues_log_batch_on_langsmithrhr)rI batch_groups batch_groups rasync_send_batch LangsmithLogger.async_send_batchsf"~~ 99; '..0K..'33)77/  1 sAA+A) A+urlendpoint api_versionc|X1;aURS5U3nURS5(aUU3$USU3$)N/)rstripendswith)rIrrrs r_add_endpoint_to_url$LangsmithLogger._add_endpoint_to_urlsP  !ZZ_%k]3C <<  U8*% %az""r rc2# USnUSnURS5nURUS5nSU0nU(aXWS'UVs/sHoSPM n n[R"S[ U 55 UR (a[R"S 5 UR RUS U 0US 9IS hvN n U R5 U RS :a1[R"SU RSU R35 g UR (a$[R"S[ U 5S35 g [R"S[ UR5S35 g s snfN![RaNn [R"SU R RSU R R35 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() r_r]r`z runs/batch x-api-key x-tenant-idrz%Sending batch of %s runs to LangsmithzB[LANGSMITH MOCK] Mock mode enabled - API calls will be interceptedpost)rjsonheadersNi,zLangsmith Error: z - z[LANGSMITH MOCK] Batch of z runs successfully mockedz Batch of z runs successfully createdzLangsmith HTTP Error: zLangsmith Layer Error - )rrrr r6rr5rArraise_for_status status_codertextrFhttpxHTTPStatusErrorrrr traceback format_exc) rIrhrlangsmith_api_baser$r(rr queue_objectelements_to_logres rr'LangsmithLogger._log_batch_on_langsmiths"))=>'(;<)oo.CD''(:LI 12 %8M "DQRML/MR#   7_9M   $$X"4499o.:H  % % '##s*$$'(<(<'=SP$$"((4S5I4JJcd#((#C$7#88RS7S($$   $ $()?)?(@AJJOOCTU     $ $*9+?+?+A*BC  suAHE4HA'E;;E9 >| L3+6('r rJcURSS5nURnUbURS5nUb [U5nU$)N standard_callback_dynamic_paramsr')rrr>r=)rIrJrr>_sampling_rates rr5LangsmithLogger._get_sampling_rate_to_use_for_request sT JJ94 @ ) $11 + 7=AA)N) %n 5 r c URSS5nUbgURURSS5URSS5URSS5URSS5URSS5SLS9nU$URnU$)z Handles key/team based logging If standard_callback_dynamic_params are provided, use those credentials. Otherwise, use the default credentials. rNr$r%r&r()r$r%r&r(r[)rrr7r8)rIrJrrhs rr3LangsmithLogger._get_credentials_to_use_for_requests JJ94 @ ) , 777"B"F"F'##C"F"F'#$D#G#G($$%E$H$H)4%'G&J&J($''8K(22Kr cnUR(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) rFr1get_event_loop is_runningrRrrun_until_completerQrunrTs rrLangsmithLogger._send_batch=s~~  1))+D  ##D$9$9$;<''(=(=(?@ 1 KK--/ 0 1sAB#B.B43B4cURSnURSnURRS5nUSU3nSU0nU(aXFS'[RRUUS9nUR 5$)Nr]r_r`z/runs/rr)rr)r8rrrBmodule_level_clientr)rIrmr$rr(rrrs r get_run_by_idLangsmithLogger.get_run_by_idOs 445HI!556JK"66::;PQ#$F6(3 12 %8M "..223  }}r c[R"[R5nUnUR S5[ U5-$)Nz%Y%m%dT%H%M%S%fZ)rnowrutcstrftimer)rIrmstid_s rrLangsmithLogger.make_dot_order_s2 \\(,, '{{-.S99r ) rHrArEr8r*r5r?rFr>)NNNNN)rNN)NNNNT)z/api/v1)*__name__ __module__ __qualname____firstlineno__rrr=r4r1TaskrrGrYboolrar7ryrsrr}rrrrrrrrrrrrrrrrrrrr__static_attributes__ __classcell__)rLs@rr"r")s,0+/,037-1 / #C=/ $C=/ %SM / "*% / &c] / / b 7HW\\#5F,G 7A,0+/,0-1&* " #C=" $C=" %SM " &c] " $ "  $" H    +E  d$- c3h" J Jx} J9 0 9v'XR)V%N8;D##"%#47# #=/=01=~(tNJ4N/O(B DcN u "38n" #"H1$ :S::r r")$r1r9rrrrrtypingrrrrrpydanticr rBlitellm._loggingr litellm._uuidr (litellm.integrations.custom_batch_loggerr *litellm.integrations.langsmith_mock_clientr r*litellm.litellm_core_utils.redact_messagesr&litellm.llms.custom_httpx.http_handlerrr$litellm.types.integrations.langsmithlitellm.types.utilsrrrr"r0r rr#s]  ',, +FP3U9y:'y:r