File size: 2,765 Bytes
8a7a6a8 ea6f2fc 8a7a6a8 ea6f2fc c300504 ea6f2fc c300504 ea6f2fc c300504 ea6f2fc 650b357 ea6f2fc c300504 ea6f2fc c300504 ea6f2fc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 |
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",
}
|