Coverage for app/doc_indexer/retrieve_document.py: 100%

1from flask import current_app 

2 

3 

4def query_database(query_text, k=3): 

5 vector_db = current_app.vector_db 

6 return vector_db.similarity_search_with_relevance_scores(query_text, k=k)