From 68bcaddb145eba8acd8e5b33edc9509583c3c550 Mon Sep 17 00:00:00 2001 From: JerryLee <223425819+Jerry2003826@users.noreply.github.com> Date: Wed, 10 Jun 2026 20:19:55 +1000 Subject: [PATCH] test(proxy): harden anthropic messages hook metadata assertion --- .../proxy_unit_tests/test_anthropic_messages_pre_call_hook.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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