Spaces:

rufimelo
/

compression-theory-viz

Runtime error

compression-theory-viz / utils /constants.py

feat: add streamlit visualization app with prompts, trajectories, and outputs

d14dfc9 5 months ago

637 Bytes

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

	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