"""
Anthropic Token Counter Tests.

Tests for the Anthropic token counter implementation using the base test suite.
"""

import os
from typing import Any, Dict, List

import pytest


from litellm.llms.anthropic.count_tokens import AnthropicTokenCounter
from litellm.llms.base_llm.base_utils import BaseTokenCounter
from tests.litellm_utils_tests.base_token_counter_test import BaseTokenCounterTest


class TestAnthropicTokenCounter(BaseTokenCounterTest):
    """Test suite for Anthropic token counter."""

    def get_token_counter(self) -> BaseTokenCounter:
        return AnthropicTokenCounter()

    def get_test_model(self) -> str:
        return "claude-haiku-4-5-20251001"

    def get_test_messages(self) -> List[Dict[str, Any]]:
        return [{"role": "user", "content": "Hello, how are you today?"}]

    def get_deployment_config(self) -> Dict[str, Any]:
        api_key = os.getenv("ANTHROPIC_API_KEY")
        if not api_key:
            pytest.skip("ANTHROPIC_API_KEY not set")
        return {
            "litellm_params": {
                "api_key": api_key,
            }
        }

    def get_custom_llm_provider(self) -> str:
        return "anthropic"