Lhumpal commited on
Commit
13cd2d2
·
verified ·
1 Parent(s): 344d17a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -1
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
- return docs_and_scores
 
 
 
 
 
 
 
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