feat: reset rag template

07b45199 · Timothy J. Baek · a31feccd · 07b45199
Commit 07b45199 authored Feb 17, 2024 by Timothy J. Baek
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

backend/apps/rag/main.py backend/apps/rag/main.py +3 -1

No files found.
--- a/backend/apps/rag/main.py
+++ b/backend/apps/rag/main.py
@@ -172,7 +172,9 @@ class RAGTemplateForm(BaseModel):
 @app.post("/template/update")
 async def update_rag_template(form_data: RAGTemplateForm, user=Depends(get_admin_user)):
    # 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}