github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-06 21:20:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

add callback import test (#13689)

Browse Source
This commit is contained in:
Bagatur 2023-11-21 15:28:49 -08:00 committed by GitHub
parent 083afba697
commit a208abe6b7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 52 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
libs/langchain/tests/unit_tests/callbacks/test_base.py Normal file
View File

@ -0,0 +1,18 @@
from langchain.callbacks.base import __all__
EXPECTED_ALL = [
"RetrieverManagerMixin",
"LLMManagerMixin",
"ChainManagerMixin",
"ToolManagerMixin",
"CallbackManagerMixin",
"RunManagerMixin",
"BaseCallbackHandler",
"AsyncCallbackHandler",
"BaseCallbackManager",
"Callbacks",
]
def test_all_imports() -> None:
assert set(__all__) == set(EXPECTED_ALL)

34
libs/langchain/tests/unit_tests/callbacks/test_manager.py Normal file
View File

@ -0,0 +1,34 @@
from langchain.callbacks.manager import __all__
EXPECTED_ALL = [
"BaseRunManager",
"RunManager",
"ParentRunManager",
"AsyncRunManager",
"AsyncParentRunManager",
"CallbackManagerForLLMRun",
"AsyncCallbackManagerForLLMRun",
"CallbackManagerForChainRun",
"AsyncCallbackManagerForChainRun",
"CallbackManagerForToolRun",
"AsyncCallbackManagerForToolRun",
"CallbackManagerForRetrieverRun",
"AsyncCallbackManagerForRetrieverRun",
"CallbackManager",
"CallbackManagerForChainGroup",
"AsyncCallbackManager",
"AsyncCallbackManagerForChainGroup",
"tracing_enabled",
"tracing_v2_enabled",
"collect_runs",
"atrace_as_chain_group",
"trace_as_chain_group",
"handle_event",
"ahandle_event",
"env_var_is_set",
"Callbacks",
]
def test_all_imports() -> None:
assert set(__all__) == set(EXPECTED_ALL)