MODEL_TYPES.py

DEFAULT_EMBED_MODEL = "all-mpnet-base-v1"
DEFAULT_EMBED_LOC = "http://localhost:5001/v1"
LLM_MODELS = {
    "gpt-3.5-turbo": None,
    "gpt-4-turbo": None,
    "gpt-4o": None,
    "gpt-4o-mini": None,
    "o1-mini": None,
    "Llama-3.2-1B": "http://localhost:5000/v1",
    "Llama-3.2-3B": "http://localhost:5000/v1",
}