RoleTypes (director/constants.py)

Module exposing: RoleTypes, DBType, LLMType, EnvPrefix

# Source: https://github.com/video-db/Director | License: MIT | Copyright Ashutosh Trivedi
# Provided via justclaw.it.com; keep this notice with the code.
from enum import Enum


class RoleTypes(str, Enum):
    system = "system"
    user = "user"
    assistant = "assistant"


class DBType(str, Enum):
    SQLITE = "sqlite"
    TURSO = "turso"
    SQL = "sql"
    POSTGRES = "postgres"


class LLMType(str, Enum):
    """Enum for LLM types"""

    OPENAI = "openai"
    ANTHROPIC = "anthropic"
    GOOGLEAI = "googleai"
    VIDEODB_PROXY = "videodb_proxy"


class EnvPrefix(str, Enum):
    """Enum for environment prefixes"""

    OPENAI_ = "OPENAI_"
    ANTHROPIC_ = "ANTHROPIC_"
    GOOGLEAI_ = "GOOGLEAI_"

DOWNLOADS_PATH="director/downloads"