codekingpro/portable-devtools

1"""**Callback handlers** allow listening to events in LangChain."""2 3from typing import TYPE_CHECKING, Any4 5from langchain_core.callbacks import (6    FileCallbackHandler,7    StdOutCallbackHandler,8    StreamingStdOutCallbackHandler,9)10from langchain_core.tracers.context import (11    collect_runs,12    tracing_v2_enabled,13)14from langchain_core.tracers.langchain import LangChainTracer15 16from langchain_classic._api import create_importer17from langchain_classic.callbacks.streaming_aiter import AsyncIteratorCallbackHandler18from langchain_classic.callbacks.streaming_stdout_final_only import (19    FinalStreamingStdOutCallbackHandler,20)21 22if TYPE_CHECKING:23    from langchain_community.callbacks.aim_callback import AimCallbackHandler24    from langchain_community.callbacks.argilla_callback import ArgillaCallbackHandler25    from langchain_community.callbacks.arize_callback import ArizeCallbackHandler26    from langchain_community.callbacks.arthur_callback import ArthurCallbackHandler27    from langchain_community.callbacks.clearml_callback import ClearMLCallbackHandler28    from langchain_community.callbacks.comet_ml_callback import CometCallbackHandler29    from langchain_community.callbacks.context_callback import ContextCallbackHandler30    from langchain_community.callbacks.flyte_callback import FlyteCallbackHandler31    from langchain_community.callbacks.human import HumanApprovalCallbackHandler32    from langchain_community.callbacks.infino_callback import InfinoCallbackHandler33    from langchain_community.callbacks.labelstudio_callback import (34        LabelStudioCallbackHandler,35    )36    from langchain_community.callbacks.llmonitor_callback import (37        LLMonitorCallbackHandler,38    )39    from langchain_community.callbacks.manager import (40        get_openai_callback,41        wandb_tracing_enabled,42    )43    from langchain_community.callbacks.mlflow_callback import MlflowCallbackHandler44    from langchain_community.callbacks.openai_info import OpenAICallbackHandler45    from langchain_community.callbacks.promptlayer_callback import (46        PromptLayerCallbackHandler,47    )48    from langchain_community.callbacks.sagemaker_callback import (49        SageMakerCallbackHandler,50    )51    from langchain_community.callbacks.streamlit import StreamlitCallbackHandler52    from langchain_community.callbacks.streamlit.streamlit_callback_handler import (53        LLMThoughtLabeler,54    )55    from langchain_community.callbacks.trubrics_callback import TrubricsCallbackHandler56    from langchain_community.callbacks.wandb_callback import WandbCallbackHandler57    from langchain_community.callbacks.whylabs_callback import WhyLabsCallbackHandler58 59# Create a way to dynamically look up deprecated imports.60# Used to consolidate logic for raising deprecation warnings and61# handling optional imports.62DEPRECATED_LOOKUP = {63    "AimCallbackHandler": "langchain_community.callbacks.aim_callback",64    "ArgillaCallbackHandler": "langchain_community.callbacks.argilla_callback",65    "ArizeCallbackHandler": "langchain_community.callbacks.arize_callback",66    "PromptLayerCallbackHandler": "langchain_community.callbacks.promptlayer_callback",67    "ArthurCallbackHandler": "langchain_community.callbacks.arthur_callback",68    "ClearMLCallbackHandler": "langchain_community.callbacks.clearml_callback",69    "CometCallbackHandler": "langchain_community.callbacks.comet_ml_callback",70    "ContextCallbackHandler": "langchain_community.callbacks.context_callback",71    "HumanApprovalCallbackHandler": "langchain_community.callbacks.human",72    "InfinoCallbackHandler": "langchain_community.callbacks.infino_callback",73    "MlflowCallbackHandler": "langchain_community.callbacks.mlflow_callback",74    "LLMonitorCallbackHandler": "langchain_community.callbacks.llmonitor_callback",75    "OpenAICallbackHandler": "langchain_community.callbacks.openai_info",76    "LLMThoughtLabeler": (77        "langchain_community.callbacks.streamlit.streamlit_callback_handler"78    ),79    "StreamlitCallbackHandler": "langchain_community.callbacks.streamlit",80    "WandbCallbackHandler": "langchain_community.callbacks.wandb_callback",81    "WhyLabsCallbackHandler": "langchain_community.callbacks.whylabs_callback",82    "get_openai_callback": "langchain_community.callbacks.manager",83    "wandb_tracing_enabled": "langchain_community.callbacks.manager",84    "FlyteCallbackHandler": "langchain_community.callbacks.flyte_callback",85    "SageMakerCallbackHandler": "langchain_community.callbacks.sagemaker_callback",86    "LabelStudioCallbackHandler": "langchain_community.callbacks.labelstudio_callback",87    "TrubricsCallbackHandler": "langchain_community.callbacks.trubrics_callback",88}89 90_import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)91 92 93def __getattr__(name: str) -> Any:94    """Look up attributes dynamically."""95    return _import_attribute(name)96 97 98__all__ = [99    "AimCallbackHandler",100    "ArgillaCallbackHandler",101    "ArizeCallbackHandler",102    "ArthurCallbackHandler",103    "AsyncIteratorCallbackHandler",104    "ClearMLCallbackHandler",105    "CometCallbackHandler",106    "ContextCallbackHandler",107    "FileCallbackHandler",108    "FinalStreamingStdOutCallbackHandler",109    "FlyteCallbackHandler",110    "HumanApprovalCallbackHandler",111    "InfinoCallbackHandler",112    "LLMThoughtLabeler",113    "LLMonitorCallbackHandler",114    "LabelStudioCallbackHandler",115    "LangChainTracer",116    "MlflowCallbackHandler",117    "OpenAICallbackHandler",118    "PromptLayerCallbackHandler",119    "SageMakerCallbackHandler",120    "StdOutCallbackHandler",121    "StreamingStdOutCallbackHandler",122    "StreamlitCallbackHandler",123    "TrubricsCallbackHandler",124    "WandbCallbackHandler",125    "WhyLabsCallbackHandler",126    "collect_runs",127    "get_openai_callback",128    "tracing_v2_enabled",129    "wandb_tracing_enabled",130]131