65 lines
2.0 KiB
Python
65 lines
2.0 KiB
Python
from fastapi import APIRouter, Depends
|
|
from pydantic import BaseModel
|
|
from typing import Optional
|
|
import aiosqlite
|
|
from database import get_db
|
|
|
|
router = APIRouter(prefix="/api/settings", tags=["settings"])
|
|
|
|
EDITABLE_KEYS = [
|
|
"ai_base_url", "ai_api_key", "ai_model", "summary_model",
|
|
"vision_model", "voice_model", "topic_analysis_prompt",
|
|
]
|
|
|
|
|
|
def _mask_key(value: str) -> str:
|
|
if not value or len(value) <= 8:
|
|
return "*" * len(value) if value else ""
|
|
return value[:3] + "*" * (len(value) - 7) + value[-4:]
|
|
|
|
|
|
@router.get("")
|
|
async def get_settings(db: aiosqlite.Connection = Depends(get_db)):
|
|
result = {}
|
|
placeholders = ",".join("?" for _ in EDITABLE_KEYS)
|
|
async with db.execute(
|
|
f"SELECT key, value FROM app_settings WHERE key IN ({placeholders})",
|
|
EDITABLE_KEYS,
|
|
) as cur:
|
|
rows = await cur.fetchall()
|
|
for row in rows:
|
|
k, v = row["key"], row["value"]
|
|
result[k] = _mask_key(v) if k == "ai_api_key" else v
|
|
for k in EDITABLE_KEYS:
|
|
if k not in result:
|
|
result[k] = ""
|
|
return result
|
|
|
|
|
|
class SettingsUpdate(BaseModel):
|
|
ai_base_url: Optional[str] = None
|
|
ai_api_key: Optional[str] = None
|
|
ai_model: Optional[str] = None
|
|
summary_model: Optional[str] = None
|
|
vision_model: Optional[str] = None
|
|
voice_model: Optional[str] = None
|
|
topic_analysis_prompt: Optional[str] = None
|
|
|
|
|
|
@router.put("")
|
|
async def update_settings(body: SettingsUpdate, db: aiosqlite.Connection = Depends(get_db)):
|
|
updates = body.model_dump(exclude_none=True)
|
|
for k, v in updates.items():
|
|
if k not in EDITABLE_KEYS:
|
|
continue
|
|
if k == "ai_api_key" and "*" in v:
|
|
continue
|
|
await db.execute(
|
|
"INSERT INTO app_settings (key, value) VALUES (?, ?) ON CONFLICT(key) DO UPDATE SET value = ?",
|
|
(k, v, v),
|
|
)
|
|
await db.commit()
|
|
from services.runtime_settings import invalidate_cache
|
|
invalidate_cache()
|
|
return {"status": "ok"}
|