api/config.py

import os
from dotenv import load_dotenv

load_dotenv()

APP_SECRET = os.getenv("APP_SECRET")

# Allowed models
ALLOWED_MODELS = [
    {"id": "gpt-4o", "name": "gpt-4o"},
    {"id": "gpt-4o-mini", "name": "gpt-4o-mini"},
    {"id": "o1-preview", "name": "o1-preview"},
    {"id": "o1-mini", "name": "o1-mini"},
    {"id": "meta-llama/Meta-Llama-3.1-405B-Instruct-Turbo", "name": "Meta-Llama-3.1-405B"},
    {"id": "meta-llama/Llama-3.2-90B-Vision-Instruct-Turbo", "name": "Llama-3.2-90B-Vision"},
    {"id": "claude-3-sonnet-20240229", "name": "claude-3-sonnet"},
    {"id": "gemini-1.5-pro", "name": "gemini-1.5-pro"},
    {"id": "flux-pro/v1.1", "name": "flux-pro/v1.1"},
    {"id": "flux-realism", "name": "flux-realism"},
    {"id": "flux-pro", "name": "flux-pro"},
    {"id": "dalle-e-3", "name": "dalle-e-3"},
    # Aliases
    {"id": "o1", "name": "o1"},
    {"id": "llama-3.1-405b", "name": "llama-3.1-405b"},
    {"id": "llama-3.2-90b", "name": "llama-3.2-90b"},
    {"id": "claude-3.5-sonnet", "name": "claude-3.5-sonnet"},
    {"id": "gemini-pro", "name": "gemini-pro"},
    {"id": "dalle-3", "name": "dalle-3"},
]

# Model to provider mapping
MODEL_PROVIDER_MAPPING = {
    # AmigoChat models
    "gpt-4o": "amigochat",
    "gpt-4o-mini": "amigochat",
    "o1-preview": "amigochat",
    "o1-mini": "amigochat",
    "meta-llama/Meta-Llama-3.1-405B-Instruct-Turbo": "amigochat",
    "meta-llama/Llama-3.2-90B-Vision-Instruct-Turbo": "amigochat",
    "claude-3-sonnet-20240229": "amigochat",
    "gemini-1.5-pro": "amigochat",
    "flux-pro/v1.1": "amigochat",
    "flux-realism": "amigochat",
    "flux-pro": "amigochat",
    "dalle-e-3": "amigochat",
    # Aliases
    "o1": "amigochat",
    "llama-3.1-405b": "amigochat",
    "llama-3.2-90b": "amigochat",
    "claude-3.5-sonnet": "amigochat",
    "gemini-pro": "amigochat",
    "dalle-3": "amigochat",
}

# Model mapping (aliases)
MODEL_MAPPING = {
    "gpt-4o": "gpt-4o",
    "gpt-4o-mini": "gpt-4o-mini",
    "o1-preview": "o1-preview",
    "o1-mini": "o1-mini",
    "meta-llama/Meta-Llama-3.1-405B-Instruct-Turbo": "meta-llama/Meta-Llama-3.1-405B-Instruct-Turbo",
    "meta-llama/Llama-3.2-90B-Vision-Instruct-Turbo": "meta-llama/Llama-3.2-90B-Vision-Instruct-Turbo",
    "claude-3-sonnet-20240229": "claude-3-sonnet-20240229",
    "gemini-1.5-pro": "gemini-1.5-pro",
    "flux-pro/v1.1": "flux-pro/v1.1",
    "flux-realism": "flux-realism",
    "flux-pro": "flux-pro",
    "dalle-e-3": "dalle-e-3",
    # Aliases
    "o1": "o1-preview",
    "llama-3.1-405b": "meta-llama/Meta-Llama-3.1-405B-Instruct-Turbo",
    "llama-3.2-90b": "meta-llama/Llama-3.2-90B-Vision-Instruct-Turbo",
    "claude-3.5-sonnet": "claude-3-sonnet-20240229",
    "gemini-pro": "gemini-1.5-pro",
    "dalle-3": "dalle-e-3",
}