diff --git a/libs/langchain/langchain/schema/runnable/utils.py b/libs/langchain/langchain/schema/runnable/utils.py index de3fbc616f7..35bb4365e5a 100644 --- a/libs/langchain/langchain/schema/runnable/utils.py +++ b/libs/langchain/langchain/schema/runnable/utils.py @@ -5,8 +5,10 @@ from langchain_core.runnables.utils import ( ConfigurableFieldSingleOption, ConfigurableFieldSpec, GetLambdaSource, + Input, IsFunctionArgDict, IsLocalDict, + Output, SupportsAdd, aadd, accepts_config, @@ -40,4 +42,6 @@ __all__ = [ "aadd", "gated_coro", "gather_with_concurrency", + "Input", + "Output", ] diff --git a/libs/langchain/tests/unit_tests/schema/runnable/test_utils.py b/libs/langchain/tests/unit_tests/schema/runnable/test_utils.py index 76eb1c3feef..b38a33676cb 100644 --- a/libs/langchain/tests/unit_tests/schema/runnable/test_utils.py +++ b/libs/langchain/tests/unit_tests/schema/runnable/test_utils.py @@ -20,6 +20,8 @@ EXPECTED_ALL = [ "get_lambda_source", "get_unique_config_specs", "indent_lines_after_first", + "Input", + "Output", ]