diff --git a/tests/proxy_unit_tests/test_anthropic_messages_pre_call_hook.py b/tests/proxy_unit_tests/test_anthropic_messages_pre_call_hook.py index 5c19d57f9b6..ecb3e183bac 100644 --- a/tests/proxy_unit_tests/test_anthropic_messages_pre_call_hook.py +++ b/tests/proxy_unit_tests/test_anthropic_messages_pre_call_hook.py @@ -77,7 +77,8 @@ def test_anthropic_messages_runs_proxy_async_pre_call_hook( assert response.json()["content"][0]["text"] == "Hello from LiteLLM" assert hook_calls == ["anthropic_messages"] mock_anthropic_messages.assert_called_once() - assert mock_anthropic_messages.call_args.kwargs["metadata"]["source"] == "unit-test" + metadata = mock_anthropic_messages.call_args.kwargs.get("metadata") + assert metadata == {"source": "unit-test"} @pytest.mark.asyncio