sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2025-02-17 08:04:21 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update global model on regenerate via API

Browse source
This commit is contained in:
Debanjum Singh Solanky 2022-06-17 00:49:06 +03:00
parent 35117af322
commit cfbd5c4ecc
1 changed files with 2 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
src/main.py
View file

@ -92,7 +92,8 @@ def search(q: str, n: Optional[int] = 5, t: Optional[SearchType] = None):
@app.get('/regenerate')
def regenerate(t: Optional[SearchType] = None):
initialize_search(config, regenerate=True, t=t)
global model
model = initialize_search(config, regenerate=True, t=t)
return {'status': 'ok', 'message': 'regeneration completed'}
@ -138,8 +139,6 @@ def chat(q: str):
def initialize_search(config: FullConfig, regenerate: bool, t: SearchType = None):
model = SearchModels()
# Initialize Org Notes Search
if (t == SearchType.Notes or t == None) and config.content_type.org:
# Extract Entries, Generate Notes Embeddings