codekingpro/portable-devtools

1from __future__ import annotations2 3from typing import TYPE_CHECKING, Any4 5from langchain_core.callbacks import Callbacks6from langchain_core.callbacks.manager import (7    AsyncCallbackManager,8    AsyncCallbackManagerForChainGroup,9    AsyncCallbackManagerForChainRun,10    AsyncCallbackManagerForLLMRun,11    AsyncCallbackManagerForRetrieverRun,12    AsyncCallbackManagerForToolRun,13    AsyncParentRunManager,14    AsyncRunManager,15    BaseRunManager,16    CallbackManager,17    CallbackManagerForChainGroup,18    CallbackManagerForChainRun,19    CallbackManagerForLLMRun,20    CallbackManagerForRetrieverRun,21    CallbackManagerForToolRun,22    ParentRunManager,23    RunManager,24    ahandle_event,25    atrace_as_chain_group,26    handle_event,27    trace_as_chain_group,28)29from langchain_core.tracers.context import (30    collect_runs,31    tracing_v2_enabled,32)33from langchain_core.utils.env import env_var_is_set34 35from langchain_classic._api import create_importer36 37if TYPE_CHECKING:38    from langchain_community.callbacks.manager import (39        get_openai_callback,40        wandb_tracing_enabled,41    )42 43# Create a way to dynamically look up deprecated imports.44# Used to consolidate logic for raising deprecation warnings and45# handling optional imports.46DEPRECATED_LOOKUP = {47    "get_openai_callback": "langchain_community.callbacks.manager",48    "wandb_tracing_enabled": "langchain_community.callbacks.manager",49}50 51_import_attribute = create_importer(__file__, deprecated_lookups=DEPRECATED_LOOKUP)52 53 54def __getattr__(name: str) -> Any:55    """Look up attributes dynamically."""56    return _import_attribute(name)57 58 59__all__ = [60    "AsyncCallbackManager",61    "AsyncCallbackManagerForChainGroup",62    "AsyncCallbackManagerForChainRun",63    "AsyncCallbackManagerForLLMRun",64    "AsyncCallbackManagerForRetrieverRun",65    "AsyncCallbackManagerForToolRun",66    "AsyncParentRunManager",67    "AsyncRunManager",68    "BaseRunManager",69    "CallbackManager",70    "CallbackManagerForChainGroup",71    "CallbackManagerForChainRun",72    "CallbackManagerForLLMRun",73    "CallbackManagerForRetrieverRun",74    "CallbackManagerForToolRun",75    "Callbacks",76    "ParentRunManager",77    "RunManager",78    "ahandle_event",79    "atrace_as_chain_group",80    "collect_runs",81    "env_var_is_set",82    "get_openai_callback",83    "handle_event",84    "trace_as_chain_group",85    "tracing_v2_enabled",86    "wandb_tracing_enabled",87]88