from enum import Enum


class CompressionLevel(Enum):
    """Token limit levels for compression experiments (powers of 2)"""

    LEVEL_16K = 16384
    LEVEL_8K = 8192
    LEVEL_4K = 4096
    LEVEL_2K = 2048
    LEVEL_1K = 1024
    LEVEL_512 = 512
    LEVEL_256 = 256


DEFAULT_COMPRESSION_LEVELS = [
    CompressionLevel.LEVEL_16K.value,
    CompressionLevel.LEVEL_8K.value,
    CompressionLevel.LEVEL_4K.value,
    CompressionLevel.LEVEL_2K.value,
    CompressionLevel.LEVEL_1K.value,
    CompressionLevel.LEVEL_512.value,
]

DEFAULT_MODEL = "claude-3-5-sonnet-20241022"

# Analysis configuration
ORIGINAL_THEOREM_PREVIEW_LENGTH = 500