Lhumpal commited on
Commit
0de6c54
·
verified ·
1 Parent(s): 5d35cb2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -35,8 +35,8 @@ def build_faiss_vectorstore(chunks):
35
 
36
  # Function to retrieve similar text
37
  def retrieve(query, vectorstore, top_k=5):
38
- docs = vectorstore.similarity_search(query, k=top_k)
39
- return docs
40
 
41
  class ChatRequest(BaseModel):
42
  message: str
 
35
 
36
  # Function to retrieve similar text
37
  def retrieve(query, vectorstore, top_k=5):
38
+ docs_and_scores = vectorstore.similarity_search_with_score(query=query, k=top_k)
39
+ return docs_and_scores
40
 
41
  class ChatRequest(BaseModel):
42
  message: str