stores/llm/LLMEnums.py

from enum import Enum

class LLMEnums(Enum):
    OPENAI    = "OPENAI"
    COHERE    = "COHERE"
    OLLAMA    = "OLLAMA"
    MISTRAL   = "MISTRAL"
    GROQ      = "GROQ"
    OPENROUTER = "OPENROUTER"
    HUGGINGFACE = "HUGGINGFACE"
    DEEPSEEK  = "DEEPSEEK"
    GEMINI    = "GEMINI"

class OpenAIEnums(Enum):
    SYSTEM    = "system"
    USER      = "user"
    ASSISTANT = "assistant"

class CoHereEnums(Enum):
    SYSTEM    = "SYSTEM"
    USER      = "USER"
    ASSISTANT = "CHATBOT"
    DOCUMENT  = "search_document"
    QUERY     = "search_query"

class DocumentTypeEnum(Enum):
    DOCUMENT = "document"
    QUERY    = "query"