48 lines
1.6 KiB
Python
48 lines
1.6 KiB
Python
import os
|
||
import sys
|
||
from dotenv import load_dotenv
|
||
|
||
load_dotenv()
|
||
|
||
|
||
def _get_base_dir() -> str:
|
||
"""获取基础目录（支持 PyInstaller 打包后的路径）"""
|
||
if getattr(sys, 'frozen', False):
|
||
# PyInstaller 打包后，exe 所在目录
|
||
return os.path.dirname(sys.executable)
|
||
else:
|
||
# 开发模式，backend 目录
|
||
return os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
||
|
||
|
||
BASE_DIR = _get_base_dir()
|
||
|
||
|
||
class Settings:
|
||
# App
|
||
APP_HOST: str = os.getenv("APP_HOST", "0.0.0.0")
|
||
APP_PORT: int = int(os.getenv("APP_PORT", "8000"))
|
||
UPLOAD_DIR: str = os.getenv("UPLOAD_DIR", os.path.join(BASE_DIR, "uploads"))
|
||
EXPORT_DIR: str = os.getenv("EXPORT_DIR", os.path.join(BASE_DIR, "exports"))
|
||
DB_DIR: str = os.getenv("DB_DIR", os.path.join(BASE_DIR, "data"))
|
||
|
||
# LLM
|
||
LLM_PROVIDER: str = os.getenv("LLM_PROVIDER", "xiaomi")
|
||
LLM_MODEL: str = os.getenv("LLM_MODEL", "MiMo-V2-Flash")
|
||
LLM_BASE_URL: str = os.getenv("LLM_BASE_URL", "https://api.xiaomimimo.com/v1")
|
||
LLM_API_KEY: str = os.getenv("LLM_API_KEY", "")
|
||
|
||
# Extraction
|
||
SEGMENT_MAX_LINES: int = int(os.getenv("SEGMENT_MAX_LINES", "500"))
|
||
STORY_MIN_LINES: int = int(os.getenv("STORY_MIN_LINES", "50"))
|
||
CONFIDENCE_THRESHOLD: float = float(os.getenv("CONFIDENCE_THRESHOLD", "0.5"))
|
||
TEMPERATURE: float = float(os.getenv("TEMPERATURE", "0.3"))
|
||
|
||
@property
|
||
def database_url(self) -> str:
|
||
os.makedirs(self.DB_DIR, exist_ok=True)
|
||
return f"sqlite+aiosqlite:///{self.DB_DIR}/story_extractor.db"
|
||
|
||
|
||
settings = Settings()
|