langchain[patch]: Migrate callbacks module to use optional imports for community (#21086)

libs/langchain/langchain/callbacks/__init__.py

@@ -6,10 +6,8 @@
 
     BaseCallbackHandler --> <name>CallbackHandler  # Example: AimCallbackHandler
 """
-import warnings
+from typing import TYPE_CHECKING, Any
-from typing import Any
 
-from langchain_core._api import LangChainDeprecationWarning
 from langchain_core.callbacks import (
     FileCallbackHandler,
     StdOutCallbackHandler,
@@ -22,28 +20,86 @@ from langchain_core.tracers.context import (
 )
 from langchain_core.tracers.langchain import LangChainTracer
 
+from langchain._api import create_importer
 from langchain.callbacks.streaming_aiter import AsyncIteratorCallbackHandler
 from langchain.callbacks.streaming_stdout_final_only import (
     FinalStreamingStdOutCallbackHandler,
 )
-from langchain.utils.interactive_env import is_interactive_env
+
+if TYPE_CHECKING:
+    from langchain_community.callbacks.aim_callback import AimCallbackHandler
+    from langchain_community.callbacks.argilla_callback import ArgillaCallbackHandler
+    from langchain_community.callbacks.arize_callback import ArizeCallbackHandler
+    from langchain_community.callbacks.arthur_callback import ArthurCallbackHandler
+    from langchain_community.callbacks.clearml_callback import ClearMLCallbackHandler
+    from langchain_community.callbacks.comet_ml_callback import CometCallbackHandler
+    from langchain_community.callbacks.context_callback import ContextCallbackHandler
+    from langchain_community.callbacks.flyte_callback import FlyteCallbackHandler
+    from langchain_community.callbacks.human import HumanApprovalCallbackHandler
+    from langchain_community.callbacks.infino_callback import InfinoCallbackHandler
+    from langchain_community.callbacks.labelstudio_callback import (
+        LabelStudioCallbackHandler,
+    )
+    from langchain_community.callbacks.llmonitor_callback import (
+        LLMonitorCallbackHandler,
+    )
+    from langchain_community.callbacks.manager import (
+        get_openai_callback,
+        wandb_tracing_enabled,
+    )
+    from langchain_community.callbacks.mlflow_callback import MlflowCallbackHandler
+    from langchain_community.callbacks.openai_info import OpenAICallbackHandler
+    from langchain_community.callbacks.promptlayer_callback import (
+        PromptLayerCallbackHandler,
+    )
+    from langchain_community.callbacks.sagemaker_callback import (
+        SageMakerCallbackHandler,
+    )
+    from langchain_community.callbacks.streamlit import StreamlitCallbackHandler
+    from langchain_community.callbacks.streamlit.streamlit_callback_handler import (
+        LLMThoughtLabeler,
+    )
+    from langchain_community.callbacks.trubrics_callback import TrubricsCallbackHandler
+    from langchain_community.callbacks.wandb_callback import WandbCallbackHandler
+    from langchain_community.callbacks.whylabs_callback import WhyLabsCallbackHandler
+
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {
+    "AimCallbackHandler": "langchain_community.callbacks.aim_callback",
+    "ArgillaCallbackHandler": "langchain_community.callbacks.argilla_callback",
+    "ArizeCallbackHandler": "langchain_community.callbacks.arize_callback",
+    "PromptLayerCallbackHandler": "langchain_community.callbacks.promptlayer_callback",
+    "ArthurCallbackHandler": "langchain_community.callbacks.arthur_callback",
+    "ClearMLCallbackHandler": "langchain_community.callbacks.clearml_callback",
+    "CometCallbackHandler": "langchain_community.callbacks.comet_ml_callback",
+    "ContextCallbackHandler": "langchain_community.callbacks.context_callback",
+    "HumanApprovalCallbackHandler": "langchain_community.callbacks.human",
+    "InfinoCallbackHandler": "langchain_community.callbacks.infino_callback",
+    "MlflowCallbackHandler": "langchain_community.callbacks.mlflow_callback",
+    "LLMonitorCallbackHandler": "langchain_community.callbacks.llmonitor_callback",
+    "OpenAICallbackHandler": "langchain_community.callbacks.openai_info",
+    "LLMThoughtLabeler": (
+        "langchain_community.callbacks.streamlit.streamlit_callback_handler"
+    ),
+    "StreamlitCallbackHandler": "langchain_community.callbacks.streamlit",
+    "WandbCallbackHandler": "langchain_community.callbacks.wandb_callback",
+    "WhyLabsCallbackHandler": "langchain_community.callbacks.whylabs_callback",
+    "get_openai_callback": "langchain_community.callbacks.manager",
+    "wandb_tracing_enabled": "langchain_community.callbacks.manager",
+    "FlyteCallbackHandler": "langchain_community.callbacks.flyte_callback",
+    "SageMakerCallbackHandler": "langchain_community.callbacks.sagemaker_callback",
+    "LabelStudioCallbackHandler": "langchain_community.callbacks.labelstudio_callback",
+    "TrubricsCallbackHandler": "langchain_community.callbacks.trubrics_callback",
+}
+
+_import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)
 
 
 def __getattr__(name: str) -> Any:
-    from langchain_community import callbacks
+    """Look up attributes dynamically."""
-
+    return _import_attribute(name)
-    # If not in interactive env, raise warning.
-    if not is_interactive_env():
-        warnings.warn(
-            "Importing this callback from langchain is deprecated. Importing it from "
-            "langchain will no longer be supported as of langchain==0.2.0. "
-            "Please import from langchain-community instead:\n\n"
-            f"`from langchain_community.callbacks import {name}`.\n\n"
-            "To install langchain-community run `pip install -U langchain-community`.",
-            category=LangChainDeprecationWarning,
-        )
-
-    return getattr(callbacks, name)
 
 
 __all__ = [

libs/langchain/langchain/callbacks/aim_callback.py

@@ -1,7 +1,33 @@
-from langchain_community.callbacks.aim_callback import (
+from typing import TYPE_CHECKING, Any
-    AimCallbackHandler,
-    BaseMetadataCallbackHandler,
-    import_aim,
-)
 
-__all__ = ["import_aim", "BaseMetadataCallbackHandler", "AimCallbackHandler"]
+from langchain._api import create_importer
+
+if TYPE_CHECKING:
+    from langchain_community.callbacks.aim_callback import (
+        AimCallbackHandler,
+        BaseMetadataCallbackHandler,
+        import_aim,
+    )
+
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {
+    "import_aim": "langchain_community.callbacks.aim_callback",
+    "BaseMetadataCallbackHandler": "langchain_community.callbacks.aim_callback",
+    "AimCallbackHandler": "langchain_community.callbacks.aim_callback",
+}
+
+_import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)
+
+
+def __getattr__(name: str) -> Any:
+    """Look up attributes dynamically."""
+    return _import_attribute(name)
+
+
+__all__ = [
+    "import_aim",
+    "BaseMetadataCallbackHandler",
+    "AimCallbackHandler",
+]

libs/langchain/langchain/callbacks/argilla_callback.py

@@ -1,3 +1,25 @@
-from langchain_community.callbacks.argilla_callback import ArgillaCallbackHandler
+from typing import TYPE_CHECKING, Any
 
-__all__ = ["ArgillaCallbackHandler"]
+from langchain._api import create_importer
+
+if TYPE_CHECKING:
+    from langchain_community.callbacks.argilla_callback import ArgillaCallbackHandler
+
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {
+    "ArgillaCallbackHandler": "langchain_community.callbacks.argilla_callback"
+}
+
+_import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)
+
+
+def __getattr__(name: str) -> Any:
+    """Look up attributes dynamically."""
+    return _import_attribute(name)
+
+
+__all__ = [
+    "ArgillaCallbackHandler",
+]

libs/langchain/langchain/callbacks/arize_callback.py

@@ -1,3 +1,25 @@
-from langchain_community.callbacks.arize_callback import ArizeCallbackHandler
+from typing import TYPE_CHECKING, Any
 
-__all__ = ["ArizeCallbackHandler"]
+from langchain._api import create_importer
+
+if TYPE_CHECKING:
+    from langchain_community.callbacks.arize_callback import ArizeCallbackHandler
+
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {
+    "ArizeCallbackHandler": "langchain_community.callbacks.arize_callback"
+}
+
+_import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)
+
+
+def __getattr__(name: str) -> Any:
+    """Look up attributes dynamically."""
+    return _import_attribute(name)
+
+
+__all__ = [
+    "ArizeCallbackHandler",
+]

libs/langchain/langchain/callbacks/arthur_callback.py

@@ -1,6 +1,24 @@
-from langchain_community.callbacks.arthur_callback import (
+from typing import TYPE_CHECKING, Any
-    ArthurCallbackHandler,
+
-)
+from langchain._api import create_importer
+
+if TYPE_CHECKING:
+    from langchain_community.callbacks.arthur_callback import ArthurCallbackHandler
+
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {
+    "ArthurCallbackHandler": "langchain_community.callbacks.arthur_callback"
+}
+
+_import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)
+
+
+def __getattr__(name: str) -> Any:
+    """Look up attributes dynamically."""
+    return _import_attribute(name)
+
 
 __all__ = [
     "ArthurCallbackHandler",

libs/langchain/langchain/callbacks/clearml_callback.py

@@ -1,5 +1,25 @@
-from langchain_community.callbacks.clearml_callback import (
+from typing import TYPE_CHECKING, Any
-    ClearMLCallbackHandler,
-)
 
-__all__ = ["ClearMLCallbackHandler"]
+from langchain._api import create_importer
+
+if TYPE_CHECKING:
+    from langchain_community.callbacks.clearml_callback import ClearMLCallbackHandler
+
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {
+    "ClearMLCallbackHandler": "langchain_community.callbacks.clearml_callback"
+}
+
+_import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)
+
+
+def __getattr__(name: str) -> Any:
+    """Look up attributes dynamically."""
+    return _import_attribute(name)
+
+
+__all__ = [
+    "ClearMLCallbackHandler",
+]

libs/langchain/langchain/callbacks/comet_ml_callback.py

@@ -1,6 +1,24 @@
-from langchain_community.callbacks.comet_ml_callback import (
+from typing import TYPE_CHECKING, Any
-    CometCallbackHandler,
+
-)
+from langchain._api import create_importer
+
+if TYPE_CHECKING:
+    from langchain_community.callbacks.comet_ml_callback import CometCallbackHandler
+
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {
+    "CometCallbackHandler": "langchain_community.callbacks.comet_ml_callback"
+}
+
+_import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)
+
+
+def __getattr__(name: str) -> Any:
+    """Look up attributes dynamically."""
+    return _import_attribute(name)
+
 
 __all__ = [
     "CometCallbackHandler",

libs/langchain/langchain/callbacks/confident_callback.py

@@ -1,3 +1,25 @@
-from langchain_community.callbacks.confident_callback import DeepEvalCallbackHandler
+from typing import TYPE_CHECKING, Any
 
-__all__ = ["DeepEvalCallbackHandler"]
+from langchain._api import create_importer
+
+if TYPE_CHECKING:
+    from langchain_community.callbacks.confident_callback import DeepEvalCallbackHandler
+
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {
+    "DeepEvalCallbackHandler": "langchain_community.callbacks.confident_callback"
+}
+
+_import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)
+
+
+def __getattr__(name: str) -> Any:
+    """Look up attributes dynamically."""
+    return _import_attribute(name)
+
+
+__all__ = [
+    "DeepEvalCallbackHandler",
+]

libs/langchain/langchain/callbacks/context_callback.py

@@ -1,5 +1,25 @@
-from langchain_community.callbacks.context_callback import (
+from typing import TYPE_CHECKING, Any
-    ContextCallbackHandler,
-)
 
-__all__ = ["ContextCallbackHandler"]
+from langchain._api import create_importer
+
+if TYPE_CHECKING:
+    from langchain_community.callbacks.context_callback import ContextCallbackHandler
+
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {
+    "ContextCallbackHandler": "langchain_community.callbacks.context_callback"
+}
+
+_import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)
+
+
+def __getattr__(name: str) -> Any:
+    """Look up attributes dynamically."""
+    return _import_attribute(name)
+
+
+__all__ = [
+    "ContextCallbackHandler",
+]

libs/langchain/langchain/callbacks/flyte_callback.py

@@ -1,5 +1,25 @@
-from langchain_community.callbacks.flyte_callback import (
+from typing import TYPE_CHECKING, Any
-    FlyteCallbackHandler,
-)
 
-__all__ = ["FlyteCallbackHandler"]
+from langchain._api import create_importer
+
+if TYPE_CHECKING:
+    from langchain_community.callbacks.flyte_callback import FlyteCallbackHandler
+
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {
+    "FlyteCallbackHandler": "langchain_community.callbacks.flyte_callback"
+}
+
+_import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)
+
+
+def __getattr__(name: str) -> Any:
+    """Look up attributes dynamically."""
+    return _import_attribute(name)
+
+
+__all__ = [
+    "FlyteCallbackHandler",
+]

libs/langchain/langchain/callbacks/human.py

@@ -1,8 +1,30 @@
-from langchain_community.callbacks.human import (
+from typing import TYPE_CHECKING, Any
-    AsyncHumanApprovalCallbackHandler,
+
-    HumanApprovalCallbackHandler,
+from langchain._api import create_importer
-    HumanRejectedException,
+
-)
+if TYPE_CHECKING:
+    from langchain_community.callbacks.human import (
+        AsyncHumanApprovalCallbackHandler,
+        HumanApprovalCallbackHandler,
+        HumanRejectedException,
+    )
+
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {
+    "HumanRejectedException": "langchain_community.callbacks.human",
+    "HumanApprovalCallbackHandler": "langchain_community.callbacks.human",
+    "AsyncHumanApprovalCallbackHandler": "langchain_community.callbacks.human",
+}
+
+_import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)
+
+
+def __getattr__(name: str) -> Any:
+    """Look up attributes dynamically."""
+    return _import_attribute(name)
+
 
 __all__ = [
     "HumanRejectedException",

libs/langchain/langchain/callbacks/infino_callback.py

@@ -1,6 +1,24 @@
-from langchain_community.callbacks.infino_callback import (
+from typing import TYPE_CHECKING, Any
-    InfinoCallbackHandler,
+
-)
+from langchain._api import create_importer
+
+if TYPE_CHECKING:
+    from langchain_community.callbacks.infino_callback import InfinoCallbackHandler
+
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {
+    "InfinoCallbackHandler": "langchain_community.callbacks.infino_callback"
+}
+
+_import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)
+
+
+def __getattr__(name: str) -> Any:
+    """Look up attributes dynamically."""
+    return _import_attribute(name)
+
 
 __all__ = [
     "InfinoCallbackHandler",

libs/langchain/langchain/callbacks/labelstudio_callback.py

@@ -1,7 +1,33 @@
-from langchain_community.callbacks.labelstudio_callback import (
+from typing import TYPE_CHECKING, Any
-    LabelStudioCallbackHandler,
-    LabelStudioMode,
-    get_default_label_configs,
-)
 
-__all__ = ["LabelStudioMode", "get_default_label_configs", "LabelStudioCallbackHandler"]
+from langchain._api import create_importer
+
+if TYPE_CHECKING:
+    from langchain_community.callbacks.labelstudio_callback import (
+        LabelStudioCallbackHandler,
+        LabelStudioMode,
+        get_default_label_configs,
+    )
+
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {
+    "LabelStudioMode": "langchain_community.callbacks.labelstudio_callback",
+    "get_default_label_configs": "langchain_community.callbacks.labelstudio_callback",
+    "LabelStudioCallbackHandler": "langchain_community.callbacks.labelstudio_callback",
+}
+
+_import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)
+
+
+def __getattr__(name: str) -> Any:
+    """Look up attributes dynamically."""
+    return _import_attribute(name)
+
+
+__all__ = [
+    "LabelStudioMode",
+    "get_default_label_configs",
+    "LabelStudioCallbackHandler",
+]

libs/langchain/langchain/callbacks/llmonitor_callback.py

@@ -1,6 +1,26 @@
-from langchain_community.callbacks.llmonitor_callback import (
+from typing import TYPE_CHECKING, Any
-    LLMonitorCallbackHandler,
+
-)
+from langchain._api import create_importer
+
+if TYPE_CHECKING:
+    from langchain_community.callbacks.llmonitor_callback import (
+        LLMonitorCallbackHandler,
+    )
+
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {
+    "LLMonitorCallbackHandler": "langchain_community.callbacks.llmonitor_callback"
+}
+
+_import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)
+
+
+def __getattr__(name: str) -> Any:
+    """Look up attributes dynamically."""
+    return _import_attribute(name)
+
 
 __all__ = [
     "LLMonitorCallbackHandler",

libs/langchain/langchain/callbacks/manager.py

@@ -1,9 +1,7 @@
 from __future__ import annotations
 
-from langchain_community.callbacks.manager import (
+from typing import TYPE_CHECKING, Any
-    get_openai_callback,
+
-    wandb_tracing_enabled,
-)
 from langchain_core.callbacks.manager import (
     AsyncCallbackManager,
     AsyncCallbackManagerForChainGroup,
@@ -35,33 +33,57 @@ from langchain_core.tracers.context import (
 )
 from langchain_core.utils.env import env_var_is_set
 
+from langchain._api import create_importer
+
+if TYPE_CHECKING:
+    from langchain_community.callbacks.manager import (
+        get_openai_callback,
+        wandb_tracing_enabled,
+    )
+
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {
+    "get_openai_callback": "langchain_community.callbacks.manager",
+    "wandb_tracing_enabled": "langchain_community.callbacks.manager",
+}
+
+_import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)
+
+
+def __getattr__(name: str) -> Any:
+    """Look up attributes dynamically."""
+    return _import_attribute(name)
+
+
 __all__ = [
-    "BaseRunManager",
+    "ahandle_event",
-    "RunManager",
+    "AsyncCallbackManagerForChainGroup",
-    "ParentRunManager",
-    "AsyncRunManager",
-    "AsyncParentRunManager",
-    "CallbackManagerForLLMRun",
-    "AsyncCallbackManagerForLLMRun",
-    "CallbackManagerForChainRun",
     "AsyncCallbackManagerForChainRun",
-    "CallbackManagerForToolRun",
+    "AsyncCallbackManagerForLLMRun",
-    "AsyncCallbackManagerForToolRun",
-    "CallbackManagerForRetrieverRun",
     "AsyncCallbackManagerForRetrieverRun",
+    "AsyncCallbackManagerForToolRun",
+    "AsyncParentRunManager",
+    "AsyncRunManager",
+    "atrace_as_chain_group",
+    "BaseRunManager",
     "CallbackManager",
     "CallbackManagerForChainGroup",
-    "AsyncCallbackManager",
+    "CallbackManagerForChainRun",
-    "AsyncCallbackManagerForChainGroup",
+    "CallbackManagerForLLMRun",
-    "tracing_enabled",
+    "CallbackManagerForRetrieverRun",
-    "tracing_v2_enabled",
+    "CallbackManagerForToolRun",
-    "collect_runs",
-    "atrace_as_chain_group",
-    "trace_as_chain_group",
-    "handle_event",
-    "ahandle_event",
     "Callbacks",
+    "AsyncCallbackManager",
+    "collect_runs",
     "env_var_is_set",
     "get_openai_callback",
+    "handle_event",
+    "ParentRunManager",
+    "RunManager",
+    "trace_as_chain_group",
+    "tracing_enabled",
+    "tracing_v2_enabled",
     "wandb_tracing_enabled",
 ]

libs/langchain/langchain/callbacks/mlflow_callback.py

@@ -1,9 +1,34 @@
-from langchain_community.callbacks.mlflow_callback import (
+from typing import TYPE_CHECKING, Any
-    MlflowCallbackHandler,
+
-    MlflowLogger,
+from langchain._api import create_importer
-    analyze_text,
+
-    construct_html_from_prompt_and_generation,
+if TYPE_CHECKING:
-)
+    from langchain_community.callbacks.mlflow_callback import (
+        MlflowCallbackHandler,
+        MlflowLogger,
+        analyze_text,
+        construct_html_from_prompt_and_generation,
+    )
+
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {
+    "analyze_text": "langchain_community.callbacks.mlflow_callback",
+    "construct_html_from_prompt_and_generation": (
+        "langchain_community.callbacks.mlflow_callback"
+    ),
+    "MlflowLogger": "langchain_community.callbacks.mlflow_callback",
+    "MlflowCallbackHandler": "langchain_community.callbacks.mlflow_callback",
+}
+
+_import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)
+
+
+def __getattr__(name: str) -> Any:
+    """Look up attributes dynamically."""
+    return _import_attribute(name)
+
 
 __all__ = [
     "analyze_text",

libs/langchain/langchain/callbacks/openai_info.py

@@ -1,3 +1,25 @@
-from langchain_community.callbacks.openai_info import OpenAICallbackHandler
+from typing import TYPE_CHECKING, Any
 
-__all__ = ["OpenAICallbackHandler"]
+from langchain._api import create_importer
+
+if TYPE_CHECKING:
+    from langchain_community.callbacks.openai_info import OpenAICallbackHandler
+
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {
+    "OpenAICallbackHandler": "langchain_community.callbacks.openai_info"
+}
+
+_import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)
+
+
+def __getattr__(name: str) -> Any:
+    """Look up attributes dynamically."""
+    return _import_attribute(name)
+
+
+__all__ = [
+    "OpenAICallbackHandler",
+]

libs/langchain/langchain/callbacks/promptlayer_callback.py

@@ -1,5 +1,27 @@
-from langchain_community.callbacks.promptlayer_callback import (
+from typing import TYPE_CHECKING, Any
-    PromptLayerCallbackHandler,
-)
 
-__all__ = ["PromptLayerCallbackHandler"]
+from langchain._api import create_importer
+
+if TYPE_CHECKING:
+    from langchain_community.callbacks.promptlayer_callback import (
+        PromptLayerCallbackHandler,
+    )
+
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {
+    "PromptLayerCallbackHandler": "langchain_community.callbacks.promptlayer_callback"
+}
+
+_import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)
+
+
+def __getattr__(name: str) -> Any:
+    """Look up attributes dynamically."""
+    return _import_attribute(name)
+
+
+__all__ = [
+    "PromptLayerCallbackHandler",
+]

libs/langchain/langchain/callbacks/sagemaker_callback.py

@@ -1,5 +1,27 @@
-from langchain_community.callbacks.sagemaker_callback import (
+from typing import TYPE_CHECKING, Any
-    SageMakerCallbackHandler,
-)
 
-__all__ = ["SageMakerCallbackHandler"]
+from langchain._api import create_importer
+
+if TYPE_CHECKING:
+    from langchain_community.callbacks.sagemaker_callback import (
+        SageMakerCallbackHandler,
+    )
+
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {
+    "SageMakerCallbackHandler": "langchain_community.callbacks.sagemaker_callback"
+}
+
+_import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)
+
+
+def __getattr__(name: str) -> Any:
+    """Look up attributes dynamically."""
+    return _import_attribute(name)
+
+
+__all__ = [
+    "SageMakerCallbackHandler",
+]

libs/langchain/langchain/callbacks/streamlit/mutable_expander.py

@@ -1,7 +1,33 @@
-from langchain_community.callbacks.streamlit.mutable_expander import (
+from typing import TYPE_CHECKING, Any
-    ChildRecord,
-    ChildType,
-    MutableExpander,
-)
 
-__all__ = ["ChildType", "ChildRecord", "MutableExpander"]
+from langchain._api import create_importer
+
+if TYPE_CHECKING:
+    from langchain_community.callbacks.streamlit.mutable_expander import (
+        ChildRecord,
+        ChildType,
+        MutableExpander,
+    )
+
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {
+    "ChildType": "langchain_community.callbacks.streamlit.mutable_expander",
+    "ChildRecord": "langchain_community.callbacks.streamlit.mutable_expander",
+    "MutableExpander": "langchain_community.callbacks.streamlit.mutable_expander",
+}
+
+_import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)
+
+
+def __getattr__(name: str) -> Any:
+    """Look up attributes dynamically."""
+    return _import_attribute(name)
+
+
+__all__ = [
+    "ChildType",
+    "ChildRecord",
+    "MutableExpander",
+]

libs/langchain/langchain/callbacks/streamlit/streamlit_callback_handler.py

@@ -1,20 +1,46 @@
-from langchain_community.callbacks.streamlit.streamlit_callback_handler import (
+from typing import TYPE_CHECKING, Any
-    CHECKMARK_EMOJI,
+
-    EXCEPTION_EMOJI,
+from langchain._api import create_importer
-    HISTORY_EMOJI,
+
-    THINKING_EMOJI,
+if TYPE_CHECKING:
-    LLMThought,
+    from langchain_community.callbacks.streamlit.streamlit_callback_handler import (
-    LLMThoughtLabeler,
+        LLMThought,
-    LLMThoughtState,
+        LLMThoughtLabeler,
-    StreamlitCallbackHandler,
+        LLMThoughtState,
-    ToolRecord,
+        StreamlitCallbackHandler,
-)
+        ToolRecord,
+    )
+
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {
+    "LLMThoughtState": (
+        "langchain_community.callbacks.streamlit.streamlit_callback_handler"
+    ),
+    "ToolRecord": (
+        "langchain_community.callbacks.streamlit.streamlit_callback_handler"
+    ),
+    "LLMThoughtLabeler": (
+        "langchain_community.callbacks.streamlit.streamlit_callback_handler"
+    ),
+    "LLMThought": (
+        "langchain_community.callbacks.streamlit.streamlit_callback_handler"
+    ),
+    "StreamlitCallbackHandler": (
+        "langchain_community.callbacks.streamlit.streamlit_callback_handler"
+    ),
+}
+
+_import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)
+
+
+def __getattr__(name: str) -> Any:
+    """Look up attributes dynamically."""
+    return _import_attribute(name)
+
 
 __all__ = [
-    "CHECKMARK_EMOJI",
-    "THINKING_EMOJI",
-    "HISTORY_EMOJI",
-    "EXCEPTION_EMOJI",
     "LLMThoughtState",
     "ToolRecord",
     "LLMThoughtLabeler",

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

@@ -1,5 +1,7 @@
 """Tracers that record execution of LangChain runs."""
 
+from typing import TYPE_CHECKING, Any
+
 from langchain_core.tracers.langchain import LangChainTracer
 from langchain_core.tracers.langchain_v1 import LangChainTracerV1
 from langchain_core.tracers.stdout import (
@@ -7,9 +9,26 @@ from langchain_core.tracers.stdout import (
     FunctionCallbackHandler,
 )
 
+from langchain._api import create_importer
 from langchain.callbacks.tracers.logging import LoggingCallbackHandler
 from langchain.callbacks.tracers.wandb import WandbTracer
 
+if TYPE_CHECKING:
+    from langchain_community.callbacks.tracers.wandb import WandbTracer
+
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {"WandbTracer": "langchain_community.callbacks.tracers.wandb"}
+
+_import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)
+
+
+def __getattr__(name: str) -> Any:
+    """Look up attributes dynamically."""
+    return _import_attribute(name)
+
+
 __all__ = [
     "ConsoleCallbackHandler",
     "FunctionCallbackHandler",

libs/langchain/langchain/callbacks/tracers/comet.py

@@ -1,6 +1,30 @@
-from langchain_community.callbacks.tracers.comet import (
+from typing import TYPE_CHECKING, Any
-    CometTracer,
-    import_comet_llm_api,
-)
 
-__all__ = ["import_comet_llm_api", "CometTracer"]
+from langchain._api import create_importer
+
+if TYPE_CHECKING:
+    from langchain_community.callbacks.tracers.comet import (
+        CometTracer,
+        import_comet_llm_api,
+    )
+
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {
+    "import_comet_llm_api": "langchain_community.callbacks.tracers.comet",
+    "CometTracer": "langchain_community.callbacks.tracers.comet",
+}
+
+_import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)
+
+
+def __getattr__(name: str) -> Any:
+    """Look up attributes dynamically."""
+    return _import_attribute(name)
+
+
+__all__ = [
+    "import_comet_llm_api",
+    "CometTracer",
+]

libs/langchain/langchain/callbacks/tracers/wandb.py

@@ -1,12 +1,32 @@
-from langchain_community.callbacks.tracers.wandb import (
+from typing import TYPE_CHECKING, Any
-    PRINT_WARNINGS,
+
-    RunProcessor,
+from langchain._api import create_importer
-    WandbRunArgs,
+
-    WandbTracer,
+if TYPE_CHECKING:
-)
+    from langchain_community.callbacks.tracers.wandb import (
+        RunProcessor,
+        WandbRunArgs,
+        WandbTracer,
+    )
+
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {
+    "RunProcessor": "langchain_community.callbacks.tracers.wandb",
+    "WandbRunArgs": "langchain_community.callbacks.tracers.wandb",
+    "WandbTracer": "langchain_community.callbacks.tracers.wandb",
+}
+
+_import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)
+
+
+def __getattr__(name: str) -> Any:
+    """Look up attributes dynamically."""
+    return _import_attribute(name)
+
 
 __all__ = [
-    "PRINT_WARNINGS",
     "RunProcessor",
     "WandbRunArgs",
     "WandbTracer",

libs/langchain/langchain/callbacks/trubrics_callback.py

@@ -1,5 +1,25 @@
-from langchain_community.callbacks.trubrics_callback import (
+from typing import TYPE_CHECKING, Any
-    TrubricsCallbackHandler,
-)
 
-__all__ = ["TrubricsCallbackHandler"]
+from langchain._api import create_importer
+
+if TYPE_CHECKING:
+    from langchain_community.callbacks.trubrics_callback import TrubricsCallbackHandler
+
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {
+    "TrubricsCallbackHandler": "langchain_community.callbacks.trubrics_callback"
+}
+
+_import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)
+
+
+def __getattr__(name: str) -> Any:
+    """Look up attributes dynamically."""
+    return _import_attribute(name)
+
+
+__all__ = [
+    "TrubricsCallbackHandler",
+]

libs/langchain/langchain/callbacks/utils.py

@@ -1,13 +1,40 @@
-from langchain_community.callbacks.utils import (
+from typing import TYPE_CHECKING, Any
-    BaseMetadataCallbackHandler,
+
-    _flatten_dict,
+from langchain._api import create_importer
-    flatten_dict,
+
-    hash_string,
+if TYPE_CHECKING:
-    import_pandas,
+    from langchain_community.callbacks.utils import (
-    import_spacy,
+        BaseMetadataCallbackHandler,
-    import_textstat,
+        _flatten_dict,
-    load_json,
+        flatten_dict,
-)
+        hash_string,
+        import_pandas,
+        import_spacy,
+        import_textstat,
+        load_json,
+    )
+
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {
+    "import_spacy": "langchain_community.callbacks.utils",
+    "import_pandas": "langchain_community.callbacks.utils",
+    "import_textstat": "langchain_community.callbacks.utils",
+    "_flatten_dict": "langchain_community.callbacks.utils",
+    "flatten_dict": "langchain_community.callbacks.utils",
+    "hash_string": "langchain_community.callbacks.utils",
+    "load_json": "langchain_community.callbacks.utils",
+    "BaseMetadataCallbackHandler": "langchain_community.callbacks.utils",
+}
+
+_import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)
+
+
+def __getattr__(name: str) -> Any:
+    """Look up attributes dynamically."""
+    return _import_attribute(name)
+
 
 __all__ = [
     "import_spacy",

libs/langchain/langchain/callbacks/wandb_callback.py

@@ -1,6 +1,24 @@
-from langchain_community.callbacks.wandb_callback import (
+from typing import TYPE_CHECKING, Any
-    WandbCallbackHandler,
+
-)
+from langchain._api import create_importer
+
+if TYPE_CHECKING:
+    from langchain_community.callbacks.wandb_callback import WandbCallbackHandler
+
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {
+    "WandbCallbackHandler": "langchain_community.callbacks.wandb_callback"
+}
+
+_import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)
+
+
+def __getattr__(name: str) -> Any:
+    """Look up attributes dynamically."""
+    return _import_attribute(name)
+
 
 __all__ = [
     "WandbCallbackHandler",

libs/langchain/langchain/callbacks/whylabs_callback.py

@@ -1,5 +1,25 @@
-from langchain_community.callbacks.whylabs_callback import (
+from typing import TYPE_CHECKING, Any
-    WhyLabsCallbackHandler,
-)
 
-__all__ = ["WhyLabsCallbackHandler"]
+from langchain._api import create_importer
+
+if TYPE_CHECKING:
+    from langchain_community.callbacks.whylabs_callback import WhyLabsCallbackHandler
+
+# Create a way to dynamically look up deprecated imports.
+# Used to consolidate logic for raising deprecation warnings and
+# handling optional imports.
+DEPRECATED_LOOKUP = {
+    "WhyLabsCallbackHandler": "langchain_community.callbacks.whylabs_callback"
+}
+
+_import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)
+
+
+def __getattr__(name: str) -> Any:
+    """Look up attributes dynamically."""
+    return _import_attribute(name)
+
+
+__all__ = [
+    "WhyLabsCallbackHandler",
+]