From ee65a4f2c79291c60d88c3d6160be97bb6ab1ac1 Mon Sep 17 00:00:00 2001
From: Debanjum Singh Solanky <debanjum@gmail.com>
Date: Wed, 14 Sep 2022 14:01:09 +0300
Subject: [PATCH] Merge /reload, /regenerate into single /update API endpoint

- Pass force=true to /update API to force regenerating index from
scratch
- Otherwise calls to the /update API endpoint will result in an
incremental update to index
---
 src/router.py | 14 ++++----------
 1 file changed, 4 insertions(+), 10 deletions(-)

diff --git a/src/router.py b/src/router.py
index d73ca331..27b237dd 100644
--- a/src/router.py
+++ b/src/router.py
@@ -137,16 +137,10 @@ def search(q: str, n: Optional[int] = 5, t: Optional[SearchType] = None, r: Opti
     return results
 
 
-@router.get('/reload')
-def reload(t: Optional[SearchType] = None):
-    state.model = configure_search(state.model, state.config, regenerate=False, t=t)
-    return {'status': 'ok', 'message': 'reload completed'}
-
-
-@router.get('/regenerate')
-def regenerate(t: Optional[SearchType] = None):
-    state.model = configure_search(state.model, state.config, regenerate=True, t=t)
-    return {'status': 'ok', 'message': 'regeneration completed'}
+@router.get('/update')
+def update(t: Optional[SearchType] = None, force: Optional[bool] = False):
+    state.model = configure_search(state.model, state.config, regenerate=force, t=t)
+    return {'status': 'ok', 'message': 'index updated completed'}
 
 
 @router.get('/beta/search')