60 lines
1.7 KiB
Python
60 lines
1.7 KiB
Python
from pydantic_settings import BaseSettings
|
||
from pathlib import Path
|
||
import os
|
||
|
||
|
||
class Settings(BaseSettings):
|
||
# 应用
|
||
APP_NAME: str = "AI 文档模板生成系统"
|
||
APP_VERSION: str = "1.0.0"
|
||
DEBUG: bool = True
|
||
|
||
# === 数据库 MySQL ===
|
||
DB_HOST: str = "localhost"
|
||
DB_PORT: int = 3306
|
||
DB_USER: str = "docforge"
|
||
DB_PASSWORD: str = "docforge123"
|
||
DB_NAME: str = "doc_forge"
|
||
@property
|
||
def DATABASE_URL(self) -> str:
|
||
return f"mysql+asyncmy://{self.DB_USER}:{self.DB_PASSWORD}@{self.DB_HOST}:{self.DB_PORT}/{self.DB_NAME}?charset=utf8mb4"
|
||
|
||
# === MinIO 文件存储 ===
|
||
MINIO_ENDPOINT: str = "localhost:9000"
|
||
MINIO_ACCESS_KEY: str = "docforge"
|
||
MINIO_SECRET_KEY: str = "docforge123"
|
||
MINIO_BUCKET_TEMPLATES: str = "doc-forge-templates"
|
||
MINIO_BUCKET_UPLOADS: str = "doc-forge-uploads"
|
||
MINIO_BUCKET_OUTPUTS: str = "doc-forge-outputs"
|
||
MINIO_USE_SSL: bool = False
|
||
|
||
# 本地缓存目录（MinIO 文件的本地临时缓存）
|
||
LOCAL_CACHE_DIR: str = "local_cache"
|
||
|
||
# 文件上传限制
|
||
MAX_UPLOAD_SIZE: int = 50 * 1024 * 1024 # 50MB
|
||
ALLOWED_EXTENSIONS: list = [".docx", ".doc", ".xlsx", ".xls", ".xlsm", ".csv", ".pdf", ".txt", ".md", ".json"]
|
||
|
||
# 加密（用于 API Key 加密）
|
||
ENCRYPTION_KEY: str = "change-this-to-a-32-byte-key-in-production!!"
|
||
|
||
# AI 模型默认配置
|
||
AI_REQUEST_TIMEOUT: int = 60
|
||
AI_MAX_RETRIES: int = 3
|
||
AI_MAX_CONCURRENT: int = 5
|
||
AI_GLOBAL_CONCURRENT: int = 10
|
||
|
||
# 服务端口
|
||
HOST: str = "0.0.0.0"
|
||
PORT: int = 8000
|
||
|
||
class Config:
|
||
env_file = ".env"
|
||
env_file_encoding = "utf-8"
|
||
|
||
|
||
settings = Settings()
|
||
|
||
# 创建本地缓存目录
|
||
os.makedirs(settings.LOCAL_CACHE_DIR, exist_ok=True)
|