From 083afba697d97ed2317c8993e1f0fea6227975e5 Mon Sep 17 00:00:00 2001
From: Bagatur <22008038+baskaryan@users.noreply.github.com>
Date: Tue, 21 Nov 2023 15:25:47 -0800
Subject: [PATCH] BUG: Add core utils imports (#13688)

---
 libs/langchain/langchain/schema/runnable/utils.py             | 4 ++++
 libs/langchain/tests/unit_tests/schema/runnable/test_utils.py | 2 ++
 2 files changed, 6 insertions(+)

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",
 ]