langchain Architecture Docs
Home / Function/ test_state_schema_with_middleware() — langchain Function Reference

test_state_schema_with_middleware() — langchain Function Reference

Architecture documentation for the test_state_schema_with_middleware() function in test_state_schema.py from the langchain codebase.

Function python LangChainCore ApiManagement calls 1

Entity Profile

Dependency Diagram

graph TD
  e32d9518_c1c6_4111_a89b_531324fd85bb["test_state_schema_with_middleware()"]
  3d4a3f09_0839_f785_dcfd_f1e7b13e3204["test_state_schema.py"]
  e32d9518_c1c6_4111_a89b_531324fd85bb -->|defined in| 3d4a3f09_0839_f785_dcfd_f1e7b13e3204
  e9c0ce14_d33f_7641_2350_cd379e713bff["before_model()"]
  e32d9518_c1c6_4111_a89b_531324fd85bb -->|calls| e9c0ce14_d33f_7641_2350_cd379e713bff
  style e32d9518_c1c6_4111_a89b_531324fd85bb fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/langchain_v1/tests/unit_tests/agents/test_state_schema.py lines 115–152

def test_state_schema_with_middleware() -> None:
    """Test that state_schema merges with middleware state schemas."""

    class UserState(AgentState[Any]):
        user_name: str

    class MiddlewareState(AgentState[Any]):
        middleware_data: str

    middleware_calls = []

    class TestMiddleware(AgentMiddleware[MiddlewareState, None]):
        state_schema = MiddlewareState

        def before_model(self, state: MiddlewareState, runtime: Runtime) -> dict[str, Any]:
            middleware_calls.append(state["middleware_data"])
            return {}

    agent = create_agent(
        model=FakeToolCallingModel(
            tool_calls=[[{"args": {"x": 5}, "id": "call_1", "name": "simple_tool"}], []]
        ),
        tools=[simple_tool],
        state_schema=UserState,
        middleware=[TestMiddleware()],
    )

    result = agent.invoke(
        {
            "messages": [HumanMessage("Test")],
            "user_name": "Alice",
            "middleware_data": "test_data",
        }
    )

    assert result["user_name"] == "Alice"
    assert result["middleware_data"] == "test_data"
    assert "test_data" in middleware_calls

Domain

Subdomains

Defined In

Calls

Source

Frequently Asked Questions

What does test_state_schema_with_middleware() do?
test_state_schema_with_middleware() is a function in the langchain codebase, defined in libs/langchain_v1/tests/unit_tests/agents/test_state_schema.py.
Where is test_state_schema_with_middleware() defined?
test_state_schema_with_middleware() is defined in libs/langchain_v1/tests/unit_tests/agents/test_state_schema.py at line 115.
What does test_state_schema_with_middleware() call?
test_state_schema_with_middleware() calls 1 function(s): before_model.

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free