from enum import Enum


class TokenUsageOwnerType(str, Enum):
    GENERATION_RECORD = "generation_record"
    CHAT_GENERATION_TASK = "chat_generation_task"
    MODULE_GENERATION_STEP = "module_generation_step"
    SHOT_REPLICATE_TASK_SET = "shot_replicate_task_set"
    SHOT_REPLICATE_SEGMENT = "shot_replicate_segment"
    UNKNOWN = "unknown"
