test_replaces_tool_with_claude_descriptor() — langchain Function Reference

Source Code

libs/partners/anthropic/tests/unit_tests/middleware/test_bash.py lines 26–61

def test_replaces_tool_with_claude_descriptor() -> None:
    """Test wrap_model_call replaces bash tool with Claude's bash descriptor."""
    from langchain.agents.middleware.types import ModelRequest

    middleware = ClaudeBashToolMiddleware()

    # Create a mock request with the bash tool (inherited from parent)
    bash_tool = middleware.tools[0]
    request = ModelRequest(
        model=MagicMock(),
        system_prompt=None,
        messages=[],
        tool_choice=None,
        tools=[bash_tool],
        response_format=None,
        state={"messages": []},
        runtime=MagicMock(),
    )

    # Mock handler that captures the modified request
    captured_request = None

    def handler(req: ModelRequest) -> MagicMock:
        nonlocal captured_request
        captured_request = req
        return MagicMock()

    middleware.wrap_model_call(request, handler)

    # The bash tool should be replaced with Claude's native bash descriptor
    assert captured_request is not None
    assert len(captured_request.tools) == 1
    assert captured_request.tools[0] == {
        "type": "bash_20250124",
        "name": "bash",
    }