schemas.py

from enum import Enum


class ChatbotVersion(str, Enum):
    # make a enum of chatbot type and string
    CHATGPT_35 = "gpt-3.5-turbo-1106"
    CHATGPT_4 = "gpt-4-1106-preview"
    CHATGPT_4O = "gpt-4o"
    CHATGPT_4O_MINI = "gpt-4o-mini"


class ServiceProvider(str, Enum):
    # make a enum of service provider and string
    OPENAI = "openai"
    AZURE = "azure"