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