src/config.py

# 应用程序配置项
import os
from dotenv import load_dotenv

# 加载 .env 文件
current_dir = os.path.dirname(os.path.abspath(__file__))
# 假设 .env 文件在项目根目录，需要调整路径
# env_path = os.path.join(current_dir, '.env')
# load_dotenv(env_path)
# 如果 .env 文件在项目根目录，则直接加载
load_dotenv()


CONFIG = {
    "MODELS": {
        'grok-2': 'grok-latest',
        'grok-2-imageGen': 'grok-latest',
        'grok-2-search': 'grok-latest',
        "grok-3": "grok-3",
        "grok-3-search": "grok-3",
        "grok-3-imageGen": "grok-3",
        "grok-3-deepsearch": "grok-3",
        "grok-3-reasoning": "grok-3"
    },
    "API": {
        "IS_TEMP_CONVERSATION": os.getenv("IS_TEMP_CONVERSATION", "true").lower() == "true",
        "IS_CUSTOM_SSO": os.getenv("IS_CUSTOM_SSO", "false").lower() == "true",
        "BASE_URL": os.getenv("API_BASE_URL", "https://grok.com"),
        "API_KEY": os.getenv("API_KEY", "sk-123456"),
        "SIGNATURE_COOKIE": None,
        "PICGO_KEY": os.getenv("PICGO_KEY") or None,
        "TUMY_KEY": os.getenv("TUMY_KEY") or None,
        "RETRY_TIME": 1000,
        "PROXY": os.getenv("PROXY") or None
    },
    "ADMIN": {
        "MANAGER_SWITCH": os.getenv("MANAGER_SWITCH") or None,
        "PASSWORD": os.getenv("ADMINPASSWORD") or None
    },
    "SERVER": {
        "COOKIE": None,
        "CF_CLEARANCE":os.getenv("CF_CLEARANCE") or None,
        "PORT": int(os.getenv("PORT", 7860))
    },
    "RETRY": {
        "RETRYSWITCH": False,
        "MAX_ATTEMPTS": 2
    },
    "SHOW_THINKING": os.getenv("SHOW_THINKING") == "true",
    "IS_THINKING": False,
    "IS_IMG_GEN": False,
    "IS_IMG_GEN2": False,
    "ISSHOW_SEARCH_RESULTS": os.getenv("ISSHOW_SEARCH_RESULTS", "true").lower() == "true"
}

# 默认请求头
DEFAULT_HEADERS = {
    'Accept': '*/*',
    'Accept-Language': 'zh-CN,zh;q=0.9',
    'Accept-Encoding': 'gzip, deflate, br, zstd',
    'Content-Type': 'text/plain;charset=UTF-8',
    'Connection': 'keep-alive',
    'Origin': 'https://grok.com',
    'Priority': 'u=1, i',
    'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/133.0.0.0 Safari/537.36',
    'Sec-Ch-Ua': '"Not(A:Brand";v="99", "Google Chrome";v="133", "Chromium";v="133"',
    'Sec-Ch-Ua-Mobile': '?0',
    'Sec-Ch-Ua-Platform': '"macOS"',
    'Sec-Fetch-Dest': 'empty',
    'Sec-Fetch-Mode': 'cors',
    'Sec-Fetch-Site': 'same-origin',
    'Baggage': 'sentry-public_key=b311e0f2690c81f25e2c4cf6d4f7ce1c'
}