From ca790449482752ccfe17b8e646384aebed27a405 Mon Sep 17 00:00:00 2001 From: William FH <13333726+hinthornw@users.noreply.github.com> Date: Sun, 16 Jul 2023 20:58:13 -0700 Subject: [PATCH] Export Tracer from callbacks (#7812) Improve discoverability --- langchain/callbacks/__init__.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/langchain/callbacks/__init__.py b/langchain/callbacks/__init__.py index a8f21e26183..0d9335b23e3 100644 --- a/langchain/callbacks/__init__.py +++ b/langchain/callbacks/__init__.py @@ -14,6 +14,7 @@ from langchain.callbacks.infino_callback import InfinoCallbackHandler from langchain.callbacks.manager import ( get_openai_callback, tracing_enabled, + tracing_v2_enabled, wandb_tracing_enabled, ) from langchain.callbacks.mlflow_callback import MlflowCallbackHandler @@ -26,6 +27,7 @@ from langchain.callbacks.streaming_stdout_final_only import ( FinalStreamingStdOutCallbackHandler, ) from langchain.callbacks.streamlit import LLMThoughtLabeler, StreamlitCallbackHandler +from langchain.callbacks.tracers.langchain import LangChainTracer from langchain.callbacks.wandb_callback import WandbCallbackHandler from langchain.callbacks.whylabs_callback import WhyLabsCallbackHandler @@ -48,11 +50,13 @@ __all__ = [ "StreamingStdOutCallbackHandler", "FinalStreamingStdOutCallbackHandler", "LLMThoughtLabeler", + "LangChainTracer", "StreamlitCallbackHandler", "WandbCallbackHandler", "WhyLabsCallbackHandler", "get_openai_callback", "tracing_enabled", + "tracing_v2_enabled", "wandb_tracing_enabled", "FlyteCallbackHandler", ]