DD8943 commited on
Commit
d9caa1b
·
verified ·
1 Parent(s): 9060e91

Se arregla search method

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -22,10 +22,11 @@ def embed_query(text):
22
 
23
  # Semantic search using FAISS
24
  def search(query, k=3):
25
- query_vec = embed_query(query)
26
- query_vec = query_vec.astype(np.float32) # Required by FAISS
27
- D, I = index.search(query_vec, k)
28
- return [chunks[i] for i in I[0]]
 
29
 
30
  # Chat modes
31
  def chat_no_rag(question):
 
22
 
23
  # Semantic search using FAISS
24
  def search(query, k=3):
25
+ query_vec = embed_query(query).astype(np.float32)
26
+ distances = np.empty((1, k), dtype=np.float32)
27
+ labels = np.empty((1, k), dtype=np.int64)
28
+ index.search(query_vec, k, distances, labels)
29
+ return [chunks[i] for i in labels[0]]
30
 
31
  # Chat modes
32
  def chat_no_rag(question):