File size: 2,996 Bytes
3979178 |
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 |
import os
import platform
import uuid
from typing import List, Dict
import httpx
from dotenv import load_dotenv
from pydantic_settings import BaseSettings
load_dotenv()
class Settings(BaseSettings):
TIMEZONE: str = "Asia/Shanghai"
READY_REFRESH_TOKEN: str = os.getenv('READY_REFRESH_TOKEN', '')
READY_TOKEN: str = os.getenv('READY_TOKEN', '')
APP_SECRET: str = os.getenv('APP_SECRET', '')
TOKEN: str = os.getenv('TOKEN', '')
PROJECT_NAME: str = os.getenv('PROJECT_NAME', 'FastAPI')
DESCRIPTION: str = os.getenv('DESCRIPTION', 'FastAPI template')
FIREBASE_API_KEY: str = os.getenv('FIREBASE_API_KEY', '')
REFRESH_TOKEN: str = os.getenv('REFRESH_TOKEN', '')
AUTHORIZATION_TOKEN: str = os.getenv('AUTHORIZATION_TOKEN', '')
ALLOWED_MODELS: List[Dict[str, str]] = [
{"id": "o1-mini", "name": "o1-mini[merlin]"},
{"id": "claude-3.5-sonnet", "name": "Claude 3.5 Sonnet[merlin]"},
{"id": "gpt-4o-64k-output", "name": "GPT 4o (Longer Output)[merlin]"},
{"id": "gemini-1.5-pro", "name": "Gemini 1.5 Pro[merlin]"},
{"id": "gpt-4o", "name": "GPT 4o[merlin]"},
{"id": "meta-llama/Meta-Llama-3.1-405B-Instruct", "name": "Llama 3.1 405B[merlin]"},
{"id": "claude-3.5-haiku", "name": "Claude 3.5 Haiku[merlin]"},
{"id": "claude-3-haiku", "name": "Claude 3 Haiku[merlin]"},
{"id": "gemini-1.5-flash", "name": "Gemini 1.5 Flash[merlin]"},
{"id": "gpt-4o-mini", "name": "GPT 4o mini[merlin]"},
{"id": "deepseek-chat", "name": "DeepSeek V3[merlin]"},
{"id": "o1", "name": "O1[merlin]"},
{"id": "o1-preview", "name": "O1 Preview[merlin]"},
{"id": "mistral-large-latest", "name": "Mistral Large[merlin]"},
{"id": "claude-3-opus", "name": "Claude 3 Opus[merlin]"},
{"id": "claude-3-sonnet", "name": "Claude 3 Sonnet[merlin]"},
{"id": "mistralai/Mixtral-8x7B-Instruct-v0.1", "name": "Mixtral[merlin]"},
{"id": "gpt-3.5-turbo", "name": "GPT 3.5[merlin]"},
]
MODEL_MAPPING: Dict[str, str] = {
"gpt-4o": "gpt-4o",
"o1-preview": "o1-preview",
"claude-3-5-sonnet": "claude-3-5-sonnet",
"o1-mini": "o1-mini",
"gemini-1.5-pro": "gemini-1.5-pro",
"gemini-2.0-flash": "gemini-2.0-flash",
}
HEADERS: Dict[str, str] = {
"authorization": f"Bearer {os.getenv('TOKEN', '')}",
"accept": "text/event-stream, text/event-stream",
"content-type": "application/json",
"x-merlin-version": "web-merlin",
"user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36",
"origin": "https://www.getmerlin.in",
"referer": "https://www.getmerlin.in/"
}
class Config:
env_file = '.env'
case_sensitive = True
_settings = None
def get_settings():
global _settings
if _settings is None:
_settings = Settings()
return _settings |