diff --git a/tests/proxy_unit_tests/test_proxy_token_counter.py b/tests/proxy_unit_tests/test_proxy_token_counter.py index 39ec4bb1887..e97b4e02f15 100644 --- a/tests/proxy_unit_tests/test_proxy_token_counter.py +++ b/tests/proxy_unit_tests/test_proxy_token_counter.py @@ -382,15 +382,24 @@ async def test_internal_token_counter_anthropic_provider_detection(): setattr(litellm.proxy.proxy_server, "llm_router", llm_router) - # Test with is_direct_request=False (simulating call from Anthropic endpoint) - response = await token_counter( - request=TokenCountRequest( - model="claude-test", - messages=[{"role": "user", "content": "hello"}], - ), - call_endpoint=True, + mock_handler = MagicMock() + mock_handler.handle_count_tokens_request = AsyncMock( + return_value={"input_tokens": 42} ) + # Test with is_direct_request=False (simulating call from Anthropic endpoint) + with patch( + "litellm.llms.anthropic.count_tokens.token_counter.anthropic_count_tokens_handler", + mock_handler, + ): + response = await token_counter( + request=TokenCountRequest( + model="claude-test", + messages=[{"role": "user", "content": "hello"}], + ), + call_endpoint=True, + ) + print("Anthropic provider test response:", response) # Verify response structure