45 lines
1.5 KiB
Python
45 lines
1.5 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from typing import Annotated
|
||
|
|
|
||
|
|
from fastapi import APIRouter, Depends, HTTPException, status
|
||
|
|
from sqlalchemy.orm import Session
|
||
|
|
|
||
|
|
from app.api.deps import get_db
|
||
|
|
from app.schemas.settings import (
|
||
|
|
ModelConnectivityTestRead,
|
||
|
|
ModelConnectivityTestRequest,
|
||
|
|
SettingsRead,
|
||
|
|
SettingsWrite,
|
||
|
|
)
|
||
|
|
from app.services.model_connectivity import probe_model_connectivity
|
||
|
|
from app.services.settings import SettingsService
|
||
|
|
|
||
|
|
router = APIRouter(prefix="/settings")
|
||
|
|
DbSession = Annotated[Session, Depends(get_db)]
|
||
|
|
|
||
|
|
|
||
|
|
@router.get("", response_model=SettingsRead)
|
||
|
|
def get_settings(db: DbSession) -> SettingsRead:
|
||
|
|
return SettingsService(db).get_settings_snapshot()
|
||
|
|
|
||
|
|
|
||
|
|
@router.put("", response_model=SettingsRead)
|
||
|
|
def update_settings(payload: SettingsWrite, db: DbSession) -> SettingsRead:
|
||
|
|
try:
|
||
|
|
return SettingsService(db).save_settings_snapshot(payload)
|
||
|
|
except ValueError as exc:
|
||
|
|
raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail=str(exc)) from exc
|
||
|
|
|
||
|
|
|
||
|
|
@router.post("/model-connectivity", response_model=ModelConnectivityTestRead)
|
||
|
|
def test_model_connectivity(payload: ModelConnectivityTestRequest, db: DbSession) -> ModelConnectivityTestRead:
|
||
|
|
resolved_payload = payload
|
||
|
|
|
||
|
|
if not payload.api_key and payload.slot:
|
||
|
|
stored_api_key = SettingsService(db).load_saved_model_api_key(payload.slot)
|
||
|
|
if stored_api_key:
|
||
|
|
resolved_payload = payload.model_copy(update={"api_key": stored_api_key})
|
||
|
|
|
||
|
|
return probe_model_connectivity(resolved_payload)
|