core[patch], langchain[patch], community[patch]: Revert #15326 (#15546)

libs/langchain/langchain/callbacks/__init__.py

@@ -44,6 +44,7 @@ from langchain_core.callbacks import (
 )
 from langchain_core.tracers.context import (
     collect_runs,
+    tracing_enabled,
     tracing_v2_enabled,
 )
 from langchain_core.tracers.langchain import LangChainTracer
@@ -79,6 +80,7 @@ __all__ = [
     "WandbCallbackHandler",
     "WhyLabsCallbackHandler",
     "get_openai_callback",
+    "tracing_enabled",
     "tracing_v2_enabled",
     "collect_runs",
     "wandb_tracing_enabled",

libs/langchain/langchain/callbacks/manager.py

@@ -30,6 +30,7 @@ from langchain_core.callbacks.manager import (
 )
 from langchain_core.tracers.context import (
     collect_runs,
+    tracing_enabled,
     tracing_v2_enabled,
 )
 from langchain_core.utils.env import env_var_is_set
@@ -52,6 +53,7 @@ __all__ = [
     "CallbackManagerForChainGroup",
     "AsyncCallbackManager",
     "AsyncCallbackManagerForChainGroup",
+    "tracing_enabled",
     "tracing_v2_enabled",
     "collect_runs",
     "atrace_as_chain_group",

libs/langchain/langchain/callbacks/tracers/__init__.py

@@ -1,6 +1,7 @@
 """Tracers that record execution of LangChain runs."""
 
 from langchain_core.tracers.langchain import LangChainTracer
+from langchain_core.tracers.langchain_v1 import LangChainTracerV1
 from langchain_core.tracers.stdout import (
     ConsoleCallbackHandler,
     FunctionCallbackHandler,
@@ -14,5 +15,6 @@ __all__ = [
     "FunctionCallbackHandler",
     "LoggingCallbackHandler",
     "LangChainTracer",
+    "LangChainTracerV1",
     "WandbTracer",
 ]

libs/langchain/langchain/callbacks/tracers/langchain_v1.py

@@ -0,0 +1,3 @@
+from langchain_core.tracers.langchain_v1 import LangChainTracerV1
+
+__all__ = ["LangChainTracerV1"]

libs/langchain/langchain/callbacks/tracers/schemas.py

@@ -1,7 +1,27 @@
 from langchain_core.tracers.schemas import (
+    BaseRun,
+    ChainRun,
+    LLMRun,
     Run,
+    RunTypeEnum,
+    ToolRun,
+    TracerSession,
+    TracerSessionBase,
+    TracerSessionV1,
+    TracerSessionV1Base,
+    TracerSessionV1Create,
 )
 
 __all__ = [
+    "BaseRun",
+    "ChainRun",
+    "LLMRun",
     "Run",
+    "RunTypeEnum",
+    "ToolRun",
+    "TracerSession",
+    "TracerSessionBase",
+    "TracerSessionV1",
+    "TracerSessionV1Base",
+    "TracerSessionV1Create",
 ]

libs/langchain/langchain/schema/callbacks/manager.py

@@ -22,11 +22,13 @@ from langchain_core.callbacks.manager import (
 from langchain_core.tracers.context import (
     collect_runs,
     register_configure_hook,
+    tracing_enabled,
     tracing_v2_enabled,
 )
 from langchain_core.utils.env import env_var_is_set
 
 __all__ = [
+    "tracing_enabled",
     "tracing_v2_enabled",
     "collect_runs",
     "trace_as_chain_group",

libs/langchain/langchain/schema/callbacks/tracers/langchain_v1.py

@@ -0,0 +1,3 @@
+from langchain_core.tracers.langchain_v1 import LangChainTracerV1, get_headers
+
+__all__ = ["get_headers", "LangChainTracerV1"]

libs/langchain/langchain/schema/callbacks/tracers/schemas.py

@@ -1,7 +1,27 @@
 from langchain_core.tracers.schemas import (
+    BaseRun,
+    ChainRun,
+    LLMRun,
     Run,
+    RunTypeEnum,
+    ToolRun,
+    TracerSession,
+    TracerSessionBase,
+    TracerSessionV1,
+    TracerSessionV1Base,
+    TracerSessionV1Create,
 )
 
 __all__ = [
+    "RunTypeEnum",
+    "TracerSessionV1Base",
+    "TracerSessionV1Create",
+    "TracerSessionV1",
+    "TracerSessionBase",
+    "TracerSession",
+    "BaseRun",
+    "LLMRun",
+    "ChainRun",
+    "ToolRun",
     "Run",
 ]

libs/langchain/tests/unit_tests/callbacks/test_imports.py

@@ -25,6 +25,7 @@ EXPECTED_ALL = [
     "WandbCallbackHandler",
     "WhyLabsCallbackHandler",
     "get_openai_callback",
+    "tracing_enabled",
     "tracing_v2_enabled",
     "collect_runs",
     "wandb_tracing_enabled",

libs/langchain/tests/unit_tests/callbacks/test_manager.py

@@ -18,6 +18,7 @@ EXPECTED_ALL = [
     "CallbackManagerForChainGroup",
     "AsyncCallbackManager",
     "AsyncCallbackManagerForChainGroup",
+    "tracing_enabled",
     "tracing_v2_enabled",
     "collect_runs",
     "atrace_as_chain_group",