g3fi?UdZddlmZddlmZddlmZddlmZm Z m Z m Z ddl m Z ddlmZddlmZdd lmZdd lmZerdd lmZdd lmZdd lmZmZddlmZm Z dZ!de"d<eddZ#de"d<eddZ$de"d<e d$dddd d%dZ%ed&dZ& d' d(dZ'd)dZ(d*dZ)gZ*de"d < d+ d,d!Z+e+e$d"#y)-zContext management for tracers.) annotations)contextmanager) ContextVar) TYPE_CHECKINGAnyLiteralcast)UUID) run_helpers)utils)LangChainTracer)RunCollectorCallbackHandler) Generator)Client)BaseCallbackHandler Callbacks)AsyncCallbackManagerCallbackManagerNrtracing_callback_vartracing_callback_v2)defaultz"ContextVar[LangChainTracer | None]tracing_v2_callback_var run_collectorz.ContextVar[RunCollectorCallbackHandler | None]run_collector_var) example_idtagsclientc#Kt|tr t|}t||||}tj |} |tj |y#tj |wxYww)ajInstruct LangChain to log all runs in context to LangSmith. Args: project_name: The name of the project. Defaults to `'default'`. example_id: The ID of the example. tags: The tags to add to the run. client: The client of the langsmith. Yields: The LangChain tracer. Example: >>> with tracing_v2_enabled(): ... # LangChain code will automatically be traced You can use this to fetch the LangSmith run URL: >>> with tracing_v2_enabled() as cb: ... chain.invoke("foo") ... run_url = cb.get_run_url() )r project_namerrN) isinstancestrr r rsetreset)rrrrcbtokens \/var/www/auto_recruiter/arenv/lib/python3.12/site-packages/langchain_core/tracers/context.pytracing_v2_enabledr''sk:*c"*% !   B $ ' ' +E-%%e,%%e,sAA7AA7A44A7c#Kt}tj|} |tj|y#tj|wxYww)zCollect all run traces in context. Yields: The run collector callback handler. Example: >>> with collect_runs() as runs_cb: chain.invoke("foo") run_id = runs_cb.traced_runs[0].id N)rrr"r#)r$r%s r& collect_runsr)SsG % &B  ! !" %E'&&s A=AAActrs|xs t}tjxs t ||}|t d|g}|St d|jDs|j||}|Sd}|S)N)rrrc3<K|]}t|tywN)r r ).0handlers r& z'_get_trace_callbacks..us7O4s) _tracing_v2_is_enabled_get_tracer_projectrgetr r anyhandlers add_handler)rrcallback_manager project_name_tracerr$s r&_get_trace_callbacksr9gs $=(;(= (,,. /&!3   #kF8,B I/88!,,V4"B I IcTtjytjS)NT)rr2ls_utilstracing_is_enabledr:r&r0r0s#""$0  & & ((r:c tj}|d}||dtd|dSt|dttj dt tjS)Nparentrr! session_nameproject) ls_rhget_tracing_contextr getattrrr2r!r<get_tracer_project)tracing_contextrun_trees r&r1r1s{//1Ox(HON;GE?>:;;  $ ' ' )  ++- .  r:zglist[tuple[ContextVar[BaseCallbackHandler | None], bool, type[BaseCallbackHandler] | None, str | None]]_configure_hookscl|| d}t|tjtd||||fy)agRegister a configure hook. Args: context_var: The context variable. inheritable: Whether the context variable is inheritable. handle_class: The callback handler class. env_var: The environment variable. Raises: ValueError: If env_var is set, handle_class must also be set to a non-None value. NzEIf env_var is set, handle_class must also be set to a non-None value.z&ContextVar[BaseCallbackHandler | None]) ValueErrorrIappendr ) context_var inheritable handle_classenv_varmsgs r&register_configure_hookrRsG$|3Uo 9; G      r:F)rNr,) r str | Nonerstr | UUID | Nonerzlist[str] | NonerzLangSmithClient | Nonereturnz&Generator[LangChainTracer, None, None])rUz2Generator[RunCollectorCallbackHandler, None, None])NNN)rrSrrTr6z-CallbackManager | AsyncCallbackManager | NonerUr)rUzbool | Literal['local'])rUr!)NN) rMzContextVar[Any | None]rNboolrOz type[BaseCallbackHandler] | NonerPrSrUNone),__doc__ __future__r contextlibr contextvarsrtypingrrrr uuidr langsmithr rCr r< langchain_core.tracers.langchainr $langchain_core.tracers.run_collectorrcollections.abcrrLangSmithClientlangchain_core.callbacks.baserr langchain_core.callbacks.managerrrr__annotations__rrr'r)r9r0r1rIrRr>r:r&rfs%"%" *'<L)3LV!c >H4?;EOTEA #(-%)!%) (-(-"(-  (- # (- , (-(-V''( $$(FJ!D 6) <6: '3   D)u=r: