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,
|
|||
|
|
}
|