Browse Source

feat: reset rag template

Timothy J. Baek 1 year ago
parent
commit
07b451995e
1 changed files with 3 additions and 1 deletions
  1. 3 1
      backend/apps/rag/main.py

+ 3 - 1
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}