feat: reset rag template

This commit is contained in:
Timothy J. Baek 2024-02-17 22:49:18 -08:00
parent a31feccd64
commit 07b451995e

View file

@ -172,7 +172,9 @@ class RAGTemplateForm(BaseModel):
@app.post("/template/update") @app.post("/template/update")
async def update_rag_template(form_data: RAGTemplateForm, user=Depends(get_admin_user)): async def update_rag_template(form_data: RAGTemplateForm, user=Depends(get_admin_user)):
# TODO: check template requirements # TODO: check template requirements
app.state.RAG_TEMPLATE = form_data.template app.state.RAG_TEMPLATE = (
form_data.template if form_data.template != "" else RAG_TEMPLATE
)
return {"status": True, "template": app.state.RAG_TEMPLATE} return {"status": True, "template": app.state.RAG_TEMPLATE}