Lhumpal commited on
Commit
f1a3c8f
·
verified ·
1 Parent(s): 78bfbb7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -43,7 +43,7 @@ def retrieve(query, vectorstore, top_k=8):
43
  docs_and_scores = vectorstore.similarity_search_with_score(query=query, k=top_k)
44
 
45
  # Filter results based on score threshold
46
- filtered_docs_and_scores = [(doc.page_content, float(score)) for doc, score in docs_and_scores if float(score) <= 0.8]
47
 
48
  # Separate docs from the (doc, score) tuples
49
  docs_content = [doc for doc, _ in filtered_docs_and_scores]
@@ -134,7 +134,7 @@ async def chat(request: ChatRequest):
134
 
135
  # ------------ Retrieve relevant text ------------
136
  rephrase_response = rephrase_response.text
137
- docs, filtered_docs_and_scores = retrieve(rephrase_response, vectorstore, top_k=8)
138
  docs = "\n\n".join(docs)
139
 
140
  # ------------ Retrievel Augmented Generation ------------
 
43
  docs_and_scores = vectorstore.similarity_search_with_score(query=query, k=top_k)
44
 
45
  # Filter results based on score threshold
46
+ filtered_docs_and_scores = [(doc.page_content, float(score)) for doc, score in docs_and_scores if float(score) <= 0.7]
47
 
48
  # Separate docs from the (doc, score) tuples
49
  docs_content = [doc for doc, _ in filtered_docs_and_scores]
 
134
 
135
  # ------------ Retrieve relevant text ------------
136
  rephrase_response = rephrase_response.text
137
+ docs, filtered_docs_and_scores = retrieve(rephrase_response, vectorstore, top_k=10)
138
  docs = "\n\n".join(docs)
139
 
140
  # ------------ Retrievel Augmented Generation ------------