diff --git a/libs/langchain_v1/langchain/agents/middleware/__init__.py b/libs/langchain_v1/langchain/agents/middleware/__init__.py index 8a8e22e44e..1b0c50bb8d 100644 --- a/libs/langchain_v1/langchain/agents/middleware/__init__.py +++ b/libs/langchain_v1/langchain/agents/middleware/__init__.py @@ -20,6 +20,7 @@ from .types import ( AgentMiddleware, AgentState, ModelRequest, + ModelResponse, after_agent, after_model, before_agent, @@ -41,6 +42,7 @@ __all__ = [ "ModelCallLimitMiddleware", "ModelFallbackMiddleware", "ModelRequest", + "ModelResponse", "PIIDetectionError", "PIIMiddleware", "PlanningMiddleware",