From 93fbf66a54150692bfd194152170eb2da792dd08 Mon Sep 17 00:00:00 2001
From: Sydney Runkle <sydney-runkle@users.noreply.github.com>
Date: Thu, 1 Oct 2026 02:30:42 +0000
Subject: [PATCH] refactor(langchain): export only model routing middleware

Co-authored-by: open-swe[bot] <open-swe@users.noreply.github.com>
---
 .../langchain/agents/middleware/__init__.py            | 10 +---------
 .../middleware/implementations/test_model_routing.py   |  8 +++++---
 2 files changed, 6 insertions(+), 12 deletions(-)

diff --git a/libs/langchain_v1/langchain/agents/middleware/__init__.py b/libs/langchain_v1/langchain/agents/middleware/__init__.py
index a79ed7315a..5d3cba0f31 100644
--- a/libs/langchain_v1/langchain/agents/middleware/__init__.py
+++ b/libs/langchain_v1/langchain/agents/middleware/__init__.py
@@ -12,12 +12,7 @@ from langchain.agents.middleware.human_in_the_loop import (
 from langchain.agents.middleware.model_call_limit import ModelCallLimitMiddleware
 from langchain.agents.middleware.model_fallback import ModelFallbackMiddleware
 from langchain.agents.middleware.model_retry import ModelRetryMiddleware
-from langchain.agents.middleware.model_routing import (
-    ModelRoutingConfig,
-    ModelRoutingInput,
-    ModelRoutingMiddleware,
-    ModelRoutingState,
-)
+from langchain.agents.middleware.model_routing import ModelRoutingMiddleware
 from langchain.agents.middleware.pii import PIIDetectionError, PIIMatch, PIIMiddleware
 from langchain.agents.middleware.provider_tool_search import ProviderToolSearchMiddleware
 from langchain.agents.middleware.shell_tool import (
@@ -77,10 +72,7 @@ __all__ = [
     "ModelRequest",
     "ModelResponse",
     "ModelRetryMiddleware",
-    "ModelRoutingConfig",
-    "ModelRoutingInput",
     "ModelRoutingMiddleware",
-    "ModelRoutingState",
     "OutputAgentState",
     "PIIDetectionError",
     "PIIMatch",
diff --git a/libs/langchain_v1/tests/unit_tests/agents/middleware/implementations/test_model_routing.py b/libs/langchain_v1/tests/unit_tests/agents/middleware/implementations/test_model_routing.py
index aae709659e..bca21ed26a 100644
--- a/libs/langchain_v1/tests/unit_tests/agents/middleware/implementations/test_model_routing.py
+++ b/libs/langchain_v1/tests/unit_tests/agents/middleware/implementations/test_model_routing.py
@@ -19,13 +19,15 @@ from langchain.agents.middleware import (
     InputAgentState,
     ModelRequest,
     ModelResponse,
-    ModelRoutingConfig,
-    ModelRoutingInput,
     ModelRoutingMiddleware,
-    ModelRoutingState,
     model_routing,
 )
 from langchain.agents.middleware.internal_call_transformer import internal_call_metadata
+from langchain.agents.middleware.model_routing import (
+    ModelRoutingConfig,
+    ModelRoutingInput,
+    ModelRoutingState,
+)
 
 
 def test_beta_warning() -> None: