Update app.py
Browse files
app.py
CHANGED
@@ -38,7 +38,14 @@ def build_faiss_vectorstore(chunks):
|
|
38 |
# Function to retrieve similar text
|
39 |
def retrieve(query, vectorstore, top_k=5):
|
40 |
docs_and_scores = vectorstore.similarity_search_with_score(query=query, k=top_k)
|
41 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
42 |
|
43 |
class ChatRequest(BaseModel):
|
44 |
message: str
|
|
|
38 |
# Function to retrieve similar text
|
39 |
def retrieve(query, vectorstore, top_k=5):
|
40 |
docs_and_scores = vectorstore.similarity_search_with_score(query=query, k=top_k)
|
41 |
+
|
42 |
+
# Correct handling of numpy float scores:
|
43 |
+
fixed_results = []
|
44 |
+
for doc, score in docs_and_scores:
|
45 |
+
if isinstance(score, np.floating):
|
46 |
+
score = float(score) # Convert numpy float to standard Python float
|
47 |
+
fixed_results.append((doc, score))
|
48 |
+
return fixed_results
|
49 |
|
50 |
class ChatRequest(BaseModel):
|
51 |
message: str
|