60 lines
1.7 KiB
Python
60 lines
1.7 KiB
Python
from __future__ import annotations
|
|
|
|
from functools import lru_cache
|
|
import os
|
|
from pathlib import Path
|
|
from typing import Optional
|
|
from pydantic_settings import BaseSettings, SettingsConfigDict
|
|
|
|
|
|
def _apply_local_supabase_env() -> None:
|
|
"""
|
|
优先加载本地 .env 中的 Supabase 配置，避免宿主环境遗留的线上变量导致鉴权失败。
|
|
仅覆盖 SUPABASE_* 相关键。
|
|
"""
|
|
env_path = Path(__file__).resolve().parents[1] / ".env"
|
|
if not env_path.exists():
|
|
return
|
|
|
|
content = env_path.read_text(encoding="utf-8").splitlines()
|
|
kv: dict[str, str] = {}
|
|
for line in content:
|
|
stripped = line.strip()
|
|
if not stripped or stripped.startswith("#") or "=" not in stripped:
|
|
continue
|
|
key, value = stripped.split("=", 1)
|
|
kv[key.strip()] = value.strip()
|
|
|
|
for key in ("SUPABASE_URL", "SUPABASE_SERVICE_ROLE_KEY", "SUPABASE_ANON_KEY"):
|
|
file_value = kv.get(key)
|
|
if not file_value:
|
|
continue
|
|
os.environ[key] = file_value
|
|
|
|
|
|
_apply_local_supabase_env()
|
|
|
|
|
|
class Settings(BaseSettings):
|
|
"""集中管理项目配置，来源于 .env / 环境变量。"""
|
|
|
|
model_config = SettingsConfigDict(env_file=".env", env_file_encoding="utf-8", extra="ignore")
|
|
|
|
supabase_url: str
|
|
supabase_anon_key: Optional[str] = None
|
|
supabase_service_role_key: str
|
|
redis_url: str = "redis://localhost:6379/0"
|
|
frontend_url: str = "http://localhost:3000"
|
|
openai_api_key: str = ""
|
|
lightrag_db_url: str
|
|
lightrag_collection: str = "wolai-docs"
|
|
|
|
|
|
@lru_cache
|
|
def get_settings() -> Settings:
|
|
"""惰性实例化设置，避免重复读取文件。"""
|
|
return Settings()
|
|
|
|
|
|
settings = get_settings()
|