from __future__ import annotations

from enum import StrEnum


class GenerationRecordConfigSourceEnum(StrEnum):
    """生成记录配置冻结来源。"""

    PROMPT_OPTIMIZE = "prompt_optimize"
    LEGACY_GENERATE_FALLBACK = "legacy_generate_fallback"
    EXISTING_FROZEN_CONFIG = "existing_frozen_config"


class GenerationRecordEventTypeEnum(StrEnum):
    """GenerationRecord 用户生成链路事件。"""

    PROMPT_CONFIG_VALIDATE_START = "PROMPT_CONFIG_VALIDATE_START"
    PROMPT_CONFIG_VALIDATE_SUCCESS = "PROMPT_CONFIG_VALIDATE_SUCCESS"
    PROMPT_CONFIG_VALIDATE_FAILED = "PROMPT_CONFIG_VALIDATE_FAILED"
    PROMPT_CONFIG_FREEZE_START = "PROMPT_CONFIG_FREEZE_START"
    PROMPT_CONFIG_FREEZE_SUCCESS = "PROMPT_CONFIG_FREEZE_SUCCESS"
    PROMPT_CONFIG_FREEZE_FAILED = "PROMPT_CONFIG_FREEZE_FAILED"
    LEGACY_CONFIG_FALLBACK_START = "LEGACY_CONFIG_FALLBACK_START"
    LEGACY_CONFIG_FALLBACK_SUCCESS = "LEGACY_CONFIG_FALLBACK_SUCCESS"
    LEGACY_CONFIG_FALLBACK_FAILED = "LEGACY_CONFIG_FALLBACK_FAILED"
    LEGACY_CONFIG_FALLBACK_SKIPPED = "LEGACY_CONFIG_FALLBACK_SKIPPED"
    GENERATION_SUBMIT_START = "GENERATION_SUBMIT_START"
    GENERATION_SUBMIT_CONFIG_READY = "GENERATION_SUBMIT_CONFIG_READY"
    GENERATION_SUBMIT_BILLING_SUCCESS = "GENERATION_SUBMIT_BILLING_SUCCESS"
    GENERATION_SUBMIT_ENQUEUE_SUCCESS = "GENERATION_SUBMIT_ENQUEUE_SUCCESS"
    GENERATION_SUBMIT_FAILED = "GENERATION_SUBMIT_FAILED"