diff --git a/libs/langchain/tests/unit_tests/callbacks/test_base.py b/libs/langchain/tests/unit_tests/callbacks/test_base.py new file mode 100644 index 00000000000..62760e032df --- /dev/null +++ b/libs/langchain/tests/unit_tests/callbacks/test_base.py @@ -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) diff --git a/libs/langchain/tests/unit_tests/callbacks/test_manager.py b/libs/langchain/tests/unit_tests/callbacks/test_manager.py new file mode 100644 index 00000000000..3ddfb4f0c7b --- /dev/null +++ b/libs/langchain/tests/unit_tests/callbacks/test_manager.py @@ -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)