caoxiaozhu
a3aa15d339
Initialize admin bootstrap settings during startup, persist username support in auth flows, and align frontend auth requests with local API behavior.
95 lines
2.7 KiB
Python
95 lines
2.7 KiB
Python
from pathlib import Path
|
|
from pydantic_settings import BaseSettings, SettingsConfigDict
|
|
from typing import Literal
|
|
|
|
|
|
REPO_ROOT = Path(__file__).resolve().parents[2]
|
|
ENV_FILE = REPO_ROOT / ".env"
|
|
|
|
|
|
def _resolve_path(value: str) -> str:
|
|
path = Path(value)
|
|
if path.is_absolute():
|
|
return str(path)
|
|
return str((REPO_ROOT / path).resolve())
|
|
|
|
|
|
class Settings(BaseSettings):
|
|
model_config = SettingsConfigDict(env_file=str(ENV_FILE), env_file_encoding="utf-8", extra="ignore")
|
|
|
|
# === 应用基础 ===
|
|
APP_NAME: str = "Jarvis"
|
|
APP_VERSION: str = "0.1.0"
|
|
DEBUG: bool = False
|
|
HOST: str = "127.0.0.1"
|
|
PORT: int = 9527
|
|
|
|
# === 安全 ===
|
|
SECRET_KEY: str = "change-me-in-production"
|
|
ALGORITHM: str = "HS256"
|
|
ACCESS_TOKEN_EXPIRE_MINUTES: int = 60 * 24 * 7
|
|
|
|
# === 数据库 ===
|
|
DATABASE_URL: str = "sqlite+aiosqlite:///./data/jarvis.db"
|
|
DATA_DIR: str = "data"
|
|
|
|
# === ChromaDB ===
|
|
CHROMA_PERSIST_DIR: str = "data/chroma"
|
|
|
|
# === LLM 配置 ===
|
|
# 支持: openai / claude / ollama / deepseek / custom
|
|
LLM_PROVIDER: Literal["openai", "claude", "ollama", "deepseek", "custom"] = "openai"
|
|
|
|
# OpenAI (默认)
|
|
OPENAI_API_KEY: str = ""
|
|
OPENAI_MODEL: str = "gpt-4o"
|
|
OPENAI_BASE_URL: str = "https://api.openai.com/v1"
|
|
|
|
# Claude
|
|
ANTHROPIC_API_KEY: str = ""
|
|
CLAUDE_MODEL: str = "claude-sonnet-4-20250514"
|
|
CLAUDE_MAX_TOKENS: int = 8192
|
|
|
|
# Ollama (本地模型)
|
|
OLLAMA_BASE_URL: str = "http://localhost:11434"
|
|
OLLAMA_MODEL: str = "llama3"
|
|
|
|
# === 定时任务 ===
|
|
SCHEDULER_ENABLED: bool = True
|
|
DAILY_PLAN_TIME: str = "00:00"
|
|
FORUM_SCAN_INTERVAL_MINUTES: int = 30
|
|
|
|
# === CORS ===
|
|
CORS_ORIGINS: list[str] = ["http://localhost:5173", "http://localhost:3000"]
|
|
|
|
# === 文件上传 ===
|
|
UPLOAD_DIR: str = "data/uploads"
|
|
MAX_UPLOAD_SIZE: int = 50 * 1024 * 1024
|
|
MINERU_LANGUAGE: Literal["ch", "en"] = "ch"
|
|
|
|
# === 管理员 bootstrap ===
|
|
ADMIN: str = ""
|
|
ADMIN_EMAIL: str = ""
|
|
ADMIN_PASSWORD: str = ""
|
|
ADMIN_FULL_NAME: str = "Administrator"
|
|
|
|
# === 向量化 ===
|
|
EMBEDDING_MODEL: str = "text-embedding-3-small"
|
|
CHUNK_SIZE: int = 500
|
|
CHUNK_OVERLAP: int = 50
|
|
|
|
# === LangSmith 可观测性 ===
|
|
LANGSMITH_TRACING: bool = False
|
|
LANGSMITH_API_KEY: str = ""
|
|
LANGSMITH_PROJECT: str = "jarvis-agent"
|
|
|
|
# === NAS 部署 ===
|
|
NAS_DATA_ROOT: str = "/data/jarvis"
|
|
|
|
|
|
settings = Settings()
|
|
settings.DATABASE_URL = settings.DATABASE_URL.replace("./data", _resolve_path("./data"), 1)
|
|
settings.DATA_DIR = _resolve_path(settings.DATA_DIR)
|
|
settings.CHROMA_PERSIST_DIR = _resolve_path(settings.CHROMA_PERSIST_DIR)
|
|
settings.UPLOAD_DIR = _resolve_path(settings.UPLOAD_DIR)
|