tdpeuter/open-webui
tdpeuter/open-webui
1
Fork 0
forked from open-webui/open-webui
Code Issues Pull requests 1 Projects 1 Releases 2 Packages 1 Activity Actions

feat: reset rag template

Browse source
This commit is contained in:
Timothy J. Baek 2024-02-17 22:49:18 -08:00
parent a31feccd64
commit 07b451995e
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
backend/apps/rag/main.py
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}