scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-11 03:38:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 84

test token ctr

Browse source
This commit is contained in:
Ishaan Jaffer 2026-01-20 17:53:53 -08:00
parent e57214dfab
commit b9de10bd27
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/proxy_unit_tests/test_proxy_token_counter.py
View file

@ -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(