58 lines
1.5 KiB
Python
58 lines
1.5 KiB
Python
"""
|
||
ParamHub 配置文件
|
||
"""
|
||
import secrets
|
||
from pathlib import Path
|
||
|
||
# 项目根目录
|
||
BASE_DIR = Path(__file__).parent
|
||
|
||
# 数据目录
|
||
DATA_DIR = BASE_DIR / 'data'
|
||
DATA_DIR.mkdir(exist_ok=True)
|
||
|
||
# 图片上传目录
|
||
IMAGES_DIR = BASE_DIR / 'static' / 'uploads'
|
||
IMAGES_DIR.mkdir(parents=True, exist_ok=True)
|
||
|
||
# 允许的图片格式
|
||
ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif', 'webp'}
|
||
|
||
# 数据文件路径
|
||
MODELS_FILE = DATA_DIR / 'models.json'
|
||
GPUS_FILE = DATA_DIR / 'gpus.json'
|
||
CPUS_FILE = DATA_DIR / 'cpus.json'
|
||
CATEGORIES_FILE = DATA_DIR / 'categories.json'
|
||
KNOWLEDGE_FILE = DATA_DIR / 'knowledge.json'
|
||
CONFIG_FILE = DATA_DIR / 'config.json'
|
||
|
||
# Flask secret key (每次启动随机生成，重启后需重新登录)
|
||
SECRET_KEY = secrets.token_hex(32)
|
||
|
||
# 默认管理员密码（首次使用，之后可从 config.json 修改）
|
||
DEFAULT_ADMIN_PASSWORD = 'admin123'
|
||
|
||
# 大模型配置（默认值）
|
||
LLM_CONFIG = {
|
||
'base_url': 'http://192.168.2.17:19007/v1',
|
||
'api_key': 'xxxx',
|
||
'model': 'auto',
|
||
'vision_model': 'gpt-4-vision-preview',
|
||
}
|
||
|
||
# 默认网站配置（包含LLM配置）
|
||
DEFAULT_CONFIG = {
|
||
'site_name': 'ParamHub',
|
||
'site_subtitle': '参数百科',
|
||
'footer_text': 'ParamHub - AI大模型与硬件参数速查平台',
|
||
'icp_number': '',
|
||
'copyright_year': '2024',
|
||
'contact_email': '',
|
||
'github_url': '',
|
||
'llm_base_url': 'http://192.168.2.17:19007/v1',
|
||
'llm_api_key': '',
|
||
'llm_model': 'auto',
|
||
'llm_vision_model': 'gpt-4-vision-preview',
|
||
'admin_password': DEFAULT_ADMIN_PASSWORD,
|
||
}
|