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

This commit is contained in:
Eugene Yurtsev 2024-04-30 16:19:13 -04:00 committed by GitHub
parent 8a62fb0570
commit aec13a6123
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
28 changed files with 802 additions and 159 deletions

View File

@ -6,10 +6,8 @@
BaseCallbackHandler --> <name>CallbackHandler # Example: AimCallbackHandler 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 ( from langchain_core.callbacks import (
FileCallbackHandler, FileCallbackHandler,
StdOutCallbackHandler, StdOutCallbackHandler,
@ -22,28 +20,86 @@ from langchain_core.tracers.context import (
) )
from langchain_core.tracers.langchain import LangChainTracer from langchain_core.tracers.langchain import LangChainTracer
from langchain._api import create_importer
from langchain.callbacks.streaming_aiter import AsyncIteratorCallbackHandler from langchain.callbacks.streaming_aiter import AsyncIteratorCallbackHandler
from langchain.callbacks.streaming_stdout_final_only import ( from langchain.callbacks.streaming_stdout_final_only import (
FinalStreamingStdOutCallbackHandler, 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: 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__ = [ __all__ = [

View File

@ -1,7 +1,33 @@
from typing import TYPE_CHECKING, Any
from langchain._api import create_importer
if TYPE_CHECKING:
from langchain_community.callbacks.aim_callback import ( from langchain_community.callbacks.aim_callback import (
AimCallbackHandler, AimCallbackHandler,
BaseMetadataCallbackHandler, BaseMetadataCallbackHandler,
import_aim, import_aim,
) )
__all__ = ["import_aim", "BaseMetadataCallbackHandler", "AimCallbackHandler"] # 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",
]

View File

@ -1,3 +1,25 @@
from typing import TYPE_CHECKING, Any
from langchain._api import create_importer
if TYPE_CHECKING:
from langchain_community.callbacks.argilla_callback import ArgillaCallbackHandler from langchain_community.callbacks.argilla_callback import ArgillaCallbackHandler
__all__ = ["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",
]

View File

@ -1,3 +1,25 @@
from typing import TYPE_CHECKING, Any
from langchain._api import create_importer
if TYPE_CHECKING:
from langchain_community.callbacks.arize_callback import ArizeCallbackHandler from langchain_community.callbacks.arize_callback import ArizeCallbackHandler
__all__ = ["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",
]

View File

@ -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__ = [ __all__ = [
"ArthurCallbackHandler", "ArthurCallbackHandler",

View File

@ -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",
]

View File

@ -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__ = [ __all__ = [
"CometCallbackHandler", "CometCallbackHandler",

View File

@ -1,3 +1,25 @@
from typing import TYPE_CHECKING, Any
from langchain._api import create_importer
if TYPE_CHECKING:
from langchain_community.callbacks.confident_callback import DeepEvalCallbackHandler from langchain_community.callbacks.confident_callback import DeepEvalCallbackHandler
__all__ = ["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",
]

View File

@ -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",
]

View File

@ -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",
]

View File

@ -1,9 +1,31 @@
from typing import TYPE_CHECKING, Any
from langchain._api import create_importer
if TYPE_CHECKING:
from langchain_community.callbacks.human import ( from langchain_community.callbacks.human import (
AsyncHumanApprovalCallbackHandler, AsyncHumanApprovalCallbackHandler,
HumanApprovalCallbackHandler, HumanApprovalCallbackHandler,
HumanRejectedException, 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__ = [ __all__ = [
"HumanRejectedException", "HumanRejectedException",
"HumanApprovalCallbackHandler", "HumanApprovalCallbackHandler",

View File

@ -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__ = [ __all__ = [
"InfinoCallbackHandler", "InfinoCallbackHandler",

View File

@ -1,7 +1,33 @@
from typing import TYPE_CHECKING, Any
from langchain._api import create_importer
if TYPE_CHECKING:
from langchain_community.callbacks.labelstudio_callback import ( from langchain_community.callbacks.labelstudio_callback import (
LabelStudioCallbackHandler, LabelStudioCallbackHandler,
LabelStudioMode, LabelStudioMode,
get_default_label_configs, get_default_label_configs,
) )
__all__ = ["LabelStudioMode", "get_default_label_configs", "LabelStudioCallbackHandler"] # 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",
]

View File

@ -1,7 +1,27 @@
from typing import TYPE_CHECKING, Any
from langchain._api import create_importer
if TYPE_CHECKING:
from langchain_community.callbacks.llmonitor_callback import ( from langchain_community.callbacks.llmonitor_callback import (
LLMonitorCallbackHandler, 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__ = [ __all__ = [
"LLMonitorCallbackHandler", "LLMonitorCallbackHandler",
] ]

View File

@ -1,9 +1,7 @@
from __future__ import annotations 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 ( from langchain_core.callbacks.manager import (
AsyncCallbackManager, AsyncCallbackManager,
AsyncCallbackManagerForChainGroup, AsyncCallbackManagerForChainGroup,
@ -35,33 +33,57 @@ from langchain_core.tracers.context import (
) )
from langchain_core.utils.env import env_var_is_set 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__ = [ __all__ = [
"BaseRunManager", "ahandle_event",
"RunManager", "AsyncCallbackManagerForChainGroup",
"ParentRunManager",
"AsyncRunManager",
"AsyncParentRunManager",
"CallbackManagerForLLMRun",
"AsyncCallbackManagerForLLMRun",
"CallbackManagerForChainRun",
"AsyncCallbackManagerForChainRun", "AsyncCallbackManagerForChainRun",
"CallbackManagerForToolRun", "AsyncCallbackManagerForLLMRun",
"AsyncCallbackManagerForToolRun",
"CallbackManagerForRetrieverRun",
"AsyncCallbackManagerForRetrieverRun", "AsyncCallbackManagerForRetrieverRun",
"AsyncCallbackManagerForToolRun",
"AsyncParentRunManager",
"AsyncRunManager",
"atrace_as_chain_group",
"BaseRunManager",
"CallbackManager", "CallbackManager",
"CallbackManagerForChainGroup", "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", "Callbacks",
"AsyncCallbackManager",
"collect_runs",
"env_var_is_set", "env_var_is_set",
"get_openai_callback", "get_openai_callback",
"handle_event",
"ParentRunManager",
"RunManager",
"trace_as_chain_group",
"tracing_enabled",
"tracing_v2_enabled",
"wandb_tracing_enabled", "wandb_tracing_enabled",
] ]

View File

@ -1,3 +1,8 @@
from typing import TYPE_CHECKING, Any
from langchain._api import create_importer
if TYPE_CHECKING:
from langchain_community.callbacks.mlflow_callback import ( from langchain_community.callbacks.mlflow_callback import (
MlflowCallbackHandler, MlflowCallbackHandler,
MlflowLogger, MlflowLogger,
@ -5,6 +10,26 @@ from langchain_community.callbacks.mlflow_callback import (
construct_html_from_prompt_and_generation, 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__ = [ __all__ = [
"analyze_text", "analyze_text",
"construct_html_from_prompt_and_generation", "construct_html_from_prompt_and_generation",

View File

@ -1,3 +1,25 @@
from typing import TYPE_CHECKING, Any
from langchain._api import create_importer
if TYPE_CHECKING:
from langchain_community.callbacks.openai_info import OpenAICallbackHandler from langchain_community.callbacks.openai_info import OpenAICallbackHandler
__all__ = ["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",
]

View File

@ -1,5 +1,27 @@
from typing import TYPE_CHECKING, Any
from langchain._api import create_importer
if TYPE_CHECKING:
from langchain_community.callbacks.promptlayer_callback import ( from langchain_community.callbacks.promptlayer_callback import (
PromptLayerCallbackHandler, PromptLayerCallbackHandler,
) )
__all__ = ["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",
]

View File

@ -1,5 +1,27 @@
from typing import TYPE_CHECKING, Any
from langchain._api import create_importer
if TYPE_CHECKING:
from langchain_community.callbacks.sagemaker_callback import ( from langchain_community.callbacks.sagemaker_callback import (
SageMakerCallbackHandler, SageMakerCallbackHandler,
) )
__all__ = ["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",
]

View File

@ -1,7 +1,33 @@
from typing import TYPE_CHECKING, Any
from langchain._api import create_importer
if TYPE_CHECKING:
from langchain_community.callbacks.streamlit.mutable_expander import ( from langchain_community.callbacks.streamlit.mutable_expander import (
ChildRecord, ChildRecord,
ChildType, ChildType,
MutableExpander, MutableExpander,
) )
__all__ = ["ChildType", "ChildRecord", "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",
]

View File

@ -1,8 +1,9 @@
from typing import TYPE_CHECKING, Any
from langchain._api import create_importer
if TYPE_CHECKING:
from langchain_community.callbacks.streamlit.streamlit_callback_handler import ( from langchain_community.callbacks.streamlit.streamlit_callback_handler import (
CHECKMARK_EMOJI,
EXCEPTION_EMOJI,
HISTORY_EMOJI,
THINKING_EMOJI,
LLMThought, LLMThought,
LLMThoughtLabeler, LLMThoughtLabeler,
LLMThoughtState, LLMThoughtState,
@ -10,11 +11,36 @@ from langchain_community.callbacks.streamlit.streamlit_callback_handler import (
ToolRecord, 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__ = [ __all__ = [
"CHECKMARK_EMOJI",
"THINKING_EMOJI",
"HISTORY_EMOJI",
"EXCEPTION_EMOJI",
"LLMThoughtState", "LLMThoughtState",
"ToolRecord", "ToolRecord",
"LLMThoughtLabeler", "LLMThoughtLabeler",

View File

@ -1,5 +1,7 @@
"""Tracers that record execution of LangChain runs.""" """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 import LangChainTracer
from langchain_core.tracers.langchain_v1 import LangChainTracerV1 from langchain_core.tracers.langchain_v1 import LangChainTracerV1
from langchain_core.tracers.stdout import ( from langchain_core.tracers.stdout import (
@ -7,9 +9,26 @@ from langchain_core.tracers.stdout import (
FunctionCallbackHandler, FunctionCallbackHandler,
) )
from langchain._api import create_importer
from langchain.callbacks.tracers.logging import LoggingCallbackHandler from langchain.callbacks.tracers.logging import LoggingCallbackHandler
from langchain.callbacks.tracers.wandb import WandbTracer 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__ = [ __all__ = [
"ConsoleCallbackHandler", "ConsoleCallbackHandler",
"FunctionCallbackHandler", "FunctionCallbackHandler",

View File

@ -1,6 +1,30 @@
from typing import TYPE_CHECKING, Any
from langchain._api import create_importer
if TYPE_CHECKING:
from langchain_community.callbacks.tracers.comet import ( from langchain_community.callbacks.tracers.comet import (
CometTracer, CometTracer,
import_comet_llm_api, import_comet_llm_api,
) )
__all__ = ["import_comet_llm_api", "CometTracer"] # 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",
]

View File

@ -1,12 +1,32 @@
from typing import TYPE_CHECKING, Any
from langchain._api import create_importer
if TYPE_CHECKING:
from langchain_community.callbacks.tracers.wandb import ( from langchain_community.callbacks.tracers.wandb import (
PRINT_WARNINGS,
RunProcessor, RunProcessor,
WandbRunArgs, WandbRunArgs,
WandbTracer, 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__ = [ __all__ = [
"PRINT_WARNINGS",
"RunProcessor", "RunProcessor",
"WandbRunArgs", "WandbRunArgs",
"WandbTracer", "WandbTracer",

View File

@ -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",
]

View File

@ -1,3 +1,8 @@
from typing import TYPE_CHECKING, Any
from langchain._api import create_importer
if TYPE_CHECKING:
from langchain_community.callbacks.utils import ( from langchain_community.callbacks.utils import (
BaseMetadataCallbackHandler, BaseMetadataCallbackHandler,
_flatten_dict, _flatten_dict,
@ -9,6 +14,28 @@ from langchain_community.callbacks.utils import (
load_json, 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__ = [ __all__ = [
"import_spacy", "import_spacy",
"import_pandas", "import_pandas",

View File

@ -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__ = [ __all__ = [
"WandbCallbackHandler", "WandbCallbackHandler",

View File

@ -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",
]