sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2024-11-23 15:38:55 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

Pass type of item via regenerate API. Default type query param to None

Browse source
This commit is contained in:
Debanjum Singh Solanky 2021-08-17 04:36:48 -07:00
parent ff7207a6bd
commit 252266b62a
1 changed files with 8 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
src/main.py
View file

@ -18,7 +18,7 @@ app = FastAPI()
@app.get('/search')
def search(q: str, n: Optional[int] = 5, t: Optional[str] = 'notes'):
def search(q: str, n: Optional[int] = 5, t: Optional[str] = None):
if q is None or q == '':
print(f'No query param (q) passed in API call to initiate search')
return {}
@ -26,7 +26,7 @@ def search(q: str, n: Optional[int] = 5, t: Optional[str] = 'notes'):
user_query = q
results_count = n
if t == 'notes':
if t == 'notes' or t == None:
# query notes
hits = asymmetric.query_notes(
user_query,
@ -44,17 +44,12 @@ def search(q: str, n: Optional[int] = 5, t: Optional[str] = 'notes'):
@app.get('/regenerate')
def regenerate():
# Extract Entries, Generate Embeddings
extracted_entries, computed_embeddings, _, _, _ = asymmetric.setup(args.input_files, args.input_filter, args.compressed_jsonl, args.embeddings, regenerate=True, verbose=args.verbose)
# Now Update State
# update state variables after regeneration complete
# minimize time the application is in inconsistent, partially updated state
global corpus_embeddings
global entries
entries = extracted_entries
corpus_embeddings = computed_embeddings
def regenerate(t: Optional[str] = None):
if t == 'notes' or t == None:
# Extract Entries, Generate Embeddings
global corpus_embeddings
global entries
entries, corpus_embeddings, _, _, _ = asymmetric.setup(args.input_files, args.input_filter, args.compressed_jsonl, args.embeddings, regenerate=True, verbose=args.verbose)
return {'status': 'ok', 'message': 'regeneration completed'}