enums.py

from enum import Enum


class AIPromptAction(Enum):
    ASSIST = "assist"
    PRE_TRANSLATE = "pre_translate"


class AIPromptOperation(Enum):
    REPLACE = "replace"
    TEST = "test"


class EditAIPromptPath(Enum):
    NAME = "/name"
    ACTION = "/action"
    AI_PROVIDER_ID = "/aiProviderId"
    AI_MODEL_ID = "/aiModelId"
    IS_ENABLED = "/isEnabled"
    ENABLED_PROJECT_IDS = "/enabledProjectIds"
    CONFIG = "/config"


class AIProviderType(Enum):
    OPEN_AI = "open_ai"
    AZURE_OPEN_AI = "azure_open_ai"
    GOOGLE_GEMINI = "google_gemini"
    MISTRAL_AI = "mistral_ai"
    ANTHROPIC = "anthropic"
    CUSTOM_AI = "custom_ai"


class EditAIProviderPath(Enum):
    NAME = "/name"
    TYPE = "/type"
    CREDENTIALS = "/credentials"
    CONFIG = "/config"
    IS_ENABLED = "/isEnabled"
    USE_SYSTEM_CREDENTIALS = "/useSystemCredentials"


class DatasetPurpose(Enum):
    TRAINING = "training"
    VALIDATION = "validation"


class EditAiCustomPlaceholderPatchPath(Enum):
    DESCRIPTION = "/description"
    PLACEHOLDER = "/placeholder"
    VALUE = "/value"


class AiPromptFineTuningJobStatus(Enum):
    CREATED = "created"
    IN_PROGRESS = "in_progress"
    CANCELED = "canceled"
    FAILED = "failed"
    FINISHED = "finished"


class AiToolType(Enum):
    FUNCTION = "function"


class AiReportType(Enum):
    TOKENS_USAGE_RAW_DATA = "tokens-usage-raw-data"


class EditAiSettingsPatchPath(Enum):
    ASSIST_ACTION_AI_PROMPT_ID = "/assistActionAiPromptId"
    EDITOR_SUGGESTION_AI_PROMPT_ID = "/editorSuggestionAiPromptId"
    SHORTCUTS = "/shortcuts"


class ListAiPromptFineTuningJobsOrderBy(Enum):
    CREATED_AT = "createdAt"
    UPDATED_AT = "updatedAt"
    STARTED_AT = "startedAt"
    FINISHED_AT = "finishedAt"


class AiReportFormat(Enum):
    CSV = "csv"
    JSON = "json"


class ListSupportedAiModelsOrderBy(Enum):
    KNOWLEDGE_CUTOFF = "knowledgeCutoff"
    RELEASE_DATE = "releaseDate"