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: