From b9de10bd277ecece2046d489a639daabdd9b0226 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Tue, 20 Jan 2026 17:53:53 -0800 Subject: [PATCH] test token ctr --- tests/proxy_unit_tests/test_proxy_token_counter.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/proxy_unit_tests/test_proxy_token_counter.py b/tests/proxy_unit_tests/test_proxy_token_counter.py index 4a2ff28cfd3..3967325e0e9 100644 --- a/tests/proxy_unit_tests/test_proxy_token_counter.py +++ b/tests/proxy_unit_tests/test_proxy_token_counter.py @@ -484,7 +484,7 @@ async def test_factory_anthropic_endpoint_calls_anthropic_counter(): # Mock the Anthropic CountTokens handler with patch( - "litellm.llms.anthropic.common_utils.AnthropicCountTokensHandler" + "litellm.llms.anthropic.count_tokens.handler.AnthropicCountTokensHandler" ) as MockHandler: mock_handler_instance = MagicMock() mock_handler_instance.handle_count_tokens_request = AsyncMock( @@ -541,7 +541,7 @@ async def test_factory_gpt4_endpoint_does_not_call_anthropic_counter(): # Mock the Anthropic CountTokens handler with patch( - "litellm.llms.anthropic.common_utils.AnthropicCountTokensHandler" + "litellm.llms.anthropic.count_tokens.handler.AnthropicCountTokensHandler" ) as MockHandler: mock_handler_instance = MagicMock() mock_handler_instance.handle_count_tokens_request = AsyncMock( @@ -600,7 +600,7 @@ async def test_factory_normal_token_counter_endpoint_does_not_call_anthropic(): # Mock the Anthropic CountTokens handler with patch( - "litellm.llms.anthropic.common_utils.AnthropicCountTokensHandler" + "litellm.llms.anthropic.count_tokens.handler.AnthropicCountTokensHandler" ) as MockHandler: mock_handler_instance = MagicMock() mock_handler_instance.handle_count_tokens_request = AsyncMock(