Update app.py
Browse files
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.
|
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=
|
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 ------------
|