24 lines
520 B
Python
24 lines
520 B
Python
|
"""GET/PUT /api/config/rag-mode — RAG mode toggle API."""
|
||
|
|
|
||
|
|
from fastapi import APIRouter
|
||
|
|
from pydantic import BaseModel
|
||
|
|
|
||
|
|
from config import get_rag_mode, set_rag_mode
|
||
|
|
|
||
|
|
router = APIRouter()
|
||
|
|
|
||
|
|
|
||
|
|
class RagModeRequest(BaseModel):
|
||
|
|
enabled: bool
|
||
|
|
|
||
|
|
|
||
|
|
@router.get("/config/rag-mode")
|
||
|
|
async def get_rag_mode_endpoint():
|
||
|
|
return {"rag_mode": get_rag_mode()}
|
||
|
|
|
||
|
|
|
||
|
|
@router.put("/config/rag-mode")
|
||
|
|
async def set_rag_mode_endpoint(request: RagModeRequest):
|
||
|
|
set_rag_mode(request.enabled)
|
||
|
|
return {"rag_mode": request.enabled}
|