sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2024-11-27 17:35:07 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Remove usage of the closure to search_notes as it's not required

Browse source
This commit is contained in:
Debanjum Singh Solanky 2021-08-16 16:52:48 -07:00
parent 649e5d1327
commit adbf157deb
1 changed files with 7 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
main.py
View file

@ -7,19 +7,6 @@ import uvicorn
app = FastAPI()
def create_search_notes(corpus_embeddings, entries, bi_encoder, cross_encoder, top_k):
"Closure to create search_notes method from initialized model, entries and embeddings"
def search_notes(query):
return asymmetric.query_notes(
query,
corpus_embeddings,
entries,
bi_encoder,
cross_encoder,
top_k)
return search_notes
@app.get('/search')
def search(q: str, n: Optional[int] = 5, t: Optional[str] = 'notes'):
@ -32,7 +19,13 @@ def search(q: str, n: Optional[int] = 5, t: Optional[str] = 'notes'):
if t == 'notes':
# query notes
hits = search_notes(user_query)
hits = asymmetric.query_notes(
q,
corpus_embeddings,
entries,
bi_encoder,
cross_encoder,
top_k)
# collate and return results
return asymmetric.collate_results(hits, entries, results_count)
@ -58,8 +51,5 @@ if __name__ == '__main__':
# Compute or Load Embeddings
corpus_embeddings = asymmetric.compute_embeddings(entries, bi_encoder, args.embeddings, args.verbose)
# Generate search_notes method from initialized model, entries and embeddings
search_notes = create_search_notes(corpus_embeddings, entries, bi_encoder, cross_encoder, top_k)
# Start Application Server
uvicorn.run(app)