NathanAlex7
42ce992a0f
Next.js 前端 + FastAPI/LangChain 后端透明 agent平台 - 含 .gitignore, 敏感文件(sessions数据/.env/venv等)已排除 - 待后续添加 trajectory 自进化机制
43 lines
1.1 KiB
Python
43 lines
1.1 KiB
Python
"""Global configuration management — JSON-based persistence."""
|
|
|
|
import json
|
|
from pathlib import Path
|
|
from typing import Any
|
|
|
|
CONFIG_FILE = Path(__file__).resolve().parent / "config.json"
|
|
|
|
_DEFAULT_CONFIG: dict[str, Any] = {
|
|
"rag_mode": False,
|
|
}
|
|
|
|
|
|
def load_config() -> dict[str, Any]:
|
|
"""Load configuration from disk, returning defaults if missing."""
|
|
if not CONFIG_FILE.exists():
|
|
return dict(_DEFAULT_CONFIG)
|
|
try:
|
|
data = json.loads(CONFIG_FILE.read_text(encoding="utf-8"))
|
|
return {**_DEFAULT_CONFIG, **data}
|
|
except Exception:
|
|
return dict(_DEFAULT_CONFIG)
|
|
|
|
|
|
def save_config(config: dict[str, Any]) -> None:
|
|
"""Persist configuration to disk."""
|
|
CONFIG_FILE.write_text(
|
|
json.dumps(config, ensure_ascii=False, indent=2),
|
|
encoding="utf-8",
|
|
)
|
|
|
|
|
|
def get_rag_mode() -> bool:
|
|
"""Get current RAG mode setting."""
|
|
return bool(load_config().get("rag_mode", False))
|
|
|
|
|
|
def set_rag_mode(enabled: bool) -> None:
|
|
"""Set RAG mode on/off."""
|
|
config = load_config()
|
|
config["rag_mode"] = enabled
|
|
save_config(config)
|