Function _assert_tool_runner_calls

tests/lib/test_bedrock.py:486–502

(calls: t.List[MockRequestCall])

def _assert_tool_runner_calls(calls: t.List[MockRequestCall]) -> None:
    assert len(calls) == 2
    for call in calls:
        assert call.request.url.path == "/model/anthropic.claude-haiku-4-5-20251001-v1:0/invoke"
        assert call.request.headers["Authorization"].startswith("AWS4-HMAC-SHA256 ")
        assert "x-amz-date" in call.request.headers
        body = json.loads(call.request.content)
        assert "model" not in body and "stream" not in body and "output_format" not in body
        assert body["anthropic_version"] == "bedrock-2023-05-31"
    second = json.loads(calls[1].request.content)
    assert [m["role"] for m in second["messages"]] == ["user", "assistant", "user"]
    assert second["messages"][1]["content"][0]["type"] == "tool_use"
    assert second["messages"][2]["content"][0] == {
        "type": "tool_result",
        "tool_use_id": "toolu_01",
        "content": "sunny in Paris",
    }


@pytest.mark.skipif(PYDANTIC_V1, reason="tool functions are only supported with pydantic v2")