DD8943 commited on
Commit
74d43d2
·
verified ·
1 Parent(s): d005cef

Solving bug with rag searching

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -23,8 +23,9 @@ def embed_query(text):
23
  # Semantic search using FAISS
24
  def search(query, k=3):
25
  query_vec = embed_query(query)
26
- scores, indices = index.search(query_vec, k)
27
- return [chunks[i] for i in indices[0]]
 
28
 
29
  # Chat modes
30
  def chat_no_rag(question):
 
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):