|
|
import os |
|
|
from dotenv import load_dotenv |
|
|
|
|
|
load_dotenv() |
|
|
|
|
|
APP_SECRET = os.getenv("APP_SECRET") |
|
|
|
|
|
|
|
|
GIZAI_BASE_URL = "https://app.giz.ai" |
|
|
GIZAI_API_ENDPOINT = f"{GIZAI_BASE_URL}/api/data/users/inferenceServer.infer" |
|
|
|
|
|
|
|
|
ALLOWED_MODELS = [ |
|
|
{"id": "chat-gemini-flash", "name": "chat-gemini-flash"}, |
|
|
{"id": "chat-gemini-pro", "name": "chat-gemini-pro"}, |
|
|
{"id": "chat-gpt4m", "name": "chat-gpt4m"}, |
|
|
{"id": "chat-gpt4", "name": "chat-gpt4"}, |
|
|
{"id": "claude-sonnet", "name": "claude-sonnet"}, |
|
|
{"id": "claude-haiku", "name": "claude-haiku"}, |
|
|
{"id": "llama-3-70b", "name": "llama-3-70b"}, |
|
|
{"id": "llama-3-8b", "name": "llama-3-8b"}, |
|
|
{"id": "mistral-large", "name": "mistral-large"}, |
|
|
{"id": "chat-o1-mini", "name": "chat-o1-mini"}, |
|
|
{"id": "flux1", "name": "flux1"}, |
|
|
{"id": "sdxl", "name": "sdxl"}, |
|
|
{"id": "sd", "name": "sd"}, |
|
|
{"id": "sd35", "name": "sd35"}, |
|
|
] |
|
|
|
|
|
|
|
|
MODEL_PROVIDER_MAPPING = { |
|
|
|
|
|
"chat-gemini-flash": "gizai", |
|
|
"chat-gemini-pro": "gizai", |
|
|
"chat-gpt4m": "gizai", |
|
|
"chat-gpt4": "gizai", |
|
|
"claude-sonnet": "gizai", |
|
|
"claude-haiku": "gizai", |
|
|
"llama-3-70b": "gizai", |
|
|
"llama-3-8b": "gizai", |
|
|
"mistral-large": "gizai", |
|
|
"chat-o1-mini": "gizai", |
|
|
"flux1": "gizai", |
|
|
"sdxl": "gizai", |
|
|
"sd": "gizai", |
|
|
"sd35": "gizai", |
|
|
|
|
|
"gemini-flash": "gizai", |
|
|
"gemini-pro": "gizai", |
|
|
"gpt-4o-mini": "gizai", |
|
|
"gpt-4o": "gizai", |
|
|
"claude-3.5-sonnet": "gizai", |
|
|
"claude-3-haiku": "gizai", |
|
|
"llama-3.1-70b": "gizai", |
|
|
"llama-3.1-8b": "gizai", |
|
|
"o1-mini": "gizai", |
|
|
"sd-1.5": "gizai", |
|
|
"sd-3.5": "gizai", |
|
|
"flux-schnell": "gizai", |
|
|
} |
|
|
|
|
|
|
|
|
MODEL_MAPPING = { |
|
|
"chat-gemini-flash": "chat-gemini-flash", |
|
|
"chat-gemini-pro": "chat-gemini-pro", |
|
|
"chat-gpt4m": "chat-gpt4m", |
|
|
"chat-gpt4": "chat-gpt4", |
|
|
"claude-sonnet": "claude-sonnet", |
|
|
"claude-haiku": "claude-haiku", |
|
|
"llama-3-70b": "llama-3-70b", |
|
|
"llama-3-8b": "llama-3-8b", |
|
|
"mistral-large": "mistral-large", |
|
|
"chat-o1-mini": "chat-o1-mini", |
|
|
"flux1": "flux1", |
|
|
"sdxl": "sdxl", |
|
|
"sd": "sd", |
|
|
"sd35": "sd35", |
|
|
|
|
|
"gemini-flash": "chat-gemini-flash", |
|
|
"gemini-pro": "chat-gemini-pro", |
|
|
"gpt-4o-mini": "chat-gpt4m", |
|
|
"gpt-4o": "chat-gpt4", |
|
|
"claude-3.5-sonnet": "claude-sonnet", |
|
|
"claude-3-haiku": "claude-haiku", |
|
|
"llama-3.1-70b": "llama-3-70b", |
|
|
"llama-3.1-8b": "llama-3-8b", |
|
|
"o1-mini": "chat-o1-mini", |
|
|
"sd-1.5": "sd", |
|
|
"sd-3.5": "sd35", |
|
|
"flux-schnell": "flux1", |
|
|
} |
|
|
|
